Golang初级项目中WebSocket实时通信实现-Golang-PHP中文网

Golang初级项目中WebSocket实时通信实现

P粉602998670

发布： 2025-09-09 09:38:01

原创

638人浏览过

使用gorilla/websocket库实现Go语言WebSocket通信，1. 安装依赖并初始化模块；2. 配置Upgrader处理HTTP升级；3. 编写handleWebSocket函数处理消息收发；4. 通过http.HandleFunc注册/ws路由；5. 前端用new WebSocket测试连接，实现简单回显交互。

golang初级项目中websocket实时通信实现

在Golang初级项目中实现WebSocket实时通信，核心是使用成熟的WebSocket库，配合标准的HTTP处理机制，建立客户端与服务端之间的双向通信通道。最常用的是gorilla/websocket库，它简单易用，文档清晰，非常适合初学者。

1. 安装WebSocket依赖库

使用Go Modules管理依赖，在项目根目录执行：

go mod init your-project-name<br>go get github.com/gorilla/websocket

登录后复制

2. 创建WebSocket升级配置

HTTP服务器不能直接处理WebSocket连接，需要将特定的HTTP请求“升级”为WebSocket协议。使用

websocket.Upgrader

登录后复制

完成这一过程。

示例代码：

立即学习“go语言免费学习笔记（深入）”；

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        // 允许所有跨域请求，开发阶段可用，生产环境应限制
        return true
    },
}

登录后复制

3. 编写WebSocket处理函数

定义一个处理函数，用于响应客户端的连接请求，并持续收发消息。

示例：

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

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()

    // 循环读取消息
    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Printf("读取消息失败: %v", err)
            break
        }

        // 打印收到的消息
        log.Printf("收到: %s", message)

        // 回显消息给客户端
        if err := conn.WriteMessage(messageType, message); err != nil {
            log.Printf("发送消息失败: %v", err)
            break
        }
    }
}

登录后复制

4. 启动HTTP服务器并注册路由

将处理函数注册到指定路由，并启动服务器。

完整main函数示例：

func main() {
    http.HandleFunc("/ws", handleWebSocket)

    log.Println("服务器启动，地址: :8080")
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("启动失败: ", err)
    }
}

登录后复制

5. 前端测试WebSocket连接

使用浏览器控制台或简单HTML页面测试连接是否成功。

前端JavaScript示例：

const ws = new WebSocket("ws://localhost:8080/ws");

ws.onopen = () => {
    console.log("连接已建立");
    ws.send("你好，Go服务器！");
};

ws.onmessage = (event) => {
    console.log("收到消息:", event.data);
};

ws.onclose = () => {
    console.log("连接已关闭");
};

登录后复制

运行Go服务后，打开浏览器开发者工具，粘贴上述代码，即可看到双向通信效果。

基本上就这些。对于初级项目，这种单连接回显模式足够理解WebSocket的基本流程：升级连接、读写消息、异常处理。后续可扩展为广播消息、用户管理、心跳机制等。

以上就是Golang初级项目中WebSocket实时通信实现的详细内容，更多请关注php中文网其它相关文章！