add driverClass option

testcontainers · Jan 14, 2024 · 7011359 · 7011359
1 parent d08e9b7
commit 7011359
Showing 1 changed file with 26 additions and 2 deletions.
diff --git a/modules/oceanbase-ce/src/main/java/org/testcontainers/containers/OceanBaseContainer.java b/modules/oceanbase-ce/src/main/java/org/testcontainers/containers/OceanBaseContainer.java
@@ -34,6 +34,7 @@ public class OceanBaseContainer extends JdbcDatabaseContainer<OceanBaseContainer
     private boolean enableFastboot;
     private String mode;
     private String tenantName = DEFAULT_TEST_TENANT_NAME;
+    private String driverClassName = "com.mysql.cj.jdbc.Driver";
 
     public OceanBaseContainer(String dockerImageName) {
         this(DockerImageName.parse(dockerImageName));
@@ -53,7 +54,7 @@ public Integer getMappedPort(int originalPort) {
 
     @Override
     public String getDriverClassName() {
-        return "com.mysql.cj.jdbc.Driver";
+        return driverClassName;
     }
 
     @Override
@@ -63,7 +64,8 @@ public String getJdbcUrl() {
 
     public String getJdbcUrl(String databaseName) {
         String additionalUrlParams = constructUrlParameters("?", "&");
-        return "jdbc:mysql://" + getHost() + ":" + getMappedPort(SQL_PORT) + "/" + databaseName + additionalUrlParams;
+        String prefix = driverClassName.contains("mysql") ? "jdbc:mysql://" : "jdbc:oceanbase://";
+        return prefix + getHost() + ":" + getMappedPort(SQL_PORT) + "/" + databaseName + additionalUrlParams;
     }
 
     @Override
@@ -124,6 +126,28 @@ public OceanBaseContainer withTenant(String tenantName) {
         return self();
     }
 
+    /**
+     * Set the driver class name.
+     *
+     * @param driverClassName the driver class name
+     * @return this
+     */
+    public OceanBaseContainer withDriverClassName(String driverClassName) {
+        if (StringUtils.isEmpty(driverClassName)) {
+            throw new IllegalArgumentException("Driver class name cannot be null or empty");
+        }
+        if (!driverClassName.contains("mysql") && !driverClassName.contains("oceanbase")) {
+            throw new IllegalArgumentException("Driver class name should contains 'mysql' or 'oceanbase'");
+        }
+        try {
+            Class.forName(driverClassName);
+        } catch (ClassNotFoundException e) {
+            throw new IllegalArgumentException("Driver class not found", e);
+        }
+        this.driverClassName = driverClassName;
+        return self();
+    }
+
     @Override
     protected void configure() {
         if (StringUtils.isNotBlank(mode)) {