From 75efe9739dd7600882365903bd1859b0ee7fb7ad Mon Sep 17 00:00:00 2001
From: zhangmei <645025773@qq.com>
Date: 星期五, 21 二月 2025 14:01:07 +0800
Subject: [PATCH] 流水导入
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBankFlowController.java | 116 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 115 insertions(+), 1 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBankFlowController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBankFlowController.java
index 9c5cdd4..fff605c 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBankFlowController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBankFlowController.java
@@ -1,19 +1,41 @@
package com.ruoyi.web.controller.api;
+import cn.afterturn.easypoi.excel.ExcelExportUtil;
+import cn.afterturn.easypoi.excel.entity.ExportParams;
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.context.AnalysisContext;
+import com.alibaba.excel.event.AnalysisEventListener;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.R;
+import com.ruoyi.common.utils.WebUtils;
+import com.ruoyi.system.export.ContractExport;
+import com.ruoyi.system.importExcel.TBankFlowImportExcel;
import com.ruoyi.system.model.TBankFlow;
import com.ruoyi.system.query.TBankFlowQuery;
import com.ruoyi.system.service.TBankFlowService;
import com.ruoyi.system.vo.TBankFlowStatisticsVo;
import io.swagger.annotations.ApiOperation;
+import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+import org.springframework.web.multipart.MultipartFile;
+
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URLEncoder;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Vector;
+import java.util.concurrent.atomic.AtomicInteger;
/**
* <p>
@@ -28,6 +50,7 @@
public class TBankFlowController {
@Autowired
private TBankFlowService flowService;
+
/**
* 获取流水列表
*/
@@ -40,8 +63,99 @@
@ApiOperation(value = "根据支付方式统计流水金额")
@PostMapping("/getPaymentStats")
- public R<TBankFlowStatisticsVo> getPaymentStats(@RequestBody TBankFlowQuery query){
+ public R<TBankFlowStatisticsVo> getPaymentStats(@RequestBody TBankFlowQuery query) {
return R.ok(flowService.getPaymentStats(query));
}
+
+ @PostMapping("importBankFlow")
+ public void importElectronic(@RequestBody MultipartFile file) {
+ try (
+ InputStream is = file.getInputStream();
+ ) {
+ List<TBankFlowImportExcel> failList = new Vector<>();
+ EasyExcel.read(is, TBankFlowImportExcel.class, new AnalysisEventListener<TBankFlowImportExcel>() {
+ private final List<TBankFlowImportExcel> list = new ArrayList<>();
+ final AtomicInteger all = new AtomicInteger();
+
+ @Override
+ public void invoke(TBankFlowImportExcel data, AnalysisContext context) {
+ all.addAndGet(1);
+ boolean isok = data.validate();
+ if (!isok) {
+ failList.add(data);
+ return;
+ }
+ list.add(data);
+ int size = list.size();
+ if (size >= 100) {
+ flowService.saveImportBatch(list, failList);
+ list.clear();
+ }
+ }
+
+ @Override
+ public void doAfterAllAnalysed(AnalysisContext context) {
+ int size = list.size();
+ if (size > 0) {
+ flowService.saveImportBatch(list, failList);
+ }
+ }
+ }).sheet().doRead();
+ // 导出导入结果
+ // ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+ // HttpServletResponse response = servletRequestAttributes.getResponse();
+ // response.setContentType("application/vnd.ms-excel;charset=utf-8");
+ // response.setHeader("Cache-Control", "no-cache, no-store, must-revalidate");
+ // if (failList.size()>0) {
+ // EasyExcel.write(response.getOutputStream(), TBankFlowImportExcel.class).sheet("Sheet1").doWrite(failList);
+ // } else {
+ // TBankFlowImportExcel result = new TBankFlowImportExcel();
+ // result.setResult("全部成功");
+ // failList.add(result);
+ // EasyExcel.write(response.getOutputStream(), TBankFlowImportExcel.class).sheet("Sheet1").doWrite(failList);
+ // }
+
+
+ if (failList.size()==0){
+ TBankFlowImportExcel result = new TBankFlowImportExcel();
+ result.setResult("全部成功");
+ failList.add(result);
+ }
+ Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), TBankFlowImportExcel.class, failList);
+ HttpServletResponse response = WebUtils.response();
+ response.setContentType("application/vnd.ms-excel");
+ response.setCharacterEncoding("utf-8");
+ ServletOutputStream outputStream = null;
+ try {
+ String fileName = URLEncoder.encode("银行流水返回结果.xls", "utf-8");
+ response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
+ response.setContentType("application/vnd.ms-excel;charset=UTF-8");
+ response.setHeader("Pragma", "no-cache");
+ response.setHeader("Cache-Control", "no-cache");
+ outputStream = response.getOutputStream();
+ workbook.write(outputStream);
+ } catch (IOException e) {
+ e.printStackTrace();
+ System.err.println("银行流水返回结果导出失败");
+ } finally {
+ try {
+ outputStream.close();
+ } catch (IOException e) {
+ e.printStackTrace();
+ }
+ }
+
+
+
+
+
+
+ } catch (Exception e) {
+ e.printStackTrace();
+ System.err.println("银行流水返回结果导出失败");
+ }
+ }
}
+
+
--
Gitblit v1.7.1