[英]BatchSqlUpdate - how to get auto generated keys
I am using spring BatchSqlUpdate to insert a set of rows.我正在使用 spring BatchSqlUpdate 插入一组行。 How do I get the auto generated keys for all of the rows inserted?如何为所有插入的行获取自动生成的键?
When doing a single insert I get the keys like this -进行单次插入时,我得到这样的键-
SqlUpdate sqlUpdate = new SqlUpdate(dataSource, sqlTemplate.toString());
sqlUpdate.setReturnGeneratedKeys(true);
KeyHolder keyHolder = new GeneratedKeyHolder();
sqlUpdate.update(new Object[] {}, keyHolder);
return keyHolder.getKey().longValue();
Thanks!谢谢!
a template proc class:模板过程 class:
public abstract class BatchPreparedStatementSetterWithKeyHolder<T> implements BatchPreparedStatementSetter {
private final List<T> beans;
/**
* @param datas
* @param returnGeneratedKeys true设置{@linkplain Statement#RETURN_GENERATED_KEYS}
*/
public BatchPreparedStatementSetterWithKeyHolder(List<T> beans) {
this.beans = beans;
}
@Override
public void setValues(PreparedStatement ps, int i) throws SQLException {
setValues(ps, beans.get(i));
}
@Override
public final int getBatchSize() {
return beans.size();
}
public void setPrimaryKey(KeyHolder keyHolder) {
List<Map<String, Object>> keys = keyHolder.getKeyList();
for (int i = 0, len = keys.size(); i < len; i++) {
setPrimaryKey(keys.get(i), beans.get(i));
}
}
protected abstract void setValues(PreparedStatement ps, T bean) throws SQLException;
protected abstract void setPrimaryKey(Map<String, Object> primaryKey, T bean);
}
a util method batchupdate:一个util方法batchupdate:
private static void generatedKeys(PreparedStatement ps, KeyHolder keyHolder) throws SQLException {
List<Map<String, Object>> keys = keyHolder.getKeyList();
ResultSet rs = ps.getGeneratedKeys();
if (rs == null) return;
try {
keys.addAll(new RowMapperResultSetExtractor<Map<String, Object>>(new ColumnMapRowMapper(), 1).extractData(rs));
} finally {
UtilIO.close(rs);
}
}
/**
* 批量更新
* @param jdbcTemplate
* @param sql
* @param pss
* @param keyHolder 存储主键,如果要存储主键,就必须传入此对象
* @return sql执行结果
* @see JdbcTemplate#batchUpdate(String, org.springframework.jdbc.core.BatchPreparedStatementSetter)
*/
public static <T> int[] batchUpdateWithKeyHolder(JdbcTemplate jdbcTemplate, final String sql, final BatchPreparedStatementSetterWithKeyHolder<T> pss) {
return jdbcTemplate.execute(new PreparedStatementCreator() {
@Override
public PreparedStatement createPreparedStatement(Connection con) throws SQLException {
return con.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
}
}, new PreparedStatementCallback<int[]>() {
@Override
public int[] doInPreparedStatement(PreparedStatement ps) throws SQLException, DataAccessException {
try {
int batchSize = pss.getBatchSize();
InterruptibleBatchPreparedStatementSetter ipss =
(pss instanceof InterruptibleBatchPreparedStatementSetter ?
(InterruptibleBatchPreparedStatementSetter) pss : null);
int[] result;
KeyHolder keyHolder = new GeneratedKeyHolder();
try {
if (JdbcUtils.supportsBatchUpdates(ps.getConnection())) {
for (int i = 0; i < batchSize; i++) {
pss.setValues(ps, i);
if (ipss != null && ipss.isBatchExhausted(i)) break;
ps.addBatch();
}
result = ps.executeBatch();
generatedKeys(ps, keyHolder);
} else {
List<Integer> rowsAffected = new ArrayList<Integer>();
for (int i = 0; i < batchSize; i++) {
pss.setValues(ps, i);
if (ipss != null && ipss.isBatchExhausted(i)) break;
rowsAffected.add(ps.executeUpdate());
generatedKeys(ps, keyHolder);
}
result = UtilObj.tointArray(rowsAffected);
}
} finally {
pss.setPrimaryKey(keyHolder);
}
return result;
} finally {
if (pss instanceof ParameterDisposer) ((ParameterDisposer) pss).cleanupParameters();
}
}
});
}
ex:前任:
UtilJdbc.batchUpdateWithKeyHolder(jdbcTemplate,
"insert into tbe_vm_node (creator_id, host, ssh_ip, ssh_user, ssh_passwd)" +
" values (?, ?, ?, ?, ?)",
new BatchPreparedStatementSetterWithKeyHolder<VmNode>(vmNodes) {
@Override
protected void setValues(PreparedStatement ps, VmNode vmNode) throws SQLException {
UtilJdbc.setValues(ps, vmNode.getCreatorId(), vmNode.getHost(),
vmNode.getSshIp(), vmNode.getSshUser(), vmNode.getSshPasswd());
}
@Override
protected void setPrimaryKey(Map<String, Object> primaryKey, VmNode vmNode) {
vmNode.setId((Long) primaryKey.get("abc"));
}
});
KeyHolder keyHolder = new GeneratedKeyHolder();
BatchSqlUpdate updateBatch = new BatchSqlUpdate();
updateBatch.setSql("Some INSERT SQL");
updateBatch.setJdbcTemplate(jdbcTemp);
updateBatch.setReturnGeneratedKeys(true);
updateBatch.updateByNamedParam( paramsMap ,keyHolder);
updateBatch.flush();
Simple answer: use the getKeyList() method简单的答案:使用 getKeyList() 方法
KeyHolder keyHolder = new GeneratedKeyHolder();
// Specify which column IDs to return, or get all columns by default
jdbcTemplate.update(INSERT_SQL, new MapSqlParameterSource(), keyHolder, new String[] {"column_name"});
List<Map<String, Object>> keyList = keyHolder.getKeyList();
Thanks for the provided solution, I implemented that and it worked great.感谢您提供的解决方案,我实现了它并且效果很好。 However, you have references to private classes (UtilIo, UtilObj) where I had to interpret what you were doing.但是,您引用了私有类(UtilIo、UtilObj),我必须在其中解释您在做什么。 I replaced "UtilIO.close(rs);"我替换了“UtilIO.close(rs);” with:和:
if (rs != null) {
rs.close();
}
And I replaced "result = UtilObj.tointArray(rowsAffected);"我替换了“result = UtilObj.tointArray(rowsAffected);” with this Java 8 code:使用此 Java 8 代码:
result = rowsAffected.stream().mapToInt(Integer::intValue).toArray();
There is no provided solution for this using BatchSqlUpdate
as far as I know, but you can always据我所知,没有提供使用BatchSqlUpdate
的解决方案,但你总是可以
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.