近年来，websocket技术已经成为了web开发中不可或缺的一部分。websocket 是一种在单个 tcp 连接上进行全双工通信的协议，它使得客户端和服务器之间的通信更加流畅和高效。如今，很多现代的web应用程序都使用了websocket技术，例如实时聊天、在线游戏以及实时数据可视化等。
Go语言作为一个现代的编程语言，自然也提供了很好的支持WebSocket技术的功能。在本文中，我们将深入探讨如何在Go中使用WebSocket技术。
WebSocket在Go语言中的实现
WebSocket 协议是由HTTP升级协议而来，因此在Go语言中使用WebSocket也需要将其与HTTP协议相结合。Go语言已经内置了net/http包，该包提供了各种HTTP相关的功能。如果我们想要在Go中使用WebSocket，只需导入net/http和github.com/gorilla/websocket库即可。
下面，我们就来看一下如何使用github.com/gorilla/websocket库来实现WebSocket。
创建WebSocket服务端
在Go语言中，创建WebSocket服务端的方式也比较简单。我们直接看下列代码，它展示了如何创建一个HTTP服务器，等待客户端的WebSocket连接：
package main
import (
"fmt"
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
func main() {
http.HandleFunc("/ws", handleWebSocket)
http.ListenAndServe(":8080", nil)
}
func handleWebSocket(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 message: %s
", message)
err = conn.WriteMessage(messageType, message)
if err != nil {
log.Println(err)
return
}
log.Printf("Sent message: %s
", message)
}
}在上面的代码中，我们首先创建了一个websocket.Upgrader对象，它负责将传入的HTTP连接升级为WebSocket连接。然后，我们创建了一个handleWebSocket函数，它会在客户端进行连接时被调用。该函数会先将传入的HTTP连接升级为WebSocket连接，然后进入循环，不断地接收来自客户端的信息，并将接收到的信息回传给客户端。其中，conn.ReadMessage()和conn.WriteMessage()分别用于读取和写入信息。
最后，我们将该handleWebSocket函数绑定到/ws路由上，并启动了HTTP服务器，以便客户端可以连接到该WebSocket服务器。如果一切正常，当我们启动上述代码后，您会看到类似如下输出：
2021/10/13 15:20:06 http: Server started at http://127.0.0.1:8080
这意味着您已经成功地创建了一个WebSocket服务器，客户端可以通过ws://localhost:8080/ws地址来连接该服务器。
创建WebSocket客户端
与创建WebSocket服务端类似，我们也可以很容易地在Go语言中创建一个WebSocket客户端。我们来看下面的代码，它描述了如何创建一个与WebSocket服务器建立连接的客户端：
package main
import (
"fmt"
"log"
"net/url"
"os"
"os/signal"
"syscall"
"github.com/gorilla/websocket"
)
func main() {
interrupt := make(chan os.Signal, 1)
signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)
u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/ws"}
log.Printf("Connecting to %s", u.String())
conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
if err != nil {
log.Fatal("dial:", err)
}
defer conn.Close()
done := make(chan struct{})
go func() {
defer close(done)
for {
messageType, message, err := conn.ReadMessage()
if err != nil {
log.Println("read:", err)
return
}
log.Printf("Received message: %s
", message)
if err := conn.WriteMessage(messageType, message); err != nil {
log.Println("write:", err)
return
}
log.Printf("Sent message: %s
", message)
}
}()
for {
select {
case <-done:
return
case <-interrupt:
log.Println("interrupt")
if err := conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")); err != nil {
log.Println("write close:", err)
return
}
select {
case <-done:
case <-time.After(time.Second):
}
return
}
}
}在上面的代码中，我们使用websocket.Dial方法来与WebSocket服务器建立连接。建立连接后，我们再进入一个无限循环，不断地发送和接收来自WebSocket服务器的信息。其中，conn.ReadMessage()和conn.WriteMessage()用于读取和写入信息。
值得注意的是，在上述代码中，我们使用了channels来等待操作完成。如果接收到interrupt信号，我们将使用websocket.CloseMessage消息发送给服务器，以关闭WebSocket连接。
结论
在本文中，我们学习了如何在Go语言中使用github.com/gorilla/websocket库来实现WebSocket。使用WebSocket技术可以为我们的Web应用程序增加更多的动态和交互，使得应用程序更加现代化和用户友好。
如果您对WebSocket技术以及如何在Go语言中使用它还有任何疑问或意见，欢迎在下面留言，与我们分享您的想法。
