Support AWS JDBC Driver for MySQL (#126)

Author: nakamura-to

codegen/src/main/java/org/seasar/doma/gradle/codegen/util/JdbcUtil.java

@@ -18,7 +18,7 @@ public final class JdbcUtil {
 
   private static final org.slf4j.Logger logger = LoggerFactory.getLogger(JdbcUtil.class);
 
-  protected static final Pattern jdbcUrlPattern = Pattern.compile("jdbc:([^:]+):");
+  protected static final Pattern jdbcUrlPattern = Pattern.compile("jdbc:([^:]+):(([^:]+)?:)?");
 
   public static Connection getConnection(DataSource dataSource) {
     try {
@@ -77,8 +77,8 @@ public static void close(ResultSet resultSet) {
   public static String inferDialectName(String url) {
     return match(
         url,
-        name -> {
-          switch (name) {
+        names -> {
+          switch (names.getFirst()) {
             case "h2":
               return "h2";
             case "hsqldb":
@@ -105,15 +105,18 @@ public static String inferDialectName(String url) {
   public static String inferDriverClassName(String url) {
     return match(
         url,
-        name -> {
-          switch (name) {
+        names -> {
+          switch (names.getFirst()) {
             case "h2":
               return "org.h2.Driver";
             case "hsqldb":
               return "org.hsqldb.jdbc.JDBCDriver";
             case "sqlite":
               return "org.sqlite.JDBC";
             case "mysql":
+              if ("aws".equals(names.getSecond())) {
+                return "software.aws.rds.jdbc.mysql.Driver";
+              }
               return "com.mysql.cj.jdbc.Driver";
             case "mariadb":
               return "org.mariadb.jdbc.Driver";
@@ -131,14 +134,15 @@ public static String inferDriverClassName(String url) {
         });
   }
 
-  protected static <R> R match(String url, Function<String, R> mapper) {
+  protected static <R> R match(String url, Function<Pair<String, String>, R> mapper) {
     if (url == null) {
       throw new CodeGenNullPointerException("url");
     }
     Matcher matcher = jdbcUrlPattern.matcher(url);
     if (matcher.lookingAt()) {
-      String name = matcher.group(1);
-      return mapper.apply(name);
+      String first = matcher.group(1);
+      String second = matcher.group(3);
+      return mapper.apply(new Pair<>(first, second));
     }
     return null;
   }

codegen/src/test/java/org/seasar/doma/gradle/codegen/util/JdbcUtilTest.java

@@ -19,6 +19,14 @@ public void testInferDialectName_mysql() throws Exception {
     assertEquals("mysql", dialectName);
   }
 
+  @Test
+  public void testInferDialectName_mysql_aws() throws Exception {
+    String dialectName =
+        JdbcUtil.inferDialectName(
+            "jdbc:mysql:aws://db-identifier.cluster-XYZ.us-east-2.rds.amazonaws.com:3306");
+    assertEquals("mysql", dialectName);
+  }
+
   @Test
   public void testInferDialectName_unknown() throws Exception {
     String dialectName = JdbcUtil.inferDialectName("jdbc:unknown://localhost/hoge");
@@ -43,6 +51,14 @@ public void testInferDriverClassName_mysql() throws Exception {
     assertEquals("com.mysql.cj.jdbc.Driver", driverClassName);
   }
 
+  @Test
+  public void testInferDriverClassName_mysql_aws() throws Exception {
+    String driverClassName =
+        JdbcUtil.inferDriverClassName(
+            "jdbc:mysql:aws://db-identifier.cluster-XYZ.us-east-2.rds.amazonaws.com:3306");
+    assertEquals("software.aws.rds.jdbc.mysql.Driver", driverClassName);
+  }
+
   @Test
   public void testInferDriverClassName_mariadb() throws Exception {
     String driverClassName = JdbcUtil.inferDriverClassName("jdbc:mariadb://localhost:3306/hoge");