From 7e33fcf7c73626ff81eeb6be7f3163c412a2a85c Mon Sep 17 00:00:00 2001
From: srcrs <srcrs@foxmail.com>
Date: Wed, 2 Aug 2023 01:56:22 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8A=A0=E9=94=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 5d45dc5..390df0b 100644
--- a/main.go
+++ b/main.go
@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"log"
 	"net/http"
+	"sync"
 	"time"
 
 	"github.com/dgraph-io/badger"
@@ -23,6 +24,7 @@ var (
 	db       *badger.DB
 	rssUrls  Config
 	upgrader = websocket.Upgrader{}
+	connMu   sync.Mutex // 定义互斥锁
 )
 
 func init() {
@@ -75,10 +77,13 @@ func wsHandler(w http.ResponseWriter, r *http.Request) {
 	if rssUrls.AutoUpdatePush > 0 {
 		go func() {
 			for {
+				connMu.Lock() // 加锁
 				if err := conn.WriteMessage(websocket.TextMessage, []byte("heartbeat")); err != nil {
 					log.Printf("heartbeat Write error: %v", err)
+					connMu.Unlock() // 解锁
 					return
 				}
+				connMu.Unlock() // 解锁
 				time.Sleep(time.Duration(10) * time.Second)
 			}
 		}()
@@ -90,13 +95,15 @@ func wsHandler(w http.ResponseWriter, r *http.Request) {
 				log.Printf("Error getting feed from Redis: %v", err)
 				continue
 			}
-
+			connMu.Lock() // 加锁
 			err = conn.WriteMessage(websocket.TextMessage, []byte(feedJSON))
 			//错误直接关闭更新
 			if err != nil {
 				log.Printf("Error sending message or Connection closed: %v", err)
+				connMu.Unlock() // 解锁
 				return
 			}
+			connMu.Unlock() // 解锁
 		}
 		//如果未配置则不自动更新
 		if rssUrls.AutoUpdatePush == 0 {