go中如何处理粘性tcp数据包？

Question

我有一个 tcp 服务器和一个客户端，服务器执行以下操作

func providerCallback(conn net.Conn) {
    reader := bufio.NewReader(conn)
    var err error
    for {
        lenbyte, _ := reader.Peek(4)
        reader.Discard(4)
        slen := int(binary.BigEndian.Uint32(lenbyte))
        data, err = reader.Peek(slen)
        process(data)
        reader.Discard(slen)
    }
}

客户端发送数据包的速度似乎比进程可以处理的快，因此我想在 bufio 中缓冲请求并稍后处理。

但是，由于bufio的大小是固定的（4096，虽然我可以增加，但还是固定的），这意味着我不能手动Reset它，因为最后可能会有一个数据包切割bufio的，如下

|普通数据... [数据包 P 的前 20 个字节] | [包P的其余部分]

|------------------- bufio 的大小 ------------------|

如何拼接被切断的数据包，并将 bufio 重新用于以后的数据包？

Answer 1

例如，

import (
    "bufio"
    "encoding/binary"
    "io"
    "net"
)

func providerCallback(conn net.Conn) error {
    rdr := bufio.NewReader(conn)
    data := make([]byte, 0, 4*1024)
    for {
        n, err := io.ReadFull(rdr, data[:4])
        data = data[:n]
        if err != nil {
            if err == io.EOF {
                break
            }
            return err
        }
        dataLen := binary.BigEndian.Uint32(data)
        if uint64(dataLen) > uint64(cap(data)) {
            data = make([]byte, 0, dataLen)
        }
        n, err = io.ReadFull(rdr, data[:dataLen])
        data = data[:n]
        if err != nil {
            return err
        }

        process(data)
    }
    return nil
}

func process([]byte) {}