Sample implementation of DB2 trusted context

Author: rPraml

ebean-datasource/pom.xml

@@ -26,6 +26,13 @@
       <scope>test</scope>
     </dependency>
 
+    <dependency>
+      <groupId>org.junit.jupiter</groupId>
+      <artifactId>junit-jupiter-params</artifactId>
+      <version>5.10.2</version>
+      <scope>test</scope>
+    </dependency>
+
     <dependency>
       <groupId>io.ebean</groupId>
       <artifactId>ebean-test-containers</artifactId>

ebean-datasource/src/test/java/io/ebean/datasource/tcdriver/TrustedContextListener.java

@@ -0,0 +1,27 @@
+package io.ebean.datasource.tcdriver;
+
+import io.ebean.datasource.DataSourceConnection;
+import io.ebean.datasource.DataSourcePool;
+import io.ebean.datasource.DataSourcePoolListener;
+
+import java.sql.SQLException;
+
+/**
+ * Listener, that sets up TrustedConnection properly
+ *
+ * @author Roland Praml, Foconis Analytics GmbH
+ */
+public class TrustedContextListener implements DataSourcePoolListener {
+
+    @Override
+    public void onAfterBorrowConnection(DataSourcePool pool, DataSourceConnection connection) throws SQLException {
+        TrustedContextTenant tenant = (TrustedContextTenant) connection.affinityId();
+        TrustedDb2Connection trustedDb2Connection = connection.unwrap(TrustedDb2Connection.class);
+        if (trustedDb2Connection.switchUser(tenant.user(), tenant.password())) {
+            trustedDb2Connection.setSchema(tenant.schema());
+            connection.clearPreparedStatementCache();
+        }
+    }
+
+
+}

ebean-datasource/src/test/java/io/ebean/datasource/tcdriver/TrustedContextTenant.java

@@ -0,0 +1,35 @@
+package io.ebean.datasource.tcdriver;
+
+/**
+ * @author Roland Praml, Foconis Analytics GmbH
+ */
+public class TrustedContextTenant {
+  private final int id;
+  private final String user;
+  private final String password;
+  private final String schema;
+
+  public TrustedContextTenant(int id, String user, String password, String schema) {
+    this.id = id;
+    this.user = user;
+    this.password = password;
+    this.schema = schema;
+  }
+
+  public int id() {
+    return id;
+  }
+
+  public String user() {
+    return user;
+  }
+
+  public String password() {
+    return password;
+  }
+
+  public String schema() {
+    return schema;
+  }
+
+}