com.codurance.lightaccess.LightAccess Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of light-access Show documentation

Non-intrusive JDBC library supporting lambdas

The newest version!

package com.codurance.lightaccess;

import com.codurance.lightaccess.connection.LAConnection;
import com.codurance.lightaccess.executables.*;

import javax.sql.DataSource;
import java.sql.SQLException;
import java.util.function.Function;

import static com.codurance.lightaccess.executables.Throwables.executeWithResource;
import static java.lang.String.format;

public class LightAccess {

    private static final String SEQUENCE_CALL_SQL = "select nextval('%s')";
    
    private DataSource ds;
                                                     
    public LightAccess(DataSource connection) {
        this.ds = connection;
    }

    public  T executeQuery(SQLQuery sqlQuery) {
        LAConnection conn = pgConnection();
        return executeWithResource(conn, () -> sqlQuery.execute(conn));
    }

    public void executeCommand(SQLCommand sqlCommand) {
        execute(sqlCommand);
    }

    public void executeDDLCommand(DDLCommand ddlCommand) {
        execute(ddlCommand);
    }

    public int nextId(String sequenceName) {
        return nextId(sequenceName, (x) -> x);
    }

    public  T nextId(String sequenceName, Function nextId) {
        LAConnection conn = pgConnection();
        return executeWithResource(conn, () -> nextId.apply(sequenceNextId(sequenceName, conn)));
    }

    private int sequenceNextId(String sequenceName, LAConnection conn) throws SQLException {
        String sql = format(SEQUENCE_CALL_SQL, sequenceName);
        return conn.callableStatement(sql)
                    .executeQuery()
                    .nextRecord()
                    .getInt(1);
    }

    private void execute(Command command) {
        LAConnection conn = pgConnection();
        executeWithResource(conn, () -> command.execute(conn));
    }

    private LAConnection pgConnection() {
        return Throwables.executeQuery(() -> new LAConnection(ds.getConnection()));
    }
}