zouyu
2023-11-04 a915c14dafebeb823294935755b6646a2ca76ff9
mes-basic/src/main/java/com/chinaztt/mes/basic/service/impl/ParamServiceImpl.java
@@ -17,11 +17,15 @@
package com.chinaztt.mes.basic.service.impl;
import cn.hutool.core.collection.CollectionUtil;
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.support.ExcelTypeEnum;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.basic.dto.ParamPlusDto;
import com.chinaztt.mes.basic.entity.Param;
import com.chinaztt.mes.basic.entity.ParamJoinTemplate;
import com.chinaztt.mes.basic.excel.ParamData;
@@ -29,12 +33,18 @@
import com.chinaztt.mes.basic.mapper.ParamMapper;
import com.chinaztt.mes.basic.service.ParamService;
import com.chinaztt.mes.basic.util.DictUtils;
import com.chinaztt.mes.common.util.easyexcel.CustemHandler;
import com.chinaztt.mes.common.util.easyexcel.EasyExcelUtils;
import com.chinaztt.mes.common.util.easyexcel.template.ParamTemplate;
import com.chinaztt.ztt.admin.api.entity.SysDictItem;
import com.chinaztt.ztt.common.core.util.R;
import lombok.AllArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;
@@ -51,12 +61,36 @@
   private DictUtils dictUtils;
   @Override
   public R<Boolean> fullSave(Param param) {
   @Transactional(rollbackFor = Exception.class)
   public R<Boolean> fullSave(ParamPlusDto param) {
      int noCount = baseMapper.selectCount(Wrappers.<Param>lambdaQuery().eq(Param::getCode, param.getCode()));
      if (noCount > 0) {
         return R.failed("编号重复");
      }
      baseMapper.insert(param);
      List<SysDictItem> second_param = dictUtils.getDict("second_param");
      List<SysDictItem> third_param = dictUtils.getDict("third_param");
      Param father=new Param();
      father.setCode(param.getCode());
      father.setParameterItem(param.getParameterItem());
      baseMapper.insert(father);
      //二级参数
      param.getSecondId().forEach(s->{
         Param second=new Param();
         SysDictItem sysDictItem = second_param.stream().filter(p -> p.getId().equals(Integer.valueOf(String.valueOf(s)))).collect(Collectors.toList()).get(0);
         second.setParameterItem(sysDictItem.getLabel());
         second.setCode(""+s);
         second.setParentId(father.getId());
         baseMapper.insert(second);
         //三级
         param.getThirdId().forEach(t->{
            Param third=new Param();
            SysDictItem sysThirdDictItem = third_param.stream().filter(p -> p.getId().equals(Integer.valueOf(String.valueOf(t)))).collect(Collectors.toList()).get(0);
            third.setParameterItem(sysThirdDictItem.getLabel());
            third.setCode(""+t);
            third.setParentId(second.getId());
            baseMapper.insert(third);
         });
      });
      return R.ok();
   }