Spring Batch如何获取ItemWriter插入的主键用于另一表更新

要实现插入tableA后获取主键并更新tableB，核心是让插入操作返回生成的主键并赋值给TableA对象，后续更新操作直接使用该对象的主键属性即可。具体步骤如下：

1. 确保TableA实体包含主键字段及Setter方法 #

你的TableA类需要定义对应tableA主键的属性（比如id），并提供Setter方法，这样Spring Batch才能将生成的主键赋值进去：

public class TableA {
    private Long id; // 对应tableA的主键id
    private String name;
    // 其他业务字段...

    // 主键的Getter和Setter方法
    public void setId(Long id) {
        this.id = id;
    }

    public Long getId() {
        return id;
    }

    // 其他字段的Getter/Setter...
}

2. 修改processTableA配置，开启主键返回 #

调整JdbcBatchItemWriter的配置，开启返回生成的主键，并指定主键列名，插入后主键会自动赋值到TableA对象的对应属性：

@Bean
public ItemWriter<TableA> processTableA() {
    return new JdbcBatchItemWriterBuilder<TableA>()
            .dataSource(dataSource)
            .sql("insert into tableA (id, name) values (tableA_id_seq.nextval, :name)")
            .beanMapped()
            // 开启返回生成的主键
            .returnGeneratedKeys(true)
            // 指定数据库表中的主键列名
            .keyColumnName("id")
            .build();
}

3. 修正updateTableB的SQL及参数映射 #

现在TableA对象已持有插入生成的主键，直接在update语句中引用:id即可，同时修正原SQL的语法错误（移除末尾多余的括号）：

@Bean
public ItemWriter<TableA> updateTableB() {
    return new JdbcBatchItemWriterBuilder<TableA>()
            .dataSource(dataSource)
            // 使用TableA的主键更新tableB的对应字段
            .sql("update tableB set tableB_key = :id where id = :another_field")
            .beanMapped()
            .build();
}

4. 确认CompositeItemWriter的执行顺序 #

CompositeItemWriter会按照delegates列表的顺序执行writer，原配置先执行processTableA完成插入并赋值主键，再执行updateTableB，顺序正确无需修改：

@Bean
public CompositeItemWriter<TableA> distributionCompositeItemWriter() {
    CompositeItemWriter<TableA> compositeItemWriter = new CompositeItemWriter<>();
    compositeItemWriter.setDelegates(Arrays.asList(processTableA(), updateTableB()));
    return compositeItemWriter;
}

关键说明 #

• returnGeneratedKeys(true)：告知JdbcBatchItemWriter执行插入后返回数据库生成的主键
• keyColumnName("id")：指定数据库表的主键列名，Spring Batch会通过bean映射将该值赋值到TableA对象的同名属性（若属性名与列名不一致，可通过@Column注解做映射）
• 必须保证TableA对象有主键属性的Setter方法，否则无法完成主键值的赋值

内容的提问来源于stack exchange，提问作者LTurTk