“linengliang”
2023-12-13 ae33e68baf21878ce145d75ac377d14c1e2b2a82
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,8 +130,19 @@
   //@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(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);
      return Result.OK("添加成功!");