extensions.sql.SqlExtensions.kt Maven / Gradle / Ivy

Go to download
/*
 *  Copyright (c) 2023-Present, Mybatis-Flex-Kotlin ([email protected]).
 *  
 *  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.
 */
@file:Suppress("unused")

package com.mybatisflex.kotlin.extensions.sql

import com.mybatisflex.core.query.*
import com.mybatisflex.kotlin.extensions.condition.and
import com.mybatisflex.kotlin.extensions.condition.emptyCondition
import com.mybatisflex.kotlin.extensions.kproperty.column
import com.mybatisflex.kotlin.scope.QueryScope
import com.mybatisflex.kotlin.scope.queryScope
import com.mybatisflex.kotlin.vec.Order
import java.util.function.Consumer
import kotlin.reflect.KProperty

/*
 * sql操作扩展
 * @author KAMOsama
 * @date 2023/8/7
 */

//like------
infix fun QueryColumn.like(value: Any): QueryCondition = this.like(value)

infix fun QueryColumn.likeRaw(value: Any): QueryCondition = this.likeRaw(value)

infix fun QueryColumn.likeLeft(value: Any): QueryCondition = this.likeLeft(value)

infix fun QueryColumn.likeRight(value: Any): QueryCondition = this.likeRight(value)

infix fun QueryColumn.notLike(other: Any): QueryCondition = this.notLike(other)

infix fun QueryColumn.notLikeRaw(other: Any): QueryCondition = this.notLikeRaw(other)

infix fun QueryColumn.notLikeLeft(other: Any): QueryCondition = this.notLikeLeft(other)

infix fun QueryColumn.notLikeRight(other: Any): QueryCondition = this.notLikeRight(other)

//comparable
infix fun QueryColumn.eq(value: Any?): QueryCondition = this.eq(value)

@Deprecated("使用 eq 和 null 进行比较可能是个错误。", ReplaceWith("this.isNull"))
infix fun QueryColumn.eq(value: Nothing?): QueryCondition = this.eq(value)

infix fun QueryColumn.ne(value: Any?): QueryCondition = this.ne(value)

@Deprecated("使用 ne 和 null 进行比较可能是个错误。", ReplaceWith("this.isNull"))
infix fun QueryColumn.ne(value: Nothing?): QueryCondition = this.ne(value)

infix fun QueryColumn.gt(value: Any?): QueryCondition = this.gt(value)

infix fun QueryColumn.ge(value: Any?): QueryCondition = this.ge(value)

infix fun QueryColumn.le(value: Any?): QueryCondition = this.le(value)

infix fun QueryColumn.lt(value: Any?): QueryCondition = this.lt(value)

//range
infix fun QueryColumn.between(pair: Pair): QueryCondition = this.between(pair.first, pair.second)

infix fun QueryColumn.between(range: ClosedRange<*>): QueryCondition = this.between(range.start, range.endInclusive)

infix fun QueryColumn.notBetween(pair: Pair): QueryCondition = this.notBetween(pair.first, pair.second)

infix fun QueryColumn.notBetween(range: ClosedRange<*>): QueryCondition =
    this.notBetween(range.start, range.endInclusive)

infix fun QueryColumn.notIn(value: Collection): QueryCondition = this.notIn(value)

infix fun QueryColumn.notIn(values: Array): QueryCondition = this.notIn(values)

/**
 * @since 1.1.0
 */
inline fun QueryColumn.notIn(scope:QueryScope.()->Unit): QueryCondition = this.notIn(queryScope(init = scope))

infix fun QueryColumn.`in`(value: Collection): QueryCondition = this.`in`(value)

infix fun QueryColumn.`in`(values: Array): QueryCondition = this.`in`(values.toList())

infix fun QueryColumn.`in`(range: IntRange): QueryCondition = this.`in`(range.toList())

/**
 * @since 1.1.0
 */
