Go语言中基于bufio.Reader实现健壮的逐行字符串读取

package main

import (
    "bufio"
    "fmt"
    "io" // 导入 io 包以使用 io.EOF
    "os"
)

// Readln 从 bufio.Reader 中读取一行数据（不包含行尾的换行符），
// 并将其作为字符串返回。
// 如果读取过程中发生错误，将返回错误信息。
// 该函数会处理 ReadLine 可能返回前缀的情况，确保返回完整的行。
func Readln(r *bufio.Reader) (string, error) {
    var (
        isPrefix bool = true // 标记是否为行的前缀
        err      error = nil // 错误信息
        line, ln []byte      // line 存储 ReadLine 返回的当前部分，ln 存储累积的完整行
    )
    for isPrefix && err == nil {
        line, isPrefix, err = r.ReadLine() // 读取一行的一部分
        ln = append(ln, line...)           // 将当前部分追加到完整行中
    }
    // 如果循环结束后，错误是 io.EOF，但我们已经读取到了一些数据，
    // 那么这不应该被视为错误，而是行的结束。
    // 只有当没有读取到任何数据时，io.EOF 才表示真正的文件结束。
    if err == io.EOF && len(ln) > 0 {
        return string(ln), nil // 有数据，返回数据，错误为nil
    }
    return string(ln), err // 返回累积的行数据和可能的错误
}

package main

import (
    "bufio"
    "fmt"
    "io"
    "os"
)

// Readln 函数定义如上...

func main() {
    filePath := "example.txt" // 假设存在一个名为 example.txt 的文件

    // 创建一个测试文件
    err := os.WriteFile(filePath, []byte("Hello, Go!\nThis is a long line that might exceed a buffer size to demonstrate the prefix handling in ReadLine.\nAnother line."), 0644)
    if err != nil {
        fmt.Printf("Error creating test file: %v\n", err)
        return
    }
    defer os.Remove(filePath) // 确保在程序结束时删除测试文件

    // 打开文件
    f, err := os.Open(filePath)
    if err != nil {
        fmt.Printf("Error opening file: %v\n", err)
        os.Exit(1)
    }
    defer f.Close() // 确保文件在使用完毕后关闭

    // 创建一个 bufio.Reader
    r := bufio.NewReader(f)

    // 逐行读取文件内容
    fmt.Println("File content:")
    for {
        s, e := Readln(r)
        if e != nil {
            if e == io.EOF {
                break // 到达文件末尾，退出循环
            }
            fmt.Printf("Error reading line: %v\n", e)
            break // 其他读取错误，退出循环
        }
        fmt.Println(s) // 打印读取到的行
    }
}