Improve batch insertion for entities with auto-increment primary keys.

Retrieve primary keys using `Statement.getGeneratedKeys()` after executing `Statement.executeBatch()`, if supported by the DBMS.

Author: nakamura-to

doma-core/src/main/java/org/seasar/doma/jdbc/command/BatchInsertCommand.java

@@ -46,4 +46,9 @@ protected int executeUpdate(PreparedStatement preparedStatement, PreparedSql sql
       throw e;
     }
   }
+
+  @Override
+  protected void postExecuteBatch(PreparedStatement preparedStatement, int position, int length) {
+    query.generateIds(preparedStatement, position, length);
+  }
 }

doma-core/src/main/java/org/seasar/doma/jdbc/command/BatchModifyCommand.java

@@ -96,6 +96,7 @@ protected int[] executeBatch(PreparedStatement preparedStatement, List<PreparedS
       if (i == sqlSize - 1 || (batchSize > 0 && (i + 1) % batchSize == 0)) {
         int[] rows = executeBatch(preparedStatement, sql);
         validateRows(preparedStatement, sql, rows);
+        postExecuteBatch(preparedStatement, pos, rows.length);
         System.arraycopy(rows, 0, updatedRows, pos, rows.length);
         pos = i + 1;
       }
@@ -118,6 +119,9 @@ protected int[] executeBatch(PreparedStatement preparedStatement, PreparedSql sq
     }
   }
 
+  protected void postExecuteBatch(
+      PreparedStatement preparedStatement, int position, int length) {}
+
   protected void log(PreparedSql sql) {
     JdbcLogger logger = query.getConfig().getJdbcLogger();
     logger.logSql(query.getClassName(), query.getMethodName(), sql);

doma-core/src/main/java/org/seasar/doma/jdbc/dialect/Dialect.java

@@ -129,6 +129,7 @@ default boolean supportsBatchExecutionReturningGeneratedValues() {
    *
    * @return {@code true}, if this object supports it
    */
+  @Deprecated
   boolean supportsIdentityReservation();
 
   /**
@@ -218,6 +219,7 @@ Sql<?> getIdentitySelectSql(
    * @throws DomaNullPointerException if either the {@code tableName} or the {@code columnName} is
    *     {@code null}
    */
+  @Deprecated
   Sql<?> getIdentityReservationSql(
       String catalogName,
       String schemaName,

doma-core/src/main/java/org/seasar/doma/jdbc/dialect/H2Dialect.java

@@ -47,6 +47,11 @@ public boolean supportsUpsertEmulationWithMergeStatement() {
     return true;
   }
 
+  @Override
+  public boolean supportsBatchExecutionReturningGeneratedValues() {
+    return true;
+  }
+
   public static class H2JdbcMappingVisitor extends H214199JdbcMappingVisitor {}
 
   public static class H2SqlLogFormattingVisitor extends H214199SqlLogFormattingVisitor {}

doma-core/src/main/java/org/seasar/doma/jdbc/dialect/MysqlDialect.java

@@ -156,6 +156,11 @@ public boolean supportsAliasInDeleteClause() {
     return true;
   }
 
+  @Override
+  public boolean supportsBatchExecutionReturningGeneratedValues() {
+    return true;
+  }
+
   @Override
   protected SqlNode toCountCalculatingSqlNode(SqlNode sqlNode) {
     switch (version) {

doma-core/src/main/java/org/seasar/doma/jdbc/dialect/PostgresDialect.java

@@ -227,6 +227,11 @@ public boolean supportsAutoGeneratedKeys() {
     return true;
   }
 
+  @Override
+  public boolean supportsBatchExecutionReturningGeneratedValues() {
+    return true;
+  }
+
   @Override
   public JdbcType<ResultSet> getResultSetType() {
     return RESULT_SET;

doma-core/src/main/java/org/seasar/doma/jdbc/id/BuiltinIdentityIdGenerator.java

@@ -19,39 +19,26 @@ public class BuiltinIdentityIdGenerator extends AbstractIdGenerator implements I
 
   @Override
   public boolean supportsBatch(IdGenerationConfig config) {
-    return config.getIdProvider().isAvailable();
+    return config.getDialect().supportsBatchExecutionReturningGeneratedValues();
   }
 
   @Override
   public boolean includesIdentityColumn(IdGenerationConfig config) {
-    if (config.getIdProvider().isAvailable()) {
-      return true;
-    }
     return config.getDialect().includesIdentityColumn();
   }
 
   @Override
   public boolean supportsAutoGeneratedKeys(IdGenerationConfig config) {
-    if (config.getIdProvider().isAvailable()) {
-      return false;
-    }
     return config.getDialect().supportsAutoGeneratedKeys();
   }
 
   @Override
   public Long generatePreInsert(IdGenerationConfig config) {
-    IdProvider idProvider = config.getIdProvider();
-    if (idProvider.isAvailable()) {
-      return idProvider.get();
-    }
     return null;
   }
 
   @Override
   public Long generatePostInsert(IdGenerationConfig config, Statement statement) {
-    if (config.getIdProvider().isAvailable()) {
-      return null;
-    }
     if (config.getDialect().supportsAutoGeneratedKeys()) {
       return getGeneratedValue(config, statement);
     }

doma-core/src/main/java/org/seasar/doma/jdbc/id/IdGenerationConfig.java

@@ -74,6 +74,7 @@ public IdProvider getIdProvider() {
     return idProvider;
   }
 
+  @Deprecated
   protected static class UnavailableIdProvider implements IdProvider {
     @Override
     public boolean isAvailable() {

doma-core/src/main/java/org/seasar/doma/jdbc/id/IdProvider.java

@@ -1,6 +1,7 @@
 package org.seasar.doma.jdbc.id;
 
 /** An identity provider. */
+@Deprecated
 public interface IdProvider {
 
   /**

doma-core/src/main/java/org/seasar/doma/jdbc/id/ReservedIdProvider.java

@@ -17,6 +17,7 @@
 import org.seasar.doma.message.Message;
 
 /** An identity provider that reserves identity values in advance. */
+@Deprecated
 public class ReservedIdProvider implements IdProvider {
 
   protected final Config config;