com.dbobjekts.statement.update.UpdateBuilderBase.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.update

import com.dbobjekts.api.*
import com.dbobjekts.api.exception.StatementBuilderException
import com.dbobjekts.jdbc.ConnectionAdapter
import com.dbobjekts.metadata.column.Column
import com.dbobjekts.metadata.column.NullableColumnAndValue
import com.dbobjekts.statement.Semaphore
import com.dbobjekts.statement.WriteQueryAccessors
import com.dbobjekts.statement.insert.InsertBuilderBase
import com.dbobjekts.statement.whereclause.EmptyWhereClause
import com.dbobjekts.statement.whereclause.SubClause

interface HasUpdateBuilder {
    /**
     * FOR DB-OBJEKTS INTERNAL USE ONLY.
     */
    fun metadata(): WriteQueryAccessors

}


class ColumnForWriteMapContainerImpl(val builder: T) {
    val data: HashSet = linkedSetOf()

    fun  put(col: Column, value: C?): T {
        if (value == null)
            data.add(NullableColumnAndValue(col, null))
        else
            data.add(col.create(value))
        return builder
    }
}

/**
 * A concrete `UpdateBuilderBase` is auto-generated for each table and provides setter methods for each column to supply the values that need to be updated.
 *
 * It is the class returned by the call to `update()` in the following example.
 *
 * ```kotlin
 * tr.update(e).children(2).married(true).where(e.name.eq("Janet"))
 * ```
 */
abstract class UpdateBuilderBase(
    internal val table: AnyTable) {
    private val ct = ColumnForWriteMapContainerImpl(this)
    internal lateinit var connection: ConnectionAdapter
    internal lateinit var semaphore: Semaphore

    @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)
    }

    /**
     * FOR INTERNAL USE ONLY
     */
    abstract fun updateRow(rowData: TableRowData<*,*>): Long

    /**
     * Opens the whereclause for this update statement.
     */
    fun where(whereClause: SubClause): Long {
        if (ct.data.isEmpty())
            throw StatementBuilderException("Need at least one column to update")
        val stm = UpdateStatementExecutor(semaphore, connection, table, ct.data.toList())
        return stm.where(whereClause ).also { connection.statementLog.logResult(it) }
    }
    /**
     * Executes the update statement for all rows in a table. Handle with care!
     */
    fun where(): Long = where(EmptyWhereClause)

}