增加关闭hub

2025-09-09 19:11:08 +08:00
parent 8d639d3b09
commit 43befdb71c
2 changed files with 18 additions and 0 deletions
--- a/internal/websocket/server.go
+++ b/internal/websocket/server.go
@@ -75,6 +75,10 @@ func (s *Server) Start() {
 	go s.hub.Run()
 }

+func (s *Server) Stop() {
+	s.hub.Close()
+}
+
 // readPump 从WebSocket连接读取消息
 func (c *Client) readPump() {
 	defer func() {