mirror of
https://github.com/restic/restic.git
synced 2024-08-26 22:27:22 +02:00
94 lines
1.7 KiB
Go
94 lines
1.7 KiB
Go
package restic
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/restic/restic/backend"
|
|
)
|
|
|
|
// for testing
|
|
var getCacheDir = GetCacheDir
|
|
|
|
type Cache struct {
|
|
base string
|
|
}
|
|
|
|
func NewCache() (*Cache, error) {
|
|
dir, err := getCacheDir()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Cache{base: dir}, nil
|
|
}
|
|
|
|
func (c *Cache) Has(t backend.Type, subtype string, id backend.ID) (bool, error) {
|
|
// try to open file
|
|
filename, err := c.filename(t, subtype, id)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
fd, err := os.Open(filename)
|
|
defer fd.Close()
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
return false, nil
|
|
}
|
|
|
|
return false, err
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
func (c *Cache) Store(t backend.Type, subtype string, id backend.ID) (io.WriteCloser, error) {
|
|
filename, err := c.filename(t, subtype, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dirname := filepath.Dir(filename)
|
|
err = os.MkdirAll(dirname, 0700)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
file, err := os.Create(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return file, nil
|
|
}
|
|
|
|
func (c *Cache) Load(t backend.Type, subtype string, id backend.ID) (io.ReadCloser, error) {
|
|
// try to open file
|
|
filename, err := c.filename(t, subtype, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return os.Open(filename)
|
|
}
|
|
|
|
// Construct file name for given Type.
|
|
func (c *Cache) filename(t backend.Type, subtype string, id backend.ID) (string, error) {
|
|
filename := id.String()
|
|
if subtype != "" {
|
|
filename += "." + subtype
|
|
}
|
|
|
|
switch t {
|
|
case backend.Snapshot:
|
|
return filepath.Join(c.base, "snapshots", filename), nil
|
|
case backend.Tree:
|
|
return filepath.Join(c.base, "trees", filename), nil
|
|
}
|
|
|
|
return "", fmt.Errorf("cache not supported for type %v", t)
|
|
}
|