Golang WebSocket实时通信实现教程

哈喽！大家好，很高兴又见面了，我是golang学习网的一名作者，今天由我给大家带来一篇《Golang WebSocket实时通信实现教程》，本文主要会讲到等等知识点，希望大家一起学习进步，也欢迎大家关注、点赞、收藏、转发! 下面就一起来看看吧！

Golang利用gorilla/websocket库可高效构建WebSocket实时通信服务，通过HTTP服务器升级连接，使用Hub管理客户端注册、消息广播与连接维护。

Golang在实时通信领域，尤其是处理WebSocket连接方面，表现得相当出色，这很大程度上得益于其原生的并发模型。用它来构建一个高性能、可扩展的实时通信服务，效率高，代码也相对简洁。

构建一个Golang WebSocket实时通信服务，核心在于管理连接、处理消息以及确保消息的正确分发。我通常会采用gorilla/websocket这个库，它功能强大且稳定。

首先，我们需要一个HTTP服务器来升级连接。这是基础。

package main

import (
    "log"
    "net/http"
    "time"

    "github.com/gorilla/websocket"
)

// 定义一些常量，用于控制心跳和超时
const (
    writeWait = 10 * time.Second    // 允许写入消息的最大时间
    pongWait = 60 * time.Second     // 等待客户端pong回复的最大时间
    pingPeriod = (pongWait * 9) / 10 // 向客户端发送ping的周期，略小于pongWait
    maxMessageSize = 512             // 允许从客户端读取的最大消息大小
)

// upgrader 用于将HTTP连接升级为WebSocket连接
var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    CheckOrigin: func(r *http.Request) bool {
        // 在生产环境中，这里应该严格检查Origin头，防止跨站WebSocket劫持。
        // 例如：return r.Header.Get("Origin") == "http://your-trusted-domain.com"
        // 暂时为了示例方便，直接返回true
        return true 
    },
}

// Client 代表一个WebSocket连接的客户端
type Client struct {
    hub *Hub
    conn *websocket.Conn
    send chan []byte // 用于从hub发送消息到客户端的缓冲通道
}

// readPump 从WebSocket连接读取消息，并发送到hub
func (c *Client) readPump() {
    defer func() {
        c.hub.unregister <- c // 客户端断开时通知hub注销
        c.conn.Close()        // 关闭WebSocket连接
    }()
    c.conn.SetReadLimit(maxMessageSize) // 设置最大读取消息大小
    c.conn.SetReadDeadline(time.Now().Add(pongWait)) // 设置读取超时
    c.conn.SetPongHandler(func(string) error { // 收到pong时重置读取超时
        c.conn.SetReadDeadline(time.Now().Add(pongWait))
        return nil
    })
    for {
        _, message, err := c.conn.ReadMessage()
        if err != nil {
            if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
                log.Printf("error reading message: %v", err)
            }
            break // 发生错误或连接关闭，退出循环
        }
        // 收到消息后，可以广播给所有客户端，或者进行其他处理
        c.hub.broadcast <- message
    }
}

// writePump 从hub接收消息，并写入WebSocket连接
func (c *Client) writePump() {
    ticker := time.NewTicker(pingPeriod) // 定时发送ping
    defer func() {
        ticker.Stop()
        c.conn.Close()

好了，本文到此结束，带大家了解了《Golang WebSocket实时通信实现教程》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！