Question

我正在尝试从Go中的tcp连接读取。我有一个特定的标头格式，我想先解析。因此，请读取具有与标头关联的信息的前12个字节。这就告诉我剩余的消息长度是一定数量的字节。在这种情况下，为1607个字节。我在读取标头字节的地方尝试了以下代码，然后尝试读取其余字节。

import ("net"
        log "github.com/sirupsen/logrus"
       "bytes"
       "encoding/binary"
)

func handleRequest(conn net.Conn) {

    // Structs to make header parsing easiers
    type encapSTHdrMsgType uint16
    type encapSTHdrMsgEncap uint16
    type encapSTHdr struct {
        MsgType       encapSTHdrMsgType
        MsgEncap      encapSTHdrMsgEncap
        MsgHdrVersion uint16
        Msgflag       uint16
        Msglen        uint32
    }

    // Make a buffer to hold header data.
    headerBuf := make([]byte, 12)

    // Read the incoming header info into the buffer.
    _, err := conn.Read(headerBuf)
    if err != nil {
        log.Debug("Error reading:", err.Error())
    }

    // Header is in big endian
    var header encapSTHdr
    headerReader := bytes.NewReader(headerBuf)
    err = binary.Read(headerReader, binary.BigEndian, &header)
    if err != nil {
        log.Debugf("Could not read header bytes into the buffer: %v", err)
    }

    messageBuf := make([]byte, header.Msglen)
    messageBufLen, err := conn.Read(messageBuf)
    if err != nil {
        log.Debugf("Error reading messages: %s", err.Error())
    }
    log.Debugf("The message buffer length is: %d", messageBufLen)
    log.Debugf("The header message length is: %d", header.Msglen) 
}

当我尝试通过创建所需长度为1607的新缓冲区来读取其余字节时，我得到了一个包含数据的新字节数组，但长度仅为1228。因此，我有两个问题：

1）为什么conn.Read（）无法读取剩余字节？

2）现在，我知道可以使用for循环，直到读取所需的字节数并在完成后中断为止，但是我想知道是否有更好的方法可以做到这一点？

Answer 1

Read在有数据可用时返回，而不是在填充缓冲区时返回。使用io.ReadFull()进行读取，直到缓冲区已满或连接读取错误。

_, err := io.ReadFull(conn, headerBuf)

...

_, err := io.ReadFull(conn, messageBuf)

在Go中使用conn.Read（）读取前几个字节后，如何继续读取其余字节？

1 个答案: