|
|
@@ -3,6 +3,7 @@ package com.xjrsoft.module.personnel.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;
|
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
|
|
import com.xjrsoft.common.model.result.RT;
|
|
|
@@ -11,6 +12,7 @@ import com.xjrsoft.common.page.PageOutput;
|
|
|
import com.xjrsoft.common.utils.VoToColumnUtil;
|
|
|
import com.xjrsoft.module.personnel.dto.AddBasePersonnelLabourCapitalDto;
|
|
|
import com.xjrsoft.module.personnel.dto.BasePersonnelLabourCapitalPageDto;
|
|
|
+import com.xjrsoft.module.personnel.dto.ImportBasePersonnelLabourCapitalDto;
|
|
|
import com.xjrsoft.module.personnel.dto.UpdateBasePersonnelLabourCapitalDto;
|
|
|
import com.xjrsoft.module.personnel.entity.BasePersonnelLabourCapital;
|
|
|
import com.xjrsoft.module.personnel.entity.BasePersonnelLabourCapitalData;
|
|
|
@@ -18,10 +20,13 @@ import com.xjrsoft.module.personnel.entity.BasePersonnelLabourCapitalTitle;
|
|
|
import com.xjrsoft.module.personnel.service.IBasePersonnelLabourCapitalService;
|
|
|
import com.xjrsoft.module.personnel.vo.BasePersonnelLabourCapitalPageVo;
|
|
|
import com.xjrsoft.module.personnel.vo.BasePersonnelLabourCapitalVo;
|
|
|
+import com.xjrsoft.module.weekly.dto.WeeklyDutySchedulePageDto;
|
|
|
+import com.xjrsoft.module.weekly.vo.WeeklyDutyScheduleListVo;
|
|
|
import com.yomahub.liteflow.util.JsonUtil;
|
|
|
import io.swagger.annotations.Api;
|
|
|
import io.swagger.annotations.ApiOperation;
|
|
|
import lombok.AllArgsConstructor;
|
|
|
+import org.springframework.http.ResponseEntity;
|
|
|
import org.springframework.web.bind.annotation.DeleteMapping;
|
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
|
@@ -33,6 +38,7 @@ import org.springframework.web.bind.annotation.RestController;
|
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
|
|
import javax.validation.Valid;
|
|
|
+import java.io.ByteArrayOutputStream;
|
|
|
import java.io.IOException;
|
|
|
import java.util.ArrayList;
|
|
|
import java.util.List;
|
|
|
@@ -107,7 +113,7 @@ public class BasePersonnelLabourCapitalController {
|
|
|
}
|
|
|
@PostMapping("/import")
|
|
|
@ApiOperation(value = "导入")
|
|
|
- public RT<Boolean> importData(@Valid @RequestBody AddBasePersonnelLabourCapitalDto dto, @RequestParam MultipartFile file) throws IOException {
|
|
|
+ public RT<Boolean> importData(@Valid ImportBasePersonnelLabourCapitalDto dto, @RequestParam("file") MultipartFile file) throws IOException {
|
|
|
List<Map<Integer, Object>> excelDataList = EasyExcel.read(file.getInputStream()).sheet().headRowNumber(dto.getDataRow() - 1).doReadSync();
|
|
|
//验证数据
|
|
|
|
|
|
@@ -132,7 +138,7 @@ public class BasePersonnelLabourCapitalController {
|
|
|
* @param excelDataList 表格数据
|
|
|
* @return 返回集合
|
|
|
*/
|
|
|
- List<BasePersonnelLabourCapitalData> initDataList(AddBasePersonnelLabourCapitalDto dto, List<Map<Integer, Object>> excelDataList){
|
|
|
+ List<BasePersonnelLabourCapitalData> initDataList(ImportBasePersonnelLabourCapitalDto dto, List<Map<Integer, Object>> excelDataList){
|
|
|
List<BasePersonnelLabourCapitalData> resultList = new ArrayList<>();
|
|
|
for (int i = 0; i < excelDataList.size(); i ++){
|
|
|
//跳过表头
|
|
|
@@ -185,7 +191,7 @@ public class BasePersonnelLabourCapitalController {
|
|
|
|
|
|
title.setName(value.toString());
|
|
|
title.setColumnNumber(column);
|
|
|
- title.setRowNumber(4);
|
|
|
+ title.setRowsNumber(4);
|
|
|
title.setMergeColumns(mergeColumns);
|
|
|
title.setMergeRows(mergeRows);
|
|
|
resultList.add(title);
|
|
|
@@ -204,7 +210,7 @@ public class BasePersonnelLabourCapitalController {
|
|
|
|
|
|
title.setName(value.toString());
|
|
|
title.setColumnNumber(column);
|
|
|
- title.setRowNumber(4);
|
|
|
+ title.setRowsNumber(4);
|
|
|
title.setMergeColumns(mergeColumns);
|
|
|
title.setMergeRows(mergeRows);
|
|
|
resultList.add(title);
|
|
|
@@ -213,4 +219,15 @@ public class BasePersonnelLabourCapitalController {
|
|
|
return resultList;
|
|
|
}
|
|
|
|
|
|
+ @GetMapping("/export")
|
|
|
+ @ApiOperation(value = "导出")
|
|
|
+
|
|
|
+ public ResponseEntity<byte[]> exportData(@Valid WeeklyDutySchedulePageDto dto, @RequestParam(defaultValue = "false") Boolean isTemplate) {
|
|
|
+ List<BasePersonnelLabourCapital> list = basePersonnelLabourCapitalService.list();
|
|
|
+ ByteArrayOutputStream bot = new ByteArrayOutputStream();
|
|
|
+ EasyExcel.write(bot, WeeklyDutyScheduleListVo.class).automaticMergeHead(false).excelType(ExcelTypeEnum.XLSX).sheet().doWrite(list);
|
|
|
+
|
|
|
+ return RT.fileStream(bot.toByteArray(), "WeeklyDutySchedule" + ExcelTypeEnum.XLSX.getValue());
|
|
|
+ }
|
|
|
+
|
|
|
}
|