在golang中将[]byte变成“虚拟”文件对象的简单方法？答案

【问题标题】：Simples way to make a []byte into a "virtual" File object in golang?在golang中将[]byte变成“虚拟”文件对象的简单方法？
【发布时间】：2020-04-20 19:29:34
【问题描述】：

我知道有一些 Go 库可以创建像 VFS 这样的整个文件系统。但我只想把字节数组做成能满足File接口的东西。

【问题讨论】：

你真的要模拟整个文件吗？我不确定一段字节如何在逻辑上实现Readdir（[]os.FileInfo{}, someErr 除外）

标签： go virtual-file

【解决方案1】：

标准库中没有现成的解决方案，但是自己做也没有那么难。

我们需要的是这个http.File接口：

type File interface {
        io.Closer
        io.Reader
        io.Seeker
        Readdir(count int) ([]os.FileInfo, error)
        Stat() (os.FileInfo, error)
}

请注意，我们可以利用bytes.Reader 来完成繁重的任务，因为它单独实现了io.Reader 和io.Seeker。 io.Closer 可以是 noop，Readdir() 可能会返回 nil, nil，因为我们正在模拟文件而不是目录，它的 Readdir() 甚至不会被调用。

“最难”的部分是模拟Stat() 以返回实现os.FileInfo 的值。

这是一个简单的模拟FileInfo：

type myFileInfo struct {
    name string
    data []byte
}

func (mif myFileInfo) Name() string       { return mif.name }
func (mif myFileInfo) Size() int64        { return int64(len(mif.data)) }
func (mif myFileInfo) Mode() os.FileMode  { return 0444 }        // Read for all
func (mif myFileInfo) ModTime() time.Time { return time.Time{} } // Return anything
func (mif myFileInfo) IsDir() bool        { return false }
func (mif myFileInfo) Sys() interface{}   { return nil }

有了它，我们就可以创建我们的模拟 http.File：

type MyFile struct {
    *bytes.Reader
    mif myFileInfo
}

func (mf *MyFile) Close() error { return nil } // Noop, nothing to do

func (mf *MyFile) Readdir(count int) ([]os.FileInfo, error) {
    return nil, nil // We are not a directory but a single file
}

func (mf *MyFile) Stat() (os.FileInfo, error) {
    return mf.mif, nil
}

使用示例（在Go Playground 上试用）：

data := []byte{0, 1, 2, 3}

mf := &MyFile{
    Reader: bytes.NewReader(data),
    mif: myFileInfo{
        name: "somename.txt",
        data: data,
    },
}

var f http.File = mf
_ = f

【讨论】：

【解决方案2】：

看起来很简单，可以自己模拟一下。

type MockFile struct {
    data    []byte
    isOpen  bool
    offset  int64
}

type MockFileInfo struct {
    mockFile *MockFile
}
func (mfi *MockFileInfo) Name() string       { return "MockFile" }
func (mfi *MockFileInfo) Size() int64        { return len(mfi.data) }
func (mfi *MockFileInfo) Mode() os.FileMode  { return os.ModeIrregular }
func (mfi *MockFileInfo) ModTime() time.Time { return time.Now() }
func (mfi *MockFileInfo) IsDir() bool        { return false }
func (mfi *MockFileInfo) Sys() interface     { return nil }

func (mf *MockFile) Read(p []byte) (n int, err error) {
    if mf.isOpen {
        n = copy(p, mf.data[mf.offset:])
        mf.offset += n
    } else {
        err = errors.New("Cannot read from closed MockFile")
    }
    return
}

func (mf *MockFile) Close() error {
    if !mf.isOpen {
        return errors.New("Cannot close an already closed MockFile")
    mf.isOpen = false
    return nil
}

func (mf *MockFile) Seek(offset int64, whence int) (ret int64, err error) {
    var relativeTo int64
    switch whence {
        case 0:
            relativeTo = 0
        case 1:
            relativeTo = mf.offset
        case 2:
            relativeTo = len(mf.data)
    }
    ret := relativeTo + offset
    if ret < 0 || ret > len(mf.data) {
        return -1, errors.New("New offset would fall outside of the MockFile")
    }
    mf.offset = ret
    return
}

func (mf *MockFile) Readdir(count int) ([]os.FileInfo, error) {
    if count <= 0 {
        return []os.FileInfo{}, nil
    }
    return []os.FileInfo{}, errors.New("MockFiles have no associated directory")
}

func (mf *MockFile) Stat() (os.FileInfo, error) {
    return MockFileInfo{mf}
}

func OpenMockFile(data []byte) *MockFile {
    mf := MockFile{data, true, 0}
}

【讨论】：

诚然，我不确定MockFile.Stat() 是否完全符合其期望。可能MockFileInfo 应该采用MockFile 而不是*MockFile。如果你os.Stat 一个真实的文件，然后改变它的大小会发生什么？文件大小变化是否反映在os.FileInfo 中？

【解决方案3】：

如果您正在寻找 gzip http2 push 的这个答案，您可以使用以下选项：

options := &http.PushOptions{
                Header: http.Header{
                    "Accept-Encoding": r.Header["Accept-Encoding"],
                },
            }
            if err := pusher.Push(filePush, options); err != nil {
                log.Printf("Failed to push: %v", err)
                return
            }

【讨论】：