一、前言

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。在 Web 应用中，常用于实现实时通信，比如在线聊天、实时数据推送等。在本文中，我将介绍如何使用 Gorilla WebSocket 库来实现 Golang 中的 WebSocket 通信。

二、环境配置

在开始之前，我们需要准备好 Golang 的环境及相关库。您可以从官方网站下载最新版的 Golang，并按照官方文档进行安装。安装完毕后，我们需要安装 Gorilla WebSocket 库。可以通过以下命令进行安装：

```
go get github.com/gorilla/websocket
```

三、实现

在开始编写代码之前，我们需要了解 WebSocket 的相关概念和原理。WebSocket 是一种基于帧的协议，包含了三种帧类型：文本、二进制和关闭。在 Gorilla WebSocket 库中，我们可以使用 Conn 类型来处理 WebSocket 连接。Conn 类型包含了读写数据的方法，可以方便地进行数据传输。

接下来，我们将编写一个简单的 WebSocket 服务器，用于处理客户端的连接和消息。我们使用 Gorilla WebSocket 库中的 Upgrader 类来升级 HTTP 连接为 WebSocket 连接，然后创建一个 goroutine 来处理连接和消息。具体实现如下：

```go
package main

import (
	"fmt"
	"log"
	"net/http"

	"github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{}

func main() {
	http.HandleFunc("/ws", handleWebSocket)
	log.Fatal(http.ListenAndServe(":8080", nil))
}

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Println("upgrade error:", err)
		return
	}

	go handleConnection(conn)
}

func handleConnection(conn *websocket.Conn) {
	defer conn.Close()

	for {
		messageType, message, err := conn.ReadMessage()
		if err != nil {
			log.Println("read error:", err)
			break
		}

		log.Printf("received message: %s", message)

		err = conn.WriteMessage(messageType, message)
		if err != nil {
			log.Println("write error:", err)
			break
		}
	}
}
```

代码说明：

- 首先，我们创建了一个名为 upgrader 的全局变量，用于将 HTTP 连接升级为 WebSocket 连接。
- 接着，我们创建了一个 handleWebSocket 函数，用于处理 WebSocket 连接请求。
- 在 handleWebSocket 函数中，我们使用 upgrader 对象将 HTTP 请求升级为 WebSocket 连接，并调用 handleConnection 函数来处理连接和消息。
- handleConnection 函数使用一个 for 循环来不断地读取客户端发送的消息，并写回相同的消息到客户端中。

现在，我们已经完成了 WebSocket 服务器的编写。接下来，我们可以使用 WebSocket 客户端来连接服务器，并测试通信是否正常。

四、测试

为了测试我们的 WebSocket 服务器，我们可以使用浏览器的 JavaScript WebSocket API 来创建一个 WebSocket 连接，并发送和接收数据。以下是一个简单的测试页面示例：

```html



    


    
    Send
    

    


```

代码说明：

- 首先，我们创建了一个 WebSocket 对象，用于连接服务器。在这里，我们将服务器地址设置为 localhost:8080/ws。
- 接着，我们定义了三个事件处理函数，分别用于处理连接成功、接收消息和连接断开事件。
- 在 send 按钮的点击事件中，我们获取用户输入的消息，并使用 WebSocket 对象的 send 方法将其发送到服务器中。
- 最后，在页面中，我们使用一个文本框和一个按钮来输入和发送消息，并在页面底部使用一个 div 元素来显示服务器返回的消息。

运行该页面后，我们可以通过输入文本框来发送消息，并在页面底部查看服务器返回的消息。如果一切正常，我们应该能够看到如下的结果：

```
Connected to server
Received message: Hello, world!
```

至此，我们已经成功地使用 Gorilla WebSocket 库实现了 Golang 中的 WebSocket 通信。你可以以此为基础，进一步扩展你的 WebSocket 应用程序。