refactor: add if and using variable that returns from the database

kevinanielsen · Jan 31, 2024 · aa20f8a · aa20f8a
1 parent af22ce9
commit aa20f8a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/handlers/docs/handleDocUpload.go b/src/handlers/docs/handleDocUpload.go
@@ -77,9 +77,13 @@ func HandleDocUpload(c *gin.Context) {
 		return
 	}
 
-	savedFileName := database.AddDoc(doc)
+	savedFileName, err := database.AddDoc(doc)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, err.Error())
+		return
+	}
 
-	err = c.SaveUploadedFile(fileHeader, util.ExPath+"/uploads/docs/"+filteredFilename)
+	err = c.SaveUploadedFile(fileHeader, util.ExPath+"/uploads/docs/"+savedFileName)
 	if err != nil {
 		c.String(http.StatusInternalServerError, "Failed to save file: %s", err.Error())
 		return

diff --git a/src/handlers/image/handleImageUpload.go b/src/handlers/image/handleImageUpload.go
@@ -85,7 +85,7 @@ func HandleImageUpload(c *gin.Context) {
 		return
 	}
 
-	err = c.SaveUploadedFile(fileHeader, util.ExPath+"/uploads/images/"+filteredFilename)
+	err = c.SaveUploadedFile(fileHeader, util.ExPath+"/uploads/images/"+savedFilename)
 	if err != nil {
 		c.String(http.StatusInternalServerError, "Failed to save file: %s", err.Error())
 		return