JDBC 批处理操作

如果将多个对同一预处理语句的调用批处理，大多数 JDBC 驱动程序会提供更好的性能。通过将更新分组到批处理中，可以限制到数据库的往返次数。

使用 JdbcTemplate 进行基本批处理操作

您可以通过实现特殊接口 BatchPreparedStatementSetter 的两个方法，并将该实现作为第二个参数传递到 batchUpdate 方法调用中来完成 JdbcTemplate 批处理。您可以使用 getBatchSize 方法提供当前批处理的大小。您可以使用 setValues 方法设置预处理语句的参数值。此方法将按照您在 getBatchSize 调用中指定的次数被调用。以下示例根据列表中的条目更新 t_actor 表，并将整个列表用作批处理。

  • Java

  • Kotlin

public class JdbcActorDao implements ActorDao {

	private JdbcTemplate jdbcTemplate;

	public void setDataSource(DataSource dataSource) {
		this.jdbcTemplate = new JdbcTemplate(dataSource);
	}

	public int[] batchUpdate(final List<Actor> actors) {
		return this.jdbcTemplate.batchUpdate(
				"update t_actor set first_name = ?, last_name = ? where id = ?",
				new BatchPreparedStatementSetter() {
					public void setValues(PreparedStatement ps, int i) throws SQLException {
						Actor actor = actors.get(i);
						ps.setString(1, actor.getFirstName());
						ps.setString(2, actor.getLastName());
						ps.setLong(3, actor.getId().longValue());
					}
					public int getBatchSize() {
						return actors.size();
					}
				});
	}

	// ... additional methods
}
class JdbcActorDao(dataSource: DataSource) : ActorDao {

	private val jdbcTemplate = JdbcTemplate(dataSource)

	fun batchUpdate(actors: List<Actor>): IntArray {
		return jdbcTemplate.batchUpdate(
				"update t_actor set first_name = ?, last_name = ? where id = ?",
				object: BatchPreparedStatementSetter {
					override fun setValues(ps: PreparedStatement, i: Int) {
						ps.setString(1, actors[i].firstName)
						ps.setString(2, actors[i].lastName)
						ps.setLong(3, actors[i].id)
					}

					override fun getBatchSize() = actors.size
				})
	}

	// ... additional methods
}

如果您处理更新流或从文件中读取，您可能有一个首选的批处理大小，但最后一个批处理可能没有该数量的条目。在这种情况下，您可以使用 InterruptibleBatchPreparedStatementSetter 接口，它允许您在输入源耗尽后中断批处理。isBatchExhausted 方法允许您发出批处理结束的信号。

使用对象列表进行批处理操作

JdbcTemplate 和 NamedParameterJdbcTemplate 都提供了一种提供批处理更新的替代方法。您不必实现特殊的批处理接口，而是在调用中将所有参数值作为列表提供。该框架遍历这些值并使用内部预处理语句设置器。API 会根据您是否使用命名参数而有所不同。对于命名参数，您需要提供一个 SqlParameterSource 数组，批处理中的每个成员对应一个条目。您可以使用 SqlParameterSourceUtils.createBatch 便利方法来创建此数组，传入一个 bean 样式对象数组（具有与参数对应的 getter 方法）、String 键控的 Map 实例（包含相应的参数作为值），或两者的混合。

以下示例展示了使用命名参数进行批处理更新。

  • Java

  • Kotlin

public class JdbcActorDao implements ActorDao {

	private NamedParameterTemplate namedParameterJdbcTemplate;

	public void setDataSource(DataSource dataSource) {
		this.namedParameterJdbcTemplate = new NamedParameterJdbcTemplate(dataSource);
	}

	public int[] batchUpdate(List<Actor> actors) {
		return this.namedParameterJdbcTemplate.batchUpdate(
				"update t_actor set first_name = :firstName, last_name = :lastName where id = :id",
				SqlParameterSourceUtils.createBatch(actors));
	}

	// ... additional methods
}
class JdbcActorDao(dataSource: DataSource) : ActorDao {

	private val namedParameterJdbcTemplate = NamedParameterJdbcTemplate(dataSource)

	fun batchUpdate(actors: List<Actor>): IntArray {
		return this.namedParameterJdbcTemplate.batchUpdate(
				"update t_actor set first_name = :firstName, last_name = :lastName where id = :id",
				SqlParameterSourceUtils.createBatch(actors));
	}

		// ... additional methods
}

对于使用经典 ? 占位符的 SQL 语句，您需要传入一个包含对象数组的列表，其中包含更新值。此对象数组必须为 SQL 语句中的每个占位符包含一个条目，并且它们必须与 SQL 语句中定义的顺序相同。

以下示例与前面的示例相同，只是它使用经典的 JDBC ? 占位符。

  • Java

  • Kotlin

public class JdbcActorDao implements ActorDao {

	private JdbcTemplate jdbcTemplate;

	public void setDataSource(DataSource dataSource) {
		this.jdbcTemplate = new JdbcTemplate(dataSource);
	}

