Go 言語 Websocket 開発ガイド: メッセージ損失の問題に対処する方法

import (
    "net/http"
    "github.com/gorilla/websocket"
)

func handleWebsocket(w http.ResponseWriter, r *http.Request) {
    // 允许跨域连接
    upgrader := websocket.Upgrader{
        CheckOrigin: func(r *http.Request) bool { return true },
    }

    // 升级HTTP连接为Websocket连接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Websocket upgrade failed: ", err)
        return
    }

    // 处理消息
    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println("Read error: ", err)
            break
        }

        // 处理消息逻辑
        handleMessage(message)

        // 回复消息
        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println("Write error: ", err)
            break
        }
    }

    // 关闭连接
    conn.Close()
}

http.HandleFunc("/websocket", handleWebsocket)
http.ListenAndServe(":8000", nil)

type Message struct {
    ID       int
    Content  string
    AckChan  chan int
}

type MessageHandler struct {
    messages map[int]Message
}

func (handler *MessageHandler) handleMessage(message Message) {
    // 处理消息逻辑
    // ...

    // 发送确认消息
    message.AckChan <- message.ID
}

messageHandler := MessageHandler{
    messages: make(map[int]Message),
}

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

    // 创建消息对象
    ackChan := make(chan int)
    msg := Message{
        ID:       len(messageHandler.messages) + 1,
        Content:  string(message),
        AckChan:  ackChan,
    }

    // 处理消息
    messageHandler.handleMessage(msg)

    // 等待确认消息
    select {
    case <-ackChan:
        // 收到确认消息
    case <-time.After(time.Second):
        // 消息发送超时，重新发送消息
        conn.WriteMessage(messageType, message)
    }

    // 回复消息
    err = conn.WriteMessage(messageType, message)
    if err != nil {
        log.Println("Write error: ", err)
        break
    }
}

type Heartbeat struct {
    PingMsg    []byte
    PongMsg    []byte
    Interval   time.Duration
}

func (h *Heartbeat) Start(conn *websocket.Conn) {
    ticker := time.NewTicker(h.Interval)
    defer ticker.Stop()

    for range ticker.C {
        // 发送心跳消息
        err := conn.WriteMessage(websocket.PingMessage, h.PingMsg)
        if err != nil {
            log.Println("Heartbeat error: ", err)
            break
        }

        // 设置心跳超时时间
        conn.SetReadDeadline(time.Now().Add(h.Interval))

        // 等待心跳回复
        _, _, err = conn.ReadMessage()
        if err != nil {
            log.Println("Heartbeat error: ", err)
            break
        }
    }
}

heartbeat := Heartbeat{
    PingMsg:    []byte("ping"),
    PongMsg:    []byte("pong"),
    Interval:   time.Second * 10,
}

go heartbeat.Start(conn)