mirror of
https://github.com/zrepl/zrepl.git
synced 2025-01-21 13:49:30 +01:00
110 lines
3.0 KiB
Go
110 lines
3.0 KiB
Go
package endpoint
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"regexp"
|
|
"strconv"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/zrepl/zrepl/zfs"
|
|
)
|
|
|
|
// returns the short name (no fs# prefix)
|
|
func makeJobAndGuidBookmarkName(prefix string, fs string, guid uint64, jobid string) (string, error) {
|
|
bmname := fmt.Sprintf(prefix+"_G_%016x_J_%s", guid, jobid)
|
|
if err := zfs.EntityNamecheck(fmt.Sprintf("%s#%s", fs, bmname), zfs.EntityTypeBookmark); err != nil {
|
|
return "", err
|
|
}
|
|
return bmname, nil
|
|
}
|
|
|
|
var jobAndGuidBookmarkRE = regexp.MustCompile(`(.+)_G_([0-9a-f]{16})_J_(.+)$`)
|
|
|
|
func parseJobAndGuidBookmarkName(fullname string, prefix string) (guid uint64, jobID JobID, _ error) {
|
|
|
|
if len(prefix) == 0 {
|
|
panic("prefix must not be empty")
|
|
}
|
|
|
|
if err := zfs.EntityNamecheck(fullname, zfs.EntityTypeBookmark); err != nil {
|
|
return 0, JobID{}, err
|
|
}
|
|
|
|
_, _, name, err := zfs.DecomposeVersionString(fullname)
|
|
if err != nil {
|
|
return 0, JobID{}, errors.Wrap(err, "decompose bookmark name")
|
|
}
|
|
|
|
match := jobAndGuidBookmarkRE.FindStringSubmatch(name)
|
|
if match == nil {
|
|
return 0, JobID{}, errors.Errorf("bookmark name does not match regex %q", jobAndGuidBookmarkRE.String())
|
|
}
|
|
if match[1] != prefix {
|
|
return 0, JobID{}, errors.Errorf("prefix component does not match: expected %q, got %q", prefix, match[1])
|
|
}
|
|
|
|
guid, err = strconv.ParseUint(match[2], 16, 64)
|
|
if err != nil {
|
|
return 0, JobID{}, errors.Wrapf(err, "parse guid component: %q", match[2])
|
|
}
|
|
|
|
jobID, err = MakeJobID(match[3])
|
|
if err != nil {
|
|
return 0, JobID{}, errors.Wrapf(err, "parse jobid component: %q", match[3])
|
|
}
|
|
|
|
return guid, jobID, nil
|
|
}
|
|
|
|
func destroyBookmarksOlderThan(ctx context.Context, fs string, mostRecent *zfs.ZFSSendArgVersion, jobID JobID, filter func(shortname string) (accept bool)) (destroyed []zfs.FilesystemVersion, err error) {
|
|
if filter == nil {
|
|
panic(filter)
|
|
}
|
|
|
|
fsp, err := zfs.NewDatasetPath(fs)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "invalid filesystem path")
|
|
}
|
|
|
|
mostRecentProps, err := mostRecent.ValidateExistsAndGetCheckedProps(ctx, fs)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "validate most recent version argument")
|
|
}
|
|
|
|
stepBookmarks, err := zfs.ZFSListFilesystemVersions(fsp, zfs.FilterFromClosure(
|
|
func(t zfs.VersionType, name string) (accept bool, err error) {
|
|
if t != zfs.Bookmark {
|
|
return false, nil
|
|
}
|
|
return filter(name), nil
|
|
}))
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "list bookmarks")
|
|
}
|
|
|
|
// cut off all bookmarks prior to mostRecent's CreateTXG
|
|
var destroy []zfs.FilesystemVersion
|
|
for _, v := range stepBookmarks {
|
|
if v.Type != zfs.Bookmark {
|
|
panic("implementation error")
|
|
}
|
|
if !filter(v.Name) {
|
|
panic("inconsistent filter result")
|
|
}
|
|
if v.CreateTXG < mostRecentProps.CreateTXG {
|
|
destroy = append(destroy, v)
|
|
}
|
|
}
|
|
|
|
// FIXME use batch destroy, must adopt code to handle bookmarks
|
|
for _, v := range destroy {
|
|
if err := zfs.ZFSDestroyIdempotent(ctx, v.ToAbsPath(fsp)); err != nil {
|
|
return nil, errors.Wrap(err, "destroy bookmark")
|
|
}
|
|
}
|
|
|
|
return destroy, nil
|
|
}
|