从bufio.Reader读取至特定字符串序列的Go语言实现

package main

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

// reader 接口定义，用于兼容 bufio.Reader 或 bytes.Buffer 等
type reader interface {
    ReadString(delim byte) (line string, err error)
}

// read 函数从读取器中读取数据，直到遇到指定的字节切片分隔符
// 返回分隔符之前的数据。
func read(r reader, delim []byte) (line []byte, err error) {
    if len(delim) == 0 {
        return nil, fmt.Errorf("delimiter cannot be empty")
    }

    var buffer bytes.Buffer // 使用 bytes.Buffer 来累积读取到的数据

    for {
        // 1. 读取直到分隔符的最后一个字节
        // 这样做是为了尽可能利用 ReadString 的高效性
        s, err := r.ReadString(delim[len(delim)-1])
        if err != nil {
            // 如果遇到 EOF，检查当前 buffer 中是否包含分隔符
            // 如果有，则返回分隔符之前的数据；否则返回 EOF 错误
            if err == io.EOF {
                buffer.WriteString(s) // 将最后一部分数据也写入 buffer
                if bytes.HasSuffix(buffer.Bytes(), delim) {
                    return buffer.Bytes()[:buffer.Len()-len(delim)], nil
                }
            }
            return nil, err // 返回其他错误或未找到分隔符的 EOF
        }

        // 2. 将读取到的字符串追加到缓冲区
        buffer.WriteString(s)

        // 3. 检查缓冲区末尾是否包含完整的字符串分隔符
        if bytes.HasSuffix(buffer.Bytes(), delim) {
            // 如果找到，则返回分隔符之前的数据
            return buffer.Bytes()[:buffer.Len()-len(delim)], nil
        }
    }
}

func main() {
    // 示例数据源
    src := bytes.NewBufferString("Hello World!delimThis is a test.delimAnother part.delimEND")
    delimiter := []byte("delim")

    fmt.Printf("使用分隔符 %q 读取数据:\n", delimiter)

    for i := 1; ; i++ {
        b, err := read(src, delimiter)
        if err != nil {
            if err == io.EOF {
                fmt.Printf("读取完成，遇到文件末尾 (EOF)。\n")
                break
            }
            log.Fatalf("读取错误: %v", err) // 遇到其他错误则终止程序
        }

        fmt.Printf("第 %d 段数据: %q\n", i, b)
    }

    // 进一步测试，例如分隔符在数据末尾，或者数据中不含分隔符
    fmt.Println("\n--- 额外测试 ---")
    src2 := bytes.NewBufferString("Data without delimiter at the end")
    b, err := read(src2, []byte("STOP"))
    if err != nil {
        if err == io.EOF {
            fmt.Printf("额外测试：读取到 EOF，未找到分隔符。已读取数据: %q\n", b)
        } else {
            log.Fatalf("额外测试错误: %v", err)
        }
    } else {
        fmt.Printf("额外测试：成功读取到分隔符，数据: %q\n", b)
    }
}