
实现两个 Go 服务之间做 WebSocket 通信要怎么做?程序本文做了介绍。
使用 WebSockets 时,通信一个程序必须充当服务器。程序
但是通信可以有许多其他程序作为客户端。在本文中,程序我们将创建一个服务器和一个客户端
WebSocket Server
服务器的通信代码非常简单。因为我们不想重新发明轮子,程序所以我们将使用gobwas模块
下面是通信运行在端口 8080 上的 WebSocket 服务器的代码
package main import ( "fmt" "github.com/gobwas/ws" "github.com/gobwas/ws/wsutil" "math/rand" "net/http" "strconv" ) func main() { fmt.Println("Server started, waiting for connection from client") http.ListenAndServe(":8080", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Println("Client connected") conn, _, _, err := ws.UpgradeHTTP(r, w) if err != nil { fmt.Println("Error starting socket server: " + err.Error()) } go func() { defer conn.Close() for { msg, op, err := wsutil.ReadClientData(conn) if err != nil { fmt.Println("Error receiving data: " + err.Error()) fmt.Println("Client disconnected") return } fmt.Println("Client message received with random number: " + string(msg)) randomNumber := strconv.Itoa(rand.Intn(100)) err = wsutil.WriteServerMessage(conn, op, []byte(randomNumber)) if err != nil { fmt.Println("Error sending data: " + err.Error()) fmt.Println("Client disconnected") return } fmt.Println("Server message send with random number " + randomNumber) } }() })) } 服务器除了等待连接之外什么都不做,打印接收到的亿华云计算程序数据并最后发送一条消息(随机整数)
如果客户端断开连接(或有人停止程序),它将打印有关此断开连接的通信信息
WebSocket Client
客户端的代码类似。同样,程序我们将使用gobwas模块
客户端正在通过端口 8080 连接到 localhost
package main import ( "context" "fmt" "github.com/gobwas/ws" "github.com/gobwas/ws/wsutil" "math/rand" "os" "strconv" "time" ) func main() { fmt.Println("Client started") for { conn,通信 _, _, err := ws.DefaultDialer.Dial(context.Background(), "ws://127.0.0.1:8080/") if err != nil { fmt.Println("Cannot connect: " + err.Error()) time.Sleep(time.Duration(5) * time.Second) continue } fmt.Println("Connected to server") for i := 0; i < 10; i++ { randomNumber := strconv.Itoa(rand.Intn(100)) msg := []byte(randomNumber) err = wsutil.WriteClientMessage(conn, ws.OpText, msg) if err != nil { fmt.Println("Cannot send: " + err.Error()) continue } fmt.Println("Client message send with random number " + randomNumber) msg, _, err := wsutil.ReadServerData(conn) if err != nil { fmt.Println("Cannot receive data: " + err.Error()) continue } fmt.Println("Server message received with random number: " + string(msg)) time.Sleep(time.Duration(5) * time.Second) } err = conn.Close() if err != nil { fmt.Println("Cannot close the connection: " + err.Error()) os.Exit(1) } fmt.Println("Disconnected from server") } } 这个客户端除了连接、发送消息(随机整数)并打印发回的程序任何内容之外没有什么特别的
测试
左侧,服务器结果 右侧,通信客户端结果
客户端每 5 秒向服务器发送一个随机数。程序服务器用另一个随机数回复
如果您停止服务器,您将在客户端看到服务器已断开连接。服务器托管如果您再次启动服务器,客户端将再次连接。
您可以以另一种方式期待相同的行为。如果停止客户端,服务器将打印一条消息,表明客户端已断开连接。如果您再次运行客户端......好吧,它再次连接
因为 Go 经常用于服务和/或微服务,所以在这些服务之间进行通信是有意义的 。
云服务器