#4 - ENH: Add support for unwrapping java.sql.Connection via unwrap(java.sql.Connection)

Author: rbygrave

src/main/java/org/avaje/datasource/delegate/ConnectionDelegator.java

@@ -13,6 +13,13 @@ public ConnectionDelegator(Connection delegate) {
     this.delegate = delegate;
   }
 
+  /**
+   * Return the underlying connection.
+   */
+  public Connection getDelegate() {
+    return delegate;
+  }
+
   @Override
   public void setSchema(String schema) throws SQLException {
     delegate.setSchema(schema);
@@ -231,6 +238,9 @@ public Struct createStruct(String typeName, Object[] attributes) throws SQLExcep
   }
 
   public <T> T unwrap(Class<T> iface) throws SQLException {
+    if (iface.equals(java.sql.Connection.class)) {
+      return (T)delegate;
+    }
     return delegate.unwrap(iface);
   }
 

src/test/java/org/avaje/datasource/pool/ConnectionPoolTest.java

@@ -95,4 +95,25 @@ public void getConnection_explicitUserPassword() throws SQLException {
     Connection another = pool.getConnection("testing", "123");
     another.close();
   }
+
+  @Test
+  public void unwrapConnection() throws SQLException {
+
+    Connection connection = pool.getConnection();
+    Connection underlying = connection.unwrap(Connection.class);
+
+    assertThat(underlying).isInstanceOf(org.h2.jdbc.JdbcConnection.class);
+    connection.close();
+  }
+
+  @Test
+  public void getDelegate() throws SQLException {
+
+    Connection connection = pool.getConnection();
+    PooledConnection pc = (PooledConnection)connection;
+    Connection underlying = pc.getDelegate();
+
+    assertThat(underlying).isInstanceOf(org.h2.jdbc.JdbcConnection.class);
+    connection.close();
+  }
 }