From 325026fade78e5bf34794954b97a8c642a0b0b72 Mon Sep 17 00:00:00 2001 From: HeZhaohui Date: Tue, 24 Feb 2026 15:52:40 +0800 Subject: [PATCH] =?UTF-8?q?Update:=20websocket=20=E5=AE=A2=E6=88=B7?= =?UTF-8?q?=E7=AB=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- client.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ main.go | 6 +++--- 2 files changed, 51 insertions(+), 3 deletions(-) create mode 100644 client.go diff --git a/client.go b/client.go new file mode 100644 index 0000000..18a8ff7 --- /dev/null +++ b/client.go @@ -0,0 +1,48 @@ +package main + +import ( + "bufio" + "fmt" + "log" + "os" + "github.com/gorilla/websocket" +) + +func main() { + url := "ws://localhost:8080/ws" + log.Printf("来都来了") + conn, _, err := websocket.DefaultDialer.Dial(url, nil) + if err != nil { + log.Fatal("哟,今天不营业呢", err) + } + defer conn.Close() + log.Println("好好好,进去了") + + go func() { + for { + _, message, err := conn.ReadMessage() + if err != nil { + log.Println("润了润了") + break + } + log.Printf("掌柜说的是 %s", message) + } + }() + + scanner := bufio.NewScanner(os.Stdin) + fmt.Print("说点什么:") + + for scanner.Scan() { + message := scanner.Text() + err := conn.WriteMessage(websocket.TextMessage, []byte(message)) + if err != nil { + log.Println("话没送出去", err) + break + } + fmt.Print("说点什么:") + } + + if err := scanner.Err(); err != nil { + log.Println("输入错误:", err) + } +} diff --git a/main.go b/main.go index d8afb12..15d5af7 100644 --- a/main.go +++ b/main.go @@ -1,13 +1,13 @@ package main import ( + "github.com/gorilla/websocket" "log" "net/http" - "github.com/gorilla/websocket" ) var upgrader = websocket.Upgrader{ - ReadBufferSize: 1024, + ReadBufferSize: 1024, WriteBufferSize: 1024, CheckOrigin: func(r *http.Request) bool { return true @@ -48,4 +48,4 @@ func main() { if err != nil { log.Fatal("下回见伙计们", err) } -} \ No newline at end of file +}