From 4966ba8f43ef256ca00cb2580e5de4d70a1bbbdb Mon Sep 17 00:00:00 2001
From: qydysky <32743305+qydysky@users.noreply.github.com>
Date: Sun, 26 Feb 2023 00:56:09 +0800
Subject: [PATCH] =?utf8?q?Improve=20=E9=87=8D=E8=BD=BD=E5=BC=B9=E5=B9=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

---
 F/api.go      | 6 ++++--
 bili_danmu.go | 1 +
 go.mod        | 2 +-
 go.sum        | 6 ++++++
 4 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/F/api.go b/F/api.go
index ed99809..c8c5b69 100644
--- a/F/api.go
+++ b/F/api.go
@@ -760,12 +760,14 @@ func (c *GetFunc) getDanmuInfo() (missKey []string) {
 		//弹幕钥
 		c.Token = j.Data.Token
 		//弹幕链接
+		var tmp []string
 		for _, v := range j.Data.HostList {
 			if v.WssPort != 443 {
-				c.WSURL = append(c.WSURL, "wss://"+v.Host+":"+strconv.Itoa(v.WssPort)+"/sub")
+				tmp = append(tmp, "wss://"+v.Host+":"+strconv.Itoa(v.WssPort)+"/sub")
 			}
-			c.WSURL = append(c.WSURL, "wss://"+v.Host+"/sub")
+			tmp = append(tmp, "wss://"+v.Host+"/sub")
 		}
+		c.WSURL = tmp
 	}
 	return
 }
diff --git a/bili_danmu.go b/bili_danmu.go
index 0d7a910..6edf113 100644
--- a/bili_danmu.go
+++ b/bili_danmu.go
@@ -105,6 +105,7 @@ func Start() {
 		//使用带tag的消息队列在功能间传递消息
 		c.C.Danmu_Main_mq.Pull_tag(msgq.FuncMap{
 			`flash_room`: func(_ any) bool { //房间重进
+				F.Get(&c.C).Get(`WSURL`)
 				select {
 				case flash_room_chan <- struct{}{}:
 				default:
diff --git a/go.mod b/go.mod
index 4f21c92..5ac278b 100644
--- a/go.mod
+++ b/go.mod
@@ -6,7 +6,7 @@ require (
 	github.com/gofrs/uuid v4.3.0+incompatible
 	github.com/gotk3/gotk3 v0.6.1
 	github.com/mdp/qrterminal/v3 v3.0.0
-	github.com/qydysky/part v0.22.18
+	github.com/qydysky/part v0.22.21
 	github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e
 	github.com/skratchdot/open-golang v0.0.0-20200116055534-eef842397966
 	golang.org/x/text v0.3.8
diff --git a/go.sum b/go.sum
index 77535df..413994a 100644
--- a/go.sum
+++ b/go.sum
@@ -39,6 +39,12 @@ github.com/qydysky/part v0.22.12 h1:g06QYeE4qwR2s+tO4UgBnYhds7Fah7nBhtIG5fZ9iB0=
 github.com/qydysky/part v0.22.12/go.mod h1:T6tQU8VYOVT+rX5v40Y7OeDWByz4fwDnyehpa+QIP2c=
 github.com/qydysky/part v0.22.18 h1:U+UJi+XHVwTr17jKfSX7yk+hGkZB+YDxoZTGSw0su0A=
 github.com/qydysky/part v0.22.18/go.mod h1:T6tQU8VYOVT+rX5v40Y7OeDWByz4fwDnyehpa+QIP2c=
+github.com/qydysky/part v0.22.19 h1:KEPFUp/TJseL3RYq/5jH4jCfVituA7vbknPOAIu/azA=
+github.com/qydysky/part v0.22.19/go.mod h1:T6tQU8VYOVT+rX5v40Y7OeDWByz4fwDnyehpa+QIP2c=
+github.com/qydysky/part v0.22.20 h1:zbYo4MzQWxcH7L7PSXF7C+8q07hHyh5HseaQHcnXaO8=
+github.com/qydysky/part v0.22.20/go.mod h1:T6tQU8VYOVT+rX5v40Y7OeDWByz4fwDnyehpa+QIP2c=
+github.com/qydysky/part v0.22.21 h1:qin7pM8pNJ4J+c7w1eHS+4//SvhWm9VPP2tGLAckwVk=
+github.com/qydysky/part v0.22.21/go.mod h1:T6tQU8VYOVT+rX5v40Y7OeDWByz4fwDnyehpa+QIP2c=
 github.com/shirou/gopsutil v3.21.11+incompatible h1:+1+c1VGhc88SSonWP6foOcLhvnKlUeu/erjjvaPEYiI=
 github.com/shirou/gopsutil v3.21.11+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMTu2PcXUg8+E1lC7eC3UO/RA=
 github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e h1:MRM5ITcdelLK2j1vwZ3Je0FKVCfqOLp5zO6trqMLYs0=
-- 
2.39.2