buckelieg.fn.db.QueryImpl Maven / Gradle / Ivy

Go to download
package buckelieg.fn.db;

import javax.annotation.Nonnull;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.function.Consumer;

import static buckelieg.fn.db.Utils.checkAnonymous;
import static buckelieg.fn.db.Utils.setStatementParameters;

@SuppressWarnings("unchecked")
final class QueryImpl extends AbstractQuery {

    private boolean isPrepared;

    QueryImpl(Connection connection, String query, Object... params) {
        super(connection, query, params);
    }

    @Override
    Statement prepareStatement(Connection connection, String query, Object... params) throws SQLException {
        return (this.isPrepared = params != null && params.length != 0) ? setStatementParameters(connection.prepareStatement(query), params) : connection.createStatement();
    }

    @Override
    public Boolean execute() {
        boolean result = withStatement(s -> isPrepared ? ((PreparedStatement) s).execute() : s.execute(checkAnonymous(query)));
        close(); // force closing this statement since we will not process any of its possible results
        return result;
    }

    @Nonnull
    @Override
    public Query poolable(boolean poolable) {
        return setPoolable(poolable);
    }

    @Nonnull
    @Override
    public Query timeout(int timeout) {
        return setTimeout(timeout);
    }

    @Nonnull
    @Override
    public Query escaped(boolean escapeProcessing) {
        return setEscapeProcessing(escapeProcessing);
    }

    @Nonnull
    @Override
    public Query skipWarnings(boolean skipWarnings) {
        return setSkipWarnings(skipWarnings);
    }

    @Nonnull
    @Override
    public Query print(Consumer printer) {
        return log(printer);
    }
}