设备数据字典fix

songhuangfeng123

2022-09-16 f4994092d9b3c6dcb60aaa0672eb283245ff5d8c

 equipment/equipment-service/src/main/java/com/gkhy/safePlatform/equipment/service/impl/KeypointEquipmentInfoServiceImpl.java

@@ -7,6 +7,8 @@
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gkhy.safePlatform.account.rpc.apimodel.AccountDepartmentService;
import com.gkhy.safePlatform.equipment.entity.*;
import com.gkhy.safePlatform.equipment.enums.EquipmentResultCodes;
import com.gkhy.safePlatform.equipment.enums.KeypointEquipmentPartTypeEnum;
import com.gkhy.safePlatform.equipment.excepiton.EquipmentException;
import com.gkhy.safePlatform.equipment.model.dto.req.*;
import com.gkhy.safePlatform.equipment.model.dto.resp.KeypointEquipmentInfoExcel;
@@ -77,7 +79,7 @@

    @Autowired
    public HttpServletResponse response;
	

   @Override
    public ResultVO queryAll(PageQuery<KeypointEquipmentInfoQueryCriteria> pageQuery) {
        Long pageIndex = pageQuery.getPageIndex();
@@ -241,6 +243,10 @@

    @Override
    public void addOrUpdate(KeypointEquipmentInfoDto infoDto) {

       // 验证枚举
        checkEnum(infoDto);

        KeypointEquipmentInfo equipmentInfo = BeanCopyUtils.copyBean(infoDto, KeypointEquipmentInfo.class);
        //保存主数据
        if (infoDto.getId() == null) { //新增
@@ -453,4 +459,14 @@
        ExcelUtil.exportExcel(map,respList , response.getOutputStream(),DateUtils.PATTERN_STANDARD);
        response.getOutputStream().close();
    }
}

    /**
     * 枚举验证
     * @param infoDto
     */
    public void checkEnum(KeypointEquipmentInfoDto infoDto){
        if (KeypointEquipmentPartTypeEnum.getByCode(infoDto.getPartType()) == null) {
            throw new EquipmentException(EquipmentResultCodes.KEYPOINT_EQUIPMENT_ENUM_PART_TYPE_NOT_EXIST);
        }
    }
}