org.mybatis.dynamic.sql.util.kotlin.KotlinUpdateBuilder.kt Maven / Gradle / Ivy

Go to download
/*
 *    Copyright ${license.git.copyrightYears} the original author or 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
 *
 *       http://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.mybatis.dynamic.sql.util.kotlin

import org.mybatis.dynamic.sql.BasicColumn
import org.mybatis.dynamic.sql.SqlColumn
import org.mybatis.dynamic.sql.update.UpdateDSL
import org.mybatis.dynamic.sql.update.UpdateModel
import org.mybatis.dynamic.sql.util.Buildable

typealias UpdateCompleter = KotlinUpdateBuilder.() -> Unit

class KotlinUpdateBuilder(private val dsl: UpdateDSL) :
    KotlinBaseBuilder, KotlinUpdateBuilder>(), Buildable {

    fun  set(column: SqlColumn): KotlinSetClauseFinisher = KotlinSetClauseFinisher(column)

    override fun build(): UpdateModel = dsl.build()

    override fun getDsl(): UpdateDSL = dsl

    override fun self(): KotlinUpdateBuilder = this

    @MyBatisDslMarker
    @Suppress("TooManyFunctions")
    inner class KotlinSetClauseFinisher(private val column: SqlColumn) {
        fun equalToNull(): KotlinUpdateBuilder =
            applyToDsl {
                set(column).equalToNull()
            }

        fun equalToConstant(constant: String): KotlinUpdateBuilder =
            applyToDsl {
                set(column).equalToConstant(constant)
            }

        fun equalToStringConstant(constant: String): KotlinUpdateBuilder =
            applyToDsl {
                set(column).equalToStringConstant(constant)
            }

        fun equalTo(value: T): KotlinUpdateBuilder = equalTo { value }

        fun equalTo(value: () -> T): KotlinUpdateBuilder =
            applyToDsl {
                set(column).equalTo(value)
            }

        fun equalTo(rightColumn: BasicColumn): KotlinUpdateBuilder =
            applyToDsl {
                set(column).equalTo(rightColumn)
            }

        fun equalToOrNull(value: T?): KotlinUpdateBuilder = equalToOrNull { value }

        fun equalToOrNull(value: () -> T?): KotlinUpdateBuilder =
            applyToDsl {
                set(column).equalToOrNull(value)
            }

        fun equalToQueryResult(subQuery: KotlinSubQueryBuilder.() -> Unit): KotlinUpdateBuilder =
            applyToDsl {
                set(column).equalTo(KotlinSubQueryBuilder().apply(subQuery))
            }

        fun equalToWhenPresent(value: () -> T?): KotlinUpdateBuilder =
            applyToDsl {
                set(column).equalToWhenPresent(value)
            }

        fun equalToWhenPresent(value: T?): KotlinUpdateBuilder = equalToWhenPresent { value }

        private fun applyToDsl(block: UpdateDSL.() -> Unit): KotlinUpdateBuilder =
            [email protected] {
                dsl.apply(block)
            }
    }
}