1 files changed, 52 insertions, 0 deletions
diff --git a/src/main/java/io/trygvis/jz14/db/NgConnectionSupplier.java b/src/main/java/io/trygvis/jz14/db/NgConnectionSupplier.java
new file mode 100644
index 0000000..4006f1d
--- /dev/null
+++ b/src/main/java/io/trygvis/jz14/db/NgConnectionSupplier.java
@@ -0,0 +1,52 @@
+package io.trygvis.jz14.db;
+
+import com.impossibl.postgres.api.jdbc.PGConnection;
+import io.trygvis.jz14.db.DbListener.PostgresConnection;
+
+import javax.sql.DataSource;
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.util.function.Supplier;
+
+public class NgConnectionSupplier implements Supplier<PostgresConnection<PGConnection>> {
+    private final DataSource dataSource;
+
+    public NgConnectionSupplier(DataSource dataSource) {
+        this.dataSource = dataSource;
+    }
+
+    @Override
+    public PostgresConnection<PGConnection> get() {
+        try {
+            Connection sqlConnection = dataSource.getConnection();
+            PGConnection pgConnection = unwrap(sqlConnection);
+            return new PostgresConnection<>(sqlConnection, pgConnection);
+        } catch (SQLException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public static PGConnection unwrap(Connection c) {
+        if (c instanceof PGConnection) {
+            return (PGConnection) c;
+        }
+
+        /* If you're using Spring, you need to add these two to properly unwrap the underlying PGConnection.
+        if (c instanceof ConnectionHandle) {
+            return unwrap(((ConnectionHandle) c).getInternalConnection());
+        }
+        if (c instanceof ConnectionProxy) {
+            return unwrap(DataSourceUtils.getTargetConnection(c));
+        }
+        */
+        Class<? extends Connection> klass = c.getClass();
+
+        Class<?>[] interfaces = klass.getInterfaces();
+        System.out.println("interfaces.length = " + interfaces.length);
+        for (Class<?> anInterface : interfaces) {
+            System.out.println("anInterface = " + anInterface);
+        }
+
+        throw new RuntimeException("Could not unwrap connection to a PGConnection: " + c.getClass());
+    }
+}