udi-wms-java/src/main/java/com/glxp/api/service/inout/IoSplitCodeService.java

package com.glxp.api.service.inout;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.github.pagehelper.PageHelper;
import com.glxp.api.dao.inout.IoSplitCodeMapper;
import com.glxp.api.entity.basic.BasicSkPrescribeDiEntity;
import com.glxp.api.entity.basic.BasicSkPrescribeEntity;
import com.glxp.api.entity.inout.IoSplitCodeEntity;
import com.glxp.api.entity.inout.IoSplitFifoCodeEntity;
import com.glxp.api.req.inout.IoSplitCodeRequest;
import com.glxp.api.service.basic.BasicSkPrescribeDiService;
import com.glxp.api.service.basic.BasicSkPrescribeService;
import com.glxp.api.util.IntUtil;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

@Service
public class IoSplitCodeService extends ServiceImpl<IoSplitCodeMapper, IoSplitCodeEntity> {

    @Resource
    BasicSkPrescribeDiService basicSkPrescribeDiService;
    @Resource
    IoSplitCodeMapper splitCodeMapper;
    @Resource
    IoSplitFifoCodeService splitFifoCodeService;
    @Resource
    BasicSkPrescribeService basicSkPrescribeService;


    public List<IoSplitCodeEntity> filterList(IoSplitCodeRequest splitCodeRequest) {
        if (splitCodeRequest == null) {
            return Collections.emptyList();
        }
        if (splitCodeRequest.getPage() != null) {
            int offset = (splitCodeRequest.getPage() - 1) * splitCodeRequest.getLimit();
            PageHelper.offsetPage(offset, splitCodeRequest.getLimit());
        }
        return super.baseMapper.filterList(splitCodeRequest);

    }


    public List<IoSplitCodeEntity> findSplitCodes(Long relId, String batchNo, String workPlaceCode) {
        List<IoSplitCodeEntity> ioSplitCodeEntities = splitCodeMapper.selectList(new LambdaQueryWrapper<IoSplitCodeEntity>()
                .eq(IoSplitCodeEntity::getRelId, relId)
                .eq(StrUtil.isNotEmpty(batchNo), IoSplitCodeEntity::getBatchNo, batchNo)
                .eq(StrUtil.isNotEmpty(workPlaceCode), IoSplitCodeEntity::getWorkPlaceCode, workPlaceCode)
                .in(IoSplitCodeEntity::getStatus, 1, 2).orderByAsc(IoSplitCodeEntity::getInBactchNo));
        return ioSplitCodeEntities;
    }


