kr.jadekim.jext.apm.es.exposed.EsApmInterceptor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jext-es-apm-exposed Show documentation

Jext Libraries

The newest version!

package kr.jadekim.jext.apm.es.exposed

import co.elastic.apm.api.Span
import kr.jadekim.jext.apm.es.apmContext
import kr.jadekim.jext.apm.es.newSpan
import org.jetbrains.exposed.sql.Transaction
import org.jetbrains.exposed.sql.statements.StatementContext
import org.jetbrains.exposed.sql.statements.StatementInterceptor

suspend fun Transaction.installEsApmTracer() {
    registerInterceptor(esApmTracer())
}

suspend fun esApmTracer(): StatementInterceptor {
    val apm = apmContext()

    return object : StatementInterceptor {

        private var span: Span? = null

        override fun beforeExecution(transaction: Transaction, context: StatementContext) {
            span = apm.transaction.newSpan(
                "db",
                transaction.db.vendor,
                context.statement.type.name,
                context.statement.prepareSQL(transaction),
            )
                .setLabel("url", transaction.db.url)
        }

        override fun afterCommit(transaction: Transaction) {
            span?.setLabel("result", "commit")?.end()
        }

        override fun afterRollback(transaction: Transaction) {
            span?.setLabel("result", "rollback")?.end()
        }
    }
}