org.fluentjdbc.DatabaseResult Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fluent-jdbc Show documentation

A Java library used to execute JDBC statements and build SQL

There is a newer version: 0.5.3

package org.fluentjdbc;

import org.fluentjdbc.DatabaseTable.RowMapper;

import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.annotation.Nullable;
import javax.annotation.ParametersAreNonnullByDefault;

@ParametersAreNonnullByDefault
public class DatabaseResult implements AutoCloseable {

    private ResultSet resultSet;
    private Map tableRows = new HashMap<>();

    public DatabaseResult(PreparedStatement stmt) throws SQLException {
        resultSet = stmt.executeQuery();
    }

    @Override
    public void close() throws SQLException {
        resultSet.close();
    }

    public boolean next() throws SQLException {
        return resultSet.next();
    }

    // TODO: This doesn't work on Android or SQL server
    public DatabaseRow table(String tableName) throws SQLException {
        if (!tableRows.containsKey(tableName)) {
            tableRows.put(tableName, new DatabaseRow(resultSet, tableName));
        }
        return tableRows.get(tableName);
    }

    public  List list(RowMapper mapper) throws SQLException {
        List result = new ArrayList<>();
        while (next()) {
            result.add(mapper.mapRow(new DatabaseRow(resultSet)));
        }
        return result;
    }

    @Nullable
    public  T single(RowMapper mapper) throws SQLException {
        if (!next()) {
            return null;
        }
        T result = mapper.mapRow(new DatabaseRow(resultSet));
        if (next()) {
            throw new IllegalStateException("More than one row returned");
        }
        return result;
    }

}