邱宇豪
2023-11-08 ce1798b6f6ef2d4e82a156a22e9646b57a2a3649
hd/pipe/secure/src/main/java/com/thhy/secure/modules/biz/hiddenDanger/service/impl/HiddenDangerServiceImpl.java
@@ -4,6 +4,7 @@
import com.github.pagehelper.PageInfo;
import com.thhy.general.common.BasicResult;
import com.thhy.general.config.SysUserInfo;
import com.thhy.general.utils.ExcelUtils;
import com.thhy.general.utils.UUIDUtils;
import com.thhy.general.utils.UserInfoUtils;
import com.thhy.secure.modules.biz.hiddenDanger.entity.HiddenDangerEntity;
@@ -13,6 +14,7 @@
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import javax.servlet.http.HttpServletResponse;
import java.util.Date;
import java.util.List;
import java.util.Map;
@@ -96,20 +98,46 @@
   }
   @Override
   public BasicResult selectNoPageList(Map<String, Object> map) {
   public void export(Map<String, Object> map, HttpServletResponse response) {
      SysUserInfo sysUserInfo = UserInfoUtils.getInstance().getUserInfo();
      map.put("companyId",sysUserInfo.getCompanyId());
      List<HiddenDangerEntity> hiddenDangerEntities = hiddenDangerMapper.selectPageList(map);
      hiddenDangerEntities.forEach(obj ->{
         String id = obj.getId();
         obj.setImgEntities(hiddenDangerMapper.selectImgList(id));
         Integer auditStatus = obj.getAuditStatus();
         Integer level = obj.getLevel();
         switch (auditStatus){
            case 0:
               obj.setAuditStatusName("未审核");
               break;
            case 1:
               obj.setAuditStatusName("审核通过");
               break;
            case 2:
               obj.setAuditStatusName("审核失败");
               break;
         }
         if (level != null){
            switch (level){
               case 1:
                  obj.setLevelName("一般隐患");
                  break;
               case 2:
                  obj.setLevelName("重大隐患");
                  break;
               case 3:
                  obj.setLevelName("不存在");
                  break;
            }
         }
      });
      return BasicResult.success(hiddenDangerEntities);
      ExcelUtils.downExcel(hiddenDangerEntities,HiddenDangerEntity.class,response,"隐患举报");
   }
   @Override
   public BasicResult selectInfo(String id) {
      return BasicResult.success(hiddenDangerMapper.selectInfo(id));
      HiddenDangerEntity hiddenDangerEntity = hiddenDangerMapper.selectInfo(id);
      hiddenDangerEntity.setImgEntities(hiddenDangerMapper.selectImgList(id));
      return BasicResult.success(hiddenDangerEntity);
   }
   @Override