Question

是否有io.ReaderAt的实施可以从io.Reader的实施创建而无需先读入[]byte或string？

Answer 1

是的，这是可能的。正如我在上面的评论中所提到的，实现是有限的，因为你不能向后搜索，也不能重新阅读已经阅读过的部分。

以下是一个示例实现：

type unbufferedReaderAt struct {
    R io.Reader
    N int64
}

func NewUnbufferedReaderAt(r io.Reader) io.ReaderAt {
    return &unbufferedReaderAt{R: r}
}

func (u *unbufferedReaderAt) ReadAt(p []byte, off int64) (n int, err error) {
    if off < u.N {
        return 0, errors.New("invalid offset")
    }
    diff := off - u.N
    written, err := io.CopyN(ioutil.Discard, u.R, diff)
    u.N += written
    if err != nil {
        return 0, err
    }

    n, err = u.R.Read(p)
    u.N += int64(n)
    return
}

使用示例：

s := strings.NewReader("hello world")

var b [5]byte
ura := NewUnbufferedReaderAt(s)
if _, err := ura.ReadAt(b[:], 0); err != nil {
    panic(err)
}
fmt.Printf("%s\n", b[:]) // prints "hello"

/*
if _, err := ura.ReadAt(b[:], 0); err != nil {
    panic(err) // panics
}
fmt.Printf("%s\n", b[:])
*/

if _, err := ura.ReadAt(b[:], 6); err != nil {
    panic(err)
}
fmt.Printf("%s\n", b[:]) // prints "world"

Answer 2

类似以下内容。注意bytes.Reader实现了ReadAt(...)方法/功能：https://golang.org/pkg/bytes/#Reader.ReadAt。因此，bytes.NewReader行本质上就是您要寻找的内容。

获取bytes.Reader：

var ioReader io.Reader
...
buff := bytes.NewBuffer([]byte{})
size, err := io.Copy(buff, ioReader)
if err != nil {
    return err
}

reader := bytes.NewReader(buff.Bytes())
// Do something with `reader`

从io.Reader创建io.ReaderAt

2 个答案: