如何使用 bufio.Scanner 在 Golang 中读取数据直到 CRLF 分隔符？

如何在 Golang 中使用 bufio 读取多行数据直到 CRLF 分隔符

Golang 的 bufio 包中，ReadBytes、ReadSlice 和 ReadString 函数是通常用于读取由特定字节或字符串分隔的数据。然而，这些函数都没有提供一种简单的方法来读取数据，直到按照 Beanstalkd 协议的要求使用 CRLF 分隔符。

使用 bufio.Scanner

更好的方法是使用 bufio.Scanner 类型创建查找 CRLF 分隔符的自定义拆分函数。这允许您读取数据，直到遇到 CRLF 分隔符。操作方法如下：

<code class="go">import (
    "bufio"
    "bytes"
)

func ScanCRLF(data []byte, atEOF bool) (advance int, token []byte, err error) {
    if atEOF && len(data) == 0 {
        return 0, nil, nil
    }
    if i := bytes.Index(data, []byte{'\r', '\n'}); i >= 0 {
        // We have a full newline-terminated line.
        return i + 2, dropCR(data[0:i]), nil
    }
    // If we're at EOF, we have a final, non-terminated line. Return it.
    if atEOF {
        return len(data), dropCR(data), nil
    }
    // Request more data.
    return 0, nil, nil
}

func dropCR(data []byte) []byte {
    if len(data) > 0 && data[len(data)-1] == '\r' {
        return data[0 : len(data)-1]
    }
    return data
}</code>

使用自定义扫描仪

现在，您可以使用自定义扫描仪包装 io.Reader 并使用它来读取数据直到 CRLF 分隔符：

<code class="go">reader := bufio.NewReader(conn)
scanner := bufio.NewScanner(reader)
scanner.Split(ScanCRLF)

for scanner.Scan() {
    fmt.Println(scanner.Text())
}

if err := scanner.Err(); err != nil {
    fmt.Printf("Invalid input: %s", err)
}</code>

替代方法：读取特定数量的字节

如果您知道预期的字节数，也可以读取特定数量的字节数据中的字节。然而，这种方法不太稳健，需要仔细验证以确保您不会读取太多或太少的数据。要使用 bufio.Reader 类型读取特定数量的字节：

<code class="go">nr_of_bytes := read_number_of_butes_somehow(conn)
buf := make([]byte, nr_of_bytes)
reader.Read(buf)</code>

处理错误条件

上述两种方法都需要仔细处理错误条件，尤其是当处理部分读取和过早的 EOF 时。确保验证输入并适当处理错误。

以上是如何使用 bufio.Scanner 在 Golang 中读取数据直到 CRLF 分隔符？的详细内容。更多信息请关注PHP中文网其他相关文章！