io.github.shmilyjxs.core.impl.DaoContextImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jdb-tool Show documentation

There is a newer version: 1.37

package io.github.shmilyjxs.core.impl;

import io.github.shmilyjxs.dialects.DBType;
import io.github.shmilyjxs.utils.LazyInitializer;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.jdbc.support.JdbcUtils;

import javax.sql.DataSource;
import java.util.Objects;
import java.util.UUID;

public class DaoContextImpl extends BaseBeanDao {

    private final DataSource dataSource;
    private final JdbcTemplate jdbcTemplate;
    private final NamedParameterJdbcTemplate namedJdbcTemplate;
    private final LazyInitializer lazyDBType;

    public DaoContextImpl(DataSource dataSource) {
        this.dataSource = Objects.requireNonNull(dataSource);
        this.jdbcTemplate = new JdbcTemplate(dataSource);
        this.namedJdbcTemplate = new NamedParameterJdbcTemplate(jdbcTemplate);
        this.lazyDBType = new LazyInitializer() {
            @Override
            protected DBType initialize() {
                try {
                    return JdbcUtils.extractDatabaseMetaData(dataSource, e -> DBType.fromProductName(e.getDatabaseProductName()));
                } catch (Exception e) {
                    throw new RuntimeException(e);
                }
            }
        };
    }

    @Override
    public DataSource getDataSource() {
        return dataSource;
    }

    @Override
    public JdbcTemplate getJdbcTemplate() {
        return jdbcTemplate;
    }

    @Override
    public NamedParameterJdbcTemplate getNamedJdbcTemplate() {
        return namedJdbcTemplate;
    }

    @Override
    public DBType getDBType() {
        return Objects.requireNonNull(lazyDBType.get());
    }

    @Override
    public Object idGenerator() {
        return UUID.randomUUID().toString();
    }
}