	public int[] batchUpdate(final List<Actor> actors) {
		List<Object[]> batch = new ArrayList<>();
		for (Actor actor : actors) {
			Object[] values = new Object[] {
					actor.getFirstName(), actor.getLastName(), actor.getId()};
			batch.add(values);
		}
		return this.jdbcTemplate.batchUpdate(
				"update t_actor set first_name = ?, last_name = ? where id = ?",
				batch);
	}

	// ... additional methods
}
class JdbcActorDao(dataSource: DataSource) : ActorDao {

	private val jdbcTemplate = JdbcTemplate(dataSource)

	fun batchUpdate(actors: List<Actor>): IntArray {
		val batch = mutableListOf<Array<Any>>()
		for (actor in actors) {
			batch.add(arrayOf(actor.firstName, actor.lastName, actor.id))
		}
		return jdbcTemplate.batchUpdate(
				"update t_actor set first_name = ?, last_name = ? where id = ?", batch)
	}

	// ... additional methods
}

我们前面描述的所有批处理更新方法都返回一个 int 数组，其中包含每个批处理条目受影响的行数。此计数由 JDBC 驱动程序报告。如果计数不可用，JDBC 驱动程序将返回 -2。

在这种情况下，当对底层 PreparedStatement 自动设置值时，每个值的相应 JDBC 类型需要从给定的 Java 类型派生。虽然这通常工作良好，但可能存在问题（例如，Map 中包含的 null 值）。Spring 默认在这种情况下调用 ParameterMetaData.getParameterType，这在您的 JDBC 驱动程序上可能很昂贵。如果您遇到应用程序的特定性能问题，您应该使用最新的驱动程序版本并考虑将 spring.jdbc.getParameterType.ignore 属性设置为 true（作为 JVM 系统属性或通过 SpringProperties 机制）。

自 6.1.2 版本起，Spring 绕过了 PostgreSQL 和 MS SQL Server 上的默认 getParameterType 解析。这是一种常见的优化，旨在避免为了参数类型解析而进一步往返于 DBMS，这在 PostgreSQL 和 MS SQL Server 上尤其是在批处理操作中已知会产生非常显著的差异。如果您遇到副作用，例如，在没有特定类型指示的情况下将字节数组设置为 null，您可以显式地将 spring.jdbc.getParameterType.ignore=false 标志设置为系统属性（如上所述），以恢复完整的 getParameterType 解析。

或者，您可以考虑显式指定相应的 JDBC 类型，方法是通过 BatchPreparedStatementSetter（如前所示），通过提供给基于 List<Object[]> 的调用的显式类型数组，通过自定义 MapSqlParameterSource 实例上的 registerSqlType 调用，通过 BeanPropertySqlParameterSource，即使是 null 值，它也能从 Java 声明的属性类型中推断出 SQL 类型，或者通过提供单个 SqlParameterValue 实例而不是纯 null 值。

多批次操作

前面批处理更新的例子处理的是批处理量非常大，以至于您希望将其分解为几个较小的批处理。您可以通过多次调用 batchUpdate 方法来使用前面提到的方法来实现这一点，但现在有一个更方便的方法。除了 SQL 语句之外，此方法还接受一个包含参数的 Collection 对象，每个批处理要进行的更新数量，以及一个 ParameterizedPreparedStatementSetter 用于设置预处理语句的参数值。该框架遍历提供的值并将更新调用分解为指定大小的批处理。

以下示例显示了使用批处理大小为 100 的批处理更新。

  • Java

  • Kotlin

public class JdbcActorDao implements ActorDao {

	private JdbcTemplate jdbcTemplate;

	public void setDataSource(DataSource dataSource) {
		this.jdbcTemplate = new JdbcTemplate(dataSource);
	}

	public int[][] batchUpdate(final Collection<Actor> actors) {
		int[][] updateCounts = jdbcTemplate.batchUpdate(
				"update t_actor set first_name = ?, last_name = ? where id = ?",
				actors,
				100,
				(PreparedStatement ps, Actor actor) -> {
					ps.setString(1, actor.getFirstName());
					ps.setString(2, actor.getLastName());
					ps.setLong(3, actor.getId().longValue());
				});
		return updateCounts;
	}

	// ... additional methods
}
class JdbcActorDao(dataSource: DataSource) : ActorDao {

	private val jdbcTemplate = JdbcTemplate(dataSource)

	fun batchUpdate(actors: List<Actor>): Array<IntArray> {
		return jdbcTemplate.batchUpdate(
					"update t_actor set first_name = ?, last_name = ? where id = ?",
					actors, 100) { ps, argument ->
			ps.setString(1, argument.firstName)
			ps.setString(2, argument.lastName)
			ps.setLong(3, argument.id)
		}
	}

	// ... additional methods
}

此调用的批处理更新方法返回一个 int 数组的数组，其中包含每个批处理的数组条目，以及每个更新受影响的行数数组。顶级数组的长度表示运行的批处理数量，第二级数组的长度表示该批处理中的更新数量。每个批处理中的更新数量应为所有批处理提供的批处理大小（除了最后一个可能较少），具体取决于提供的更新对象的总数。每个更新语句的更新计数是由 JDBC 驱动程序报告的。如果计数不可用，JDBC 驱动程序将返回 -2。

© . This site is unofficial and not affiliated with VMware.