org.fluentjdbc.DatabaseBulkInsertBuilderWithPk 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.util.ExceptionUtil;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.List;
import java.util.function.BiConsumer;
import java.util.function.Function;

import static org.fluentjdbc.DatabaseStatement.addBatch;

/**
 * Variation of {@link DatabaseBulkInsertBuilder} which executes the statement with a request
 * to return generated primary keys from the database and calls the registered update function
 * for each object in the batch with the corresponding generated primary key
 */
public class DatabaseBulkInsertBuilderWithPk {

    private final String primaryKeyColumn;
    private final DatabaseTable table;
    private final List updateFields;
    private final List> updateParameters;
    private final BiConsumer primaryKeyCallback;
    private final Iterable objects;

    public DatabaseBulkInsertBuilderWithPk(
            Iterable objects,
            DatabaseTable table,
            List updateFields,
            List> updateParameters,
            String primaryKeyColumn,
            BiConsumer primaryKeyCallback
    ) {
        this.objects = objects;
        this.table = table;
        this.updateFields = updateFields;
        this.updateParameters = updateParameters;
        this.primaryKeyColumn = primaryKeyColumn;
        this.primaryKeyCallback = primaryKeyCallback;
    }

    public void execute(Connection connection) {
        String insertStatement = table.createInsertSql(updateFields);
        try (PreparedStatement statement = connection.prepareStatement(insertStatement, new String[] { primaryKeyColumn })) {
            addBatch(statement, objects, updateParameters);
            statement.executeBatch();

            ResultSet generatedKeys = statement.getGeneratedKeys();
            int i=0;
            for (T object : objects) {
                i++;
                if (!generatedKeys.next()) {
                    throw new IllegalStateException("Could not find generated keys for row: " + i);
                }
                primaryKeyCallback.accept(object, generatedKeys.getLong(1));
            }
        } catch (SQLException e) {
            throw ExceptionUtil.softenCheckedException(e);
        }

    }
}