如何从JdbcTemplate batchUpdate拆分获取成败、错误、异常的计数与ID？

首先咱们先把几个核心概念理清楚，避免后续混淆：

• 成功ID：更新语句正常执行且确实修改了数据（更新行数≥1）的记录ID
• 错误ID：因为SQL错误（比如数值超出范围、主键冲突等）被IGNORE关键字跳过的记录ID
• 失败ID：更新语句执行成功但没有匹配到任何行（更新行数=0）的记录ID
• 异常：执行批量更新时抛出的非SQL错误类异常（比如数据库连接中断、SQL语法错误等，这类会直接终止批量操作）

接下来一步步实现需求：

1. 绑定ID与更新参数 #

首先要把每个待更新的参数和对应的记录ID绑定，这样后续能根据批量更新的结果数组，精准对应到具体ID。可以用自定义类或者Map来存储：

// 自定义类存储更新记录与ID
class UpdateRecord {
    private Long id;
    private String column1;
    private Integer column2;

    public UpdateRecord(Long id, String column1, Integer column2) {
        this.id = id;
        this.column1 = column1;
        this.column2 = column2;
    }

    // Getter方法
    public Long getId() { return id; }
    public String getColumn1() { return column1; }
    public Integer getColumn2() { return column2; }
}

// 填充待更新的记录列表
List<UpdateRecord> updateRecords = new ArrayList<>();
updateRecords.add(new UpdateRecord(1L, "val1", 100));
updateRecords.add(new UpdateRecord(2L, "val2", 999999999)); // 假设该值超出字段范围
updateRecords.add(new UpdateRecord(3L, "val3", 200)); // 假设该ID不存在于表中

2. 执行批量更新并捕获核心结果 #

执行批量更新时，先捕获会终止批量操作的严重异常，同时获取executeBatch()返回的更新行数数组：

Connection conn = null;
PreparedStatement pstmt = null;
int[] updateCounts = null;
List<Long> exceptionIds = new ArrayList<>();

try {
    conn = getYourDatabaseConnection(); // 替换成你的数据库连接获取逻辑
    String sql = "UPDATE IGNORE your_table SET column1 = ?, column2 = ? WHERE id = ?";
    pstmt = conn.prepareStatement(sql);

    // 批量添加更新语句
    for (UpdateRecord record : updateRecords) {
        pstmt.setString(1, record.getColumn1());
        pstmt.setInt(2, record.getColumn2());
        pstmt.setLong(3, record.getId());
        pstmt.addBatch();
    }

    // 执行批量更新
    updateCounts = pstmt.executeBatch();
} catch (SQLException e) {
    // 这里捕获的是会终止批量的严重异常（比如连接断开、SQL语法错误）
    // 简单处理：将所有待更新ID加入异常列表，也可根据批次执行进度筛选未执行的ID
    exceptionIds = updateRecords.stream().map(UpdateRecord::getId).collect(Collectors.toList());
    e.printStackTrace();
} finally {
    // 关闭资源
    if (pstmt != null) pstmt.close();
    if (conn != null) conn.close();
}

3. 解析结果，区分各类ID与计数 #

updateCounts数组的每个元素对应批次中对应位置的更新行数，但值为0时可能是「无匹配行」或「被IGNORE跳过的错误行」，需要通过MySQL的警告信息区分：

List<Long> successIds = new ArrayList<>();
List<Long> failIds = new ArrayList<>();
List<Long> errorIds = new ArrayList<>();

if (updateCounts != null && updateCounts.length == updateRecords.size()) {
    // 获取MySQL返回的所有警告信息（IGNORE会把错误转为警告）
    SQLWarning warning = pstmt.getWarnings();
    Map<Long, Boolean> idHasError = new HashMap<>();

    // 遍历警告，解析出被跳过的错误行ID
    while (warning != null) {
        String message = warning.getMessage();
        // 匹配MySQL警告中的行号（格式示例："Row 2 was skipped due to data errors"）
        Pattern pattern = Pattern.compile("Row (\\d+) was skipped");
        Matcher matcher = pattern.matcher(message);
        if (matcher.find()) {
            int rowIndex = Integer.parseInt(matcher.group(1)) - 1; // 转为数组的0索引
            if (rowIndex >= 0 && rowIndex < updateRecords.size()) {
                Long errorId = updateRecords.get(rowIndex).getId();
                idHasError.put(errorId, true);
                errorIds.add(errorId);
            }
        }
        warning = warning.getNextWarning();
    }

    // 遍历更新计数，分类匹配对应的ID
    for (int i = 0; i < updateCounts.length; i++) {
        UpdateRecord record = updateRecords.get(i);
        Long id = record.getId();
        if (idHasError.containsKey(id)) {
            continue; // 已标记为错误ID，跳过
        }
        if (updateCounts[i] > 0) {
            successIds.add(id);
        } else {
            failIds.add(id);
        }
    }
}

// 统计各类计数
int successCount = successIds.size();
int failCount = failIds.size();
int errorCount = errorIds.size();
int exceptionCount = exceptionIds.size();

关键注意事项 #

• rewriteBatchedStatements=true的兼容性：开启该参数后，MySQL会合并批量语句，但警告信息中的行号仍与你添加批次的顺序一致，所以行号提取逻辑有效。
• 警告消息格式适配：不同MySQL版本的警告消息格式可能略有差异，需根据实际测试结果调整正则表达式。
• 事务与原子性：使用IGNORE后，错误行不会终止批量操作，成功行的修改会生效，若需要原子性需结合事务逻辑调整。
• 异常重试逻辑：若捕获到严重异常，可根据业务需求对exceptionIds中的ID进行重试。

内容的提问来源于stack exchange，提问作者Explain Down Vote