feat: add hibernate dialects for mysql if user has not set by default (…

…#161) Signed-off-by: Sarthak160 <rocksarthak45@gmail.com>
keploy · Dec 6, 2023 · f9225f2 · f9225f2
1 parent d9e5be5
commit f9225f2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/agent/src/main/java/io/keploy/advice/ksql/RegisterDriverAdvice.java b/agent/src/main/java/io/keploy/advice/ksql/RegisterDriverAdvice.java
@@ -26,6 +26,7 @@ public static void enterMethod(@Advice.Origin Method method, @Advice.Argument(va
                     KDriver.Dialect = "org.hibernate.dialect.PostgreSQLDialect";
                     break;
                 case "com.mysql.cj.jdbc.Driver":
+                case "com.mysql.jdbc.Driver":
                     KDriver.Dialect = "org.hibernate.dialect.MySQLDialect";
                     break;
                 case "oracle.jdbc.driver.OracleDriver":

diff --git a/agent/src/main/java/io/keploy/advice/ksql/RegisterDriverAdvice_Interceptor.java b/agent/src/main/java/io/keploy/advice/ksql/RegisterDriverAdvice_Interceptor.java
@@ -33,6 +33,7 @@ public static String execute(@SuperCall Callable<String> client, @Origin Method
                     KDriver.Dialect = "org.hibernate.dialect.PostgreSQLDialect";
                     break;
                 case "com.mysql.cj.jdbc.Driver":
+                case "com.mysql.jdbc.Driver":
                     KDriver.Dialect = "org.hibernate.dialect.MySQLDialect";
                     break;
                 case "oracle.jdbc.driver.OracleDriver":

diff --git a/integration/src/main/java/io/keploy/ksql/KDatabaseMetaData.java b/integration/src/main/java/io/keploy/ksql/KDatabaseMetaData.java
@@ -82,6 +82,7 @@ public String getDatabaseProductName() throws SQLException {
             case "org.postgresql.Driver":
                 return "PostgreSQL";
             case "com.mysql.cj.jdbc.Driver":
+            case "com.mysql.jdbc.Driver":
                 return "MySQL";
             case "org.h2.Driver":
                 return "H2";