Add Kotlin Support for new select items

Author: jeffgbutler

src/main/kotlin/org/mybatis/dynamic/sql/util/kotlin/KotlinSelectBuilder.kt

@@ -76,6 +76,30 @@ class KotlinSelectBuilder(private val fromGatherer: QueryExpressionDSL.FromGathe
     fun unionAll(unionAll: KotlinUnionBuilder.() -> Unit): Unit =
         unionAll(KotlinUnionBuilder(getDsl().unionAll()))
 
+    fun forUpdate() {
+        getDsl().forUpdate()
+    }
+
+    fun forNoKeyUpdate() {
+        getDsl().forNoKeyUpdate()
+    }
+
+    fun forShare() {
+        getDsl().forShare()
+    }
+
+    fun forKeyShare() {
+        getDsl().forKeyShare()
+    }
+
+    fun skipLocked() {
+        getDsl().skipLocked()
+    }
+
+    fun nowait() {
+        getDsl().nowait()
+    }
+
     override fun build(): SelectModel = getDsl().build()
 
     override fun getDsl(): KQueryExpressionDSL = invalidIfNull(dsl, "ERROR.27") //$NON-NLS-1$

src/test/kotlin/org/mybatis/dynamic/sql/util/kotlin/model/ModelBuilderTest.kt

@@ -46,4 +46,50 @@ class ModelBuilderTest {
 
         assertThat(provider.selectStatement).isEqualTo("select distinct id, description from Table where id = :p1")
     }
+
+    @Test
+    fun testSelectBuilderForUpdate() {
+        val provider = select(id, description) {
+            from(table)
+            where { id isEqualTo 3 }
+            forUpdate()
+            skipLocked()
+        }.render(RenderingStrategies.SPRING_NAMED_PARAMETER)
+
+        assertThat(provider.selectStatement).isEqualTo("select id, description from Table where id = :p1 for update skip locked")
+    }
+
+    @Test
+    fun testSelectBuilderForShare() {
+        val provider = select(id, description) {
+            from(table)
+            where { id isEqualTo 3 }
+            forShare()
+            nowait()
+        }.render(RenderingStrategies.SPRING_NAMED_PARAMETER)
+
+        assertThat(provider.selectStatement).isEqualTo("select id, description from Table where id = :p1 for share nowait")
+    }
+
+    @Test
+    fun testSelectBuilderForKeyShare() {
+        val provider = select(id, description) {
+            from(table)
+            where { id isEqualTo 3 }
+            forKeyShare()
+        }.render(RenderingStrategies.SPRING_NAMED_PARAMETER)
+
+        assertThat(provider.selectStatement).isEqualTo("select id, description from Table where id = :p1 for key share")
+    }
+
+    @Test
+    fun testSelectBuilderForKeyNoKeyUpdate() {
+        val provider = select(id, description) {
+            from(table)
+            where { id isEqualTo 3 }
+            forNoKeyUpdate()
+        }.render(RenderingStrategies.SPRING_NAMED_PARAMETER)
+
+        assertThat(provider.selectStatement).isEqualTo("select id, description from Table where id = :p1 for no key update")
+    }
 }