Examples with JdbcDataSourceConfiguration com.palantir.atlasdb.jdbc.config.JdbcDataSourceConfiguration used on opensource projects

Example 1 with JdbcDataSourceConfiguration

use of com.palantir.atlasdb.jdbc.config.JdbcDataSourceConfiguration in project atlasdb by palantir.

the class JdbcKeyValueService method create.

public static JdbcKeyValueService create(JdbcKeyValueConfiguration config) {
    JdbcDataSourceConfiguration dataSourceConfig = config.getDataSourceConfig();
    SQLDialect sqlDialect = SQLDialect.valueOf(dataSourceConfig.getSqlDialect());
    DataSource dataSource = dataSourceConfig.createDataSource();
    Settings settings = new Settings();
    settings.setRenderNameStyle(RenderNameStyle.AS_IS);
    final JdbcKeyValueService kvs = new JdbcKeyValueService(settings, sqlDialect, dataSource, config.getTablePrefix(), config.getRowBatchSize(), config.getBatchSizeForReads(), config.getBatchSizeForMutations());
    kvs.run((Function<DSLContext, Void>) ctx -> {
        String partialSql = ctx.createTable(kvs.METADATA_TABLE).column(TABLE_NAME, VARCHAR.nullable(false)).column(METADATA, BLOB.nullable(false)).getSQL();
        int endIndex = partialSql.lastIndexOf(')');
        String fullSql = partialSql.substring(0, endIndex) + "," + " CONSTRAINT pk_" + kvs.METADATA_TABLE.getName() + " PRIMARY KEY (" + TABLE_NAME.getName() + ")" + partialSql.substring(endIndex);
        try {
            ctx.execute(fullSql);
        } catch (DataAccessException e) {
            kvs.handleTableCreationException(e);
        }
        return null;
    });
    return kvs;
}