Spring Batch - 一次写入 2 个表

Question

我有一个包含地址列表的 Person 对象。我的 Spring Batch 应用程序将 Person JSON 记录转换为 Person POJO 并将它们写入数据库。

通常，我会使用 JdbcBatchItemWriter，但我发现它仅限于写入 2 个单独的表。我需要写入人员和地址表。

Hibernate 可以一步完成此操作，但我在 Spring Bathc 中没有看到类似的功能。

这个SO问题Multiple itemwriters in Spring Batch建议使用CompositeItemWriter，但问题是，我希望我的Address表包含一个名为Person_id的列，以将地址链接回Person。我无法通过两次写入来做到这一点。

有什么建议么？当然，Batch 可以满足更复杂的写入需求

Answer 1

我从评论中看到，您不能使用 ORM 或保存 Person 及其地址的服务。在这种情况下，您可以使用自定义编写器：

import java.util.List;

import javax.sql.DataSource;

import org.springframework.batch.item.ItemWriter;
import org.springframework.jdbc.core.JdbcTemplate;

public class PersonWriter implements ItemWriter<Person> {

    private JdbcTemplate jdbcTemplate;

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

    @Override
    public void write(List<? extends Person> items) {
        for (Person person : items) {
            jdbcTemplate.update("INSERT INTO Person ...");
            for (Address address : person.addresses) {
                jdbcTemplate.update("INSERT INTO Address ... where person_id = ?", person.getId()); 
            }
        }
    }
}

另一种方法是委托给两个 jdbc 批处理项目编写器：

import java.util.Collections;
import java.util.List;

import org.springframework.batch.item.ItemWriter;
import org.springframework.batch.item.database.JdbcBatchItemWriter;

public class PersonItemWriter implements ItemWriter<Person> {

    private JdbcBatchItemWriter<Person> personWriter;
    private JdbcBatchItemWriter<Address> addressWriter;

    public PersonItemWriter(JdbcBatchItemWriter<Person> personWriter, JdbcBatchItemWriter<Address> addressWriter) {
        this.personWriter = personWriter;
        this.addressWriter = addressWriter;
    }

    @Override
    public void write(List<? extends Person> persons) throws Exception {
        for (Person person : persons) {
            personWriter.write(Collections.singletonList(person));
            List<Address> addresses = person.getAddresses();
            for (Address address : addresses) {
                address.setPersonId(person.getId());
            }
            addressWriter.write(addresses);
        }
    }
}