向io.Reader添加前缀

package main

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

type PrefixReader struct {
    prefix string
    reader io.Reader
}

func (r *PrefixReader) Read(p []byte) (n int, err error) {
    // 先读取前缀
    if len(r.prefix) > 0 {
        n = copy(p, []byte(r.prefix))
        r.prefix = ""
        return n, nil
    }

    // 读取底层的数据流
    return r.reader.Read(p)
}

func main() {
    // 创建一个字符串作为数据流
    data := "Hello, World!"

    // 创建一个带有前缀的Reader
    reader := &PrefixReader{
        prefix: "Prefix: ",
        reader: strings.NewReader(data),
    }

    // 读取数据流并输出
    buf := make([]byte, len(data)+len(reader.prefix))
    n, err := reader.Read(buf)
    if err != nil && err != io.EOF {
        fmt.Println("Error:", err)
        return
    }
    fmt.Println(string(buf[:n]))
}