com.firefly.kotlin.ext.db.AsyncHttpContextTransactionalManager.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firefly-kotlin-ext Show documentation

There is a newer version: 5.0.0-dev6

package com.firefly.kotlin.ext.db

import com.firefly.db.SQLClient
import com.firefly.db.SQLConnection
import com.firefly.kotlin.ext.common.CoroutineLocal
import com.firefly.server.http2.router.RoutingContext
import kotlinx.coroutines.experimental.future.await
import kotlinx.coroutines.experimental.withTimeout
import java.util.concurrent.CompletableFuture
import java.util.concurrent.TimeUnit

/**
 * Manage transaction in the HTTP request lifecycle.
 *
 * @author Pengtao Qiu
 */
class AsyncHttpContextTransactionalManager(
    val requestCtx: CoroutineLocal,
    val sqlClient: SQLClient
                                          ) : AsyncTransactionalManager {

    val transactionKey = "_currentKotlinTransaction"

    override suspend fun getConnection(time: Long, unit: TimeUnit): SQLConnection = withTimeout(unit.toMillis(time)) {
        if (requestCtx.get() == null) {
            sysLogger.debug("get new db connection from pool")
            sqlClient.connection.await()
        } else {
            var sqlConn = createConnectionIfEmpty().await()
            if (sqlConn.connection.isClosed) {
                val future = sqlClient.connection
                sqlConn = future.await()
                requestCtx.get()?.attributes?.put(transactionKey, future)
                sqlConn
            } else {
                sqlConn
            }
        }
    }

    @Suppress("UNCHECKED_CAST")
    override suspend fun getCurrentConnection(time: Long, unit: TimeUnit): SQLConnection? = withTimeout(unit.toMillis(time)) {
        (requestCtx.get()?.attributes?.get(transactionKey) as CompletableFuture?)?.await()
    }

    @Suppress("UNCHECKED_CAST")
    private fun createConnectionIfEmpty(): CompletableFuture =
        requestCtx.get()?.attributes?.computeIfAbsent(transactionKey) {
            sysLogger.debug("init new db connection")
            sqlClient.connection
        } as CompletableFuture

}