mirror of
https://github.com/kopia/kopia.git
synced 2026-01-25 06:48:48 -05:00
4e705726fe
* cache: refactored reusable portion of cache into separate package * repo: plumbed through caching for remote repository clients * repo: plumb through cache in the unit tests * cache: ensure we only allow absolute cache paths, fixed cache path resolution for remote repositories
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/kopia/kopia/internal/ctxutil"
|
|
"github.com/kopia/kopia/repo/blob"
|
|
"github.com/kopia/kopia/repo/blob/filesystem"
|
|
)
|
|
|
|
// Storage is the storage interface required by the cache and is implemented by the filesystem Storage.
|
|
type Storage interface {
|
|
blob.Storage
|
|
TouchBlob(ctx context.Context, contentID blob.ID, threshold time.Duration) error
|
|
}
|
|
|
|
// NewStorageOrNil returns cache.Storage backed by the provided directory.
|
|
func NewStorageOrNil(ctx context.Context, cacheDir string, maxBytes int64, subdir string) (Storage, error) {
|
|
if maxBytes <= 0 || cacheDir == "" {
|
|
return nil, nil
|
|
}
|
|
|
|
if !filepath.IsAbs(cacheDir) {
|
|
return nil, errors.Errorf("cache dir %q was not absolute", cacheDir)
|
|
}
|
|
|
|
contentCacheDir := filepath.Join(cacheDir, subdir)
|
|
|
|
if _, err := os.Stat(contentCacheDir); os.IsNotExist(err) {
|
|
if mkdirerr := os.MkdirAll(contentCacheDir, 0o700); mkdirerr != nil {
|
|
return nil, errors.Wrap(mkdirerr, "error creating cache directory")
|
|
}
|
|
}
|
|
|
|
fs, err := filesystem.New(ctxutil.Detach(ctx), &filesystem.Options{
|
|
Path: contentCacheDir,
|
|
DirectoryShards: []int{2},
|
|
})
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error initializing filesystem cache")
|
|
}
|
|
|
|
return fs.(Storage), nil
|
|
}
|