spring-jdbc - 如何使用Spring JDBC模板批处理在数据库中进行多次插入？

Question

我需要一次将数千条记录插入数据库中。我在我的应用程序中使用spring JDBC模板。

下面是我到目前为止编写的代码，可以一次性执行所有插入操作。因此，如果我拥有10,000个用户，则将它们一次插入。但是我想要的是分批执行它们，例如一批执行500条记录，依此类推。

@Override
public void saveBatch(final List<Employee> employeeList) {
    final int batchSize = 500;

    getJdbcTemplate().batchUpdate(QUERY_SAVE,
            new BatchPreparedStatementSetter() {
                @Override
                public void setValues(PreparedStatement ps, int i)
                        throws SQLException {
                    Employee employee = employeeList.get(i);
                    ps.setString(1, employee.getFirstname());
                    ps.setString(2, employee.getLastname());
                    ps.setString(3, employee.getEmployeeIdOnSourceSystem());
                }

                @Override
                public int getBatchSize() {
                    return employeeList.size();
                }
            });

}

我如何更改上面的代码，以使我们可以将批量大小设为500，而不是employeeList.size()批量大小，然后执行它们，然后再执行下一个500，依此类推？

请帮忙。

Answer 1

我不确定是否可以单独使用JDBC模板来做到这一点。也许您可以通过将大列表切成批处理大小的块来逐步调用batchUpdate方法。

在这里看看:

@Override
public void saveBatch(final List<Employee> employeeList) {
    final int batchSize = 500;

    for (int j = 0; j < employeeList.size(); j += batchSize) {

        final List<Employee> batchList = employeeList.subList(j, j + batchSize > employeeList.size() ? employeeList.size() : j + batchSize);

        getJdbcTemplate().batchUpdate(QUERY_SAVE,
            new BatchPreparedStatementSetter() {
                @Override
                public void setValues(PreparedStatement ps, int i)
                        throws SQLException {
                    Employee employee = batchList.get(i);
                    ps.setString(1, employee.getFirstname());
                    ps.setString(2, employee.getLastname());
                    ps.setString(3, employee.getEmployeeIdOnSourceSystem());
                }

                @Override
                public int getBatchSize() {
                    return batchList.size();
                }
            });

    }
}