dev.monosoul.jooq.settings.JooqDockerPluginSettings.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.14
package dev.monosoul.jooq.settings

import dev.monosoul.jooq.container.GenericDatabaseContainer
import dev.monosoul.jooq.util.CodegenClasspathAwareClassLoaders
import org.gradle.api.Action
import org.gradle.api.tasks.Nested
import org.testcontainers.shaded.org.apache.commons.lang3.builder.EqualsBuilder.reflectionEquals
import org.testcontainers.shaded.org.apache.commons.lang3.builder.HashCodeBuilder.reflectionHashCode

sealed class JooqDockerPluginSettings : SettingsElement {
    @get:Nested
    internal abstract val database: Database

    internal abstract fun runWithDatabaseCredentials(
        classloaders: CodegenClasspathAwareClassLoaders,
        block: (CodegenClasspathAwareClassLoaders, DatabaseCredentials) -> Unit,
    )

    internal abstract fun copy(): JooqDockerPluginSettings

    class WithContainer private constructor(
        override val database: Database.Internal,
        @get:Nested internal val image: Image,
    ) : JooqDockerPluginSettings(),
        DbAware,
        ImageAware {
        private constructor(database: Database.Internal) : this(database, Image(database))
        constructor(customizer: Action = Action { }) : this(Database.Internal()) {
            customizer.execute(this)
        }

        override fun runWithDatabaseCredentials(
            classloaders: CodegenClasspathAwareClassLoaders,
            block: (CodegenClasspathAwareClassLoaders, DatabaseCredentials) -> Unit,
        ) {
            val dbContainer =
                GenericDatabaseContainer(
                    image = image,
                    database = database,
                    jdbcAwareClassLoader = classloaders.buildscriptInclusive,
                ).also { it.start() }

            try {
                block(
                    classloaders,
                    DatabaseCredentials(
                        jdbcDriverClassName = dbContainer.driverClassName,
                        jdbcUrl = dbContainer.jdbcUrl,
                        username = dbContainer.username,
                        password = dbContainer.password,
                    ),
                )
            } finally {
                dbContainer.stop()
            }
        }

        override fun copy(): WithContainer =
            WithContainer(
                database = database.run { copy(jdbc = jdbc.copy()) },
                image = image.copy(),
            )

        override fun db(customizer: Action) = customizer.execute(database)

        override fun image(customizer: Action) = customizer.execute(image)

        override fun equals(other: Any?) = reflectionEquals(this, other)

        override fun hashCode(): Int = reflectionHashCode(this)
    }

    class WithoutContainer private constructor(
        override val database: Database.External,
    ) : JooqDockerPluginSettings(),
        DbAware {
        constructor(customizer: Action = Action { }) : this(Database.External()) {
            customizer.execute(this)
        }

        override fun runWithDatabaseCredentials(
            classloaders: CodegenClasspathAwareClassLoaders,
            block: (CodegenClasspathAwareClassLoaders, DatabaseCredentials) -> Unit,
        ) {
            block(
                classloaders,
                DatabaseCredentials(
                    jdbcDriverClassName = database.jdbc.driverClassName,
                    jdbcUrl = database.getJdbcUrl(),
                    username = database.username,
                    password = database.password,
                ),
            )
        }

        override fun copy(): WithoutContainer =
            WithoutContainer(
                database = database.run { copy(jdbc = jdbc.copy()) },
            )

        override fun db(customizer: Action) = customizer.execute(database)

        override fun equals(other: Any?) = reflectionEquals(this, other)

        override fun hashCode(): Int = reflectionHashCode(this)
    }
}