“linengliang”
2023-11-20 38c6f00b5efbbb9ce21669a887e03d6cb8ce3df9
lxzn-module-eam/src/main/java/org/jeecg/modules/eam/controller/EquipmentCategoryController.java
@@ -10,6 +10,7 @@
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import org.jeecg.common.api.vo.CommonGenericTree;
@@ -25,6 +26,10 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;
import org.jeecg.modules.system.entity.SysDict;
import org.jeecg.modules.system.entity.SysDictItem;
import org.jeecg.modules.system.service.ISysDictItemService;
import org.jeecg.modules.system.service.ISysDictService;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
import org.jeecgframework.poi.excel.entity.ExportParams;
@@ -54,6 +59,10 @@
public class EquipmentCategoryController extends JeecgController<EquipmentCategory, IEquipmentCategoryService> {
   @Autowired
   private IEquipmentCategoryService equipmentCategoryService;
   @Autowired
   private ISysDictService sysDictService;
   @Autowired
   private ISysDictItemService sysDictItemService;
   /**
    * 分页列表查询
@@ -121,32 +130,18 @@
   //@RequiresPermissions("org.jeecg.modules.demo:mom_eam_equipment_category:add")
   @PostMapping(value = "/add")
   public Result<String> add(@RequestBody EquipmentCategory equipmentCategory) {
      if(!StringUtils.isNotBlank( equipmentCategory.getParentId())){
         equipmentCategory.setParentId("-1");
         if("生产设备".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("producter");
         } else if("试验器".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("tester");
         }else if("数控回转库".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("filesCarousel");
         }else if("非标吊索具".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("nonStandardLiftingLocks");
         }else if("运输车".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("truck");
         }else if("起重设备".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("crane");
         }else if("平衡机".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("balanceMachine");
         }else if("焊机".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("weldingMachine");
         }else if("压力容器".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("pressureVessel");
         }else if("环保设备".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("envProEquipment");
         }else if("车台".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("ride");
         }else if("检测、计量设备".equals(equipmentCategory.getName())){
            equipmentCategory.setEquipmentCategoryUda1("checkoutEquipment");
      if(equipmentCategory.getParentId().equals("-1")){
         SysDict sysDict = sysDictService.getOne(new QueryWrapper<SysDict>().eq("dict_code","equipment_category"),false);
         if(ObjectUtils.isNotNull(sysDict)){
            List<SysDictItem> sysDictItems = sysDictItemService.selectItemsByMainId(sysDict.getId());
            for(SysDictItem sysDictItem:sysDictItems){
               if(equipmentCategory.getName().equals(sysDictItem.getItemText())){
                  equipmentCategory.setEquipmentCategoryUda1(sysDictItem.getItemValue());
               }
            }
         }
         else {
            return Result.error("添加失败,需要去维护数据字典设备大类!");
         }
      }
      equipmentCategoryService.save(equipmentCategory);