io.descoped.rawdata.provider.postgres.PostgresRawdataClientInitializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rawdata-client-provider-postgres Show documentation
Rawdata Client Provider PostgreSQL Database
The newest version!
package io.descoped.rawdata.provider.postgres;

import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import io.descoped.rawdata.api.RawdataClient;
import io.descoped.rawdata.api.RawdataClientInitializer;
import io.descoped.rawdata.provider.postgres.tx.TransactionFactory;
import io.descoped.service.provider.api.ProviderName;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.PrintWriter;
import java.lang.reflect.InvocationTargetException;
import java.util.Map;
import java.util.Properties;
import java.util.Set;

@ProviderName("postgres")
public class PostgresRawdataClientInitializer implements RawdataClientInitializer {

    private static final Logger LOG = LoggerFactory.getLogger(PostgresRawdataClientInitializer.class);

    @Override
    public String providerId() {
        return "postgres";
    }

    @Override
    public Set configurationKeys() {
        return Set.of(
                "postgres.driver.host",
                "postgres.driver.port",
                "postgres.driver.user",
                "postgres.driver.password",
                "postgres.driver.database",
                "rawdata.postgres.consumer.prefetch-size",
                "rawdata.postgres.consumer.prefetch-poll-interval-when-empty"
        );
    }

    @Override
    public RawdataClient initialize(Map configMap) {
        boolean enableH2DatabaseDriver = Boolean.parseBoolean(configMap.get("h2.enabled"));

        int consumerPrefetchSize = Integer.parseInt(configMap.get("rawdata.postgres.consumer.prefetch-size"));
        int dbPrefetchPollIntervalWhenEmptyMilliseconds = Integer.parseInt(configMap.get("rawdata.postgres.consumer.prefetch-poll-interval-when-empty"));

        if (!enableH2DatabaseDriver) {
            HikariDataSource dataSource = openPostgresDataSource(
                    configMap.get("postgres.driver.host"),
                    configMap.get("postgres.driver.port"),
                    configMap.get("postgres.driver.user"),
                    configMap.get("postgres.driver.password"),
                    configMap.get("postgres.driver.database")
            );

            return new PostgresRawdataClient(new PostgresTransactionFactory(dataSource), consumerPrefetchSize, dbPrefetchPollIntervalWhenEmptyMilliseconds);

        } else {
            HikariDataSource dataSource = openH2DataSource(
                    configMap.get("h2.driver.url"),
                    "sa",
                    "sa"
            );

            try {
                Class transactionFactoryClass = (Class) Class.forName("io.descoped.rawdata.provider.postgres.H2TransactionFactory");
                TransactionFactory transactionFactory = transactionFactoryClass.getDeclaredConstructor(HikariDataSource.class).newInstance(dataSource);
                return new PostgresRawdataClient(transactionFactory, consumerPrefetchSize, dbPrefetchPollIntervalWhenEmptyMilliseconds);

            } catch (ClassNotFoundException | InstantiationException | IllegalAccessException |
                     InvocationTargetException | NoSuchMethodException e) {
                throw new RuntimeException(e);
            }
        }
    }

    // https://github.com/brettwooldridge/HikariCP
    static HikariDataSource openPostgresDataSource(String postgresDbDriverHost, String postgresDbDriverPort, String postgresDbDriverUser, String postgresDbDriverPassword, String postgresDbDriverDatabase) {
        LOG.info("Configured database: postgres");
        Properties props = new Properties();
        props.setProperty("dataSourceClassName", "org.postgresql.ds.PGSimpleDataSource");
        props.setProperty("dataSource.serverName", postgresDbDriverHost);
        props.setProperty("dataSource.portNumber", postgresDbDriverPort);
        props.setProperty("dataSource.user", postgresDbDriverUser);
        props.setProperty("dataSource.password", postgresDbDriverPassword);
        props.setProperty("dataSource.databaseName", postgresDbDriverDatabase);
        props.put("dataSource.logWriter", new PrintWriter(System.out));

        HikariConfig config = new HikariConfig(props);
        config.setAutoCommit(false);
        config.setMaximumPoolSize(10);
        HikariDataSource datasource = new HikariDataSource(config);

        return datasource;
    }

    static HikariDataSource openH2DataSource(String jdbcUrl, String username, String password) {
        LOG.info("Configured database: h2");
        Properties props = new Properties();
        props.setProperty("jdbcUrl", jdbcUrl);
        props.setProperty("username", username);
        props.setProperty("password", password);
        props.put("dataSource.logWriter", new PrintWriter(System.out));

        HikariConfig config = new HikariConfig(props);
        config.setAutoCommit(false);
        config.setMaximumPoolSize(10);
        HikariDataSource datasource = new HikariDataSource(config);

        return datasource;
    }


}