io.quarkiverse.doma.runtime.DomaRecorder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quarkus-doma Show documentation
Simplify your database accesses with compile-time code generation
The newest version!
package io.quarkiverse.doma.runtime;

import java.lang.annotation.Annotation;
import java.nio.file.Path;
import java.util.List;
import java.util.Objects;
import java.util.function.Supplier;

import jakarta.enterprise.inject.Default;

import org.seasar.doma.jdbc.Config;
import org.seasar.doma.jdbc.ConfigSupport;
import org.seasar.doma.jdbc.criteria.Entityql;
import org.seasar.doma.jdbc.criteria.NativeSql;

import io.quarkiverse.doma.runtime.devmode.HotReplacementScriptFileLoader;
import io.quarkiverse.doma.runtime.devmode.HotReplacementSqlFileRepository;
import io.quarkus.agroal.DataSource;
import io.quarkus.arc.Arc;
import io.quarkus.arc.runtime.BeanContainerListener;
import io.quarkus.runtime.LaunchMode;
import io.quarkus.runtime.annotations.Recorder;

@Recorder
public class DomaRecorder {

    private static volatile List hotReplacementResourcesDirs;

    public static void setHotReplacementResourcesDirs(List resourcesDirs) {
        hotReplacementResourcesDirs = Objects.requireNonNull(resourcesDirs);
    }

    public BeanContainerListener configure(DomaSettings domaSettings, LaunchMode launchMode) {
        Objects.requireNonNull(domaSettings);
        Objects.requireNonNull(launchMode);
        return beanContainer -> {
            DomaProducer producer = beanContainer.beanInstance(DomaProducer.class);

            if (launchMode == LaunchMode.DEVELOPMENT) {
                producer.setSqlFileRepository(
                        new HotReplacementSqlFileRepository(hotReplacementResourcesDirs));
                producer.setScriptFileLoader(
                        new HotReplacementScriptFileLoader(hotReplacementResourcesDirs));
            } else {
                producer.setSqlFileRepository(domaSettings.sqlFileRepository.create());
                producer.setScriptFileLoader(ConfigSupport.defaultScriptFileLoader);
            }

            producer.setNaming(domaSettings.naming.naming());
            producer.setExceptionSqlLogType(domaSettings.exceptionSqlLogType);
            producer.setNamedSqlLoadScripts(domaSettings.asNamedSqlLoadScripts());
        };
    }

    public Supplier configureConfig(DomaSettings.DataSourceSettings settings) {
        Objects.requireNonNull(settings);
        return () -> {
            DomaConfig.Core core = Arc.container().instance(DomaConfig.Core.class).get();
            DataSourceNameResolver dataSourceNameResolver = Arc.container().instance(DataSourceNameResolver.class).get();
            DataSourceResolver dataSourceResolver = Arc.container().instance(DataSourceResolver.class).get();
            String candidateName = settings.name;
            return DomaConfig.builder(core)
                    .setDataSourceNameSupplier(() -> dataSourceNameResolver.resolve(candidateName))
                    .setDataSourceResolver(dataSourceResolver)
                    .setDialect(settings.dialect.create())
                    .setBatchSize(settings.batchSize)
                    .setFetchSize(settings.fetchSize)
                    .setMaxRows(settings.maxRows)
                    .setQueryTimeout(settings.queryTimeout)
                    .build();
        };
    }

    public Supplier configureEntityql(DomaSettings.DataSourceSettings settings) {
        Objects.requireNonNull(settings);
        return () -> {
            Config config = resolveConfig(settings);
            return new Entityql(config);
        };
    }

    public Supplier configureNativeSql(DomaSettings.DataSourceSettings settings) {
        Objects.requireNonNull(settings);
        return () -> {
            Config config = resolveConfig(settings);
            return new NativeSql(config);
        };
    }

    private Config resolveConfig(DomaSettings.DataSourceSettings settings) {
        Annotation qualifier = resolveQualifier(settings);
        return Arc.container().instance(Config.class, qualifier).get();
    }

    private Annotation resolveQualifier(DomaSettings.DataSourceSettings settings) {
        if (settings.isDefault) {
            return Default.Literal.INSTANCE;
        }
        return new DataSource.DataSourceLiteral(settings.name);
    }
}