    /**
     * 处方自动赋码
     */
    public void prescribeAutoTagCode(BasicSkPrescribeEntity basicSkPrescribeEntity) {

        List<BasicSkPrescribeDiEntity> basicSkPrescribeDiEntities = basicSkPrescribeDiService.findByPrescribe(basicSkPrescribeEntity.getAdNum());
        //处方赋码状态 1：未赋码，2：部分赋码，3：全部赋码
        for (BasicSkPrescribeDiEntity basicSkPrescribeDiEntity : basicSkPrescribeDiEntities) {

            //1.按照先进先出原则，查询拆零表，获取拆零表ID
            List<IoSplitCodeEntity> ioSplitCodeEntities = findSplitCodes(basicSkPrescribeDiEntity.getRelId(), basicSkPrescribeDiEntity.getBatchNo(), basicSkPrescribeEntity.getWorkPlaceCode());

            //2.如果拆零表为空，则自动从预出库队列中获拉取数据
            if (CollUtil.isEmpty(ioSplitCodeEntities)) {
                addFifoCode(basicSkPrescribeDiEntity.getRelId(), basicSkPrescribeDiEntity.getBatchNo(), IntUtil.value(basicSkPrescribeDiEntity.getMeasureCount()));
            }
            ioSplitCodeEntities = findSplitCodes(basicSkPrescribeDiEntity.getRelId(), basicSkPrescribeDiEntity.getBatchNo(), basicSkPrescribeEntity.getWorkPlaceCode());

            //3.如果拆零表不为空，则开始扣除数量
            int count = 0;
            int unCount = 0;  //剩余未扣减数量
            unCount = IntUtil.value(basicSkPrescribeDiEntity.getMeasureCount());
            for (IoSplitCodeEntity ioSplitCodeEntity : ioSplitCodeEntities) {
                count = IntUtil.value(ioSplitCodeEntity.getRemainCount()) - unCount;
                if (count > 0) {
                    ioSplitCodeEntity.setRemainCount(count);
                    break;
                } else if (count == 0) {
                    ioSplitCodeEntity.setRemainCount(0);
                } else {
                    ioSplitCodeEntity.setRemainCount(0);
                    unCount = -count;
                }
            }

            if (unCount > 0) {
                //4.拆零表数量不足，则从预出库队列中获取数据
                Integer fifoCount = addFifoCode(basicSkPrescribeDiEntity.getRelId(), basicSkPrescribeDiEntity.getBatchNo(), unCount);
                if (fifoCount == 0) {
                    //预出库队列数量不足
                    basicSkPrescribeDiEntity.setTagStatus(2);
                    basicSkPrescribeDiEntity.setTagMsg("预出库队列码数量不足，赋码失败！");
                } else {
                    //再次执行赋码
                    ioSplitCodeEntities = findSplitCodes(basicSkPrescribeDiEntity.getRelId(), basicSkPrescribeDiEntity.getBatchNo(), basicSkPrescribeEntity.getWorkPlaceCode());
                    for (IoSplitCodeEntity ioSplitCodeEntity : ioSplitCodeEntities) {
                        count = IntUtil.value(ioSplitCodeEntity.getRemainCount()) - unCount;
                        if (count > 0) {
                            ioSplitCodeEntity.setRemainCount(count);
                            break;
                        } else if (count == 0) {
                            ioSplitCodeEntity.setRemainCount(0);
                        } else {
                            ioSplitCodeEntity.setRemainCount(0);
                            unCount = -count;
                        }
                    }
                    if (unCount > 0) {
                        basicSkPrescribeDiEntity.setTagStatus(2);
                        basicSkPrescribeDiEntity.setTagMsg("预出库队列码数量不足，赋码失败！");
                    } else {
                        basicSkPrescribeDiEntity.setTagStatus(3);
                        basicSkPrescribeDiEntity.setTagMsg("赋码完成！");
                    }
                }
            }
            basicSkPrescribeDiService.updateById(basicSkPrescribeDiEntity);
        }
        basicSkPrescribeEntity.setTagStatus(checkStatus(basicSkPrescribeDiEntities));
        basicSkPrescribeService.updateById(basicSkPrescribeEntity);
    }


    /**
     * 处方手动赋码
     */
    public void prescribeManuTagCode(BasicSkPrescribeEntity basicSkPrescribeEntity) {

    }


    /**
     * 从预出库队列里面取出数据，添加至拆零表
     */
    public Integer addFifoCode(Long relId, String batchNo, Integer needCount) {

        List<IoSplitFifoCodeEntity> splitFifoCodeEntities = splitFifoCodeService
                .list(new LambdaQueryWrapper<IoSplitFifoCodeEntity>()
                        .eq(IoSplitFifoCodeEntity::getRelId, relId)
                        .eq(StrUtil.isNotEmpty(batchNo), IoSplitFifoCodeEntity::getBatchNo, batchNo)
                        .orderByAsc(IoSplitFifoCodeEntity::getInBactchNo)
                );
        Integer count = 0;
        List<IoSplitCodeEntity> addSplitCodeEntities = new ArrayList<>();
        List<Integer> removeIds = new ArrayList<>();
        for (IoSplitFifoCodeEntity splitFifoCodeEntity : splitFifoCodeEntities) {
            count = IntUtil.value(splitFifoCodeEntity.getTotalCount()) + count;
            IoSplitCodeEntity splitCodeEntity = new IoSplitCodeEntity();
            BeanUtils.copyProperties(splitFifoCodeEntity, splitCodeEntity);
            splitCodeEntity.setStatus(1);
            splitCodeEntity.setId(null);
            addSplitCodeEntities.add(splitCodeEntity);
            removeIds.add(splitFifoCodeEntity.getId());
            if (count >= needCount) {
                break;
            }
        }
        if (removeIds.size() > 0) {
            splitFifoCodeService.removeByIds(removeIds);
        }
        if (addSplitCodeEntities.size() > 0) {
            splitCodeMapper.insertBatch(addSplitCodeEntities);
        }
        return count;

    }


    public int checkStatus(List<BasicSkPrescribeDiEntity> list) {
        boolean hasTwo = false;
        boolean hasOne = false;
        for (BasicSkPrescribeDiEntity di : list) {
            int tagStatus = di.getTagStatus();
            if (tagStatus == 2) {
                hasTwo = true;
            } else if (tagStatus == 1) {
                hasOne = true;
            } else if (tagStatus != 3) {
                throw new IllegalArgumentException("Invalid tagStatus value: " + tagStatus);
            }
        }
        if (hasTwo) {
            return 2;
        } else if (hasOne) {
            return 1;
        } else {
            return 3;
        }
    }

}