|
@@ -1,6 +1,7 @@
|
|
|
package com.xjrsoft.module.weekly.controller;
|
|
|
|
|
|
import cn.dev33.satoken.annotation.SaCheckPermission;
|
|
|
+import cn.hutool.core.bean.BeanUtil;
|
|
|
import com.alibaba.excel.EasyExcel;
|
|
|
import com.alibaba.excel.support.ExcelTypeEnum;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
@@ -12,6 +13,7 @@ import com.xjrsoft.common.utils.VoToColumnUtil;
|
|
|
import com.xjrsoft.module.weekly.dto.WeeklyDutySchedulePageDto;
|
|
|
import com.xjrsoft.module.weekly.entity.WeeklyDutySchedule;
|
|
|
import com.xjrsoft.module.weekly.service.IWeeklyDutyScheduleService;
|
|
|
+import com.xjrsoft.module.weekly.vo.WeeklyDutyScheduleListVo;
|
|
|
import com.xjrsoft.module.weekly.vo.WeeklyDutySchedulePageVo;
|
|
|
import io.swagger.annotations.Api;
|
|
|
import io.swagger.annotations.ApiOperation;
|
|
@@ -27,6 +29,7 @@ import org.springframework.web.multipart.MultipartFile;
|
|
|
import javax.validation.Valid;
|
|
|
import java.io.ByteArrayOutputStream;
|
|
|
import java.io.IOException;
|
|
|
+import java.text.SimpleDateFormat;
|
|
|
import java.util.ArrayList;
|
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
@@ -75,9 +78,17 @@ public class WeeklyDutyScheduleController {
|
|
|
@GetMapping("/export")
|
|
|
@ApiOperation(value = "导出")
|
|
|
public ResponseEntity<byte[]> exportData(@Valid WeeklyDutySchedulePageDto dto, @RequestParam(defaultValue = "false") Boolean isTemplate) {
|
|
|
- List<WeeklyDutySchedulePageVo> customerList = isTemplate != null && isTemplate ? new ArrayList<>() : ((PageOutput<WeeklyDutySchedulePageVo>) page(dto).getData()).getList();
|
|
|
+ List<WeeklyDutySchedule> list = weeklyDutyScheduleService.list();
|
|
|
+ List<WeeklyDutyScheduleListVo> customerList = new ArrayList<>();
|
|
|
+ SimpleDateFormat sdf = new SimpleDateFormat("yyyy年MM月dd日");
|
|
|
+ for (WeeklyDutySchedule vo : list) {
|
|
|
+ WeeklyDutyScheduleListVo listVo = BeanUtil.toBean(vo, WeeklyDutyScheduleListVo.class);
|
|
|
+ listVo.setDateBirth(sdf.format(vo.getDateBirth()));
|
|
|
+
|
|
|
+ customerList.add(listVo);
|
|
|
+ }
|
|
|
ByteArrayOutputStream bot = new ByteArrayOutputStream();
|
|
|
- EasyExcel.write(bot, WeeklyDutySchedulePageVo.class).automaticMergeHead(false).excelType(ExcelTypeEnum.XLSX).sheet().doWrite(customerList);
|
|
|
+ EasyExcel.write(bot, WeeklyDutyScheduleListVo.class).automaticMergeHead(false).excelType(ExcelTypeEnum.XLSX).sheet().doWrite(customerList);
|
|
|
|
|
|
return RT.fileStream(bot.toByteArray(), "WeeklyDutySchedule" + ExcelTypeEnum.XLSX.getValue());
|
|
|
}
|