yishiqihuasheng
2 years ago
parent
ae9b52ad19
commit
cf81570b40
@ -0,0 +1,442 @@ |
||||
package com.dxhy.core.thread; |
||||
|
||||
import cn.hutool.core.date.DateUtil; |
||||
import com.alibaba.excel.EasyExcel; |
||||
import com.alibaba.excel.ExcelWriter; |
||||
import com.alibaba.excel.write.builder.ExcelWriterBuilder; |
||||
import com.alibaba.excel.write.metadata.WriteSheet; |
||||
import com.alibaba.excel.write.metadata.WriteTable; |
||||
import com.alibaba.fastjson.JSONArray; |
||||
import com.alibaba.fastjson.JSONObject; |
||||
import com.dxhy.common.service.impl.TabulateRedisServiceImpl; |
||||
import com.dxhy.common.util.CustomCellWriteHandler; |
||||
import com.dxhy.common.util.ExceptionUtil; |
||||
import com.dxhy.common.util.StringHelper; |
||||
import com.dxhy.common.utils.DateUtils; |
||||
import com.dxhy.core.entity.TDxExcelExportlog; |
||||
import com.dxhy.core.feign.IFeginWebsocket; |
||||
import com.dxhy.core.rabbitmq.Sender; |
||||
import com.dxhy.core.service.export.PoolPaperService; |
||||
import com.dxhy.core.service.export.RzFpxxService; |
||||
import com.dxhy.core.service.singletable.IExcelExportlogService; |
||||
import com.dxhy.core.util.JsonResponseUtils; |
||||
import com.dxhy.oss.service.FileService; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
|
||||
import java.io.File; |
||||
import java.io.FileOutputStream; |
||||
import java.io.OutputStream; |
||||
import java.util.*; |
||||
|
||||
//import com.alibaba.excel.metadata.Sheet;
|
||||
//import com.alibaba.excel.metadata.Table;
|
||||
|
||||
/** |
||||
* @author user |
||||
*/ |
||||
@Slf4j |
||||
@SuppressWarnings({"AliDeprecation", "AlibabaUndefineMagicConstant", "AlibabaMethodTooLong"}) |
||||
public class ManualAthensThread extends BaseThread { |
||||
|
||||
/** |
||||
* 默认要导出第几页的数据 |
||||
*/ |
||||
private final int default_curr = 1; |
||||
/** |
||||
* 默认每页导出多少条数据 |
||||
*/ |
||||
private final int default_size = 5000; |
||||
/** |
||||
* 文件名前缀 |
||||
*/ |
||||
private final String fileNamePrefix = "手工认证"; |
||||
/** |
||||
* 文件名后缀 |
||||
*/ |
||||
private final String fileNameSurfix = ".xlsx"; |
||||
/** |
||||
* 文件名中间分隔符 |
||||
*/ |
||||
private final String fileSplit = "_"; |
||||
/** |
||||
* excel的标题 |
||||
*/ |
||||
private String title = "手工认证发票"; |
||||
/** |
||||
* 导出excel的列名 |
||||
*/ |
||||
private final String[] titleColumns = {"序号", "发票代码", "发票号码", "发票状态", "开票日期","购方名称", "销方名称", "金额", "税额", |
||||
"抵扣税额", "签收状态", "签收日期", "签收方式", "签收人","凭证号","报账单号", "过账时间", "影像号", "账期", "报账单号", "所属组织"}; |
||||
|
||||
/** |
||||
* 导出excel的列名 |
||||
*/ |
||||
private final String[] detailTitleColumns = |
||||
{"序号", "发票代码", "发票号码", "货物或应税劳务名称", "规格型号", "单位", "数量", "单价", "金额", "税额", "税率", "车牌号", "类型", "通行日期起", "通行日期止"}; |
||||
|
||||
// private Sheet sheet;
|
||||
// private Table table;
|
||||
private WriteSheet writeSheet; |
||||
private WriteTable writeTable; |
||||
private ExcelWriter writer = null; |
||||
|
||||
private final IExcelExportlogService exportLogService; |
||||
|
||||
private final List<TDxExcelExportlog> exportLogs; |
||||
|
||||
private final RzFpxxService rzFpxxService; |
||||
|
||||
private final FileService fileService; |
||||
|
||||
private final Sender sender; |
||||
|
||||
private final IFeginWebsocket feginWebsocket; |
||||
private final TabulateRedisServiceImpl tabulateRedisService; |
||||
|
||||
public ManualAthensThread(IExcelExportlogService exportLogService, List<TDxExcelExportlog> exportLogs, |
||||
RzFpxxService rzFpxxService, FileService fileService, Sender sender, IFeginWebsocket feginWebsocket, |
||||
TabulateRedisServiceImpl tabulateRedisService) { |
||||
this.exportLogService = exportLogService; |
||||
this.exportLogs = exportLogs; |
||||
this.rzFpxxService = rzFpxxService; |
||||
this.fileService = fileService; |
||||
this.sender = sender; |
||||
this.feginWebsocket = feginWebsocket; |
||||
this.tabulateRedisService = tabulateRedisService; |
||||
} |
||||
|
||||
@Override |
||||
public void run() { |
||||
for (TDxExcelExportlog exportLog : exportLogs) { |
||||
// 日志表主键
|
||||
Long id = exportLog.getId(); |
||||
String userid = exportLog.getUserAccount(); |
||||
String userName = exportLog.getUserName(); |
||||
// 导出条件
|
||||
String condition = exportLog.getConditions(); |
||||
// 数据源名称
|
||||
String dbName = exportLog.getDbName(); |
||||
if (StringHelper.isBlank(dbName)) { |
||||
dbName = ""; |
||||
} |
||||
String createDate = DateUtils.dateToStrLong(exportLog.getCreateDate()); |
||||
|
||||
JSONObject msg = new JSONObject(); |
||||
|
||||
// 捕获异常，防止某一条日志导出出错，影响其它日志的导出
|
||||
try { |
||||
exportLogService.updateStart(id, dbName); |
||||
|
||||
// 应该保存的参数
|
||||
JSONObject prams = JSONObject.parseObject(condition); |
||||
String businessCode = prams.getString("businessCode"); |
||||
String bzdh = prams.getString("bzdh"); |
||||
String company = prams.getString("company"); |
||||
String companyCode = prams.getString("companyCode"); |
||||
String keyWord = prams.getString("keyWord"); |
||||
String gfsh = prams.getString("gfsh"); |
||||
String fpzt = prams.getString("invoiceStatus"); |
||||
String invoiceType = prams.getString("invoiceType"); |
||||
String qszt = prams.getString("qszt"); |
||||
String qsr = prams.getString("qsr"); |
||||
String qsfs = prams.getString("qsfs"); |
||||
String rzzt = prams.getString("rzzt"); |
||||
String rzlx = prams.getString("rzlx"); |
||||
String rzr = prams.getString("rzr"); |
||||
String cjksyf = prams.getString("cjksyf"); |
||||
String cjjsyf = prams.getString("cjjsyf"); |
||||
String gfTaxNo = prams.getString("gfTaxNo"); |
||||
String inAccountStatus = prams.getString("inAccountStatus"); |
||||
String inAccountMonth = prams.getString("inAccountMonth"); |
||||
JSONArray companyCodeArray = prams.getJSONArray("companyCodes"); |
||||
List<String> companyCodes = companyCodeArray.toJavaList(String.class); |
||||
// 参数转换为需要的形式
|
||||
Map<String, Object> pramsMap = new HashMap<>(16); |
||||
pramsMap.put("dbName", dbName); |
||||
List<String> gfshList = new ArrayList<>(); |
||||
if (!StringHelper.isBlank(gfsh)) { |
||||
gfshList = StringHelper.stringToList(gfsh); |
||||
pramsMap.put("gfsh", gfshList); |
||||
} else { |
||||
pramsMap.put("gfsh", gfshList); |
||||
} |
||||
if (cjksyf != null && !"".equals(cjksyf)) { |
||||
pramsMap.put("cjksyf", DateUtils.getMorning(DateUtils.strToDate(cjksyf))); |
||||
} |
||||
if (cjjsyf != null && !"".equals(cjjsyf)) { |
||||
pramsMap.put("cjjsyf", DateUtils.getNight(DateUtils.strToDate(cjjsyf))); |
||||
} |
||||
if (prams.get("kpksrq") != null && !"".equals(prams.get("kpksrq"))) { |
||||
pramsMap.put("kpksrq", |
||||
DateUtils.getMorning(DateUtils.strToDate(prams.get("kpksrq").toString()))); |
||||
} |
||||
if (prams.get("kpjsrq") != null && !"".equals(prams.get("kpjsrq"))) { |
||||
pramsMap.put("kpjsrq", DateUtils.getNight(DateUtils.strToDate(prams.get("kpjsrq").toString()))); |
||||
} |
||||
if (prams.get("skssq") != null && !"".equals(prams.get("skssq")) && "1".equals(prams.get("rzzt"))) { |
||||
pramsMap.put("skssq", prams.get("skssq").toString().replace("-", "")); |
||||
} else { |
||||
pramsMap.put("skssq", ""); |
||||
} |
||||
if (prams.get("qsyf") == null && "".equals(prams.get("qsyf")) && !"1".equals(prams.get("qszt"))) { |
||||
pramsMap.put("qsyf", ""); |
||||
}else { |
||||
pramsMap.put("qsyf", prams.get("qsyf")); |
||||
} |
||||
if (prams.get("inAccountMonth") == null && "".equals(prams.get("inAccountMonth")) |
||||
&& !"1".equals(prams.get("inAccountStatus"))) { |
||||
pramsMap.put("inAccountMonth", ""); |
||||
}else { |
||||
pramsMap.put("inAccountMonth", inAccountMonth); |
||||
} |
||||
pramsMap.put("keyWord", keyWord); |
||||
pramsMap.put("qszt", qszt); |
||||
pramsMap.put("invoiceType", invoiceType); |
||||
pramsMap.put("qsr", qsr); |
||||
pramsMap.put("qsfs", qsfs); |
||||
pramsMap.put("fpzt", fpzt); |
||||
pramsMap.put("rzzt", rzzt); |
||||
pramsMap.put("rzlx", rzlx); |
||||
pramsMap.put("rzr", rzr); |
||||
pramsMap.put("inAccountStatus", inAccountStatus); |
||||
pramsMap.put("company", company); |
||||
pramsMap.put("invoiceSource",prams.getString("invoiceSource")); |
||||
pramsMap.put("companyCodes",companyCodes); |
||||
pramsMap.put("companyCode",companyCode); |
||||
pramsMap.put("snVoucherNumber", prams.getString("snVoucherNumber")); |
||||
pramsMap.put("postingTimeEnd", prams.getString("postingTimeEnd")); |
||||
pramsMap.put("postingTimeStart", prams.getString("postingTimeStart")); |
||||
pramsMap.put("imageNumber", prams.getString("imageNumber")); |
||||
if (prams.getString("accountPeriod") != null && !"".equals(prams.getString("accountPeriod"))) { |
||||
String accountPeriod = prams.getString("accountPeriod").substring(0,7); |
||||
pramsMap.put("accountPeriod", accountPeriod); |
||||
} |
||||
// 导出文件名
|
||||
StringBuilder ftpFileName = new StringBuilder(); |
||||
ftpFileName.append(fileNamePrefix); |
||||
ftpFileName.append(DateUtils.dateToMonthStr(new Date())); |
||||
ftpFileName.append(fileSplit); |
||||
if ("99".equals(gfTaxNo)) { |
||||
if (gfshList.size() > 0) { |
||||
ftpFileName.append(rzFpxxService.getblocName(gfshList.get(0))); |
||||
ftpFileName.append(fileSplit); |
||||
} |
||||
} else { |
||||
if (gfshList.size() > 0) { |
||||
for (String s : gfshList) { |
||||
ftpFileName.append(rzFpxxService.getGfName(s)); |
||||
ftpFileName.append(fileSplit); |
||||
} |
||||
} |
||||
} |
||||
|
||||
ftpFileName.append(DateUtils.getNowTimeStamp()); |
||||
title = ftpFileName.toString(); |
||||
ftpFileName.append(fileNameSurfix); |
||||
|
||||
// 本地文件存储路径
|
||||
StringBuilder excelFile = new StringBuilder(); |
||||
excelFile.append(fileService.getLocalPathDefault()); |
||||
excelFile.append(ftpFileName.toString()); |
||||
|
||||
File filePath = new File(getFilePath(excelFile.toString())); |
||||
if (!filePath.exists()) { |
||||
filePath.mkdirs(); |
||||
} |
||||
|
||||
File file = new File(excelFile.toString()); |
||||
if (!file.exists()) { |
||||
file.delete(); |
||||
} |
||||
|
||||
if (!file.exists()) { |
||||
file.createNewFile(); |
||||
} |
||||
OutputStream out = new FileOutputStream(excelFile.toString()); |
||||
// writer = new ExcelWriter(null, out, ExcelTypeEnum.XLSX, true, new CellWriteHandler());
|
||||
ExcelWriterBuilder builder = EasyExcel.write(excelFile.toString()); |
||||
builder.registerWriteHandler(new CustomCellWriteHandler()); |
||||
writer = builder.build(); |
||||
// 递归导出数据到excel 发票信息
|
||||
this.cicleExport(pramsMap, default_curr, excelFile.toString(), out, writer); |
||||
|
||||
writer.finish(); |
||||
// FTP文件存储路径
|
||||
String[] toDay = DateUtil.today().split("-"); |
||||
String ftpFilePath = fileService.getPathPrefix() + userName + "/" + "excelFile/" + toDay[0] + "/" |
||||
+ toDay[1] + "/" + toDay[2] + "/" + ftpFileName.toString(); |
||||
// 文件上传ftp
|
||||
fileService.uploadFile(ftpFilePath, excelFile.toString()); |
||||
|
||||
// 设置成功消息内容
|
||||
msg.put("title", super.getSuccTitle(title)); |
||||
msg.put("content", super.getSuccContent(createDate)); |
||||
msg.put("url", super.getUrl(id)); |
||||
msg.put("useraccount", userName); |
||||
|
||||
// 记录导出完成状态
|
||||
exportLogService.updateSucc(id, ftpFilePath, dbName); |
||||
|
||||
} catch (Exception e) { |
||||
String errmsg = ExceptionUtil.getExceptionDeteil(e, 4000); |
||||
// 设置失败消息内容
|
||||
msg.put("title", super.getFailTitle(title)); |
||||
msg.put("content", super.getFailContent(createDate, errmsg)); |
||||
msg.put("url", ""); |
||||
msg.put("useraccount", userName); |
||||
|
||||
log.error("", e); |
||||
// 记录导出出错状态
|
||||
exportLogService.updateFail(id, errmsg, dbName); |
||||
} finally { |
||||
// 发送导出消息到mq
|
||||
String responseResult = JsonResponseUtils.responseResult(msg); |
||||
sender.sendToMsg(responseResult); |
||||
|
||||
// 申请成功后像websocket发送申请信息
|
||||
feginWebsocket.setMessageToWebsocket(userName); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 递归导出数据到excel文件，防止jvm oom |
||||
* |
||||
* @param pramsMap |
||||
* 查询数据参数集合 |
||||
* @param curr |
||||
* 当前第几页 |
||||
* @param excelFile |
||||
* 导出的文件 |
||||
* @param writer |
||||
* @throws Exception |
||||
* |
||||
*/ |
||||
|
||||
private boolean cicleExport(Map<String, Object> pramsMap, int curr, String excelFile, OutputStream out, |
||||
ExcelWriter writer) throws Exception { |
||||
// 查询指定页数的数据
|
||||
JSONObject pageData = rzFpxxService.selectWithParams(pramsMap, curr, default_size); |
||||
boolean hasNext = pageData.getBooleanValue("hasNext"); |
||||
List<String[]> dataList = (List<String[]>)pageData.get("datalist"); |
||||
boolean firstPage = false; |
||||
if (curr == default_curr) { |
||||
firstPage = true; |
||||
} |
||||
|
||||
// 数据不是最后一页
|
||||
if (hasNext) { |
||||
int nextPage = pageData.getIntValue("nextPage"); |
||||
// 写入当前页数据
|
||||
exportListArrayToExcel(excelFile, title, titleColumns, dataList, firstPage, false, 1, "发票信息", out, writer); |
||||
// 更新页数
|
||||
return cicleExport(pramsMap, nextPage, excelFile, out, writer); |
||||
} else { |
||||
// 写入最后一页数据
|
||||
exportListArrayToExcel(excelFile, title, titleColumns, dataList, firstPage, true, 1, "发票信息", out, writer); |
||||
return hasNext; |
||||
} |
||||
|
||||
} |
||||
|
||||
/** |
||||
* 递归导出数据到excel文件，防止jvm oom |
||||
* |
||||
* @param pramsMap |
||||
* 查询数据参数集合 |
||||
* @param curr |
||||
* 当前第几页 |
||||
* @param excelFile |
||||
* 导出的文件 |
||||
* @param writer |
||||
* @throws Exception |
||||
* |
||||
*/ |
||||
|
||||
private boolean detailCicleExport(Map<String, Object> pramsMap, int curr, String excelFile, OutputStream out, |
||||
ExcelWriter writer) throws Exception { |
||||
// 查询指定页数的数据
|
||||
JSONObject pageData = rzFpxxService.getTaxRebateList(pramsMap, curr, default_size); |
||||
boolean hasNext = pageData.getBooleanValue("hasNext"); |
||||
List<String[]> dataList = (List<String[]>)pageData.get("datalist"); |
||||
boolean firstPage = false; |
||||
if (curr == default_curr) { |
||||
firstPage = true; |
||||
} |
||||
|
||||
// 数据不是最后一页
|
||||
if (hasNext) { |
||||
int nextPage = pageData.getIntValue("nextPage"); |
||||
// 写入当前页数据
|
||||
exportListArrayToExcel(excelFile, title, detailTitleColumns, dataList, firstPage, false, 2, "货物信息", out, |
||||
writer); |
||||
// 更新页数
|
||||
return detailCicleExport(pramsMap, nextPage, excelFile, out, writer); |
||||
} else { |
||||
// 写入最后一页数据
|
||||
exportListArrayToExcel(excelFile, title, detailTitleColumns, dataList, firstPage, true, 2, "货物信息", out, |
||||
writer); |
||||
return hasNext; |
||||
} |
||||
|
||||
} |
||||
|
||||
public void exportListArrayToExcel(String excelFile, String title, String[] titleColumns, List<String[]> dataList, |
||||
boolean firstPage, boolean lastPage, int sheetNo, String sheetName, OutputStream out, ExcelWriter writer) { |
||||
|
||||
String[] datas; |
||||
if (firstPage) { |
||||
writeSheet = new WriteSheet(); |
||||
writeSheet.setSheetNo(sheetNo); |
||||
writeSheet.setSheetName(sheetName); |
||||
writeTable = new WriteTable(); |
||||
writeTable.setTableNo(sheetNo); |
||||
List<List<String>> titles = new ArrayList<>(); |
||||
datas = titleColumns; |
||||
int var11 = titleColumns.length; |
||||
|
||||
for (int var12 = 0; var12 < var11; ++var12) { |
||||
String titleColumn = datas[var12]; |
||||
titles.add(Collections.singletonList(titleColumn)); |
||||
} |
||||
|
||||
writeTable.setHead(titles); |
||||
List<List<String>> list = new ArrayList<>(); |
||||
|
||||
for (String[] strings : dataList) { |
||||
datas = strings; |
||||
list.add(Arrays.asList(datas)); |
||||
} |
||||
|
||||
this.writer.write(list, writeSheet, writeTable); |
||||
} else { |
||||
List<List<String>> list = new ArrayList<>(); |
||||
|
||||
for (String[] strings : dataList) { |
||||
datas = strings; |
||||
list.add(Arrays.asList(datas)); |
||||
} |
||||
|
||||
this.writer.write(list, writeSheet, writeTable); |
||||
} |
||||
|
||||
} |
||||
|
||||
private static String getFilePath(String fileName) { |
||||
StringBuilder filePath = new StringBuilder(); |
||||
if (!fileName.contains(":")) { |
||||
filePath.append("/"); |
||||
} |
||||
|
||||
String[] filePaths = fileName.split("/"); |
||||
|
||||
for (int i = 0; i < filePaths.length - 1; ++i) { |
||||
if (!StringHelper.isBlank(filePaths[i])) { |
||||
filePath.append(filePaths[i]); |
||||
filePath.append("/"); |
||||
} |
||||
} |
||||
|
||||
return filePath.toString(); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue