Go语言中从io.Reader高效读取UTF-8编码字符串的方法

package main

import (
    "bytes"
    "fmt"
    "io"
    "os"
)

func main() {
    // 1. 打开一个文件作为io.Reader的示例
    // 实际应用中，这里可以是 net.Conn, http.Response.Body, 或其他任何io.Reader
    filePath := "example.txt"
    // 为了演示，先创建一个示例文件
    err := os.WriteFile(filePath, []byte("你好，世界！This is a UTF-8 string example."), 0644)
    if err != nil {
        fmt.Printf("创建示例文件失败: %v\n", err)
        return
    }
    defer os.Remove(filePath) // 确保示例文件在程序结束时被清理

    file, err := os.Open(filePath)
    if err != nil {
        fmt.Printf("打开文件失败: %v\n", err)
        return
    }
    defer file.Close() // 确保文件句柄被关闭

    // 2. 创建一个bytes.Buffer实例
    // bytes.Buffer的零值是一个可以直接使用的缓冲区
    var buf bytes.Buffer

    // 3. 将io.Reader的内容拷贝到bytes.Buffer中
    // io.Copy是一个非常通用的函数，用于将数据从io.Reader复制到io.Writer
    // buf实现了io.Writer接口，因此可以直接作为io.Copy的目标
    n, err := io.Copy(&buf, file) // 注意：io.Copy的第一个参数是io.Writer，第二个是io.Reader
    if err != nil {
        fmt.Printf("从文件读取数据失败: %v\n", err)
        return
    }
    fmt.Printf("成功从文件读取了 %d 字节。\n", n)

    // 4. 使用bytes.Buffer的String()方法获取字符串
    // String()方法将缓冲区内的所有字节转换为字符串
    contentString := buf.String()
    fmt.Printf("文件内容（字符串形式）:\n%s\n", contentString)

    // 替代方法：使用ReadFrom()
    // bytes.Buffer也提供了一个ReadFrom方法，功能与io.Copy类似，但更直接
    fmt.Println("\n--- 使用 ReadFrom 方法 ---")
    file2, err := os.Open(filePath) // 重新打开文件，因为file已经被读取到末尾
    if err != nil {
        fmt.Printf("重新打开文件失败: %v\n", err)
        return
    }
    defer file2.Close()

    var buf2 bytes.Buffer
    bytesRead, err := buf2.ReadFrom(file2) // ReadFrom的参数是io.Reader
    if err != nil {
        fmt.Printf("使用ReadFrom读取失败: %v\n", err)
        return
    }
    fmt.Printf("成功使用ReadFrom读取了 %d 字节。\n", bytesRead)
    fmt.Printf("文件内容（ReadFrom结果）:\n%s\n", buf2.String())
}