feat: 基本广播服务

- 由Hub接收/push/{topic}的请求并解析信息体广播到对应的Client

interval/server/ws/handler.go

@@ -2,7 +2,6 @@ package ws
 
 import (
 	"context"
-	"io"
 	"log"
 	"net/http"
 	"time"
@@ -23,53 +22,11 @@ func Handler(ctx context.Context, h *Hub) http.HandlerFunc {
 			return
 		}
 
-		c := NewClient(r.RemoteAddr, conn, ctx)
+		c := NewClient(r.RemoteAddr, conn, h, ctx)
 		log.Println("Client", r.RemoteAddr, "connected.")
-		h.RegisterClient(c)
 
-		go echo(c, h)
+		go c.ReadLoop()
+		go c.WriteLoop()
 		go heartbeat(c)
 	}
 }
-
-func echo(c *Client, h *Hub) {
-	defer func() {
-		if c.Conn != nil {
-			log.Println("Closing WebSocket connection")
-			h.UnregisterClient(c)
-			c.Cancel()
-			_ = c.Conn.Close(websocket.StatusNormalClosure, "echo finished")
-		}
-	}()
-
-	for {
-		typ, r, err := c.Conn.Reader(c.Ctx)
-
-		if err != nil {
-			if websocket.CloseStatus(err) == websocket.StatusNormalClosure {
-				log.Println("WebSocket connection closed normally")
-			} else {
-				log.Println("WebSocket reader error:", err)
-			}
-
-			return
-		}
-
-		w, err := c.Conn.Writer(c.Ctx, typ)
-		if err != nil {
-			log.Println("WebSocket writer error:", err)
-			return
-		}
-
-		_, err = io.Copy(w, r)
-		if err != nil {
-			log.Println("WebSocket copy error:", err)
-			return
-		}
-
-		if err = w.Close(); err != nil {
-			log.Println("WebSocket writer close error:", err)
-			return
-		}
-	}
-}