spring batch 使用ItemReaderAdapter导致读写无限循环问题

最新推荐文章于 2023-05-18 15:01:02 发布

lannal_w

最新推荐文章于 2023-05-18 15:01:02 发布

阅读量2.3k

点赞数

分类专栏： java 文章标签： spring batch

本文链接：https://blog.csdn.net/topxixi/article/details/75455699

版权

java 专栏收录该内容

16 篇文章 1 订阅

订阅专栏

当同步数据时reader 或者 writer 调用现有的 service时候，会用到ItemReaderAdapter或者ItemWriterAdapter

例如

ItemReaderAdapter adapter = new ItemReaderAdapter();
adapter.setTargetObject(orderService);
adapter.setTargetMethod("dealOrder");

但是上面的写法，会导致reader 和 writer 无限循环，不能停止因此需要reader 计数 重新实现 ItemStream
public class OrderAdapterReader extends AbstractItemCountingItemStreamItemReader {

    @Autowired
    private OrderService orderService;
    @Autowired
    private HubOrderSqlOperation hubOrderSqlOperation;

    private List<MongoOrder> mongoOrderList;

    private final String ecName = "orderReaderContext";

    public OrderAdapterReader() {
        setName(ecName);
    }

    @Override
    protected MongoOrder doRead() throws Exception {
        MongoOrder mongoOrder = mongoOrderList.get(getCurrentItemCount() - 1);
        return mongoOrder;
    }

    @Override
    protected void doOpen() throws Exception {
        mongoOrderList = (List<MongoOrder>) orderService.getDayOrders().getData();
        if (CollectionUtils.isEmpty(mongoOrderList)){
            setMaxItemCount(0);
        }else{
            List<Long> orderIds = mongoOrderList.stream().map(MongoOrder::getId).collect(Collectors.toList());
            hubOrderSqlOperation.deleteOrderByIds(orderIds);
            setMaxItemCount(mongoOrderList.size());
        }

    }

    @Override
    protected void doClose() throws Exception {
        if (!CollectionUtils.isEmpty(mongoOrderList)){
            mongoOrderList.clear();
        }
        setMaxItemCount(0);
        setCurrentItemCount(0);
    }
}