JdbcDatabaseConnection.kt

/*
 * Copyright (C) 2023 Ctrip.com.
 *
 * 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 com.ctrip.sqllin.driver

import java.sql.Connection

import java.sql.PreparedStatement
import java.util.concurrent.atomic.AtomicBoolean

/**
 * JDBC implementation of [DatabaseConnection] for JVM platforms.
 *
 * Uses java.sql.Connection for database operations.
 *
 * @author Yuang Qiao
 */
internal class JdbcDatabaseConnection(private val connection: Connection) : AbstractJdbcDatabaseConnection() {

    override fun execSQL(sql: String, bindParams: Array<out Any?>?) {
        bindParamsToSQL(sql, bindParams).use {
            it.execute()
        }
    }

    override fun executeInsert(sql: String, bindParams: Array<out Any?>?) {
        executeUpdate(sql, bindParams)
    }

    override fun executeUpdateDelete(sql: String, bindParams: Array<out Any?>?) {
        executeUpdate(sql, bindParams)
    }

    private fun executeUpdate(sql: String, bindParams: Array<out Any?>?): Int = bindParamsToSQL(sql, bindParams).use {
        it.executeUpdate()
    }

    override fun query(sql: String, bindParams: Array<out Any?>?): CommonCursor =
        bindParamsToSQL(sql, bindParams)
            .executeQuery()
            ?.let { JdbcCursor(it) }
            ?: throw IllegalStateException("The query result is null.")

    private val isTransactionSuccess = AtomicBoolean(false)

    override fun beginTransaction() {
        if (isTransactionSuccess.get())
            isTransactionSuccess.set(false)
        connection.autoCommit = false
    }

    override fun setTransactionSuccessful() {
        isTransactionSuccess.set(true)
    }

    override fun endTransaction() = try {
        if (isTransactionSuccess.get())
            connection.commit()
        else
            connection.rollback()
    } finally {
        connection.autoCommit = true
        isTransactionSuccess.set(false)
    }

    override fun close() = connection.close()

    override val isClosed: Boolean
        get() = connection.isClosed

    override fun createStatement(sql: String): PreparedStatement = connection.prepareStatement(sql)
}