GolangWebSocket实时通信开发示例

gorilla/websocket

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