From 868460f6eac37b858d0698c9e57023b04fc87db2 Mon Sep 17 00:00:00 2001
From: zhaoqiaobo <583671871@qq.com>
Date: Fri, 14 Jun 2024 15:16:45 +0800
Subject: [PATCH] =?UTF-8?q?feat(all):=20=E5=A2=9E=E5=8A=A0=E6=8A=A5?=
 =?UTF-8?q?=E8=A1=A8=E5=AF=BC=E5=87=BA=E6=95=B0=E9=87=8F=E9=99=90=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/main/java/com/logpm/report/util/ReportExcelUtil.java  | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/blade-service/logpm-report/src/main/java/com/logpm/report/util/ReportExcelUtil.java b/blade-service/logpm-report/src/main/java/com/logpm/report/util/ReportExcelUtil.java
index 1b862b28a..625bddc6b 100644
--- a/blade-service/logpm-report/src/main/java/com/logpm/report/util/ReportExcelUtil.java
+++ b/blade-service/logpm-report/src/main/java/com/logpm/report/util/ReportExcelUtil.java
@@ -9,6 +9,7 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.logpm.report.constant.ReportConstants;
 import com.logpm.report.service.ExportReader;
 import org.apache.poi.ss.formula.functions.T;
+import org.springblade.core.log.exception.ServiceException;
 
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
@@ -87,6 +88,9 @@ public class ReportExcelUtil {
 		ReportExcelUtil.setExportHeader(response, fileName);
 		// 获取需要导出的总条数
 		Long count = service.getCount(query);
+		if(count > 500000L){
+			throw new ServiceException("导出数据不能大于50w，请增加检索条件");
+		}
 		// 每个 sheet 放多少条数据
 		Double sheetDataNum = onceSheetDataNum.doubleValue();
 		// 总共要几个 sheet