diff --git a/bonus-business/src/main/java/com/bonus/digital/controller/PlanManagementController.java b/bonus-business/src/main/java/com/bonus/digital/controller/PlanManagementController.java index 1b9e9ce..a5386ae 100644 --- a/bonus-business/src/main/java/com/bonus/digital/controller/PlanManagementController.java +++ b/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 list = planManagementService.getPlanManagementList(planManagementVo); + ExcelUtil util = new ExcelUtil<>(PlanManagementVo.class); + util.exportExcel(response, list, "计划管理"); + } catch (Exception e) { + logger.error(e.toString(), e); + } + } + }