Implements Entityql.

Author: orekyuu

doma-core/src/main/java/org/seasar/doma/jdbc/criteria/statement/EntityqlMultiInsertStatement.java

@@ -13,6 +13,7 @@
 import org.seasar.doma.jdbc.criteria.metamodel.PropertyMetamodel;
 import org.seasar.doma.jdbc.entity.EntityType;
 import org.seasar.doma.jdbc.query.AutoMultiInsertQuery;
+import org.seasar.doma.jdbc.query.DuplicateKeyType;
 import org.seasar.doma.jdbc.query.Query;
 
 public class EntityqlMultiInsertStatement<ENTITY>
@@ -23,6 +24,7 @@ public class EntityqlMultiInsertStatement<ENTITY>
   private final EntityMetamodel<ENTITY> entityMetamodel;
   private final List<ENTITY> entities;
   private final InsertSettings settings;
+  private DuplicateKeyType duplicateKeyType = DuplicateKeyType.EXCEPTION;
 
   public EntityqlMultiInsertStatement(
       Config config,
@@ -35,6 +37,26 @@ public EntityqlMultiInsertStatement(
     this.settings = Objects.requireNonNull(settings);
   }
 
+  /**
+   * Create statement that inserts or updates
+   *
+   * @return statement
+   */
+  public EntityqlMultiInsertStatement<ENTITY> onDuplicateKeyUpdate() {
+    this.duplicateKeyType = DuplicateKeyType.UPDATE;
+    return this;
+  }
+
+  /**
+   * Create statement that inserts or ignore
+   *
+   * @return statement
+   */
+  public EntityqlMultiInsertStatement<ENTITY> onDuplicateKeyIgnore() {
+    this.duplicateKeyType = DuplicateKeyType.IGNORE;
+    return this;
+  }
+
   /**
    * {@inheritDoc}
    *
@@ -64,6 +86,7 @@ protected Command<MultiResult<ENTITY>> createCommand() {
     query.setExcludedPropertyNames(
         settings.exclude().stream().map(PropertyMetamodel::getName).toArray(String[]::new));
     query.setMessage(settings.getComment());
+    query.setDuplicateKeyType(this.duplicateKeyType);
     query.prepare();
     InsertCommand command =
         config.getCommandImplementors().createInsertCommand(EXECUTE_METHOD, query);

integration-test-java/src/test/java/org/seasar/doma/it/criteria/EntityqlMultiInsertTest.java

@@ -7,6 +7,9 @@
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
+import java.util.Map;
+import java.util.function.Function;
+import java.util.stream.Collectors;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.seasar.doma.it.Dbms;
@@ -61,6 +64,105 @@ void test() {
     assertEquals(2, departments2.size());
   }
 
+  @Test
+  @Run(unless = {Dbms.ORACLE, Dbms.SQLSERVER, Dbms.H2})
+  void testIgnore() {
+    Department_ d = new Department_();
+
+    Department department = new Department();
+    department.setDepartmentId(99);
+    department.setDepartmentNo(99);
+    department.setDepartmentName("aaa");
+    department.setLocation("bbb");
+
+    Department department2 = new Department();
+    department2.setDepartmentId(100);
+    department2.setDepartmentNo(100);
+    department2.setDepartmentName("ccc");
+    department2.setLocation("ddd");
+
+    Department department3 = new Department();
+    department3.setDepartmentId(101);
+    department3.setDepartmentNo(101);
+    department3.setDepartmentName("eee");
+    department3.setLocation("fff");
+
+    entityql.insertMulti(d, Arrays.asList(department, department2)).execute();
+
+    department.setDepartmentName("aaa_updated");
+    department.setLocation("bbb_updated");
+
+    List<Department> departments = Arrays.asList(department, department3);
+    MultiResult<Department> result =
+        entityql.insertMulti(d, departments).onDuplicateKeyIgnore().execute();
+    assertEquals(departments, result.getEntities());
+    assertEquals(1, result.getCount());
+
+    List<Integer> ids = departments.stream().map(Department::getDepartmentId).collect(toList());
+
+    Map<Integer, Department> departments2 =
+        entityql
+            .from(d)
+            .where(c -> c.in(d.departmentId, ids))
+            .orderBy(c -> c.asc(d.departmentId))
+            .stream()
+            .collect(Collectors.toMap(Department::getDepartmentId, Function.identity()));
+    assertEquals(2, departments2.size());
+    assertEquals("aaa", departments2.get(99).getDepartmentName());
+  }
+
+  @Test
+  @Run(unless = {Dbms.ORACLE, Dbms.SQLSERVER, Dbms.H2})
+  void testUpdate() {
+    Department_ d = new Department_();
+
+    Department department = new Department();
+    department.setDepartmentId(99);
+    department.setDepartmentNo(99);
+    department.setDepartmentName("aaa");
+    department.setLocation("bbb");
+
+    Department department2 = new Department();
+    department2.setDepartmentId(100);
+    department2.setDepartmentNo(100);
+    department2.setDepartmentName("ccc");
+    department2.setLocation("ddd");
+
+    Department department3 = new Department();
+    department3.setDepartmentId(101);
+    department3.setDepartmentNo(101);
+    department3.setDepartmentName("eee");
+    department3.setLocation("fff");
+
+    entityql.insertMulti(d, Arrays.asList(department, department2)).execute();
+
+    department.setDepartmentName("aaa_updated");
+    department.setLocation("bbb_updated");
+
+    List<Department> departments = Arrays.asList(department, department3);
+    MultiResult<Department> result =
+        entityql.insertMulti(d, departments).onDuplicateKeyUpdate().execute();
+    assertEquals(departments, result.getEntities());
+    if (dialect.getName().equals("mysql") || dialect.getName().equals("mariadb")) {
+      assertEquals(3, result.getCount());
+    } else {
+      assertEquals(2, result.getCount());
+    }
+
+    List<Integer> ids = departments.stream().map(Department::getDepartmentId).collect(toList());
+
+    Map<Integer, Department> departments2 =
+        entityql
+            .from(d)
+            .where(c -> c.in(d.departmentId, ids))
+            .orderBy(c -> c.asc(d.departmentId))
+            .stream()
+            .collect(Collectors.toMap(Department::getDepartmentId, Function.identity()));
+    assertEquals(2, departments2.size());
+    assertEquals("aaa_updated", departments2.get(99).getDepartmentName());
+    assertEquals("eee", departments2.get(101).getDepartmentName());
+  }
+
   @Test
   @Run(unless = {Dbms.ORACLE})
   void empty() {