From d293aa53e1c734d8f2b75c1992134d0b99d21917 Mon Sep 17 00:00:00 2001 From: rfyiamcool Date: Tue, 23 Jan 2024 15:02:44 +0800 Subject: [PATCH] feat: format message type Signed-off-by: rfyiamcool --- examples/echo/client.go | 4 ++-- examples/echo/server.go | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/echo/client.go b/examples/echo/client.go index 7d870bd..d53b81c 100644 --- a/examples/echo/client.go +++ b/examples/echo/client.go @@ -41,12 +41,12 @@ func main() { go func() { defer close(done) for { - _, message, err := c.ReadMessage() + mt, message, err := c.ReadMessage() if err != nil { log.Println("read:", err) return } - log.Printf("recv: %s", message) + log.Printf("recv: %s, type: %s", message, websocket.FormatMessageType(mt)) } }() diff --git a/examples/echo/server.go b/examples/echo/server.go index f9a0b7b..9804e6b 100644 --- a/examples/echo/server.go +++ b/examples/echo/server.go @@ -33,7 +33,8 @@ func echo(w http.ResponseWriter, r *http.Request) { log.Println("read:", err) break } - log.Printf("recv: %s", message) + + log.Printf("recv: %s, type: %s", message, websocket.FormatMessageType(mt)) err = c.WriteMessage(mt, message) if err != nil { log.Println("write:", err)