如何使用 golang 将字节数组转换为虚拟文件对象？

在 Go 中，将字节数组转换为虚拟文件对象以实现 `http.File` 接口并不容易，但可以通过自定义实现来完成。需要实现 `io.Reader`、`io.Seeker`、`io.Closer` 和 `stat()` 方法，其中 `stat()` 返回自定义的 `os.FileInfo` 实现。 可以使用 `bytes.Reader` 和自定义的 `myfileinfo` 结构来实现 `io.Reader` 和 `io.Seeker`，而 `io.Closer` 可以是一个空函数。`stat()` 方法需要返回一个自定义的 `os.FileInfo` 实现，例如 `myfileinfo`，它提供文件大小、名称和模式等信息。

我知道有一些 Go 库可以创建整个文件系统，例如 VFS。但我只想将字节数组制作成可以实现 File 接口的东西。

解决方案

标准库中没有现成的解决方案，但自己实现并不难。

我们需要的是这个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

看起来很简单，可以自己模拟这个。

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}
}

好了，本文到此结束，带大家了解了《如何使用 golang 将字节数组转换为虚拟文件对象？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！