优化

bonus-business/src/main/java/com/bonus/digital/controller/PlanManagementController.java

@@ -2,12 +2,10 @@ package com.bonus.digital.controller;
 
 import com.bonus.common.core.controller.BaseController;
 import com.bonus.common.core.domain.AjaxResult;
-import com.bonus.common.core.domain.entity.SysUser;
 import com.bonus.common.core.page.TableDataInfo;
 import com.bonus.common.utils.SecurityUtils;
 import com.bonus.common.utils.poi.ExcelUtil;
 import com.bonus.digital.dao.PlanManagementVo;
-import com.bonus.digital.dao.WorkloadCategoryVo;
 import com.bonus.digital.service.PlanManagementService;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
@@ -24,6 +22,7 @@ import java.io.InputStream;
 import java.net.URLEncoder;
 import java.nio.charset.StandardCharsets;
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author 马三炮
@@ -186,5 +185,23 @@ public class PlanManagementController extends BaseController {
         }
     }
 
+    /**
+     * 导出计划管理
+     */
+    @PostMapping("/exportPlanManagement")
+    public void exportPlanManagement(HttpServletResponse response, PlanManagementVo planManagementVo) {
+        try {
+            if (SecurityUtils.getLoginUser().getDeptId()!=null){
+                Long deptId = SecurityUtils.getDeptId();
+                planManagementVo.setInspectionStationId(deptId.intValue());
+            }
+            List<PlanManagementVo> list = planManagementService.getPlanManagementList(planManagementVo);
+            ExcelUtil<PlanManagementVo> util = new ExcelUtil<>(PlanManagementVo.class);
+            util.exportExcel(response, list, "计划管理");
+        } catch (Exception e) {
+            logger.error(e.toString(), e);
+        }
+    }
+
 
 }