Merge pull request #1146 from momosetkn/feat/doma-kotlin-insertMulti

Support upsert with insertMulti in doma-kotlin

Author: nakamura-to

doma-kotlin/src/main/kotlin/org/seasar/doma/kotlin/jdbc/criteria/statement/KEntityqlMultiInsertStatement.kt

@@ -8,6 +8,14 @@ class KEntityqlMultiInsertStatement<ENTITY>(
     private val statement: EntityqlMultiInsertStatement<ENTITY>,
 ) : KStatement<MultiResult<ENTITY>> {
 
+    fun onDuplicateKeyUpdate(): KStatement<MultiResult<ENTITY>> {
+        return KEntityqlMultiUpsertStatement(statement.onDuplicateKeyUpdate())
+    }
+
+    fun onDuplicateKeyIgnore(): KStatement<MultiResult<ENTITY>> {
+        return KEntityqlMultiUpsertStatement(statement.onDuplicateKeyIgnore())
+    }
+
     override fun execute(): MultiResult<ENTITY> {
         return statement.execute()
     }

doma-kotlin/src/test/kotlin/org/seasar/doma/jdbc/criteria/KEntityqlInsertMultiTest.kt

@@ -0,0 +1,64 @@
+package org.seasar.doma.jdbc.criteria
+
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.Test
+import org.seasar.doma.jdbc.criteria.entity.Emp
+import org.seasar.doma.jdbc.criteria.entity.Emp_
+import org.seasar.doma.jdbc.criteria.mock.MockConfig
+import java.math.BigDecimal
+
+internal class KEntityqlInsertMultiTest {
+    private val config = MockConfig().apply {
+        dialect = org.seasar.doma.jdbc.dialect.PostgresDialect()
+    }
+
+    private val entityql = org.seasar.doma.kotlin.jdbc.criteria.KEntityql(config)
+
+    private val items = listOf(
+        Emp().apply {
+            id = 1
+            name = "aaa"
+            salary = BigDecimal("1000")
+            version = 1
+        },
+        Emp().apply {
+            id = 2
+            name = "eee"
+            salary = BigDecimal("2000")
+            version = 2
+        },
+    )
+
+    @Test
+    fun insertMultiMultiInto() {
+        val e = Emp_()
+        val stmt = entityql.insertMulti(e, items)
+        val sql = stmt.asSql()
+        Assertions.assertEquals(
+            "insert into EMP (ID, NAME, SALARY, VERSION) values (1, 'aaa', 1000, 1), (2, 'eee', 2000, 2)",
+            sql.formattedSql,
+        )
+    }
+
+    @Test
+    fun insertMultiMultiOnDuplicateKeyUpdate() {
+        val e = Emp_()
+        val stmt = entityql.insertMulti(e, items).onDuplicateKeyUpdate()
+        val sql = stmt.asSql()
+        Assertions.assertEquals(
+            "insert into EMP as target (ID, NAME, SALARY, VERSION) values (1, 'aaa', 1000, 1), (2, 'eee', 2000, 2) on conflict (ID) do update set NAME = excluded.NAME, SALARY = excluded.SALARY, VERSION = excluded.VERSION",
+            sql.formattedSql,
+        )
+    }
+
+    @Test
+    fun insertMultiOnDuplicateKeyIgnore() {
+        val e = Emp_()
+        val stmt = entityql.insertMulti(e, items).onDuplicateKeyIgnore()
+        val sql = stmt.asSql()
+        Assertions.assertEquals(
+            "insert into EMP as target (ID, NAME, SALARY, VERSION) values (1, 'aaa', 1000, 1), (2, 'eee', 2000, 2) on conflict do nothing",
+            sql.formattedSql,
+        )
+    }
+}