org.komapper.r2dbc.R2dbcDataOperator.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of komapper-r2dbc Show documentation
Komapper R2DBC Module
There is a newer version: 5.0.0
package org.komapper.r2dbc

import io.r2dbc.spi.Row
import io.r2dbc.spi.Statement
import org.komapper.core.DataOperator
import kotlin.reflect.KType

interface R2dbcDataOperator : DataOperator {
    /**
     * Returns the value.
     *
     * @param row the row
     * @param index the index
     * @param type the value type
     * @return the value
     */
    fun  getValue(row: Row, index: Int, type: KType): T?

    /**
     * Returns the value.
     *
     * @param row the row
     * @param columnLabel the column label
     * @param type the value type
     * @return the value
     */
    fun  getValue(row: Row, columnLabel: String, type: KType): T?

    /**
     * Sets the value.
     *
     * @param statement the statement
     * @param index the index
     * @param value the value
     * @param type the value type
     */
    fun  setValue(statement: Statement, index: Int, value: T?, type: KType)

    /**
     * Returns the data type.
     *
     * @param type the value type
     * @return the data type
     */
    fun  getDataType(type: KType): R2dbcDataType

    /**
     * Returns the data type or null.
     *
     * @param type the value type
     * @return the data type or null
     */
    fun  getDataTypeOrNull(type: KType): R2dbcDataType?
}

class DefaultR2dbcDataOperator(private val dialect: R2dbcDialect, private val dataTypeProvider: R2dbcDataTypeProvider) :
    R2dbcDataOperator {
    override fun  getValue(row: Row, index: Int, type: KType): T? {
        val dataType = getDataType(type)
        return dataType.getValue(row, index)
    }

    override fun  getValue(row: Row, columnLabel: String, type: KType): T? {
        val dataType = getDataType(type)
        return dataType.getValue(row, columnLabel)
    }

    override fun  setValue(statement: Statement, index: Int, value: T?, type: KType) {
        val dataType = getDataType(type)
        val bindMarker = dialect.getBinder()
        return bindMarker.bind(statement, index, value, dataType)
    }

    override fun  formatValue(value: T?, type: KType, masking: Boolean): String {
        return if (masking) {
            dialect.mask
        } else {
            val dataType = getDataTypeOrNull(type)
            dataType?.toString(value) ?: value.toString()
        }
    }

    override fun  getDataType(type: KType): R2dbcDataType {
        return getDataTypeOrNull(type) ?: error(
            "The dataType is not found for the type \"${type}\".",
        )
    }

    override fun  getDataTypeOrNull(type: KType): R2dbcDataType? {
        return dataTypeProvider.get(type)
    }

    override fun  getDataTypeName(type: KType): String {
        val dataType = getDataType(type)
        return dataType.name
    }
}