|
|
@@ -35,7 +35,10 @@ import java.io.File; |
|
|
|
import java.io.FileInputStream; |
|
|
|
import java.io.FileNotFoundException; |
|
|
|
import java.io.InputStream; |
|
|
|
import java.util.*; |
|
|
|
import java.util.ArrayList; |
|
|
|
import java.util.Arrays; |
|
|
|
import java.util.Date; |
|
|
|
import java.util.List; |
|
|
|
|
|
|
|
/** |
|
|
|
* @Author ChengWang |
|
|
@@ -73,8 +76,8 @@ public class QueryAirDataListService { |
|
|
|
String userId = user.getId(); |
|
|
|
//校验 |
|
|
|
JsonResult jsonResult = this.check(id); |
|
|
|
if(0 != jsonResult.getCode()){ |
|
|
|
log.info("生成excel表格数据校验失败:{}",jsonResult.getMsg()); |
|
|
|
if (0 != jsonResult.getCode()) { |
|
|
|
log.info("生成excel表格数据校验失败:{}", jsonResult.getMsg()); |
|
|
|
return jsonResult; |
|
|
|
} |
|
|
|
AirDataExport airDataExport = this.checkAirDataExport(id); |
|
|
@@ -83,11 +86,11 @@ public class QueryAirDataListService { |
|
|
|
} |
|
|
|
//获取任务名称 |
|
|
|
Inspection inspection = inspectionMapper.selectById(id); |
|
|
|
if(ObjectUtil.isNull(inspection)){ |
|
|
|
return JsonResult.error(AirDataEnum.INSPECTION_IS_NOT_EXIST.getCode(),AirDataEnum.INSPECTION_IS_NOT_EXIST.getMsg()); |
|
|
|
if (ObjectUtil.isNull(inspection)) { |
|
|
|
return JsonResult.error(AirDataEnum.INSPECTION_IS_NOT_EXIST.getCode(), AirDataEnum.INSPECTION_IS_NOT_EXIST.getMsg()); |
|
|
|
} |
|
|
|
String name = inspection.getName(); |
|
|
|
if(StringUtils.isEmpty(name)){ |
|
|
|
if (StringUtils.isEmpty(name)) { |
|
|
|
log.info("对应的任务名称为空"); |
|
|
|
throw new ServiceException(ServiceExceptionEnum.FILE_DATA_TIME_IS_ERROR); |
|
|
|
} |
|
|
@@ -96,14 +99,14 @@ public class QueryAirDataListService { |
|
|
|
.eq(AirData::getMark, MarkEnum.VALID.getCode()) |
|
|
|
.eq(AirData::getInspectionId, id) |
|
|
|
.orderByAsc(AirData::getCUtcTime)); |
|
|
|
if(CollectionUtils.isEmpty(airDataList) || airDataList.size() == 0){ |
|
|
|
return JsonResult.error(AirDataEnum.AIR_DATA_IS_NOT_EXIST.getCode(),AirDataEnum.AIR_DATA_IS_NOT_EXIST.getMsg()); |
|
|
|
if (CollectionUtils.isEmpty(airDataList) || airDataList.size() == 0) { |
|
|
|
return JsonResult.error(AirDataEnum.AIR_DATA_IS_NOT_EXIST.getCode(), AirDataEnum.AIR_DATA_IS_NOT_EXIST.getMsg()); |
|
|
|
} |
|
|
|
//数据转换 |
|
|
|
List<AirDataExcelVo> airDataExcelVoList = new ArrayList<>(); |
|
|
|
airDataList.forEach(x->{ |
|
|
|
airDataList.forEach(x -> { |
|
|
|
AirDataExcelVo vo = new AirDataExcelVo(); |
|
|
|
BeanUtils.copyProperties(x,vo); |
|
|
|
BeanUtils.copyProperties(x, vo); |
|
|
|
vo.setSo2(x.getSo2()); |
|
|
|
vo.setNo2(x.getNo2()); |
|
|
|
vo.setO3(x.getO3()); |
|
|
@@ -116,37 +119,37 @@ public class QueryAirDataListService { |
|
|
|
//对列表进行排序 |
|
|
|
//airDataExcelVoList.stream().sorted(Comparator.comparing(AirDataExcelVo::getCUtcTime)).collect(Collectors.toList()); |
|
|
|
|
|
|
|
String fileName = inspection.getName()+"+"+inspection.getCode()+".xlsx"; |
|
|
|
String fileName = inspection.getName() + "+" + inspection.getCode() + ".xlsx"; |
|
|
|
String filePath = UploadFileConfig.uploadFolder + "/data/" + fileName; |
|
|
|
File fd = new File(UploadFileConfig.uploadFolder + "/data"); |
|
|
|
//目录是否存在 |
|
|
|
if(!fd.exists()){ |
|
|
|
if (!fd.exists()) { |
|
|
|
fd.mkdirs(); |
|
|
|
} |
|
|
|
//生成excel |
|
|
|
JsonResult result = this.easyExcel(airDataExcelVoList,inspection,filePath); |
|
|
|
JsonResult result = this.easyExcel(airDataExcelVoList, inspection, filePath); |
|
|
|
if (0 != result.getCode()) { |
|
|
|
log.info("导出数据业务:生成excel失败:{}", result.getMsg()); |
|
|
|
return result; |
|
|
|
} |
|
|
|
//上传到oss |
|
|
|
String ossUrl = this.uploadOss(filePath,fileName); |
|
|
|
String ossUrl = this.uploadOss(filePath, fileName); |
|
|
|
// 删除原文件 |
|
|
|
FileUtils.deleteFile(filePath); |
|
|
|
//数据导入记录表中 |
|
|
|
Integer count = this.addAirDataExport(id,ossUrl,tenantId,userId); |
|
|
|
if(count <= 0){ |
|
|
|
Integer count = this.addAirDataExport(id, ossUrl, tenantId, userId); |
|
|
|
if (count <= 0) { |
|
|
|
log.info("添加大气数据报告导出记录失败"); |
|
|
|
} |
|
|
|
//查询记录表返回结果 |
|
|
|
AirDataExport airDataExportResult = airDataExportMapper.selectOne(Wrappers.<AirDataExport>lambdaQuery() |
|
|
|
.eq(AirDataExport::getInspectionId, id) |
|
|
|
.eq(AirDataExport::getMark, MarkEnum.VALID.getCode()) |
|
|
|
.eq(AirDataExport::getExportType,"data")); |
|
|
|
.eq(AirDataExport::getExportType, "data")); |
|
|
|
return JsonResult.success(airDataExportResult); |
|
|
|
} |
|
|
|
|
|
|
|
private Integer addAirDataExport(String id, String ossUrl, String tenantId,String userId) { |
|
|
|
private Integer addAirDataExport(String id, String ossUrl, String tenantId, String userId) { |
|
|
|
AirDataExport airDataExport = new AirDataExport(); |
|
|
|
airDataExport.setInspectionId(id); |
|
|
|
airDataExport.setTenantId(tenantId); |
|
|
@@ -166,7 +169,7 @@ public class QueryAirDataListService { |
|
|
|
try { |
|
|
|
in = new FileInputStream(file); |
|
|
|
//上传到oss |
|
|
|
aliyunOSSUtil.uploadFile2OSS(in,fileName); |
|
|
|
aliyunOSSUtil.uploadFile2OSS(in, fileName); |
|
|
|
//通过文件名获取文件完整路径 |
|
|
|
ossUrl = aliyunOSSUtil.getFileUrl(fileName); |
|
|
|
} catch (FileNotFoundException e) { |
|
|
@@ -178,6 +181,7 @@ public class QueryAirDataListService { |
|
|
|
|
|
|
|
/** |
|
|
|
* 大气报告导出记录表数据校验 |
|
|
|
* |
|
|
|
* @param id |
|
|
|
* @return |
|
|
|
*/ |
|
|
@@ -192,17 +196,18 @@ public class QueryAirDataListService { |
|
|
|
|
|
|
|
/** |
|
|
|
* 生成excel表格 |
|
|
|
* |
|
|
|
* @param voArrayList |
|
|
|
* @param inspection |
|
|
|
* @return |
|
|
|
*/ |
|
|
|
public JsonResult easyExcel(List<AirDataExcelVo> voArrayList,Inspection inspection,String filePath) { |
|
|
|
log.info("文件输出目录及格式:filePathName={}",filePath); |
|
|
|
public JsonResult easyExcel(List<AirDataExcelVo> voArrayList, Inspection inspection, String filePath) { |
|
|
|
log.info("文件输出目录及格式:filePathName={}", filePath); |
|
|
|
List<List<String>> heads = new ArrayList<>(); |
|
|
|
String totalName = "项目名称:"+ inspection.getName(); |
|
|
|
String totalName = "项目名称:" + inspection.getName(); |
|
|
|
String[] nameString = SystemConstant.EXCEL_SECOND_NAME; |
|
|
|
for (String excelName : nameString) { |
|
|
|
heads.add(Arrays.asList(totalName,excelName)); |
|
|
|
heads.add(Arrays.asList(totalName, excelName)); |
|
|
|
} |
|
|
|
//需要写出的表格(对应实体类) |
|
|
|
try { |
|
|
@@ -219,8 +224,8 @@ public class QueryAirDataListService { |
|
|
|
} |
|
|
|
|
|
|
|
private JsonResult check(String id) { |
|
|
|
if(StringUtils.isEmpty(id)){ |
|
|
|
return JsonResult.error(AirDataEnum.INSPECTION_ID_IS_NULL.getCode(),AirDataEnum.INSPECTION_ID_IS_NULL.getMsg()); |
|
|
|
if (StringUtils.isEmpty(id)) { |
|
|
|
return JsonResult.error(AirDataEnum.INSPECTION_ID_IS_NULL.getCode(), AirDataEnum.INSPECTION_ID_IS_NULL.getMsg()); |
|
|
|
} |
|
|
|
return JsonResult.success(); |
|
|
|
} |