visitor.outputvisitor.OracleOutPutVisitor.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of easysql Show documentation
Kotlin ORM
There is a newer version: 1.1.10
package visitor.outputvisitor

import ast.expr.SqlIdentifierExpr
import ast.expr.SqlPropertyExpr
import ast.limit.SqlLimit
import ast.statement.select.SqlWithItem
import ast.statement.select.SqlWithSelect
import ast.statement.upsert.SqlUpsert

class OracleOutPutVisitor : SqlOutPutVisitor() {
    override fun visitSqlLimit(sqlLimit: SqlLimit) {
        sqlBuilder.append("OFFSET ${sqlLimit.offset} ROWS FETCH FIRST ${sqlLimit.limit} ROWS ONLY")
    }

    override fun printWithRecursive() {}

    override fun visitSqlUpsert(sqlUpsert: SqlUpsert) {
        sqlBuilder.append("MERGE INTO ")
        visitSqlExpr(sqlUpsert.table!!)
        sqlBuilder.append(" ${quote}t1${quote}")

        sqlBuilder.append(" USING (")
        sqlBuilder.append("SELECT ")
        sqlUpsert.columns.forEachIndexed { index, item ->
            visitSqlExpr(sqlUpsert.value[index])
            sqlBuilder.append(" AS ")
            visitSqlExpr(item)
            if (index < sqlUpsert.columns.size - 1) {
                sqlBuilder.append(",")
                sqlBuilder.append(" ")
            }
        }
        sqlBuilder.append(" FROM ${quote}dual${quote}) ${quote}t2${quote}")

        sqlBuilder.append("\nON (")
        sqlUpsert.primaryColumns.forEachIndexed { index, item ->
            sqlBuilder.append("${quote}t1${quote}.")
            visitSqlExpr(item)
            sqlBuilder.append(" = ")
            sqlBuilder.append("${quote}t2${quote}.")
            visitSqlExpr(item)
            if (index < sqlUpsert.primaryColumns.size - 1) {
                sqlBuilder.append(" AND ")
            }
        }
        sqlBuilder.append(")")

        sqlBuilder.append("\nWHEN MATCHED THEN UPDATE SET ")
        printList(sqlUpsert.updateColumns) {
            sqlBuilder.append("${quote}t1${quote}.")
            visitSqlExpr(it)
            sqlBuilder.append(" = ")
            sqlBuilder.append("${quote}t2${quote}.")
            visitSqlExpr(it)
        }

        sqlBuilder.append("\nWHEN NOT MATCHED THEN INSERT")

        sqlBuilder.append(" (")
        printList(sqlUpsert.columns) {
            sqlBuilder.append("${quote}t1${quote}.")
            visitSqlExpr(it)
        }
        sqlBuilder.append(")")

        sqlBuilder.append(" VALUES")
        sqlBuilder.append(" (")
        printList(sqlUpsert.value, ::visitSqlExpr)
        sqlBuilder.append(")")
    }
}