From 23d68effd45eb187c17f32bcec3b5496ff070936 Mon Sep 17 00:00:00 2001
From: momo <1012651275@qq.com>
Date: Mon, 21 Aug 2023 13:42:11 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E6=96=87=E4=BB=B6=E4=B8=8A=E4=BC=A0?=
 =?UTF-8?q?=E5=A4=A7=E5=B0=8F=E9=99=90=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/upload/internal/logic/uploadlogologic.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/server/upload/internal/logic/uploadlogologic.go b/server/upload/internal/logic/uploadlogologic.go
index bd495c61..0318e245 100644
--- a/server/upload/internal/logic/uploadlogologic.go
+++ b/server/upload/internal/logic/uploadlogologic.go
@@ -95,6 +95,12 @@ func (l *UploadLogoLogic) UploadLogo(req *types.UploadLogoReq, userinfo *auth.Us
 		return resp.SetStatus(basic.CodeFileUploadErr, "file upload err,file is not image")
 	}
 
+	// 限制上传文件大小 50k
+	maxSize := 100 * 1024
+	if fileHeader.Size > int64(maxSize) {
+		return resp.SetStatus(basic.CodeFileUploadErr, "file upload err,The file size exceeds the maximum limit of 100k")
+	}
+
 	// 读取数据流
 	ioData, err := io.ReadAll(fileObject)
 	if err != nil {