mirror of
https://github.com/restic/restic.git
synced 2024-08-25 05:37:25 +02:00
This enables the backends to request the calculation of a backend-specific hash. For the currently supported backends this will always be MD5. The hash calculation happens as early as possible, for pack files this is during assembly of the pack file. That way the hash would even capture corruptions of the temporary pack file on disk.
148 lines
3.6 KiB
Go
148 lines
3.6 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
"hash"
|
|
"io"
|
|
|
|
"github.com/restic/restic/internal/errors"
|
|
"github.com/restic/restic/internal/restic"
|
|
)
|
|
|
|
// Backend implements a mock backend.
|
|
type Backend struct {
|
|
CloseFn func() error
|
|
IsNotExistFn func(err error) bool
|
|
SaveFn func(ctx context.Context, h restic.Handle, rd restic.RewindReader) error
|
|
OpenReaderFn func(ctx context.Context, h restic.Handle, length int, offset int64) (io.ReadCloser, error)
|
|
StatFn func(ctx context.Context, h restic.Handle) (restic.FileInfo, error)
|
|
ListFn func(ctx context.Context, t restic.FileType, fn func(restic.FileInfo) error) error
|
|
RemoveFn func(ctx context.Context, h restic.Handle) error
|
|
TestFn func(ctx context.Context, h restic.Handle) (bool, error)
|
|
DeleteFn func(ctx context.Context) error
|
|
LocationFn func() string
|
|
HasherFn func() hash.Hash
|
|
}
|
|
|
|
// NewBackend returns new mock Backend instance
|
|
func NewBackend() *Backend {
|
|
be := &Backend{}
|
|
return be
|
|
}
|
|
|
|
// Close the backend.
|
|
func (m *Backend) Close() error {
|
|
if m.CloseFn == nil {
|
|
return nil
|
|
}
|
|
|
|
return m.CloseFn()
|
|
}
|
|
|
|
// Location returns a location string.
|
|
func (m *Backend) Location() string {
|
|
if m.LocationFn == nil {
|
|
return ""
|
|
}
|
|
|
|
return m.LocationFn()
|
|
}
|
|
|
|
// Hasher may return a hash function for calculating a content hash for the backend
|
|
func (m *Backend) Hasher() hash.Hash {
|
|
if m.HasherFn == nil {
|
|
return nil
|
|
}
|
|
|
|
return m.HasherFn()
|
|
}
|
|
|
|
// IsNotExist returns true if the error is caused by a missing file.
|
|
func (m *Backend) IsNotExist(err error) bool {
|
|
if m.IsNotExistFn == nil {
|
|
return false
|
|
}
|
|
|
|
return m.IsNotExistFn(err)
|
|
}
|
|
|
|
// Save data in the backend.
|
|
func (m *Backend) Save(ctx context.Context, h restic.Handle, rd restic.RewindReader) error {
|
|
if m.SaveFn == nil {
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
return m.SaveFn(ctx, h, rd)
|
|
}
|
|
|
|
// Load runs fn with a reader that yields the contents of the file at h at the
|
|
// given offset.
|
|
func (m *Backend) Load(ctx context.Context, h restic.Handle, length int, offset int64, fn func(rd io.Reader) error) error {
|
|
rd, err := m.openReader(ctx, h, length, offset)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = fn(rd)
|
|
if err != nil {
|
|
_ = rd.Close() // ignore secondary errors closing the reader
|
|
return err
|
|
}
|
|
return rd.Close()
|
|
}
|
|
|
|
func (m *Backend) openReader(ctx context.Context, h restic.Handle, length int, offset int64) (io.ReadCloser, error) {
|
|
if m.OpenReaderFn == nil {
|
|
return nil, errors.New("not implemented")
|
|
}
|
|
|
|
return m.OpenReaderFn(ctx, h, length, offset)
|
|
}
|
|
|
|
// Stat an object in the backend.
|
|
func (m *Backend) Stat(ctx context.Context, h restic.Handle) (restic.FileInfo, error) {
|
|
if m.StatFn == nil {
|
|
return restic.FileInfo{}, errors.New("not implemented")
|
|
}
|
|
|
|
return m.StatFn(ctx, h)
|
|
}
|
|
|
|
// List items of type t.
|
|
func (m *Backend) List(ctx context.Context, t restic.FileType, fn func(restic.FileInfo) error) error {
|
|
if m.ListFn == nil {
|
|
return nil
|
|
}
|
|
|
|
return m.ListFn(ctx, t, fn)
|
|
}
|
|
|
|
// Remove data from the backend.
|
|
func (m *Backend) Remove(ctx context.Context, h restic.Handle) error {
|
|
if m.RemoveFn == nil {
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
return m.RemoveFn(ctx, h)
|
|
}
|
|
|
|
// Test for the existence of a specific item.
|
|
func (m *Backend) Test(ctx context.Context, h restic.Handle) (bool, error) {
|
|
if m.TestFn == nil {
|
|
return false, errors.New("not implemented")
|
|
}
|
|
|
|
return m.TestFn(ctx, h)
|
|
}
|
|
|
|
// Delete all data.
|
|
func (m *Backend) Delete(ctx context.Context) error {
|
|
if m.DeleteFn == nil {
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
return m.DeleteFn(ctx)
|
|
}
|
|
|
|
// Make sure that Backend implements the backend interface.
|
|
var _ restic.Backend = &Backend{}
|