Update uploadpic.go (#208)

文件已存在则跳过保存
Hoshinonyaruko · Dec 4, 2023 · bb985a0 · bb985a0
1 parent 86edcdf
commit bb985a0
Showing 1 changed file with 20 additions and 9 deletions.
diff --git 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()