golang的net.conn.read参数如何接收长度未知的数据?

func (c *IPConn) Read(b []byte) (int, error)

如果我要读取的字节流是未知长度,该怎么办?这里面这个b是不是只能在var声明变量的时候写死长度?

net包它有木有像别的编程语言里面的socket一样以类似于chunk的方式读取?(也就是说从socket缓冲区中每次读小部分数据,直到缓冲区没有数据为止,而不是一次性读整个数据,不然怕大文件会造成卡死)

阅读 15.2k
1 个回答

这种需求一般要先定义好包结构,不然就会有沾包问题出现了。基本数据类型(int|bool|byte)可以按照次序pack,对断按照次序读取就好,如果非基本类型,比如string或者binary,可以使用size+data的形式。read时,先读取size,获取之后的data数据长度,再去读取就好了。大体如下:

func (p *BinaryProtocol) ReadBinary() ([]byte, error) {
    size, e := p.ReadI32()
    if e != nil {
        return nil, e
    }
    if size < 0 {
        return nil, ErrProtocolLen
    }
    if uint64(size) > p.trans.RemainingBytes() {
        return nil, ErrProtocolLen
    }

    isize := int(size)
    buf := make([]byte, isize)
    _, err := io.ReadFull(p.trans, buf)
    return buf, NewBasicExceptionFromError(err)
}

或者很多开发者采用的特殊字符分割的方式,每次读取一个字符,遇到特定字符认为是包截止。

func (p *PacketSample) ReadFull(proto ProtocolIF) error {
    var code int8
    var e error
    var start bool = true

    for code != PacketFlagEnd {
        code, e = proto.ReadByte()
        if e != nil {
            return e
        }
        if start && code != PacketFlagStart {
            return errors.New("Using PacketSample, But not start with " + string(PacketFlagStart))
        }
        // store the byte
        p.WriteByte(code)
        start = false
    }
    return nil
}
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题