com.dbobjekts.statement.insert.InsertBuilderBase.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of db-objekts-core Show documentation

There is a newer version: 0.6.0-RC2

package com.dbobjekts.statement.insert

import com.dbobjekts.api.TableRowData
import com.dbobjekts.statement.Semaphore
import com.dbobjekts.jdbc.ConnectionAdapter
import com.dbobjekts.metadata.column.Column
import com.dbobjekts.statement.update.ColumnForWriteMapContainerImpl

/**
 * A concrete `InsertBuilderBase` is auto-generated for each table and provides setter methods for each column to supply the values that need to be inserted.
 *
 * It is the class returned by the call to `insert(..)` in the following example.
 *
 * ```kotlin
 *  tr.insert(EmployeeAddressRow(johnsId, addressId, AddressType.HOME))
 * ```
 */
abstract class InsertBuilderBase() {
    internal lateinit var connection: ConnectionAdapter
    internal lateinit var semaphore: Semaphore

    private val ct = ColumnForWriteMapContainerImpl(this)

    @Suppress("UNCHECKED_CAST")
    protected fun  put(col: Column, value: C?): B {
        add(col, value)
        return this as B
    }

    protected fun  add(col: Column, value: C?) {
        ct.put(col, value)
    }

    protected fun  mandatory(col: Column, value: C?) {
        ct.put(col, value)
    }

    abstract fun insertRow(rowData: TableRowData<*, *>): Long

    /**
     * Executes the insert statement and persists a new row to the table.
     * @return the primary key of the new row if it is auto-generated. Otherwise, it returns the result of the underlying [java.sql.PreparedStatement.execute] call
     */
    fun execute(): Long = InsertStatementExecutor(semaphore, connection, ct.data.toList(), connection.vendor.properties).execute()
        .also { connection.statementLog.logResult(it) }
}