50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package cache
|
|
|
|
import (
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/pojntfx/stfs/internal/pathext"
|
|
"github.com/pojntfx/stfs/pkg/config"
|
|
"github.com/spf13/afero"
|
|
)
|
|
|
|
func NewCacheFilesystem(
|
|
base afero.Fs,
|
|
root string,
|
|
cacheType string,
|
|
ttl time.Duration,
|
|
cacheDir string,
|
|
) (afero.Fs, error) {
|
|
switch cacheType {
|
|
case config.FileSystemCacheTypeMemory:
|
|
if pathext.IsRoot(root) {
|
|
return afero.NewCacheOnReadFs(base, afero.NewMemMapFs(), ttl), nil
|
|
}
|
|
|
|
return afero.NewCacheOnReadFs(afero.NewBasePathFs(base, root), afero.NewMemMapFs(), ttl), nil
|
|
case config.FileSystemCacheTypeDir:
|
|
if err := os.RemoveAll(cacheDir); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := os.MkdirAll(cacheDir, os.ModePerm); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if pathext.IsRoot(root) {
|
|
return afero.NewCacheOnReadFs(base, afero.NewBasePathFs(afero.NewOsFs(), cacheDir), ttl), nil
|
|
}
|
|
|
|
return afero.NewCacheOnReadFs(afero.NewBasePathFs(base, root), afero.NewBasePathFs(afero.NewOsFs(), cacheDir), ttl), nil
|
|
case config.NoneKey:
|
|
if pathext.IsRoot(root) {
|
|
return base, nil
|
|
}
|
|
|
|
return afero.NewBasePathFs(base, root), nil
|
|
default:
|
|
return nil, config.ErrFileSystemCacheTypeUnsupported
|
|
}
|
|
}
|