极客笔记MySQL BatchSqlUpdate – 如何获取自动生成的键
当我们在使用Java编程语言与MySQL数据库进行交互时，我们经常需要在数据库中插入多条记录。使用MySQL的Batch插入可以极大地提高性能。Batch插入方法需要我们提供一个由多个SQL语句组成的列表，一次性执行所有的语句，可以避免多次与数据库进行通信的开销。在这种情况下，我们可能需要获取实例化的记录的自动生成的键。在这篇文章中，我们将学习如何在BatchSQLUpdate中获取自动生成的键。
阅读更多：MySQL 教程
Spring JdbcTemplate
Spring JdbcTemplate是Spring Framework的核心组件之一，它简化了JDBC编程的复杂性。Spring JdbcTemplate提供了大量的API来帮助我们执行SQL语句并处理结果。BatchSqlUpdate是其实现之一。
BatchSqlUpdate是Spring JdbcTemplate的一个典型用法，它允许我们一次性执行多个SQL语句，可以通过使用PreparedStatement批量插入或更新。BatchSqlUpdate中的API，用于进行此类操作，有：
public class BatchSqlUpdate extends AbstractSqlUpdate implements SqlProvider {
public BatchSqlUpdate(){}
public BatchSqlUpdate(DataSource dataSource, String sql){}
public BatchSqlUpdate(JdbcTemplate jdbcTemplate, String sql){}
public void setBatchSize(int batchSize){}
public void setParameters(List<?> parameters){}
public int[] executeBatch(){}
}
我们可以使用setBatchSize方法设置一次性要插入的记录数，并使用setParameters方法设置我们要插入的记录列表。最后，我们可以通过调用executeBatch()方法来执行批量插入操作。
Batch插入示例
下面的示例将向MySQL数据库中的student表插入3条记录。每个记录有两个字段 – 名称和年龄。我们将使用BatchSqlUpdate进行插入。
1. 准备数据
首先，我们准备我们要插入的数据。在我们的示例中，我们将创建一个Java List对象，其元素类型为Object []。每个数组元素都是一个由两个子元素组成的对象数组，表示姓名和年龄：
List<Object[]> students = new ArrayList<>();
students.add(new Object[]{"John", 20});
students.add(new Object[]{"Tom", 19});
students.add(new Object[]{"Jane", 21});
2. 创建BatchSqlUpdate实例
接下来，我们创建一个BatchSqlUpdate实例，用于将数据插入MySQL数据库中的student表中。在创建BatchSqlUpdate实例时，我们需要传递一个数据源和要执行的SQL语句。在我们的示例中，我们的SQL语句如下：
INSERT INTO student(name, age) VALUES(?, ?)
这个语句可以用来插入记录到student表中。注意该语句中的问号，它们是占位符，稍后将由我们的数据列表的实际值替换。
DataSource dataSource = ... // 数据源
BatchSqlUpdate batchUpdate = new BatchSqlUpdate(dataSource, "INSERT INTO student(name, age) VALUES(?, ?)");
3. 设置批量更新大小和插入数据
现在，我们还需要设置批量更新大小（batch size）和要插入的数据。在我们的示例中，我们将设置批量更新大小为2，意味着我们将一次性插入2条记录。为了设置这个值，我们调用BatchSqlUpdate的setBatchSize()方法。
batchUpdate.setBatchSize(2);
然后，我们将设置要插入的数据。在我们的示例中，我们将使用students列表来代表我们要插入的数据，通过调用BatchSqlUpdate的setParameters()方法来设置它：
batchUpdate.setParameters(students);
4. 执行批量插入
现在，我们已经准备好了执行批量插入操作。我们可以调用BatchSqlUpdate的executeBatch()方法执行插入操作。执行批量插入操作后，我们将得到一个int[]类型的数组，它包含每个SQL语句影响的行数。在我们的示例中，我们将对student表中的三个记录执行批量插入操作。因此，该数组应该具有以下值：
int[] result = batchUpdate.executeBatch();
// result为 [1, 1, 1]
请注意，这个数组只包含每个SQL语句影响的行数。在我们的示例中，每个SQL语句只影响了一行。如果您的SQL语句影响了多行，那么数组中的元素数也将增加。
获取自动生成的键
在某些情况下，我们需要获取批处理插入操作后，每个自动生成的主键。我们可以通过使用JdbcTemplate中的一些方法来获取这些主键。在下面的示例中，我们将演示如何获取自动生成的主键。
1. 准备数据
我们将使用与上面示例中相同的数据，向MySQL数据库中的student表插入3条记录。
2. 创建BatchSqlUpdate实例
我们将创建与上面示例中相同的BatchSqlUpdate实例，用于将数据插入MySQL数据库的student表中。
DataSource dataSource = ... // 数据源
BatchSqlUpdate batchUpdate = new BatchSqlUpdate(dataSource, "INSERT INTO student(name, age) VALUES(?, ?)");
batchUpdate.setBatchSize(2);
3. 定义KeyHolder
我们需要定义一个KeyHolder实例来存储自动生成的主键。在我们的示例中，我们将使用GeneratedKeyHolder来存储主键：
KeyHolder keyHolder = new GeneratedKeyHolder();
4. 设置批量更新大小、插入数据和键持有者
接下来，我们需要设置批量更新大小（batch size）、要插入的数据和键持有者。要设置这些值，我们采用与上面示例中相同的方式：
batchUpdate.setBatchSize(2);
batchUpdate.setParameters(students);
batchUpdate.setGeneratedKeysColumnNames(new String[] { "id" }); //指定自动生成的主键的列名
batchUpdate.setReturnGeneratedKeys(true);
batchUpdate.update(keyHolder);
请注意，我们在此处调用了BatchSqlUpdate的update()方法来执行批量插入操作。该方法也可将KeyHolder实例作为参数进行传递，以捕获自动生成的键。
5. 获取自动生成的键
现在，我们可以通过KeyHolder实例来获取自动生成的主键。我们可以调用getKey()方法来获取单个主键值：
long firstGeneratedId = (long) keyHolder.getKeyList().get(0).get("id");
long secondGeneratedId = (long) keyHolder.getKeyList().get(1).get("id");
long thirdGeneratedId = (long) keyHolder.getKeyList().get(2).get("id");
这些值将分别存储每个插入操作的自动生成的主键。在我们的示例中，它们可能会分别是1、2和3（如果MySQL数据库配置为自动递增主键）。
总结
在本文中，我们学习了如何使用Spring JdbcTemplate的BatchSqlUpdate来批量插入多个数据。我们还学习了如何在BatchSqlUpdate中获取自动生成的键。BatchSqlUpdate提供了一个将多个SQL语句一次性插入给定数据的快速和高效的方法。当我们需要向数据库中插入多个记录时，BatchSqlUpdate是一个很好的选择。