mirror of
https://github.com/offen/docker-volume-backup.git
synced 2024-09-19 21:20:28 +02:00
132 lines
3.4 KiB
Go
132 lines
3.4 KiB
Go
|
package local
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"os"
|
||
|
"path"
|
||
|
"path/filepath"
|
||
|
"time"
|
||
|
|
||
|
"github.com/offen/docker-volume-backup/internal/storage"
|
||
|
"github.com/offen/docker-volume-backup/internal/utilities"
|
||
|
)
|
||
|
|
||
|
type localStorage struct {
|
||
|
*storage.StorageBackend
|
||
|
latestSymlink string
|
||
|
}
|
||
|
|
||
|
// NewStorageBackend creates and initializes a new local storage backend.
|
||
|
func NewStorageBackend(archivePath string, latestSymlink string, logFunc storage.Log) storage.Backend {
|
||
|
return &localStorage{
|
||
|
StorageBackend: &storage.StorageBackend{
|
||
|
DestinationPath: archivePath,
|
||
|
Log: logFunc,
|
||
|
},
|
||
|
latestSymlink: latestSymlink,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Name return the name of the storage backend
|
||
|
func (b *localStorage) Name() string {
|
||
|
return "Local"
|
||
|
}
|
||
|
|
||
|
// Copy copies the given file to the local storage backend.
|
||
|
func (b *localStorage) Copy(file string) error {
|
||
|
if _, err := os.Stat(b.DestinationPath); os.IsNotExist(err) {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
_, name := path.Split(file)
|
||
|
|
||
|
if err := utilities.CopyFile(file, path.Join(b.DestinationPath, name)); err != nil {
|
||
|
return b.Log(storage.ERROR, b.Name(), "Copy: Error copying file to local archive! %w", err)
|
||
|
}
|
||
|
b.Log(storage.INFO, b.Name(), "Stored copy of backup `%s` in local archive `%s`.", file, b.DestinationPath)
|
||
|
|
||
|
if b.latestSymlink != "" {
|
||
|
symlink := path.Join(b.DestinationPath, b.latestSymlink)
|
||
|
if _, err := os.Lstat(symlink); err == nil {
|
||
|
os.Remove(symlink)
|
||
|
}
|
||
|
if err := os.Symlink(name, symlink); err != nil {
|
||
|
return b.Log(storage.ERROR, b.Name(), "Copy: error creating latest symlink! %w", err)
|
||
|
}
|
||
|
b.Log(storage.INFO, b.Name(), "Created/Updated symlink `%s` for latest backup.", b.latestSymlink)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// Prune rotates away backups according to the configuration and provided deadline for the local storage backend.
|
||
|
func (b *localStorage) Prune(deadline time.Time, pruningPrefix string) (*storage.PruneStats, error) {
|
||
|
globPattern := path.Join(
|
||
|
b.DestinationPath,
|
||
|
fmt.Sprintf("%s*", pruningPrefix),
|
||
|
)
|
||
|
globMatches, err := filepath.Glob(globPattern)
|
||
|
if err != nil {
|
||
|
return nil, b.Log(storage.ERROR, b.Name(),
|
||
|
"Prune: Error looking up matching files using pattern %s! %w",
|
||
|
globPattern,
|
||
|
err,
|
||
|
)
|
||
|
}
|
||
|
|
||
|
var candidates []string
|
||
|
for _, candidate := range globMatches {
|
||
|
fi, err := os.Lstat(candidate)
|
||
|
if err != nil {
|
||
|
return nil, b.Log(storage.ERROR, b.Name(),
|
||
|
"Prune: Error calling Lstat on file %s! %w",
|
||
|
candidate,
|
||
|
err,
|
||
|
)
|
||
|
}
|
||
|
|
||
|
if fi.Mode()&os.ModeSymlink != os.ModeSymlink {
|
||
|
candidates = append(candidates, candidate)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
var matches []string
|
||
|
for _, candidate := range candidates {
|
||
|
fi, err := os.Stat(candidate)
|
||
|
if err != nil {
|
||
|
return nil, b.Log(storage.ERROR, b.Name(),
|
||
|
"Prune: Error calling stat on file %s! %w",
|
||
|
candidate,
|
||
|
err,
|
||
|
)
|
||
|
}
|
||
|
if fi.ModTime().Before(deadline) {
|
||
|
matches = append(matches, candidate)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
stats := &storage.PruneStats{
|
||
|
Total: uint(len(candidates)),
|
||
|
Pruned: uint(len(matches)),
|
||
|
}
|
||
|
|
||
|
b.DoPrune(b.Name(), len(matches), len(candidates), "local backup(s)", func() error {
|
||
|
var removeErrors []error
|
||
|
for _, match := range matches {
|
||
|
if err := os.Remove(match); err != nil {
|
||
|
removeErrors = append(removeErrors, err)
|
||
|
}
|
||
|
}
|
||
|
if len(removeErrors) != 0 {
|
||
|
return b.Log(storage.ERROR, b.Name(),
|
||
|
"Prune: %d error(s) deleting local files, starting with: %w",
|
||
|
len(removeErrors),
|
||
|
utilities.Join(removeErrors...),
|
||
|
)
|
||
|
}
|
||
|
return nil
|
||
|
})
|
||
|
|
||
|
return stats, nil
|
||
|
}
|