package com.zy.asrs.importexcle;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.fastjson.JSON;
import com.core.exception.CoolException;
import com.zy.asrs.entity.LocDetl;
import com.zy.asrs.entity.LocMast;
import com.zy.asrs.mapper.LocDetlMapper;
import com.zy.asrs.service.LocMastService;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * @author pang.jiabao
 * @description 监听立库已有物料excel数据导入
 * @createDate 2023/10/9 11:13
 */
@Slf4j
public class LocMatListener extends AnalysisEventListener<LocMat> {

    /**
     * 每隔1000条存储数据库，实际使用中可以3000条，然后清理list ，方便内存回收
     */
    private static final int BATCH_COUNT = 400;

    private static int num = 1;

    private int count = 0;

    List<LocMat> list = new ArrayList<>();
    /**
     * 假设这个是一个DAO，当然有业务逻辑这个也可以是一个service。当然如果不用存储这个对象没用。
     */
    private final LocMastService locMastService;
    private final LocDetlMapper locDetlMapper;
    private final Long userId;

    /**
     * 如果使用了spring,请使用这个构造方法。每次创建Listener的时候需要把spring管理的类传进来
     */
    public LocMatListener(LocMastService locMastService,LocDetlMapper locDetlMapper, Long userId) {
        this.locMastService = locMastService;
        this.locDetlMapper = locDetlMapper;
        this.userId = userId;
    }

    /**
     * 这个每一条数据解析都会来调用
     */
    @SneakyThrows
    @Override
    public void invoke(LocMat data, AnalysisContext context) {
        log.info("解析到第 {} 条数据:{}", ++count, JSON.toJSONString(data));

        list.add(data);

        // 达到BATCH_COUNT了，需要去存储一次数据库，防止数据几万条数据在内存，容易OOM
        if (list.size() >= BATCH_COUNT) {
            saveData();
            // 存储完成清理 list
            list.clear();
        }
    }

    /**
     * 所有数据解析完成了 都会来调用
     */
    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        // 这里也要保存数据，确保最后遗留的数据也存储到数据库
        if (list.size() > 0) {
            saveData();
        }
        log.info("所有数据解析完成！");
    }

    /**
     * 加上存储数据库
     */
    private void saveData() {
        log.info("{}条数据，开始存储数据库！", list.size());
        List<LocDetl> locDetls = new ArrayList<>();
            list.forEach(locMat -> {
                LocMast locMast = locMastService.selectById(locMat.getA());
                if (!locMast.getLocSts().equals("O")) {
                    throw new CoolException(locMat.getA() + ":库位不为空！");
                }
                Date now = new Date();
                if ("F".equals(locMat.getB())) {
                    LocDetl locDetl = new LocDetl();
                    locDetl.setLocNo(locMat.getA());
                    locDetl.setZpallet("test" + num);
                    locDetl.setAnfme(1.0);
                    locDetl.setMatnr("3μm×10mm");
                    locDetl.setMaktx("3μm×10mm");
                    locDetl.setBatch("t" + num);
                    locDetl.setSpecs("1");
                    locDetl.setModel("t" + num);
                    locDetl.setColor(locMat.getC()); // 木箱类型
                    locDetl.setBrand("t" + num);
                    locDetl.setPrice(800.0);
                    locDetl.setOrigin("中");
                    locDetl.setWeight(900.0);
                    locDetl.setVolume(1000.0);
                    locDetl.setModiTime(now);
                    locDetl.setModiUser(userId);
                    locDetl.setAppeTime(now);
                    locDetl.setAppeUser(userId);

                    // 添加进库存明细列表
                    locDetls.add(locDetl);
                    num ++;
                }

                // 更新库位状态
                locMast.setLocSts(locMat.getB());
                locMastService.updateById(locMast);
            });
            // 批量插入库存明细
            if (!locDetls.isEmpty()) {
                locDetlMapper.batchDetls(locDetls);
            }

        log.info("存储数据库成功！");
    }

    /**
     *解析出现错误会进入该方法 具体看源代码或文档
     */
    @Override
    public void onException(Exception exception, AnalysisContext context) throws Exception {
        log.error("处理异常：" + exception.getMessage());
        throw exception;
    }
}