From f7e6917203b2eca4ff0b5e6afbdc313d204e2223 Mon Sep 17 00:00:00 2001
From: yuelengqingchen <2279839557@qq.com>
Date: Sun, 3 Dec 2023 17:11:06 +0800
Subject: [PATCH] Update uploadpic.go
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

文件已存在则跳过保存
---
 server/uploadpic.go | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

diff --git a/server/uploadpic.go b/server/uploadpic.go
index d1a1a019..1897cf17 100644
--- a/server/uploadpic.go
+++ b/server/uploadpic.go
@@ -81,11 +81,17 @@ func UploadBase64ImageHandler(rateLimiter *RateLimiter) gin.HandlerFunc {
 			return
 		}
 
-		err = os.WriteFile(savePath, imageBytes, 0644)
-		if err != nil {
-			c.JSON(http.StatusInternalServerError, gin.H{"error": "error saving file"})
-			return
+		//如果文件存在则跳过
+		if _, err := os.Stat(savePath); os.IsNotExist(err) {
+			err = os.WriteFile(savePath, imageBytes, 0644)
+			if err != nil {
+				c.JSON(http.StatusInternalServerError, gin.H{"error": "error saving file"})
+				return
+			}
+		} else {
+			mylog.Println("File already exists, skipping save.")
 		}
+		
 		var serverPort string
 		serverAddress := config.GetServer_dir()
 		frpport := config.GetFrpPort()
@@ -146,11 +152,16 @@ func UploadBase64RecordHandler(rateLimiter *RateLimiter) gin.HandlerFunc {
 			c.JSON(http.StatusInternalServerError, gin.H{"error": "error creating directory"})
 			return
 		}
-
-		err = os.WriteFile(savePath, RecordBytes, 0644)
-		if err != nil {
-			c.JSON(http.StatusInternalServerError, gin.H{"error": "error saving file"})
-			return
+		
+		//如果文件存在则跳过
+		if _, err := os.Stat(savePath); os.IsNotExist(err) {
+			err = os.WriteFile(savePath, RecordBytes, 0644)
+			if err != nil {
+				c.JSON(http.StatusInternalServerError, gin.H{"error": "error saving file"})
+				return
+			}
+		} else {
+			mylog.Println("File already exists, skipping save.")
 		}
 
 		serverAddress := config.GetServer_dir()