golang websocket详解

使用Golang实现WebSocket通信 WebSocket是一种全双工的通信协议，允许客户端和服务器之间进行实时的双向通信。在开发Web应用程序时，很多情况下需要实现实时通信功能，例如聊天室、实时数据展示等。而Golang作为一种高性能的编程语言，也提供了内置的`net/http`包以及`gorilla/websocket`包来支持WebSocket的开发。本文将介绍使用Golang实现WebSocket通信的基本步骤和注意事项。 ## 准备工作 在开始之前，我们首先需要安装Golang的开发环境，并且对于Golang的基本语法和HTTP协议有一定的了解。同时，我们还需要安装`gorilla/websocket`包，这个包提供了更加方便的WebSocket相关操作方法。可以通过执行以下命令来安装： ``` go get github.com/gorilla/websocket ``` ## 建立WebSocket服务器 下面的代码演示了如何使用Golang建立一个WebSocket服务器： ```go package main import ( "log" "net/http" "github.com/gorilla/websocket" ) var upgrader = websocket.Upgrader{} // 使用默认的upgrader func echo(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Println(err) return } defer conn.Close() for { messageType, message, err := conn.ReadMessage() if err != nil { log.Println(err) return } log.Printf("Received: %s\n", message) err = conn.WriteMessage(messageType, message) if err != nil { log.Println(err) return } } } func main() { http.HandleFunc("/ws", echo) log.Fatal(http.ListenAndServe(":8080", nil)) } ``` **代码解析** - 创建一个`gorilla/websocket`的默认upgrader对象，用于将普通的HTTP连接升级为WebSocket连接。 - 定义一个处理WebSocket通信的回调函数`echo`。在该函数中，先通过upgrader将HTTP连接升级为WebSocket连接。然后使用`ReadMessage`方法读取客户端发送的消息，并打印在服务端日志中。最后使用`WriteMessage`方法将服务端接收到的消息原样返回给客户端。 - 使用`http.HandleFunc`函数将路径`/ws`映射到`echo`处理函数。 - 使用`http.ListenAndServe`函数启动HTTP服务器。 ## 建立WebSocket客户端 下面的代码演示了如何使用Golang建立一个WebSocket客户端并发送消息： ```go package main import ( "fmt" "log" "net/url" "os" "os/signal" "time" "github.com/gorilla/websocket" ) func main() { interrupt := make(chan os.Signal, 1) signal.Notify(interrupt, os.Interrupt) u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/ws"} log.Printf("connecting to %s", u.String()) c, _, err := websocket.DefaultDialer.Dial(u.String(), nil) if err != nil { log.Fatal("dial:", err) } defer c.Close() done := make(chan struct{}) go func() { defer close(done) for { _, message, err := c.ReadMessage() if err != nil { log.Println("read:", err) return } log.Printf("received: %s", message) } }() ticker := time.NewTicker(time.Second) defer ticker.Stop() for { select { case <-done: return case t := <-ticker.C: err := c.WriteMessage(websocket.TextMessage, []byte(fmt.Sprintf("message sent at %v", t))) if err != nil { log.Println("write:", err) return } case <-interrupt: log.Println("interrupt") err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")) if err != nil { log.Println("write close:", err) return } select { case <-done: case <-time.After(time.Second): } return } } } ``` **代码解析** - 创建一个WebSocket客户端，连接到`localhost:8080/ws`。 - 启动一个协程用于读取服务端发送的消息，并打印在客户端日志中。 - 使用定时器每隔一秒发送一条消息到服务端。 - 监听操作系统的中断信号，当接收到中断信号后，调用`WriteMessage`方法发送关闭消息给服务端，并在一秒内等待服务端的关闭响应。 ## 运行 在命令行中分别运行两个程序：WebSocket服务器和WebSocket客户端。运行之后，可以看到服务端打印出接收到的消息，并且客户端每秒发送一条消息到服务端。 ## 小结 通过本文的介绍，我们学习了如何使用Golang建立一个简单的WebSocket服务器和客户端，并进行消息的发送和接收。但是需要注意的是，在实际应用中，我们需要处理更多的错误情况、并发连接数较大的情况等。同时，可以根据具体需求来扩展和优化代码。希望本文对于你理解和使用Golang的WebSocket有所帮助。