klp-oa/klp-wms/src/main/java/com/klp/service/impl/WmsTransferOrderItemServiceImpl.java

package com.klp.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.klp.common.core.page.TableDataInfo;
import com.klp.common.core.domain.PageQuery;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import com.klp.domain.bo.WmsTransferOrderItemBo;
import com.klp.domain.vo.WmsTransferOrderItemVo;
import com.klp.domain.WmsTransferOrderItem;
import com.klp.domain.WmsRawMaterial;
import com.klp.domain.WmsProduct;
import com.klp.domain.WmsMaterialCoil;
import com.klp.mapper.WmsTransferOrderItemMapper;
import com.klp.mapper.WmsRawMaterialMapper;
import com.klp.mapper.WmsProductMapper;
import com.klp.mapper.WmsMaterialCoilMapper;
import com.klp.service.IWmsTransferOrderItemService;

import java.util.List;
import java.util.Map;
import java.util.Collection;
import java.util.Objects;
import java.util.ArrayList;

import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.klp.common.utils.StringUtils;

/**
 * 调拨单明细Service业务层处理
 *
 * @author klp
 * @date 2026-03-27
 */
@RequiredArgsConstructor
@Service
public class WmsTransferOrderItemServiceImpl implements IWmsTransferOrderItemService {

    private final WmsTransferOrderItemMapper baseMapper;
    private final WmsRawMaterialMapper rawMaterialMapper;
    private final WmsProductMapper productMapper;
    private final WmsMaterialCoilMapper coilMapper;

    /**
     * 查询调拨单明细
     */
    @Override
    public WmsTransferOrderItemVo queryById(Long itemId){
        return baseMapper.selectVoById(itemId);
    }

    /**
     * 查询调拨单明细列表
     */
    @Override
    public TableDataInfo<WmsTransferOrderItemVo> queryPageList(WmsTransferOrderItemBo bo, PageQuery pageQuery) {
        LambdaQueryWrapper<WmsTransferOrderItem> lqw = buildQueryWrapper(bo);
        Page<WmsTransferOrderItemVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
        return TableDataInfo.build(result);
    }

    /**
     * 查询调拨单明细列表
     */
    @Override
    public List<WmsTransferOrderItemVo> queryList(WmsTransferOrderItemBo bo) {
        LambdaQueryWrapper<WmsTransferOrderItem> lqw = buildQueryWrapper(bo);
        return baseMapper.selectVoList(lqw);
    }

    private LambdaQueryWrapper<WmsTransferOrderItem> buildQueryWrapper(WmsTransferOrderItemBo bo) {
        Map<String, Object> params = bo.getParams();
        LambdaQueryWrapper<WmsTransferOrderItem> lqw = Wrappers.lambdaQuery();
        lqw.eq(bo.getTransferId() != null, WmsTransferOrderItem::getTransferId, bo.getTransferId());
        lqw.eq(bo.getCoilId() != null, WmsTransferOrderItem::getCoilId, bo.getCoilId());
        lqw.eq(bo.getItemIdBefore() != null, WmsTransferOrderItem::getItemIdBefore, bo.getItemIdBefore());
        lqw.eq(bo.getItemIdAfter() != null, WmsTransferOrderItem::getItemIdAfter, bo.getItemIdAfter());
        lqw.eq(bo.getMaterialTypeBefore() != null, WmsTransferOrderItem::getMaterialTypeBefore, bo.getMaterialTypeBefore());
        lqw.eq(bo.getMaterialTypeAfter() != null, WmsTransferOrderItem::getMaterialTypeAfter, bo.getMaterialTypeAfter());
        lqw.eq(bo.getWarehouseIdBefore() != null, WmsTransferOrderItem::getWarehouseIdBefore, bo.getWarehouseIdBefore());
        lqw.eq(bo.getWarehouseIdAfter() != null, WmsTransferOrderItem::getWarehouseIdAfter, bo.getWarehouseIdAfter());
        return lqw;
    }

    /**
     * 新增调拨单明细
     */
    @Override
    public Boolean insertByBo(WmsTransferOrderItemBo bo) {
        WmsTransferOrderItem add = BeanUtil.toBean(bo, WmsTransferOrderItem.class);
        validEntityBeforeSave(add);
        boolean flag = baseMapper.insert(add) > 0;
        if (flag) {
            bo.setItemId(add.getItemId());
        }
        return flag;
    }

    /**
     * 修改调拨单明细
     */
    @Override
    public Boolean updateByBo(WmsTransferOrderItemBo bo) {
        WmsTransferOrderItem update = BeanUtil.toBean(bo, WmsTransferOrderItem.class);
        validEntityBeforeSave(update);
        return baseMapper.updateById(update) > 0;
    }

    /**
     * 保存前的数据校验
     */
    private void validEntityBeforeSave(WmsTransferOrderItem entity){
        //TODO 做一些数据校验,如唯一约束
    }

