golang 处理TCP粘包问题

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
func Encode(data []byte) []byte {
    buf := bytes.NewBuffer(nil)
    binary.Write(buf, binary.BigEndian, uint32(len(data)))
    binary.Write(buf, binary.BigEndian, data)
    return buf.Bytes()
}
 
func Decode(data []byte) ([]byte, error) {
    reader := bufio.NewReader(bytes.NewBuffer(data))
    peek, err := reader.Peek(4)
    if err != nil {
        return nil, err
    }
    buffer := bytes.NewBuffer(peek)
    var length uint32
    err = binary.Read(buffer, binary.LittleEndian, &length)
    if err != nil {
        return nil, err
    }
    if uint32(reader.Buffered()) < length+4 {
        return nil, errors.New("长度不足")
    }
    data = make([]byte, length+4)
    _, err = reader.Read(data)
    if err != nil {
        return nil, err
    }
    return data[4:], nil
}

  

posted @   闲云-野鹤  阅读(270)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示