dev.monosoul.jooq.container.GenericDatabaseContainer.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jooq-gradle-plugin Show documentation
Generates jOOQ classes using dockerized database
There is a newer version: 6.1.5
package dev.monosoul.jooq.container

import dev.monosoul.jooq.settings.Database
import dev.monosoul.jooq.settings.Image
import org.slf4j.LoggerFactory
import org.testcontainers.containers.JdbcDatabaseContainer
import org.testcontainers.containers.output.Slf4jLogConsumer
import org.testcontainers.containers.wait.strategy.HostPortWaitStrategy
import org.testcontainers.dockerclient.DockerClientProviderStrategy
import org.testcontainers.utility.DockerImageName
import java.sql.Driver
import java.util.concurrent.atomic.AtomicBoolean
import java.util.concurrent.locks.ReentrantLock
import kotlin.concurrent.withLock
import kotlin.reflect.KCallable
import kotlin.reflect.full.declaredMembers
import kotlin.reflect.jvm.isAccessible

class GenericDatabaseContainer(
    private val image: Image,
    private val database: Database.Internal,
    private val jdbcAwareClassLoader: ClassLoader,
) : JdbcDatabaseContainer(
        image.let {
            failFastAlways.set(false)
            DockerImageName.parse(it.name)
        },
    ) {
    private val driverLoadLock = ReentrantLock()
    private var driver: Driver? = null

    init {
        withLogConsumer(
            Slf4jLogConsumer(
                LoggerFactory.getLogger("JooqGenerationDb[$dockerImageName]"),
            ),
        )
        withEnv(image.envVars)
        withExposedPorts(database.port)
        setWaitStrategy(HostPortWaitStrategy())
        image.command.takeUnless { it.isNullOrBlank() }?.run(::withCommand)
    }

    override fun getDriverClassName() = database.jdbc.driverClassName

    override fun getJdbcUrl() = database.getJdbcUrl(host, getMappedPort(database.port))

    override fun getUsername() = database.username

    override fun getPassword() = database.password

    override fun getTestQueryString() = image.testQuery

    override fun getDatabaseName() = database.name

    override fun getJdbcDriverInstance(): Driver {
        if (driver == null) {
            driverLoadLock.withLock {
                if (driver == null) {
                    driver = getNewJdbcDriverInstance()
                }
            }
        }

        return driver!!
    }

    private fun getNewJdbcDriverInstance() =
        try {
            @Suppress("DEPRECATION")
            jdbcAwareClassLoader.loadClass(driverClassName).newInstance() as Driver
        } catch (e: Exception) {
            when (e) {
                is InstantiationException, is IllegalAccessException, is ClassNotFoundException -> {
                    throw NoDriverFoundException("Could not get Driver", e)
                }
                else -> throw e
            }
        }

    private companion object {
        /**
         * Workaround for https://github.com/testcontainers/testcontainers-java/issues/6441
         */
        val failFastAlways =
            DockerClientProviderStrategy::class
                .declaredMembers
                .single { it.name == "FAIL_FAST_ALWAYS" }
                .apply { isAccessible = true }
                .let {
                    @Suppress("UNCHECKED_CAST")
                    it as KCallable
                }.call()
    }
}