From: qydysky <qydysky@foxmail.com>
Date: Sun, 21 May 2023 01:29:53 +0000 (+0800)
Subject: Fix 直播流回放连接限制
X-Git-Tag: v0.9.6~8
X-Git-Url: http://127.0.0.1:8081/?a=commitdiff_plain;h=c2e7bc8306a098b83d0b03f92696ddd518662a05;p=bili_danmu%2F.git

Fix 直播流回放连接限制
---

diff --git a/Reply/F.go b/Reply/F.go
index 2b70284..58f3f09 100644
--- a/Reply/F.go
+++ b/Reply/F.go
@@ -1239,15 +1239,10 @@ func init() {
 		// 直播流播放器
 		c.C.SerF.Store(path+"player/", func(w http.ResponseWriter, r *http.Request) {
 			// 直播流回放连接限制
-			if maxCon > 0 {
-				if limitCon.Add(1) > maxCon {
-					limitCon.Add(-1)
-					w.WriteHeader(http.StatusTooManyRequests)
-					_, _ = w.Write([]byte("已达到设定最大连接数"))
-					return
-				} else {
-					defer limitCon.Add(-1)
-				}
+			if maxCon > 0 && limitCon.Load() >= maxCon {
+				w.WriteHeader(http.StatusTooManyRequests)
+				_, _ = w.Write([]byte("已达到设定最大连接数"))
+				return
 			}
 
 			p := strings.TrimPrefix(r.URL.Path, path+"player/")
@@ -1277,7 +1272,6 @@ func init() {
 				if limitCon.Add(1) > maxCon {
 					limitCon.Add(-1)
 					w.WriteHeader(http.StatusTooManyRequests)
-					_, _ = w.Write([]byte("已达到设定最大连接数"))
 					return
 				} else {
 					defer limitCon.Add(-1)