diff --git a/adc-da-slrs/src/main/java/com/adc/da/slrs/wgdCensusLeo/controller/WgdCgfyController.java b/adc-da-slrs/src/main/java/com/adc/da/slrs/wgdCensusLeo/controller/WgdCgfyController.java
index ca9b942d..fc9aee94 100644
--- a/adc-da-slrs/src/main/java/com/adc/da/slrs/wgdCensusLeo/controller/WgdCgfyController.java
+++ b/adc-da-slrs/src/main/java/com/adc/da/slrs/wgdCensusLeo/controller/WgdCgfyController.java
@@ -1,18 +1,37 @@
package com.adc.da.slrs.wgdCensusLeo.controller;
+import com.adc.da.excel.poi.excel.entity.ExportParams;
+import com.adc.da.excel.poi.excel.entity.enums.ExcelType;
import com.adc.da.http.PageInfo;
import com.adc.da.http.ResponseMessage;
import com.adc.da.http.Result;
+import com.adc.da.slrs.msgDynamicInfo.common.ReadExcel;
+import com.adc.da.slrs.processModel.utils.ExcelExportUtilByWk;
+import com.adc.da.slrs.processModel.utils.ExcelImportUtilByWk;
import com.adc.da.slrs.wgdCensusLeo.entity.WgdCgfy;
+import com.adc.da.slrs.wgdCensusLeo.entity.WgdWbtj;
import com.adc.da.slrs.wgdCensusLeo.service.IWgdCgfyService;
+import com.adc.da.util.exception.AdcDaBaseException;
+import com.adc.da.util.utils.StringUtils;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import io.swagger.annotations.ApiOperation;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.util.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import io.swagger.annotations.Api;
import com.adc.da.base.web.BaseController;
+import org.springframework.web.multipart.MultipartFile;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.Arrays;
import java.util.List;
+import java.util.UUID;
/**
*
@@ -42,6 +61,66 @@ public class WgdCgfyController extends BaseController {
return Result.success(pageInfo);
}
+ @ApiOperation("导出采购费用信息统计")
+ @GetMapping("/exportAllCgfys")
+ public void exportAllAssocInfos(String idList, String exportName, HttpServletResponse response, HttpServletRequest request){//查询所有
+ OutputStream os = null;
+ Workbook workbook = null;
+ try {
+ if(StringUtils.isEmpty(exportName) ||exportName.equals("null")){
+ exportName="采购费用信息统计";
+ }
+ response.setHeader("Content-Disposition",
+ "attachment; filename=" + ReadExcel.encodeFileName(exportName+".xlsx", request));
+ response.setContentType("application/force-download");
+ ExportParams exportParams = new ExportParams();
+ exportParams.setType(ExcelType.XSSF);
+ exportParams.setSheetName(exportName);
+ List datas=new ArrayList<>();
+ if (null!=idList) {
+ List ids = new ArrayList<>(Arrays.asList(idList.split(",")));
+ QueryWrapper wrapper=new QueryWrapper<>();
+ wrapper.in("id",ids);
+ datas = iWgdCgfyService.list(wrapper);
+ } else {
+ datas = iWgdCgfyService.getAllWgdCgfys();
+ }
+ workbook = ExcelExportUtilByWk.exportWithData(exportParams,WgdWbtj.class,datas);
+ os = response.getOutputStream();
+ workbook.write(os);
+ os.flush();
+ } catch (IOException e) {
+ throw new AdcDaBaseException("下载文件失败,请重试");
+ } finally {
+ IOUtils.closeQuietly(os);
+ }
+ }
+
+ @ApiOperation(value = "|SarFileSplitItemsEO|导入")
+ @PostMapping(value = "/importAllCgfys")
+ public ResponseMessage> importSplitItemsZip(@RequestParam(value = "file", required = false) MultipartFile file) throws Exception {
+ List res= ExcelImportUtilByWk.readExcelpublic(WgdCgfy.class,file);
+
+ QueryWrapper queryWrapper = new QueryWrapper<>();
+ queryWrapper.select(WgdCgfy.class, info -> !info.getColumn().equals("id"));
+ List wgdAssocInfos=iWgdCgfyService.list(queryWrapper);
+ List remove=new ArrayList<>();
+ res.forEach(wgdAssocInfo -> {
+ if (wgdAssocInfos.contains(wgdAssocInfo)){
+ remove.add(wgdAssocInfo);
+ }
+ });
+ res.removeAll(remove);
+ res.forEach(wgdAssocInfo -> {
+ wgdAssocInfo.setId(String.valueOf(UUID.randomUUID()));
+ });
+ if (iWgdCgfyService.saveOrUpdateBatch(res)) {
+ return Result.success();
+ }else {
+ return Result.error();
+ }
+ }
+
@GetMapping("/getCgfyById")
public ResponseMessage