实现封面缓存

internal/service/cover_svc.go

@@ -1,13 +1,10 @@
 package service
 
 import (
-	"butterfliu/config"
 	"butterfliu/internal/model"
 	"butterfliu/internal/repository"
 	"butterfliu/internal/util"
 	"errors"
-	"path"
-	"strconv"
 )
 
 type CoverService struct {
@@ -37,9 +34,11 @@ func (s *CoverService) GetSongCover(id int) (string, error) {
 	if err != nil {
 		return "", err
 	}
-	util.ExtractCover(file.Path, id)
-	conf := config.LoadConfig()
-	return path.Join(conf.GetCachePath("cover"), strconv.Itoa(id)+".jpg"), nil
+	cover, err := util.ExtractCover(file.Path, id)
+	if err != nil {
+		return "", err
+	}
+	return cover, nil
 }
 
 func (s *CoverService) GetAlbumCover(id int) (string, error) {

internal/util/ffmpeg.go

@@ -1,20 +1,30 @@
 package util
 
 import (
+	"butterfliu/config"
+	"os"
 	"os/exec"
+	"path"
 	"strconv"
 )
 
-func ExtractCover(filePath string, id int) bool {
-	if _, err := exec.LookPath("ffmpeg"); err != nil {
-		return false
+func ExtractCover(filePath string, id int) (string, error) {
+	config := config.LoadConfig()
+	coverPath := path.Join(config.GetCachePath("cover"), strconv.Itoa(id)+".jpg")
+
+	_, err := os.Stat(coverPath)
+	if err == nil {
+		return coverPath, nil
+	}
+
+	if _, err := exec.LookPath("ffmpeg"); err != nil {
+		return "", err
 	}
-	coverPath := "cache/cover/" + strconv.Itoa(id) + ".jpg"
 	cmd := exec.Command("ffmpeg", "-i", filePath, "-an", "-vcodec", "mjpeg", "-q:v", "5", "-f", "image2", "-y", coverPath)
 	if err := cmd.Run(); err != nil {
-		return false
+		return "", err
 	}
 
-	return true
+	return coverPath, nil
 
 }