no.ssb.rawdata.provider.state.h2.H2TransactionFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rawdata-state-provider-h2 Show documentation

Service Provider API

The newest version!

package no.ssb.rawdata.provider.state.h2;

import com.zaxxer.hikari.HikariDataSource;
import no.ssb.rawdata.api.persistence.PersistenceException;
import no.ssb.rawdata.api.persistence.Transaction;
import no.ssb.rawdata.api.persistence.TransactionFactory;

import java.sql.Connection;
import java.sql.SQLException;
import java.util.concurrent.CompletableFuture;
import java.util.function.Function;

public class H2TransactionFactory implements TransactionFactory {

    final HikariDataSource dataSource;

    public H2TransactionFactory(HikariDataSource dataSource) {
        this.dataSource = dataSource;
    }

    @Override
    public  CompletableFuture runAsyncInIsolatedTransaction(Function retryable, boolean readOnly) {
        return CompletableFuture.supplyAsync(() -> retryable.apply(createTransaction(readOnly)));
    }

    @Override
    public H2Transaction createTransaction(boolean readOnly) throws PersistenceException {
        try {
            Connection connection = dataSource.getConnection();
            connection.setAutoCommit(false);
            return new H2Transaction(connection);
        } catch (SQLException e) {
            throw new PersistenceException(e);
        }
    }



    @Override
    public void close() {
        dataSource.close();
    }
}