|
@@ -1,14 +1,20 @@
|
|
|
package com.ruoyi.powerdistribution.controller;
|
|
|
|
|
|
import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.stream.Collectors;
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
|
|
|
+import com.ruoyi.common.core.domain.entity.SysDictData;
|
|
|
+import com.ruoyi.powerdistribution.domain.PdmRiskRole;
|
|
|
import com.ruoyi.powerdistribution.domain.dto.PdmWorkPlanDTO;
|
|
|
import com.ruoyi.powerdistribution.domain.dto.RpaWorkPlanDTO;
|
|
|
import com.ruoyi.powerdistribution.domain.vo.WorkNumByCompanyStatistics;
|
|
|
import com.ruoyi.powerdistribution.domain.vo.WorkNumStatisticsHome;
|
|
|
import com.ruoyi.powerdistribution.domain.vo.WorkNumStatisticsHomeChild;
|
|
|
import com.ruoyi.powerdistribution.domain.vo.WorkRiskStatistics;
|
|
|
+import com.ruoyi.powerdistribution.service.IPdmRiskRoleService;
|
|
|
+import com.ruoyi.system.service.ISysDictDataService;
|
|
|
import io.swagger.annotations.ApiOperation;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
@@ -39,6 +45,11 @@ public class PdmWorkPlanController extends BaseController
|
|
|
@Autowired
|
|
|
private IPdmWorkPlanService pdmWorkPlanService;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ ISysDictDataService sysDictDataService;
|
|
|
+ @Autowired
|
|
|
+ private IPdmRiskRoleService pdmRiskRoleService;
|
|
|
+
|
|
|
/**
|
|
|
* 查询作业计划列表
|
|
|
*/
|
|
@@ -60,6 +71,25 @@ public class PdmWorkPlanController extends BaseController
|
|
|
public void export(HttpServletResponse response, PdmWorkPlanDTO pdmWorkPlan)
|
|
|
{
|
|
|
List<PdmWorkPlan> list = pdmWorkPlanService.selectPdmWorkPlanList(pdmWorkPlan);
|
|
|
+
|
|
|
+ SysDictData queryDictData = new SysDictData();
|
|
|
+ queryDictData.setDictType("pdm_cancel_type");
|
|
|
+ Map<String,String> cancelMap = sysDictDataService.selectDictDataList(queryDictData).stream().collect(Collectors.toMap(SysDictData::getDictValue, SysDictData::getDictLabel));
|
|
|
+ queryDictData.setDictType("power_exception_type");
|
|
|
+ Map<String,String> exceptionMap = sysDictDataService.selectDictDataList(queryDictData).stream().collect(Collectors.toMap(SysDictData::getDictValue, SysDictData::getDictLabel));
|
|
|
+ queryDictData.setDictType("pdm_temporary_type");
|
|
|
+ Map<String,String> temporaryMap = sysDictDataService.selectDictDataList(queryDictData).stream().collect(Collectors.toMap(SysDictData::getDictValue, SysDictData::getDictLabel));
|
|
|
+
|
|
|
+ Map<String,String> riskMap = pdmRiskRoleService.selectPdmRiskRoleList(null).stream().collect(Collectors.toMap(PdmRiskRole::getRoleCode, PdmRiskRole::getRoleName));
|
|
|
+ riskMap.put("0","未分析");
|
|
|
+ riskMap.put("1","正常");
|
|
|
+
|
|
|
+ for(PdmWorkPlan workPlan:list){
|
|
|
+ workPlan.setCancelType(cancelMap.get(workPlan.getCancelType()));
|
|
|
+ workPlan.setExceptionType(exceptionMap.get(workPlan.getExceptionType()));
|
|
|
+ workPlan.setTemporaryType(temporaryMap.get(workPlan.getTemporaryType()));
|
|
|
+ workPlan.setRiskType(riskMap.get(workPlan.getRiskType()));
|
|
|
+ }
|
|
|
ExcelUtil<PdmWorkPlan> util = new ExcelUtil<PdmWorkPlan>(PdmWorkPlan.class);
|
|
|
util.exportExcel(response, list, "作业计划数据");
|
|
|
}
|