Support UserDefinedCriteria for kotlin

Author: momosetkn

doma-kotlin/src/main/kotlin/org/seasar/doma/kotlin/jdbc/criteria/declaration/KComparisonDeclaration.kt

@@ -15,7 +15,9 @@
  */
 package org.seasar.doma.kotlin.jdbc.criteria.declaration
 
+import org.seasar.doma.jdbc.criteria.declaration.UserDefinedCriteriaContext
 import org.seasar.doma.jdbc.criteria.metamodel.PropertyMetamodel
+import org.seasar.doma.jdbc.dialect.Dialect
 
 abstract class KComparisonDeclaration<DECLARATION : org.seasar.doma.jdbc.criteria.declaration.ComparisonDeclaration>(protected val declaration: DECLARATION) {
 
@@ -94,4 +96,29 @@ abstract class KComparisonDeclaration<DECLARATION : org.seasar.doma.jdbc.criteri
     fun not(block: () -> Unit) {
         declaration.not(block)
     }
+
+    fun <EXTENSION> extension(
+        construct: (KUserDefinedCriteriaContext) -> EXTENSION,
+        extensionDeclaration: EXTENSION.() -> Unit,
+    ) {
+        declaration.extension(
+            { context -> construct(context.toKotlin()) },
+            extensionDeclaration
+        )
+    }
+
+    private fun UserDefinedCriteriaContext.toKotlin(): KUserDefinedCriteriaContext {
+        return KUserDefinedCriteriaContext { builderBlock ->
+            this.add { builder ->
+                builderBlock(object : KUserDefinedCriteriaContext.Builder {
+                    override fun appendSql(sql: String) = builder.appendSql(sql)
+                    override fun cutBackSql(length: Int) = builder.cutBackSql(length)
+                    override fun appendExpression(propertyMetamodel: PropertyMetamodel<*>) =
+                        builder.appendExpression(propertyMetamodel)
+                    override val dialect: Dialect
+                        get() = builder.dialect
+                })
+            }
+        }
+    }
 }

doma-kotlin/src/main/kotlin/org/seasar/doma/kotlin/jdbc/criteria/declaration/KUserDefinedCriteriaContext.kt

@@ -0,0 +1,33 @@
+/*
+ * Copyright Doma Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.seasar.doma.kotlin.jdbc.criteria.declaration
+
+import org.seasar.doma.jdbc.criteria.metamodel.PropertyMetamodel
+import org.seasar.doma.jdbc.dialect.Dialect
+
+fun interface KUserDefinedCriteriaContext {
+    fun add(builderBlock: Builder.() -> Unit)
+
+    interface Builder {
+        fun appendSql(sql: String)
+
+        fun cutBackSql(length: Int)
+
+        fun appendExpression(propertyMetamodel: PropertyMetamodel<*>)
+
+        val dialect: Dialect
+    }
+}

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

@@ -687,6 +687,26 @@ internal class KQueryDslSqlSelectTest {
         )
     }
 
+    @Test
+    fun where_extension() {
+        val e = Emp_()
+        val stmt = dsl
+            .from(e)
+            .where {
+                eq(e.id, 1)
+                extension(::MyExtension) {
+                    likeMultiple(e.name, "A", "B", "C")
+                    eq(e.id, 1)
+                }
+            }
+            .select(e.id)
+        val sql = stmt.asSql()
+        assertEquals(
+            "select t0_.ID from EMP t0_ where t0_.ID = 1 and t0_.NAME like '%A%' escape '\\' and t0_.NAME like '%B%' escape '\\' and t0_.NAME like '%C%' escape '\\' and t0_.ID = 1",
+            sql.formattedSql,
+        )
+    }
+
     @Test
     fun innerJoin() {
         val e = Emp_()

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

@@ -0,0 +1,35 @@
+/*
+ * Copyright Doma Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.seasar.doma.jdbc.criteria
+
+import org.seasar.doma.jdbc.criteria.metamodel.PropertyMetamodel
+import org.seasar.doma.kotlin.jdbc.criteria.declaration.KUserDefinedCriteriaContext
+import org.seasar.doma.kotlin.jdbc.criteria.expression.KExpressions
+
+data class MyExtension(val context: KUserDefinedCriteriaContext) {
+    fun likeMultiple(entityMetamodel: PropertyMetamodel<*>, vararg patterns: String) {
+        context.add {
+            patterns.forEach { pattern ->
+                appendExpression(entityMetamodel)
+                appendSql(" like ")
+                appendExpression(KExpressions.literal("%" + pattern + "%"))
+                appendSql(" escape '\\'")
+                appendSql(" and ")
+            }
+            cutBackSql(5)
+        }
+    }
+}