Update: websocket 客户端

This commit is contained in:
2026-02-24 15:52:40 +08:00
parent e92361eb80
commit 325026fade
2 changed files with 51 additions and 3 deletions
+48
View File
@@ -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)
}
}
+3 -3
View File
@@ -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)
}
}
}