org.fluentjdbc.DatabaseBulkInsertBuilder 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 org.joda.time.DateTime;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class DatabaseBulkInsertBuilder extends DatabaseStatement {

    private List objects = new ArrayList<>();
    private InsertMapper insertMapper;
    private List fieldNames = new ArrayList<>();
    private DatabaseTable table;

    public DatabaseBulkInsertBuilder(DatabaseTable table) {
        this.table = table;
    }

    public DatabaseBulkInsertBuilder(DatabaseTable table, List objects) {
        this(table);
        this.objects.addAll(objects);
    }

    public void addFieldNames(String... newFieldNames) {
        this.fieldNames.addAll(Arrays.asList(newFieldNames));
    }

    public void execute(Connection connection) {
        String insertStatement = "insert into " + table.getTableName() +
                    " (" + join(",", fieldNames)
                    + ") values ("
                    + join(",", repeat("?", fieldNames.size())) + ")";


        DateTime now = DateTime.now();
        try (PreparedStatement statement = connection.prepareStatement(insertStatement)) {
            for (T object : objects) {
                Inserter inserter = new Inserter(statement, fieldNames);
                insertMapper.mapRow(inserter, object);

                // HACK: There must be a better way of doing this!
                if (table instanceof DatabaseTableWithTimestamps) {
                    inserter.setField("updated_at", now);
                    inserter.setField("created_at", now);
                }
                statement.addBatch();
            }

            statement.executeBatch();
        } catch (SQLException e) {
            throw ExceptionUtil.softenCheckedException(e);
        }
    }

    public DatabaseBulkInsertBuilder insert(InsertMapper insertMapper) {
        this.insertMapper = insertMapper;
        return this;
    }
}