Spring Batch：如何用JpaItemWriter实现同表不同po_id的数据插入？

要实现从PRODUCT表读取po_id=1001的数据并以po_id=1002插入同表，仅修改JpaItemWriter是不够的——默认情况下JpaItemWriter会对传入的实体执行merge操作（更新原记录），而非插入新记录。你需要添加一个ItemProcessor来复制原对象并修改属性，再配合JpaItemWriter完成插入。

1. 实现ItemProcessor复制并修改数据 #

创建处理器类，复制读取到的Product对象，设置新的po_id并重置主键（确保JPA识别为新实体执行插入）：

import org.springframework.batch.item.ItemProcessor;

public class ProductCopyProcessor implements ItemProcessor<Product, Product> {

    @Override
    public Product process(Product original) throws Exception {
        // 实例化新的Product对象，避免修改原读取的实体
        Product newProduct = new Product();
        
        // 复制原对象的所有业务属性（根据你的Product实体类调整字段）
        newProduct.setName(original.getName());
        newProduct.setPrice(original.getPrice());
        newProduct.setOrder(original.getOrder()); // 如果存在与ORDER表的关联，同步复制
        
        // 设置目标po_id
        newProduct.setPoId(1002);
        
        // 重置主键（如果主键是自增策略，设为null让JPA自动生成）
        newProduct.setId(null);
        
        return newProduct;
    }
}

2. 配置Step关联Processor和Writer #

将处理器加入你的Step配置，确保数据先经过处理再写入：

@Bean
public Step productCopyStep(StepBuilderFactory stepBuilderFactory) {
    return stepBuilderFactory.get("productCopyStep")
            .<Product, Product>chunk(10) // 按批次处理，可根据性能调整大小
            .reader(jpaCursorItemReader())
            .processor(productCopyProcessor())
            .writer(jpaItemWriter())
            .build();
}

@Bean
public ProductCopyProcessor productCopyProcessor() {
    return new ProductCopyProcessor();
}

3. JpaItemWriter的调整 #

你的JpaItemWriter基础配置无需大幅修改，但要确保传入的是新实例（如上述Processor生成的对象）。默认情况下，JPA会识别无主键的实体执行persist（插入）操作，所以原Writer代码保持即可：

public JpaItemWriter<Product> jpaItemWriter() {
    JpaItemWriter<Product> jpaItemWriter = new JpaItemWriter<>();
    jpaItemWriter.setEntityManagerFactory(postgresqlEntityManagerFactory);
    return jpaItemWriter;
}

注意事项 #

• 若你的Product实体主键不是自增，需确保新对象的主键值唯一，避免主键冲突；
• 若存在与ORDER表的关联关系，需根据业务需求确认是否同步复制关联对象，或调整关联逻辑。

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