package com.gkhy.hazmat.framework.listener;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.gkhy.hazmat.common.config.IdTableNameHandler;
import com.gkhy.hazmat.common.enums.CodePrexEnum;
import com.gkhy.hazmat.common.enums.OperateStatusEnum;
import com.gkhy.hazmat.common.listener.EntryEvent;
import com.gkhy.hazmat.system.domain.HzHazmat;
import com.gkhy.hazmat.system.domain.HzHazmatFlow;
import com.gkhy.hazmat.system.domain.HzProduct;
import com.gkhy.hazmat.system.domain.HzProductFlow;
import com.gkhy.hazmat.system.service.HzHazmatFlowService;
import com.gkhy.hazmat.system.service.HzHazmatService;
import com.gkhy.hazmat.system.service.HzProductFlowService;
import com.gkhy.hazmat.system.service.HzProductService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.event.EventListener;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

@Component
public class CustomListener {
    @Autowired
    private HzHazmatService hazmatService;
    @Autowired
    private HzProductService productService;
    @Autowired
    private HzHazmatFlowService hazmatFlowService;
    @Autowired
    private HzProductFlowService productFlowService;

    @Async("threadPoolTaskExecutor")
    @EventListener
    public void addFlow(EntryEvent hazmatEntryEvent){

        String codePrex=hazmatEntryEvent.getCodePrex();
        if(codePrex.equals(CodePrexEnum.MATERIAL.getCode())){
            generateHazmatFlow(hazmatEntryEvent);
        }else{
            generateProductFlow(hazmatEntryEvent);
        }
    }

    private void generateHazmatFlow(EntryEvent hazmatEntryEvent) {
        Integer pageSize=100;
        Integer pageIndex=1;
        List<HzHazmatFlow> hazmatFlowList=new ArrayList<>();
        //设置分表id
        IdTableNameHandler.setCurrentId(hazmatEntryEvent.getCompanyId());
        while (true){
            LambdaQueryWrapper<HzHazmat> eq = Wrappers.<HzHazmat>lambdaQuery().eq(true, HzHazmat::getEntryId, hazmatEntryEvent.getEntryRecordId());
            eq.last(" limit "+((pageIndex-1)*pageSize)+","+pageSize);
            List<HzHazmat> hazmatList=hazmatService.list(eq);
            if(hazmatList.isEmpty()){
                break;
            }
            hazmatFlowList=hazmatList.stream().map(item -> {
                //生成流向
                HzHazmatFlow hazmatFlow=new HzHazmatFlow();
                hazmatFlow.setHazmatId(item.getId());
                hazmatFlow.setBasicId(item.getBasicId());
                hazmatFlow.setState(OperateStatusEnum.ENTRY.getCode());
                hazmatFlow.setCompanyId(hazmatEntryEvent.getCompanyId());
                hazmatFlow.setNum(item.getRemaining());
                hazmatFlow.setCreateId(hazmatEntryEvent.getUserId());
                return hazmatFlow;
            }).collect(Collectors.toList());
            hazmatFlowService.saveBatch(hazmatFlowList);
            hazmatFlowList.clear();
            pageIndex++;
        }
        IdTableNameHandler.removeCurrentId();
    }

    private void generateProductFlow(EntryEvent hazmatEntryEvent) {
        Integer pageSize=100;
        Integer pageIndex=1;
        List<HzProductFlow> productFlowList=new ArrayList<>();
        IdTableNameHandler.setCurrentId(hazmatEntryEvent.getCompanyId());
        while (true){
            LambdaQueryWrapper<HzProduct> eq = Wrappers.<HzProduct>lambdaQuery().eq(true, HzProduct::getEntryId, hazmatEntryEvent.getEntryRecordId());
            eq.last(" limit "+((pageIndex-1)*pageSize)+","+pageSize);
            List<HzProduct> productList=productService.list(eq);
            if(productList.isEmpty()){
                break;
            }
            productFlowList=productList.stream().map(item -> {
                //生成流向
                HzProductFlow productFlow=new HzProductFlow();
                productFlow.setProductId(item.getId());
                productFlow.setBasicId(item.getBasicId());
                productFlow.setState(OperateStatusEnum.ENTRY.getCode());
                productFlow.setCompanyId(hazmatEntryEvent.getCompanyId());
                productFlow.setNum(item.getRemaining());
                productFlow.setCreateId(hazmatEntryEvent.getUserId());
                return productFlow;
            }).collect(Collectors.toList());
            productFlowService.saveBatch(productFlowList);
            productFlowList.clear();
            pageIndex++;
        }
        IdTableNameHandler.removeCurrentId();
    }
}