tstor/src/host/vfs/memory.go

package vfs

import (
	"bytes"
	"context"
	"io/fs"
	"path"
)

var _ Filesystem = &MemoryFs{}

type MemoryFs struct {
	name  string
	files map[string]*MemoryFile
}

// Info implements Filesystem.
func (fs *MemoryFs) Info() (fs.FileInfo, error) {
	return newDirInfo(fs.name), nil
}

// IsDir implements Filesystem.
func (fs *MemoryFs) IsDir() bool {
	return true
}

// Name implements Filesystem.
func (fs *MemoryFs) Name() string {
	return fs.name
}

// Type implements Filesystem.
func (mfs *MemoryFs) Type() fs.FileMode {
	return fs.ModeDir
}

func NewMemoryFS(name string, files map[string]*MemoryFile) *MemoryFs {
	return &MemoryFs{
		name:  name,
		files: files,
	}
}

func (m *MemoryFs) Open(ctx context.Context, filename string) (File, error) {
	return getFile(m.files, filename)
}

func (fs *MemoryFs) ReadDir(ctx context.Context, path string) ([]fs.DirEntry, error) {
	return listDirFromFiles(fs.files, path)
}

// Stat implements Filesystem.
func (mfs *MemoryFs) Stat(ctx context.Context, filename string) (fs.FileInfo, error) {
	file, ok := mfs.files[filename]
	if !ok {
		return nil, ErrNotExist
	}
	return newFileInfo(path.Base(filename), file.Size()), nil
}

// Unlink implements Filesystem.
func (fs *MemoryFs) Unlink(ctx context.Context, filename string) error {
	return ErrNotImplemented
}

var _ File = (*MemoryFile)(nil)

type MemoryFile struct {
	name string
	data *bytes.Reader
}

func NewMemoryFile(name string, data []byte) *MemoryFile {
	return &MemoryFile{
		name: name,
		data: bytes.NewReader(data),
	}
}

func (d *MemoryFile) Stat() (fs.FileInfo, error) {
	return newFileInfo(d.name, int64(d.data.Len())), nil
}

func (d *MemoryFile) Size() int64 {
	return int64(d.data.Len())
}

func (d *MemoryFile) IsDir() bool {
	return false
}

func (d *MemoryFile) Close(ctx context.Context) (err error) {
	return
}

// Read implements File.
func (d *MemoryFile) Read(ctx context.Context, p []byte) (n int, err error) {
	return d.data.Read(p)
}

// ReadAt implements File.
func (d *MemoryFile) ReadAt(ctx context.Context, p []byte, off int64) (n int, err error) {
	return d.data.ReadAt(p, off)
}