解決Go語言Websocket應用程式中的資料遺失問題

func readPump(conn *websocket.Conn, ch chan []byte) {
    for {
        _, message, err := conn.ReadMessage()
        if err != nil {
            break
        }
        ch <- message
    }
    close(ch)
}

func writePump(conn *websocket.Conn, ch chan []byte) {
    for message := range ch {
        err := conn.WriteMessage(websocket.TextMessage, message)
        if err != nil {
            break
        }
    }
    conn.Close()
}

func main() {
    ch := make(chan []byte, 10)

    // 启动读取协程
    go readPump(conn, ch)

    // 启动写入协程
    go writePump(conn, ch)
}

func writeMessage(conn *websocket.Conn, message []byte) error {
    messageSize := len(message)
    if messageSize > maxMessageSize {
        return errors.New("Message too large")
    }

    // 计算分片数量
    fragmentSize := (messageSize / maxFragmentSize) + 1

    for i := 0; i < fragmentSize; i++ {
        start := i * maxFragmentSize
        end := start + maxFragmentSize

        // 切割数据片段
        if end > messageSize {
            end = messageSize
        }
        fragment := message[start:end]

        // 写入分片
        err := conn.WriteMessage(websocket.TextMessage, fragment)
        if err != nil {
            return err
        }
    }

    return nil
}

func main() {
    message := []byte("Hello, world!")
    err := writeMessage(conn, message)
    if err != nil {
        log.Println(err)
    }
}

type Message struct {
    ID    int    `json:"id"`
    Data  []byte `json:"data"`
    Checksum uint16 `json:"checksum"`
}

func writeMessage(conn *websocket.Conn, message Message) error {
    // 计算校验和
    checksum := calculateChecksum(message.Data)

    // 添加校验和信息
    message.Checksum = checksum

    // 序列化消息
    body, err := json.Marshal(message)
    if err != nil {
        return err
    }

    // 发送消息
    err = conn.WriteMessage(websocket.TextMessage, body)
    if err != nil {
        return err
    }

    return nil
}

func readMessage(conn *websocket.Conn) (Message, error) {
    var message Message

    // 接收消息
    _, body, err := conn.ReadMessage()
    if err != nil {
        return message, err
    }

    // 反序列化消息
    err = json.Unmarshal(body, &message)
    if err != nil {
        return message, err
    }

    // 校验消息
    if message.Checksum != calculateChecksum(message.Data) {
        return message, errors.New("Checksum error")
    }

    return message, nil
}

func calculateChecksum(data []byte) uint16 {
    checksum := uint16(0)
    for i := 0; i < len(data); i++ {
        checksum += uint16(data[i])
    }
    return checksum
}

func main() {
    message := Message{
        ID:   1,
        Data: []byte("Hello, world!"),
    }
    err := writeMessage(conn, message)
    if err != nil {
        log.Println(err)
    }

    rcvMessage, err := readMessage(conn)
    if err != nil {
        log.Println(err)
    } else {
        log.Println(rcvMessage)
    }
}