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

package main

import (
    "bytes"
    "fmt"
    "io"
    "log"
    "strings"
)

func main() {
    // 1. 模拟一个io.Reader
    // 在实际应用中，这可能是一个 *os.File、*net.Conn 或其他实现了 io.Reader 接口的对象。
    // 这里我们使用 strings.NewReader 来模拟一个包含UTF-8字符串的源。
    sourceReader := strings.NewReader("你好，Go语言！这是一个包含UTF-8字符的字符串。\nHello, Go!")

    // 2. 创建一个 bytes.Buffer 实例
    // bytes.Buffer 的零值是可用的，无需调用构造函数。
    var buffer bytes.Buffer

    // 3. 使用 io.Copy 将 io.Reader 的内容复制到 bytes.Buffer 中
    // io.Copy 函数的第一个参数是 io.Writer，第二个参数是 io.Reader。
    // bytes.Buffer 实现了 io.Writer 接口（通过其指针方法），因此我们需要传递 &buffer。
    // 另一个等效的方法是 buffer.ReadFrom(sourceReader)。
    n, err := io.Copy(&buffer, sourceReader)
    if err != nil {
        log.Fatalf("Error copying data: %v", err)
    }
    fmt.Printf("成功从源读取并复制了 %d 字节到缓冲区。\n", n)

    // 4. 调用 bytes.Buffer 的 String() 方法获取 UTF-8 编码的字符串
    // String() 方法会将缓冲区中的所有字节内容解释为 UTF-8 字符串并返回。
    readString := buffer.String()

    fmt.Printf("从缓冲区读取到的字符串：\"%s\"\n", readString)
    fmt.Printf("字符串的字节长度：%d\n", len(readString)) // len() 返回字节长度
    // 对于包含多字节UTF-8字符的字符串，可以通过转换为 []rune 来获取字符数
    fmt.Printf("字符串的字符数：%d\n", len([]rune(readString)))

    // 进一步操作：bytes.Buffer 也可以作为 io.Reader
    // 例如，我们可以再次从 buffer 中读取数据
    fmt.Println("\n--- 再次从缓冲区读取数据 ---")
    anotherBuffer := new(bytes.Buffer) // 另一种创建 bytes.Buffer 的方式
    _, err = io.Copy(anotherBuffer, &buffer) // 此时 buffer 作为 io.Reader
    if err != nil && err != io.EOF { // io.Copy 在源耗尽时可能返回 io.EOF，通常不需要作为错误处理
        log.Fatalf("Error reading from buffer: %v", err)
    }
    fmt.Printf("再次从缓冲区读取到的字符串：\"%s\"\n", anotherBuffer.String())
    // 注意：一旦从 buffer 中读取数据，这些数据就会从 buffer 中被“消费”掉。
    // 第一次 io.Copy(&buffer, sourceReader) 将数据写入 buffer。
    // 第二次 io.Copy(anotherBuffer, &buffer) 将数据从 buffer 读取到 anotherBuffer，
    // 此时原始的 buffer 内部数据已被清空。
    fmt.Printf("原始缓冲区现在的内容：%q\n", buffer.String()) // 此时原始 buffer 应该为空
}