实现两个 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 经常用于服务和/或微服务,所以在这些服务之间进行通信是有意义的 。