org.icij.extract.mysql.FunctionalDataSource Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of extract-cli Show documentation

There is a newer version: 7.3.0

package org.icij.extract.mysql;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;

public class FunctionalDataSource extends DataSourceDecorator {

	public static FunctionalDataSource cast(final DataSource dataSource) {
		if (dataSource instanceof FunctionalDataSource) {
			return (FunctionalDataSource) dataSource;
		} else {
			return new FunctionalDataSource(dataSource);
		}
	}

	public FunctionalDataSource(final DataSource dataSource) {
		super(dataSource);
	}

	public void withConnection(final CheckedConsumer consumer) throws SQLException {
		try (final Connection c = getConnection()) {
			consumer.acceptThrows(c);
		}
	}

	public void withConnectionUnchecked(final CheckedConsumer consumer) {
		try {
			withConnection(consumer);
		} catch (final SQLException e) {
			throw new RuntimeException(e);
		}
	}

	public  R withConnection(final CheckedFunction function) throws SQLException {
		try (final Connection c = getConnection()) {
			return function.applyThrows(c);
		}
	}

	public  R withConnectionUnchecked(final CheckedFunction function) {
		try {
			return withConnection(function);
		} catch (final SQLException e) {
			throw new RuntimeException(e);
		}
	}

	public void withStatement(final String statement, final CheckedConsumer consumer)
			throws SQLException {
		withConnection(c -> {
			try (final PreparedStatement q = c.prepareStatement(statement)){
				consumer.acceptThrows(q);
			}
		});
	}

	public void withStatementUnchecked(final String statement, final CheckedConsumer consumer) {
		withConnectionUnchecked(c -> {
			try (final PreparedStatement q = c.prepareStatement(statement)){
				consumer.acceptThrows(q);
			}
		});
	}

	public  R withStatement(final String statement, final CheckedFunction function)
			throws SQLException {
		return withConnection(c -> {
			try (final PreparedStatement q = c.prepareStatement(statement)) {
				return function.applyThrows(q);
			}
		});
	}

	public  R withStatementUnchecked(final String statement, final CheckedFunction function) {
		return withConnectionUnchecked(c -> {
			try (final PreparedStatement q = c.prepareStatement(statement)) {
				return function.applyThrows(q);
			}
		});
	}
}