Go语言中将io.Reader高效转换为字符串的策略与实践

package main

import (
    "fmt"
    "io"
    "strings"
)

// 模拟一个io.Reader
type MockReader struct {
    data string
    pos  int
}

func (m *MockReader) Read(p []byte) (n int, err error) {
    if m.pos >= len(m.data) {
        return 0, io.EOF
    }
    n = copy(p, m.data[m.pos:])
    m.pos += n
    return n, nil
}

func main() {
    // 假设我们有一个io.Reader，例如来自http.Response.Body
    // 这里使用MockReader模拟
    reader := &MockReader{data: "Hello, Go! This is a test string from an io.Reader."}

    // 使用strings.Builder进行转换
    var builder strings.Builder
    _, err := io.Copy(&builder, reader)
    if err != nil {
        fmt.Printf("Error copying to builder: %v\n", err)
        return
    }

    resultString := builder.String()
    fmt.Println("使用 strings.Builder 转换结果:")
    fmt.Println(resultString)
    fmt.Printf("字符串长度: %d\n", len(resultString))
}

package main

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

// 模拟一个io.Reader
type MockReader struct {
    data string
    pos  int
}

func (m *MockReader) Read(p []byte) (n int, err) {
    if m.pos >= len(m.data) {
        return 0, io.EOF
    }
    n = copy(p, m.data[m.pos:])
    m.pos += n
    return n, nil
}

func main() {
    reader := &MockReader{data: "Hello, Go! This is another test string from an io.Reader."}

    // 使用bytes.Buffer进行转换
    var buf bytes.Buffer
    _, err := buf.ReadFrom(reader)
    if err != nil {
        fmt.Printf("Error reading from reader to buffer: %v\n", err)
        return
    }

    resultString := buf.String() // 这里会发生一次数据拷贝
    fmt.Println("\n使用 bytes.Buffer 转换结果:")
    fmt.Println(resultString)
    fmt.Printf("字符串长度: %d\n", len(resultString))
}

// 仅为演示其原理，强烈不建议在生产环境中使用！
package main

import (
    "bytes"
    "fmt"
    "io"
    "unsafe"
)

// 模拟一个io.Reader
type MockReader struct {
    data string
    pos  int
}

func (m *MockReader) Read(p []byte) (n int, err) {
    if m.pos >= len(m.data) {
        return 0, io.EOF
    }
    n = copy(p, m.data[m.pos:])
    m.pos += n
    return n, nil
}

func main() {
    reader := &MockReader{data: "Hello, unsafe world! Be careful."}

    var buf bytes.Buffer
    _, err := buf.ReadFrom(reader)
    if err != nil {
        fmt.Printf("Error reading from reader to buffer: %v\n", err)
        return
    }

    // 获取bytes.Buffer内部的字节切片
    b := buf.Bytes()

    // 使用unsafe包将[]byte转换为string
    // ！！极其危险，强烈不推荐！！
    s := *(*string)(unsafe.Pointer(&b))

    fmt.Println("\n使用 unsafe 包转换结果 (强烈不推荐):")
    fmt.Println(s)
    fmt.Printf("字符串长度: %d\n", len(s))

    // 演示其危险性：如果底层[]byte发生改变，字符串也会改变
    // 这违反了Go字符串不可变性原则
    buf.WriteString(" Appended data.") // 修改了buf的底层数据
    fmt.Println("修改缓冲区后，字符串 s 的内容:")
    fmt.Println(s) // s 的内容也可能随之改变，或导致程序崩溃！
}