renkang
2025-05-13 0a99001be87811ebb884b8f3f491f48054a22330
server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/YwMaterialServiceImpl.java
@@ -1,6 +1,8 @@
package com.doumee.service.business.impl;
import com.alibaba.nacos.shaded.org.checkerframework.checker.units.qual.A;
import com.doumee.biz.system.SystemDictDataBiz;
import com.doumee.core.annotation.excel.ExcelImporter;
import com.doumee.core.constants.ResponseStatus;
import com.doumee.core.exception.BusinessException;
import com.doumee.core.model.LoginUserInfo;
@@ -9,11 +11,14 @@
import com.doumee.core.model.PageWrap;
import com.doumee.core.utils.Constants;
import com.doumee.core.utils.Utils;
import com.doumee.dao.business.CategoryMapper;
import com.doumee.dao.business.YwMaterialMapper;
import com.doumee.dao.business.YwOutinboundMapper;
import com.doumee.dao.business.YwOutinboundRecordMapper;
import com.doumee.dao.business.model.*;
import com.doumee.dao.system.MultifileMapper;
import com.doumee.dao.system.dto.ImportMaterialDTO;
import com.doumee.dao.system.dto.ImportSystemUserDTO;
import com.doumee.dao.system.model.Multifile;
import com.doumee.service.business.YwMaterialService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
@@ -22,16 +27,19 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.github.yulichang.base.MPJBaseMapper;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import javafx.scene.paint.Material;
import org.apache.commons.lang.StringUtils;
import org.checkerframework.checker.units.qual.A;
import org.apache.shiro.SecurityUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.web.multipart.MultipartFile;
import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.*;
import java.util.stream.Collectors;
/**
 * 运维资产信息表Service实现
@@ -51,6 +59,9 @@
    private YwOutinboundRecordMapper ywOutinboundRecordMapper;
    @Autowired
    private CategoryMapper categoryMapper;
    @Autowired
    private SystemDictDataBiz systemDictDataBiz;
    @Override
@@ -67,8 +78,17 @@
        ywMaterial.setId(null);
        if(StringUtils.isBlank(ywMaterial.getCode())){
            //自动生成 TODO 存在问题
            Long countCode = ywMaterialMapper.selectCount(new QueryWrapper<YwMaterial>().lambda().eq(YwMaterial::getAutoCode, Constants.ONE));
            YwMaterial lastAutoMaterial = ywMaterialMapper.selectOne(new QueryWrapper<YwMaterial>().lambda().eq(YwMaterial::getAutoCode,Constants.ONE).last("limit 1 "));
            Long countCode = 0L;
            String nextCode = StringUtils.leftPad(Long.toString(countCode + 1),4,"0");
            if(Objects.nonNull(lastAutoMaterial)){
                Long maxCode = Long.valueOf(lastAutoMaterial.getCode().replace("P",""));
                nextCode = StringUtils.leftPad(Long.toString(maxCode + 1),4,"0");
                while (ywMaterialMapper.selectCount(new QueryWrapper<YwMaterial>().lambda().eq(YwMaterial::getCode, "P"+nextCode))>Constants.ZERO){
                    countCode = countCode + 1 ;
                    nextCode = StringUtils.leftPad(Long.toString(countCode),4,"0");
                }
            }
            ywMaterial.setCode("P"+nextCode);
            ywMaterial.setAutoCode(Constants.ONE);
        }else{
@@ -247,4 +267,67 @@
        QueryWrapper<YwMaterial> wrapper = new QueryWrapper<>(ywMaterial);
        return ywMaterialMapper.selectCount(wrapper);
    }
    @Override
    public Integer importMaterialBatch(MultipartFile file,LoginUserInfo loginUserInfo) {
        try {
            ExcelImporter ie = new ExcelImporter(file, 0, 0);
            List<ImportMaterialDTO> dataList = ie.getDataList(ImportMaterialDTO.class, null);
            if (CollectionUtils.isEmpty(dataList)) {
                throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "对不起,录入数据为空!");
            }
            List<Category> allCategory = categoryMapper.selectList(new QueryWrapper<Category>().lambda()
                    .eq(Category::getIsdeleted,Constants.ZERO).eq(Category::getIsdeleted,Constants.ZERO)
                    .eq(Category::getType,Constants.SEVEN)
            );
            if(com.github.xiaoymin.knife4j.core.util.CollectionUtils.isEmpty(allCategory)){
                throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "对不起,无分类基础数据!");
            }
            List<Category> categoryList = allCategory.stream().filter(i->Objects.isNull(i.getParentId())).collect(Collectors.toList());
            for (Category category: categoryList) {
                category.setChildCategoryList(
                        allCategory.stream().filter(i->Objects.nonNull(i.getParentId())&&Constants.equalsInteger(i.getParentId(),category.getId())).collect(Collectors.toList())
                );
            }
            List<YwMaterial> ywMaterialList = new ArrayList<>();
            for (int i = 0; i < dataList.size(); i++) {
                ImportMaterialDTO importMaterialDTO = dataList.get(i);
                if(Objects.isNull(importMaterialDTO)
                    || StringUtils.isBlank(importMaterialDTO.getName())
                    || StringUtils.isBlank(importMaterialDTO.getCategoryName())
                    || StringUtils.isBlank(importMaterialDTO.getCategoryChildName())
                    || StringUtils.isBlank(importMaterialDTO.getQrcode())
                ){
                    throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "对不起,第"+(i+1)+"行必填项缺失!");
                }
                YwMaterial ywMaterial = new YwMaterial();
                BeanUtils.copyProperties(importMaterialDTO,ywMaterial);
                Optional<Category> categoryOptional = categoryList.stream().filter(j->j.getName().equals(importMaterialDTO.getCategoryName())).findAny();
                if(categoryOptional.isPresent()){
                    Category category = categoryOptional.get();
                    ywMaterial.setParentCateId(category.getId());
                    Optional<Category> childCategoryOptional = category.getChildCategoryList().stream().filter(j->j.getName().equals(importMaterialDTO.getCategoryChildName())).findAny();
                    if(childCategoryOptional.isPresent()){
                        ywMaterial.setCateId(childCategoryOptional.get().getId());
                    }else{
                        throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "对不起,第"+(i+1)+"行物料子分类未查询到!");
                    }
                }else{
                    throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(), "对不起,第"+(i+1)+"行物料分类未查询到!");
                }
                ywMaterialList.add(ywMaterial);
            }
            for (YwMaterial ywMaterial:ywMaterialList) {
                ywMaterial.setLoginUserInfo(loginUserInfo);
                this.create(ywMaterial);
            }
        } catch (Exception e) {
            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),e.getMessage());
        }
        return null;
    }
}