fix: add concurrency for database

Author: melonSorbet

src/service/DeleteMediaById.go

@@ -23,8 +23,11 @@ func DeleteMediaById(w http.ResponseWriter, r *http.Request) {
 	}
 
 	for i := 0; i < len(ids); i++ {
-		err := storage.DeleteMediaById(ids[i])
-		if err != nil {
+		ch := make(chan storage.DeleteResult)
+		go storage.DeleteMediaById(ids[i], ch)
+		result := <-ch
+
+		if result.Err != nil {
 			http.Error(w, "Could not find "+ids[i], http.StatusBadRequest)
 			return
 		}

src/service/GetMediaById.go

@@ -25,12 +25,16 @@ func GetMediaById(w http.ResponseWriter, r *http.Request) {
 	}
 
 	for i := 0; i < len(ids); i++ {
-		media, err := storage.GetMediaById(ids[i])
-		if err != nil {
+		ch := make(chan storage.GetResult)
+		go storage.GetMediaByIdDb(ids[i], ch)
+
+		result := <-ch // Properly receive from the channel
+		if result.Err != nil {
 			http.Error(w, "Could not find "+ids[i], http.StatusBadRequest)
 			return
 		}
-		mediaArray = append(mediaArray, media)
+
+		mediaArray = append(mediaArray, result.Media)
 	}
 
 	response, _ := json.Marshal(mediaArray)

src/service/PostMedia.go

@@ -90,19 +90,20 @@ func SaveMedia(handler *multipart.FileHeader, w http.ResponseWriter, mime string
 		return "", err
 	}
 	width, height := models.GetWidthAndHeigthOfImage(file)
-
+	ch := make(chan storage.AddResult)
 	media := models.Media{
 		FileType:         mime,
 		Width:            width,
 		Height:           height,
 		OriginalFileName: handler.Filename,
 	}
 
-	fileId, filepath, err := storage.AddNewMedia(media)
+	go storage.AddNewMedia(media, ch)
+	result := <-ch
+	filepath, fileId, err := result.FileUrl, result.MediaId, result.Err
 	if err != nil {
 		return "", err
 	}
-
 	dst, err := os.Create(filepath)
 	if err != nil {
 		http.Error(w, "Failed to create file", http.StatusInternalServerError)

src/storage/storage.go

@@ -9,6 +9,18 @@ import (
 	"os"
 )
 
+type GetResult struct {
+	Media models.Media
+	Err   error
+}
+type DeleteResult struct {
+	Err error
+}
+type AddResult struct {
+	MediaId string
+	FileUrl string
+	Err     error
+}
 type User struct {
 	ID       uint   `gorm:"primaryKey"`
 	Username string `gorm:"unique"`
@@ -48,38 +60,65 @@ func SetupStorage() error {
 	fmt.Println("Tables created successfully!")
 	return nil
 }
-
-func AddNewMedia(media models.Media) (string, string, error) {
-
+func AddNewMedia(media models.Media, ch chan AddResult) {
+	// Perform the database insertion
 	result := database.Debug().Create(&media)
 	if result.Error != nil {
-		fmt.Println("Could not create entry:")
-		return "", "", result.Error
+		fmt.Println("Could not create entry:", result.Error)
+		ch <- struct {
+			MediaId string
+			FileUrl string
+			Err     error
+		}{"", "", result.Error} // Send error through the channel
+		return
 	}
-	return media.MediaId, media.FileUrl, nil
-}
 
-func GetMediaById(fileId string) (models.Media, error) {
+	// Send the result (mediaId, fileUrl, and nil error) through the channel
+	ch <- struct {
+		MediaId string
+		FileUrl string
+		Err     error
+	}{media.MediaId, media.FileUrl, nil} // Send mediaId, fileUrl, and nil error
+}
+func GetMediaByIdDb(fileId string, ch chan GetResult) {
 	var media models.Media
 
 	result := database.Where("media_id = ?", fileId).First(&media)
 	if result.Error != nil {
 		log.Printf("Error fetching media by ID: %v", result.Error)
-		return models.Media{}, result.Error
+		ch <- struct {
+			Media models.Media
+			Err   error
+		}{models.Media{}, result.Error} // Send the error through the channel
+		return
 	}
-	return media, nil
+
+	ch <- struct {
+		Media models.Media
+		Err   error
+	}{media, nil} // Send the media and nil error through the channel
 }
 
-func DeleteMediaById(fileId string) error {
-	media, err := GetMediaById(fileId)
-	if err != nil {
-		fmt.Printf("Could not Get Media %v", err)
-		return err
+func DeleteMediaById(fileId string, deleteCh chan DeleteResult) {
+	ch := make(chan GetResult)
+	go GetMediaByIdDb(fileId, ch)
+	result := <-ch
+
+	if result.Err != nil {
+		fmt.Printf("Could not Get Media %v", result.Err)
+		deleteCh <- struct {
+			Err error
+		}{result.Err}
+		return
 	}
-	err = os.Remove(media.FileUrl)
-	if err != nil {
-		fmt.Printf("Could not find Media / Delete it %v", err)
+	result.Err = os.Remove(result.Media.FileUrl)
+	if result.Err != nil {
+		fmt.Printf("Could not find Media / Delete it %v", result.Err)
+		return
 	}
-	database.Where("media_id = ?", fileId).First(&media).Delete(&models.Media{})
-	return nil
+	database.Where("media_id = ?", fileId).First(&result.Media).Delete(&models.Media{})
+
+	deleteCh <- struct {
+		Err error
+	}{nil}
 }