如何在Golang中开发简单的WebSocket通信

Golang中WebSocket开发的基础准备

WebSocket是一种全双工通信协议，允许客户端和服务端建立持久连接，实现双向实时数据传输，相比传统的HTTP轮询，能大幅减少不必要的请求开销，非常适合即时聊天、实时数据推送等场景。在Golang中开发WebSocket功能，通常不需要从零实现协议细节，使用成熟的第三方库可以快速完成开发，其中gorilla/websocket是最常用的选择。

首先需要通过包管理工具安装gorilla/websocket库，执行以下命令即可完成安装：

go get github.com/gorilla/websocket

搭建WebSocket服务端

配置升级器与基础参数

WebSocket连接需要先通过HTTP协议完成握手升级，gorilla/websocket提供了Upgrader结构体来处理升级逻辑，我们可以配置升级器的相关参数，比如允许跨域、设置读取和写入缓冲区大小等。

以下是一个基础的服务端配置示例：

package main

import (
	"log"
	"net/http"

	"github.com/gorilla/websocket"
)

// 定义升级器，配置允许所有来源跨域，实际生产环境可限制具体来源
var upgrader = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
	ReadBufferSize:  1024,
	WriteBufferSize: 1024,
}

实现连接处理与消息收发

接下来需要编写处理WebSocket连接的函数，该函数会完成HTTP到WebSocket的升级，然后循环读取客户端发送的消息，并做出响应。同时需要注意处理连接关闭的情况，避免程序异常。

// 处理WebSocket连接的函数
func handleWebSocket(w http.ResponseWriter, r *http.Request) {
	// 升级HTTP连接为WebSocket连接
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Printf("连接升级失败: %v", err)
		return
	}
	// 函数退出时关闭连接
	defer conn.Close()

	log.Println("新的WebSocket连接建立")

	// 循环读取客户端消息
	for {
		// 读取消息类型和消息内容
		messageType, message, err := conn.ReadMessage()
		if err != nil {
			log.Printf("读取消息失败: %v", err)
			break
		}
		log.Printf("收到客户端消息: %s", message)

		// 向客户端回复消息
		err = conn.WriteMessage(messageType, []byte("服务端已收到消息: "+string(message)))
		if err != nil {
			log.Printf("回复消息失败: %v", err)
			break
		}
	}
}

启动HTTP服务注册路由

最后需要启动HTTP服务，将WebSocket处理函数注册到对应的路由上，服务端就可以正常接收客户端的连接请求了。

func main() {
	// 注册WebSocket路由
	http.HandleFunc("/ws", handleWebSocket)

	// 启动HTTP服务，监听8080端口
	log.Println("WebSocket服务启动，监听端口8080")
	err := http.ListenAndServe(":8080", nil)
	if err != nil {
		log.Fatalf("服务启动失败: %v", err)
	}
}

编写简单的WebSocket客户端

我们可以使用Golang编写客户端来测试服务端的通信功能，客户端同样使用gorilla/websocket库来建立连接和收发消息。

package main

import (
	"log"
	"time"

	"github.com/gorilla/websocket"
)

func main() {
	// 建立WebSocket连接，连接到本地8080端口的/ws路由
	url := "ws://127.0.0.1:8080/ws"
	conn, _, err := websocket.DefaultDialer.Dial(url, nil)
	if err != nil {
		log.Fatalf("连接服务端失败: %v", err)
	}
	defer conn.Close()

	// 启动一个goroutine用于接收服务端的消息
	go func() {
		for {
			messageType, message, err := conn.ReadMessage()
			if err != nil {
				log.Printf("接收消息失败: %v", err)
				return
			}
			log.Printf("收到服务端消息: %s, 消息类型: %d", message, messageType)
		}
	}()

	// 每隔2秒向服务端发送一条消息，发送5次后退出
	for i := 0; i < 5; i++ {
		err := conn.WriteMessage(websocket.TextMessage, []byte("这是客户端发送的第"+string(rune('1'+i))+"条消息"))
		if err != nil {
			log.Printf("发送消息失败: %v", err)
			break
		}
		time.Sleep(2 * time.Second)
	}
}

并发场景下的注意事项

当多个客户端同时连接到服务端时，需要注意并发安全问题。gorilla/websocket的conn.ReadMessage和conn.WriteMessage都不是并发安全的，因此不要多个goroutine同时调用同一个连接的写方法，也不要同时调用读方法和写方法。

如果需要在多个goroutine中向同一个连接写消息，可以通过一个消息通道来统一处理写操作，示例如下：

// 定义连接结构体，包含连接实例和写消息通道
type Client struct {
	conn *websocket.Conn
	send chan []byte
}

// 启动连接的写协程，统一处理写消息操作
func (c *Client) writePump() {
	defer c.conn.Close()
	for message := range c.send {
		err := c.conn.WriteMessage(websocket.TextMessage, message)
		if err != nil {
			log.Printf("写消息失败: %v", err)
			return
		}
	}
}

常见问题与解决方法

• 连接升级失败：检查Upgrader的CheckOrigin配置，如果是生产环境需要设置允许的具体来源，而不是返回true。
• 消息读取失败：通常是连接已经被客户端关闭，需要在读取失败时退出循环，关闭连接。
• 跨域问题：如果客户端和服务端不在同一个域，需要正确配置Upgrader的CheckOrigin，或者服务端设置对应的跨域响应头。

通过以上步骤，我们就可以在Golang中完成简单的WebSocket通信开发，实现客户端和服务端的实时双向消息传输，后续可以根据实际需求扩展消息处理逻辑，比如实现多客户端消息广播、消息格式校验等功能。

GolangWebSocketgorilla_websocket并发通信实时消息修改时间：2026-07-14 20:09:30