se.ugli.habanero.j.Habanero Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of habanero-java Show documentation
Habanero for Java
There is a newer version: 1.8.1.1
package se.ugli.habanero.j;

import static se.ugli.habanero.j.internal.CloseUtil.close;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;

import javax.sql.DataSource;

import se.ugli.habanero.j.datasource.H2DataSource;
import se.ugli.habanero.j.internal.PrepareArgumentsCommand;
import se.ugli.habanero.j.internal.SingleValueIterator;
import se.ugli.habanero.j.util.Option;

public final class Habanero {

	public static Habanero apply(final DataSource dataSource) {
		return new Habanero(dataSource);
	}

	public static Habanero apply() {
		return new Habanero(new H2DataSource());
	}

	public final DataSource dataSource;

	private Habanero(final DataSource dataSource) {
		this.dataSource = dataSource;
	}

	public  Iterable queryMany(final Class type, final String sql, final Object... args) {
		return queryMany(new SingleValueIterator(type), sql, args);
	}

	private static class TypedMapIdentityIterator extends TypedMapIterator {

		@Override
		protected TypedMap nextObject(final TypedMap typedMap) {
			return typedMap;
		}

	}

	public  Iterable queryMany(final GroupFunction groupFunction, final String sql, final Object... args) {
		final Iterable tuples = queryMany(new TypedMapIdentityIterator(), sql, args);
		return groupFunction.createObjects(tuples);
	}

	public  Iterable queryMany(final ResultSetIterator iterator, final String sql, final Object... args) {
		Connection connection = null;
		PreparedStatement statement = null;
		ResultSet resultSet = null;
		try {
			connection = dataSource.getConnection();
			statement = connection.prepareStatement(sql);
			PrepareArgumentsCommand.apply(statement).exec(args);
			resultSet = statement.executeQuery();
			iterator.init(resultSet);
			final List result = new ArrayList();
			while (iterator.hasNext())
				result.add(iterator.next());
			return Collections.unmodifiableCollection(result);
		} catch (final SQLException e) {
			throw new HabaneroException(e);
		} finally {
			close(resultSet, statement, connection);
		}
	}

	public  Option queryOne(final Class type, final String sql, final Object... args) {
		return queryOne(new SingleValueIterator(type), sql, args);
	}

	public  Option queryOne(final GroupFunction groupFunction, final String sql, final Object... args) {
		final Iterable result = queryMany(groupFunction, sql, args);
		final Iterator objectIterator = result.iterator();
		if (objectIterator.hasNext())
			return Option.apply(objectIterator.next());
		return Option.none();
	}

	public  Option queryOne(final ResultSetIterator resultSetiterator, final String sql, final Object... args) {
		final Iterable result = queryMany(resultSetiterator, sql, args);
		final Iterator objectIterator = result.iterator();
		if (objectIterator.hasNext())
			return Option.apply(objectIterator.next());
		return Option.none();
	}

	public int update(final String sql, final Object... args) {
		Connection connection = null;
		PreparedStatement statement = null;
		try {
			connection = dataSource.getConnection();
			statement = connection.prepareStatement(sql);
			PrepareArgumentsCommand.apply(statement).exec(args);
			return statement.executeUpdate();
		} catch (final SQLException e) {
			throw new HabaneroException(e);
		} finally {
			close(statement, connection);
		}
	}

	public boolean execute(final String sql) {
		Connection connection = null;
		PreparedStatement statement = null;
		try {
			connection = dataSource.getConnection();
			statement = connection.prepareStatement(sql);
			return statement.execute();
		} catch (final SQLException e) {
			throw new HabaneroException(e);
		} finally {
			close(statement, connection);
		}
	}

}