Add PGConnection accessor to PGSQLOutput

impossibl · Oct 28, 2020 · 45b16e6 · 45b16e6
1 parent 6da2884
commit 45b16e6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/driver/src/main/java/com/impossibl/postgres/jdbc/PGSQLOutput.java b/driver/src/main/java/com/impossibl/postgres/jdbc/PGSQLOutput.java
@@ -28,6 +28,7 @@
  */
 package com.impossibl.postgres.jdbc;
 
+import com.impossibl.postgres.api.jdbc.PGConnection;
 import com.impossibl.postgres.system.Context;
 import com.impossibl.postgres.types.Type;
 import com.impossibl.postgres.utils.guava.ByteStreams;
@@ -75,6 +76,10 @@ public PGSQLOutput(Context context) {
     this.attributeValues = new ArrayList<>();
   }
 
+  public PGConnection getConnection() {
+    return (PGDirectConnection) context.unwrap();
+  }
+
   public Type[] getAttributeTypes() {
     return attributeTypes.toArray(EMPTY_TYPES);
   }