package cn.gistack.sm.sjztmd.feign;
|
|
import cn.gistack.resource.feign.IOssClient;
|
import cn.gistack.resource.vo.MyMultipartFile;
|
import cn.gistack.sm.sjztmd.entity.AttResBase;
|
import cn.gistack.sm.sjztmd.feign.IAttResBaseClient;
|
import cn.gistack.sm.sjztmd.feign.ISjztMdClient;
|
import cn.gistack.sm.sjztmd.service.IAttResBaseService;
|
import cn.gistack.sm.sjztmd.util.MyDateUtils;
|
import cn.gistack.sm.sjztmd.word.service.ISjztmdService;
|
import cn.gistack.sm.sjztmd.word.util.WordUtil;
|
import cn.gistack.sm.sjztmd.word.vo.DzkInfo;
|
import cn.gistack.sm.sjztmd.word.vo.OverDetail;
|
import cn.gistack.sm.sjztmd.word.vo.SzInfo;
|
import cn.gistack.sm.sjztmd.word.vo.TotalInfo;
|
import lombok.AllArgsConstructor;
|
import org.springblade.core.tenant.annotation.NonDS;
|
import org.springframework.core.io.InputStreamResource;
|
import org.springframework.http.MediaType;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.multipart.MultipartFile;
|
import springfox.documentation.annotations.ApiIgnore;
|
|
import java.io.ByteArrayInputStream;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
import java.net.URLEncoder;
|
import java.text.SimpleDateFormat;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
@NonDS
|
@ApiIgnore
|
@RestController
|
@AllArgsConstructor
|
public class SjztMdClientImpl implements ISjztMdClient {
|
|
private final ISjztmdService sjztmdService;
|
private final IOssClient ossClient;
|
|
@Override
|
@GetMapping(GENERATE_DAY_REPORT_FLOOD)
|
public String generateDayReportFlood(String isShow) throws Exception{
|
|
|
|
//获取总览、大中库、明细数据
|
List<TotalInfo> totalList = sjztmdService.getTotalInfo(isShow);
|
List<DzkInfo> dzkList = sjztmdService.getDzkInfo(isShow);
|
List<OverDetail> overDetailList = sjztmdService.getOverDetailInfo(isShow);
|
List<SzInfo> szInfoList = sjztmdService.getSzInfo(isShow);
|
|
HashMap<String,String> over10Params = new HashMap<>();
|
//small_start_stag=0&small_end_stag=2&mid_start_stag=0&mid_end_stag=2&big_start_stag=0&big_end_stag=2
|
over10Params.put("small_start_stag", "0.1");
|
over10Params.put("mid_start_stag", "0.1");
|
over10Params.put("big_start_stag", "0.1");
|
|
HashMap<String,String> normalParams = new HashMap<>();
|
//small_start_stag=0&small_end_stag=2&mid_start_stag=0&mid_end_stag=2&big_start_stag=0&big_end_stag=2
|
normalParams.put("small_start_stag", "0");
|
normalParams.put("mid_start_stag", "0");
|
normalParams.put("big_start_stag", "0");
|
|
List<TotalInfo> totalOverList = sjztmdService.getTotalOverInfo(isShow,normalParams);
|
List<TotalInfo> totalOver10List = sjztmdService.getTotalOverInfo(isShow,over10Params);
|
byte[] data = null;
|
if (MyDateUtils.isTodayInFloodSeason()){
|
data = WordUtil.GetOverInfo(totalList,dzkList,overDetailList,szInfoList);
|
}else {
|
data= WordUtil.GetNotInSeasonOverInfo(totalList,dzkList,overDetailList,szInfoList,totalOverList,totalOver10List);
|
|
}
|
// 创建响应实体并设置响应头，将输出流作为响应体返回给客户端
|
// InputStreamResource resource = new InputStreamResource(new ByteArrayInputStream(data));
|
String time = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
|
String fileName = time + "全省水库防汛情况.docx";
|
MultipartFile file = new MyMultipartFile("file",fileName, "text/plain", data);
|
String path = ossClient.uploadFile(fileName, file);
|
return path;
|
}
|
|
private void createFile(String filePath,byte[] bytes){
|
// 将byte[]写入文件
|
try (FileOutputStream fos = new FileOutputStream(filePath)) {
|
fos.write(bytes);
|
System.out.println("文件已成功生成！");
|
} catch (IOException e) {
|
e.printStackTrace();
|
System.out.println("生成文件时发生错误。");
|
}
|
}
|
}
|