    /**
     * 批量删除调拨单明细
     */
    @Override
    public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
        if(isValid){
            //TODO 做一些业务上的校验,判断是否需要校验
        }
        return baseMapper.deleteBatchIds(ids) > 0;
    }

    /**
     * 根据itemId和itemType匹配或新增物料
     * itemType: raw_material-原料, product-成品
     */
    @Override
    public Long matchOrCreateMaterial(Long itemId, String itemType) {
        if (Objects.equals(itemType, "raw_material")) {
            return matchOrCreateFromRawMaterial(itemId);
        } else if (Objects.equals(itemType, "product")) {
            return matchOrCreateFromProduct(itemId);
        }
        throw new IllegalArgumentException("无效的itemType: " + itemType);
    }

    private Long matchOrCreateFromRawMaterial(Long itemId) {
        WmsRawMaterial rawMaterial = rawMaterialMapper.selectById(itemId);
        if (rawMaterial == null) {
            throw new IllegalArgumentException("原料不存在: " + itemId);
        }

        LambdaQueryWrapper<WmsProduct> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(WmsProduct::getProductName, rawMaterial.getRawMaterialName())
            .eq(WmsProduct::getSpecification, rawMaterial.getSpecification())
            .eq(WmsProduct::getManufacturer, rawMaterial.getManufacturer())
            .eq(WmsProduct::getMaterial, rawMaterial.getMaterial())
            .eq(WmsProduct::getSurfaceTreatmentDesc, rawMaterial.getSurfaceTreatmentDesc())
            .eq(WmsProduct::getZincLayer, rawMaterial.getZincLayer())
            .last("LIMIT 1");

        WmsProduct existProduct = productMapper.selectOne(wrapper);
        if (existProduct != null) {
            return existProduct.getProductId();
        }

        WmsProduct newProduct = new WmsProduct();
        newProduct.setProductName(rawMaterial.getRawMaterialName());
        newProduct.setSpecification(rawMaterial.getSpecification());
        newProduct.setManufacturer(rawMaterial.getManufacturer());
        newProduct.setMaterial(rawMaterial.getMaterial());
        newProduct.setSurfaceTreatmentDesc(rawMaterial.getSurfaceTreatmentDesc());
        newProduct.setZincLayer(rawMaterial.getZincLayer());
        newProduct.setDelFlag(0);
        productMapper.insert(newProduct);
        return newProduct.getProductId();
    }

    private Long matchOrCreateFromProduct(Long itemId) {
        WmsProduct product = productMapper.selectById(itemId);
        if (product == null) {
            throw new IllegalArgumentException("产品不存在: " + itemId);
        }

        LambdaQueryWrapper<WmsRawMaterial> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(WmsRawMaterial::getRawMaterialName, product.getProductName())
            .eq(WmsRawMaterial::getSpecification, product.getSpecification())
            .eq(WmsRawMaterial::getManufacturer, product.getManufacturer())
            .eq(WmsRawMaterial::getMaterial, product.getMaterial())
            .eq(WmsRawMaterial::getSurfaceTreatmentDesc, product.getSurfaceTreatmentDesc())
            .eq(WmsRawMaterial::getZincLayer, product.getZincLayer())
            .last("LIMIT 1");

        WmsRawMaterial existRawMaterial = rawMaterialMapper.selectOne(wrapper);
        if (existRawMaterial != null) {
            return existRawMaterial.getRawMaterialId();
        }

        WmsRawMaterial newRawMaterial = new WmsRawMaterial();
        newRawMaterial.setRawMaterialName(product.getProductName());
        newRawMaterial.setSpecification(product.getSpecification());
        newRawMaterial.setManufacturer(product.getManufacturer());
        newRawMaterial.setMaterial(product.getMaterial());
        newRawMaterial.setSurfaceTreatmentDesc(product.getSurfaceTreatmentDesc());
        newRawMaterial.setZincLayer(product.getZincLayer());
        newRawMaterial.setDelFlag(0);
        rawMaterialMapper.insert(newRawMaterial);
        return newRawMaterial.getRawMaterialId();
    }

    /**
     * 批量新增调拨单明细
     */
    @Override
    public Boolean batchInsert(WmsTransferOrderItemBo bo) {
        Long transferId = bo.getTransferId();
        List<Long> coilIds = bo.getCoilIds();
        if (transferId == null || coilIds == null || coilIds.isEmpty()) {
            throw new IllegalArgumentException("transferId和coilIds不能为空");
        }

        List<WmsMaterialCoil> coils = coilMapper.selectBatchIds(coilIds);
        if (coils.isEmpty()) {
            throw new IllegalArgumentException("钢卷不存在");
        }

        List<WmsTransferOrderItem> items = new ArrayList<>();
        for (WmsMaterialCoil coil : coils) {
            WmsTransferOrderItem item = new WmsTransferOrderItem();
            item.setTransferId(transferId);
            item.setCoilId(coil.getCoilId());
            item.setItemIdBefore(coil.getItemId());
            // 根据 itemType 设置 materialType: raw_material -> 1, product -> 2
            if ("raw_material".equals(coil.getItemType())) {
                item.setMaterialTypeBefore(1L);
            } else if ("product".equals(coil.getItemType())) {
                item.setMaterialTypeBefore(2L);
            }
            item.setWarehouseIdBefore(coil.getWarehouseId());
            items.add(item);
        }

        return baseMapper.insertBatch(items);
    }
}