在Go语言中从Reader读取数据直到特定字符串分隔符

package main

import (
    "bytes"
    "fmt"
    "io" // 导入 io 包以使用 io.Reader 接口
    "log"
)

// reader 接口定义了 ReadString 方法，用于抽象底层的读取器
// 这里使用 io.Reader 接口更为通用，但为了与 ReadString(delim byte) 行为保持一致，
// 我们可以使用 bufio.Reader 或自定义一个包含 ReadString 的接口。
// 为了简化示例，我们假设传入的 r 能够提供 ReadString(byte) 的能力，
// 例如 bufio.Reader 或 bytes.Buffer 包装后的 reader。
// 实际应用中，如果需要更通用，可以考虑逐字节读取或使用 bufio.Scanner。
type reader interface {
    ReadString(delim byte) (line string, err error)
}

// read 函数从 r 中读取数据，直到遇到完整的 delim 字符串
func read(r reader, delim []byte) (line []byte, err error) {
    // 检查分隔符是否为空，空分隔符会导致无限循环或不明确的行为
    if len(delim) == 0 {
        return nil, fmt.Errorf("分隔符不能为空")
    }

    for {
        // 1. 使用 ReadString 优化读取：
        // 每次读取都尝试直到分隔符的最后一个字节。
        // 这样做可以避免逐字节读取的低效，并利用 ReadString 内部的优化。
        s, err := r.ReadString(delim[len(delim)-1])
        if err != nil {
            // 如果遇到 EOF 且没有读取到任何数据，则返回 EOF
            // 如果在读取过程中遇到其他错误，则直接返回
            if err == io.EOF && len(line) == 0 && len(s) == 0 {
                return nil, io.EOF
            }
            // 如果在 EOF 之前已经读取了部分数据，则将这部分数据返回，并返回 EOF
            if err == io.EOF {
                line = append(line, []byte(s)...)
                // 检查最终是否以分隔符结尾
                if bytes.HasSuffix(line, delim) {
                    return line[:len(line)-len(delim)], nil
                }
                // 如果没有以分隔符结尾，但已经到文件末尾，则返回所有剩余数据和 EOF
                return line, io.EOF
            }
            return nil, err
        }

        // 2. 将读取到的字符串追加到累积的字节切片中
        line = append(line, []byte(s)...)

        // 3. 检查当前累积的数据是否以完整的分隔符结尾
        if bytes.HasSuffix(line, delim) {
            // 如果匹配成功，返回分隔符之前的数据
            return line[:len(line)-len(delim)], nil
        }
    }
}

func main() {
    // 示例用法：从一个 bytes.Buffer 中读取数据
    // 注意：bytes.Buffer 实现了 ReadString 方法，因此可以直接作为 read 函数的参数。
    // 如果使用 bufio.Reader，则需要 `bufio.NewReader(bytes.NewBufferString(...))`
    src := bytes.NewBufferString("123deli456elim789delimABCdelimDEF")

    // 定义要查找的分隔符
    delimiter := []byte("delim")

    fmt.Printf("开始从数据源读取，分隔符：%q\n", delimiter)

    for {
        // 调用自定义的 read 函数
        b, err := read(src, delimiter)
        if err != nil {
            // 遇到 io.EOF 时退出循环
            if err == io.EOF {
                fmt.Println("读取结束 (EOF)")
                // 如果 EOF 前还有数据，打印出来
                if len(b) > 0 {
                    fmt.Printf("剩余数据：%q\n", b)
                }
                break
            }
            // 处理其他错误
            log.Fatalf("读取过程中发生错误: %v", err)
        }

        // 打印读取到的内容（分隔符之前的部分）
        fmt.Printf("读取到：%q\n", b)
    }
}