refactor: 使用引用计数的互斥锁优化 Emby 和 Jellyfin 的并发控制 (#82)

在 VideosHandler 中引入 mutexWithRefCount 类型，确保同一 item ID 的处理任务在并发情况下不会重复执行。通过原子操作管理引用计数，避免内存泄漏，提升性能和稳定性。

Author: EkkoG

internal/handler/emby.go

@@ -21,10 +21,17 @@ import (
 	"strconv"
 	"strings"
 	"sync"
+	"sync/atomic"
 
 	"github.com/gin-gonic/gin"
 )
 
+// 带引用计数的互斥锁
+type mutexWithRefCount struct {
+	mu       sync.Mutex
+	refCount int32 // 使用 atomic 操作
+}
+
 // Emby服务器处理器
 type EmbyServerHandler struct {
 	server            *emby.EmbyServer       // Emby 服务器
@@ -261,12 +268,26 @@ func (embyServerHandler *EmbyServerHandler) VideosHandler(ctx *gin.Context) {
 
 	// 并发控制：确保同一个 item ID 只有一个任务在运行
 	// 将整个处理流程放在锁内，避免重复查询和重复获取重定向 URL
-	var mu *sync.Mutex
+	var muWrapper *mutexWithRefCount
 	if itemID != "" {
-		mutex, _ := embyServerHandler.playbackInfoMutex.LoadOrStore(itemID, &sync.Mutex{})
-		mu = mutex.(*sync.Mutex)
-		mu.Lock()
-		defer mu.Unlock()
+		// 加载或创建 mutex wrapper
+		value, _ := embyServerHandler.playbackInfoMutex.LoadOrStore(itemID, &mutexWithRefCount{})
+		muWrapper = value.(*mutexWithRefCount)
+
+		// 增加引用计数
+		atomic.AddInt32(&muWrapper.refCount, 1)
+
+		// 锁定并处理
+		muWrapper.mu.Lock()
+		defer func() {
+			muWrapper.mu.Unlock()
+			// 减少引用计数
+			refCount := atomic.AddInt32(&muWrapper.refCount, -1)
+			// 如果没有其他 goroutine 在使用，删除这个 mutex 以避免内存泄漏
+			if refCount == 0 {
+				embyServerHandler.playbackInfoMutex.Delete(itemID)
+			}
+		}()
 		logging.Debugf("开始处理 item %s 的 VideosHandler 请求", itemID)
 	}
 

internal/handler/jellyfin.go

@@ -21,6 +21,7 @@ import (
 	"strconv"
 	"strings"
 	"sync"
+	"sync/atomic"
 
 	"github.com/gin-gonic/gin"
 )
@@ -94,7 +95,7 @@ func (jellyfinHandler *JellyfinHandler) GetImageCacheRegexp() *regexp.Regexp {
 	return constants.JellyfinRegexp.Cache.Image
 }
 
-func (JellyfinHandler) GetSubtitleCacheRegexp() *regexp.Regexp {
+func (*JellyfinHandler) GetSubtitleCacheRegexp() *regexp.Regexp {
 	return constants.JellyfinRegexp.Cache.Subtitle
 }
 
@@ -234,12 +235,26 @@ func (jellyfinHandler *JellyfinHandler) VideosHandler(ctx *gin.Context) {
 
 	// 并发控制：确保同一个 item ID 只有一个任务在运行
 	// 将整个处理流程放在锁内，避免重复查询和重复获取重定向 URL
-	var mu *sync.Mutex
+	var muWrapper *mutexWithRefCount
 	if itemID != "" {
-		mutex, _ := jellyfinHandler.playbackInfoMutex.LoadOrStore(itemID, &sync.Mutex{})
-		mu = mutex.(*sync.Mutex)
-		mu.Lock()
-		defer mu.Unlock()
+		// 加载或创建 mutex wrapper
+		value, _ := jellyfinHandler.playbackInfoMutex.LoadOrStore(itemID, &mutexWithRefCount{})
+		muWrapper = value.(*mutexWithRefCount)
+
+		// 增加引用计数
+		atomic.AddInt32(&muWrapper.refCount, 1)
+
+		// 锁定并处理
+		muWrapper.mu.Lock()
+		defer func() {
+			muWrapper.mu.Unlock()
+			// 减少引用计数
+			refCount := atomic.AddInt32(&muWrapper.refCount, -1)
+			// 如果没有其他 goroutine 在使用，删除这个 mutex 以避免内存泄漏
+			if refCount == 0 {
+				jellyfinHandler.playbackInfoMutex.Delete(itemID)
+			}
+		}()
 		logging.Debugf("开始处理 item %s 的 VideosHandler 请求", itemID)
 	}