infix fun QueryColumn.inList(other: Collection>): QueryCondition {
    if (other.isEmpty()) {
        return emptyCondition()
    }
    return if (other.size == 1) this.eq(other.first()) else this.`in`(other)
}

/**
 * @since 1.1.0
 */
infix fun QueryColumn.inArray(other: Array>): QueryCondition {
    if (other.isEmpty()) {
        return emptyCondition()
    }
    return if (other.size == 1) this.eq(other[0]) else this.`in`(other)
}

/**
 * @since 1.1.0
 */
infix fun  QueryColumn.inRange(other: ClosedRange>): QueryCondition =
    if (other.endInclusive == other.start) this.eq(other.start) else this.between(
        other.start,
        other.endInclusive
    )

/**
 * @since 1.1.1
 */
infix fun QueryColumn.`as`(alias: String): QueryColumn = this.`as`(alias)

/**
 * @since 1.1.0
 */
inline fun QueryColumn.`in`(scope:QueryScope.()->Unit): QueryCondition = this.`in`(queryScope(init = scope))

fun  Pair.inPair(vararg others: Pair): QueryCondition =
    this inPair others.toList()

infix fun  Pair.inPair(others: Iterable>): QueryCondition =
    others.map { this.first.eq(it.first) and this.second.eq(it.second) }
        .reduceIndexed { i, c1, c2 -> (if (i == 1) Brackets(c1) else c1).or(c2) }

fun  Pair, C>.inTriple(vararg others: Pair, A>): QueryCondition =
    this inTriple others.toList()

infix fun  Pair, C>.inTriple(others: Iterable, A>>): QueryCondition =
    others.map { this.first.first.eq(it.first.first) and this.first.second.eq(it.first.second) and this.second.eq(it.second) }
        .reduceIndexed { i, c1, c2 -> (if (i == 1) Brackets(c1) else c1).or(c2) }

//join
@Deprecated(
    "核心库已废除 Joiner.as 方法。详情请看官方库描述。",
    replaceWith = ReplaceWith("this.`as`(alias)"),
    level = DeprecationLevel.ERROR,
)
infix fun  Joiner.`as`(alias: String?): Joiner = this.`as`(alias)

infix fun  Joiner.on(on: String?): W = this.on(on)

infix fun  Joiner.on(on: QueryCondition?): W = this.on(on)

infix fun  Joiner.on(consumer: Consumer): W = this.on(consumer)

operator fun QueryColumn.unaryPlus(): QueryOrderBy = this.asc()

operator fun QueryColumn.unaryMinus(): QueryOrderBy = this.desc()

// operator
operator fun QueryColumn.plus(other: QueryColumn): QueryColumn = add(other)

operator fun QueryColumn.plus(other: Number): QueryColumn = add(other)

operator fun QueryColumn.plus(other: KProperty): QueryColumn = add(other.column)

operator fun QueryColumn.minus(other: QueryColumn): QueryColumn = subtract(other)

operator fun QueryColumn.minus(other: Number): QueryColumn = subtract(other)

operator fun QueryColumn.minus(other: KProperty): QueryColumn = subtract(other.column)

operator fun QueryColumn.times(other: QueryColumn): QueryColumn = multiply(other)

operator fun QueryColumn.times(other: Number): QueryColumn = multiply(other)

operator fun QueryColumn.times(other: KProperty): QueryColumn = multiply(other.column)

operator fun QueryColumn.div(other: QueryColumn): QueryColumn = divide(other)

operator fun QueryColumn.div(other: Number): QueryColumn = divide(other)

operator fun QueryColumn.div(other: KProperty): QueryColumn = divide(other.column)

// order
fun QueryColumn.toOrd(order: Order = Order.ASC): QueryOrderBy = when (order) {
    Order.ASC -> asc()
    Order.DESC -> desc()
}

operator fun QueryCondition.not(): QueryCondition = QueryMethods.not(this)