mirror of
https://github.com/zrepl/zrepl.git
synced 2024-12-23 15:38:49 +01:00
6f441c55dc
``` cs@cstp:[~/zrepl/zrepl]: artifacts/zrepl-linux-amd64 zfs-abstraction release-stale --dry-run panic: runtime error: invalid memory address or nil pointer dereference [signal SIGSEGV: segmentation violation code=0x1 addr=0x0 pc=0x9de971] goroutine 1 [running]: github.com/zrepl/zrepl/endpoint.listStaleFiltering(0xc00012b700, 0x5, 0x8, 0x0, 0x13ccb58) /endpoint/endpoint_zfs_abstraction.go:736 +0x281 github.com/zrepl/zrepl/endpoint.ListStale(0xe3ae20, 0xc00026b740, 0x0, 0xe27c60, 0x13ccb58, 0xc0001d8690, 0x0, 0x0, 0x0, 0x1, ...) /endpoint/endpoint_zfs_abstraction.go:698 +0x3fd github.com/zrepl/zrepl/client.doZabsReleaseStale(0xe3ae20, 0xc00026b740, 0x13a28a0, 0xc000151be0, 0x0, 0x1, 0x5705b4, 0xc0002686e0) /client/zfsabstractions_release.go:83 +0x1a0 github.com/zrepl/zrepl/cli.(*Subcommand).run(0x13a28a0, 0xc000264c80, 0xc000151be0, 0x0, 0x1) /cli/cli.go:104 +0xf5 github.com/spf13/cobra.(*Command).execute(0xc000264c80, 0xc000151bd0, 0x1, 0x1, 0xc000264c80, 0xc000151bd0) GOROOT/pkg/mod/github.com/spf13/cobra@v0.0.2/command.go:760 +0x2aa github.com/spf13/cobra.(*Command).ExecuteC(0x13a43c0, 0x0, 0x0, 0x0) GOROOT/pkg/mod/github.com/spf13/cobra@v0.0.2/command.go:846 +0x2ea github.com/spf13/cobra.(*Command).Execute(...) GOROOT/pkg/mod/github.com/spf13/cobra@v0.0.2/command.go:794 github.com/zrepl/zrepl/cli.Run() /cli/cli.go:151 +0x2d main.main() /main.go:24 +0x20 ```
839 lines
22 KiB
Go
839 lines
22 KiB
Go
package endpoint
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"math"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/zrepl/zrepl/util/envconst"
|
|
"github.com/zrepl/zrepl/util/semaphore"
|
|
"github.com/zrepl/zrepl/zfs"
|
|
)
|
|
|
|
type AbstractionType string
|
|
|
|
// Implementation note:
|
|
// There are a lot of exhaustive switches on AbstractionType in the code base.
|
|
// When adding a new abstraction type, make sure to search and update them!
|
|
const (
|
|
AbstractionStepBookmark AbstractionType = "step-bookmark"
|
|
AbstractionStepHold AbstractionType = "step-hold"
|
|
AbstractionLastReceivedHold AbstractionType = "last-received-hold"
|
|
AbstractionReplicationCursorBookmarkV1 AbstractionType = "replication-cursor-bookmark-v1"
|
|
AbstractionReplicationCursorBookmarkV2 AbstractionType = "replication-cursor-bookmark-v2"
|
|
)
|
|
|
|
var AbstractionTypesAll = map[AbstractionType]bool{
|
|
AbstractionStepBookmark: true,
|
|
AbstractionStepHold: true,
|
|
AbstractionLastReceivedHold: true,
|
|
AbstractionReplicationCursorBookmarkV1: true,
|
|
AbstractionReplicationCursorBookmarkV2: true,
|
|
}
|
|
|
|
// Implementation Note:
|
|
// Whenever you add a new accessor, adjust AbstractionJSON.MarshalJSON accordingly
|
|
type Abstraction interface {
|
|
GetType() AbstractionType
|
|
GetFS() string
|
|
GetName() string
|
|
GetFullPath() string
|
|
GetJobID() *JobID // may return nil if the abstraction does not have a JobID
|
|
GetCreateTXG() uint64
|
|
GetFilesystemVersion() zfs.FilesystemVersion
|
|
String() string
|
|
// destroy the abstraction: either releases the hold or destroys the bookmark
|
|
Destroy(context.Context) error
|
|
json.Marshaler
|
|
}
|
|
|
|
func (t AbstractionType) Validate() error {
|
|
switch t {
|
|
case AbstractionStepBookmark:
|
|
return nil
|
|
case AbstractionStepHold:
|
|
return nil
|
|
case AbstractionLastReceivedHold:
|
|
return nil
|
|
case AbstractionReplicationCursorBookmarkV1:
|
|
return nil
|
|
case AbstractionReplicationCursorBookmarkV2:
|
|
return nil
|
|
default:
|
|
return errors.Errorf("unknown abstraction type %q", t)
|
|
}
|
|
}
|
|
|
|
func (t AbstractionType) MustValidate() error {
|
|
if err := t.Validate(); err != nil {
|
|
panic(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type AbstractionJSON struct{ Abstraction }
|
|
|
|
var _ json.Marshaler = (*AbstractionJSON)(nil)
|
|
|
|
func (a AbstractionJSON) MarshalJSON() ([]byte, error) {
|
|
type S struct {
|
|
Type AbstractionType
|
|
FS string
|
|
Name string
|
|
FullPath string
|
|
JobID *JobID // may return nil if the abstraction does not have a JobID
|
|
CreateTXG uint64
|
|
FilesystemVersion zfs.FilesystemVersion
|
|
String string
|
|
}
|
|
v := S{
|
|
Type: a.Abstraction.GetType(),
|
|
FS: a.Abstraction.GetFS(),
|
|
Name: a.Abstraction.GetName(),
|
|
FullPath: a.Abstraction.GetFullPath(),
|
|
JobID: a.Abstraction.GetJobID(),
|
|
CreateTXG: a.Abstraction.GetCreateTXG(),
|
|
FilesystemVersion: a.Abstraction.GetFilesystemVersion(),
|
|
String: a.Abstraction.String(),
|
|
}
|
|
return json.Marshal(v)
|
|
}
|
|
|
|
type AbstractionTypeSet map[AbstractionType]bool
|
|
|
|
func AbstractionTypeSetFromStrings(sts []string) (AbstractionTypeSet, error) {
|
|
ats := make(map[AbstractionType]bool, len(sts))
|
|
for i, t := range sts {
|
|
at := AbstractionType(t)
|
|
if err := at.Validate(); err != nil {
|
|
return nil, errors.Wrapf(err, "invalid abstraction type #%d %q", i+1, t)
|
|
}
|
|
ats[at] = true
|
|
}
|
|
return ats, nil
|
|
}
|
|
|
|
func (s AbstractionTypeSet) ContainsAll(q AbstractionTypeSet) bool {
|
|
for k := range q {
|
|
if _, ok := s[k]; !ok {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (s AbstractionTypeSet) ContainsAnyOf(q AbstractionTypeSet) bool {
|
|
for k := range q {
|
|
if _, ok := s[k]; ok {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (s AbstractionTypeSet) String() string {
|
|
sts := make([]string, 0, len(s))
|
|
for i := range s {
|
|
sts = append(sts, string(i))
|
|
}
|
|
sts = sort.StringSlice(sts)
|
|
return strings.Join(sts, ",")
|
|
}
|
|
|
|
func (s AbstractionTypeSet) Validate() error {
|
|
for k := range s {
|
|
if err := k.Validate(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type BookmarkExtractor func(fs *zfs.DatasetPath, v zfs.FilesystemVersion) Abstraction
|
|
|
|
// returns nil if the abstraction type is not bookmark-based
|
|
func (t AbstractionType) BookmarkExtractor() BookmarkExtractor {
|
|
switch t {
|
|
case AbstractionStepBookmark:
|
|
return StepBookmarkExtractor
|
|
case AbstractionReplicationCursorBookmarkV1:
|
|
return ReplicationCursorV1Extractor
|
|
case AbstractionReplicationCursorBookmarkV2:
|
|
return ReplicationCursorV2Extractor
|
|
case AbstractionStepHold:
|
|
return nil
|
|
case AbstractionLastReceivedHold:
|
|
return nil
|
|
default:
|
|
panic(fmt.Sprintf("unimpl: %q", t))
|
|
}
|
|
}
|
|
|
|
type HoldExtractor = func(fs *zfs.DatasetPath, v zfs.FilesystemVersion, tag string) Abstraction
|
|
|
|
// returns nil if the abstraction type is not hold-based
|
|
func (t AbstractionType) HoldExtractor() HoldExtractor {
|
|
switch t {
|
|
case AbstractionStepBookmark:
|
|
return nil
|
|
case AbstractionReplicationCursorBookmarkV1:
|
|
return nil
|
|
case AbstractionReplicationCursorBookmarkV2:
|
|
return nil
|
|
case AbstractionStepHold:
|
|
return StepHoldExtractor
|
|
case AbstractionLastReceivedHold:
|
|
return LastReceivedHoldExtractor
|
|
default:
|
|
panic(fmt.Sprintf("unimpl: %q", t))
|
|
}
|
|
}
|
|
|
|
type ListZFSHoldsAndBookmarksQuery struct {
|
|
FS ListZFSHoldsAndBookmarksQueryFilesystemFilter
|
|
// What abstraction types should match (any contained in the set)
|
|
What AbstractionTypeSet
|
|
|
|
// The output for the query must satisfy _all_ (AND) requirements of all fields in this query struct.
|
|
|
|
// if not nil: JobID of the hold or bookmark in question must be equal
|
|
// else: JobID of the hold or bookmark can be any value
|
|
JobID *JobID
|
|
|
|
// zero-value means any CreateTXG is acceptable
|
|
CreateTXG CreateTXGRange
|
|
|
|
// Number of concurrently queried filesystems. Must be >= 1
|
|
Concurrency int64
|
|
}
|
|
|
|
type CreateTXGRangeBound struct {
|
|
CreateTXG uint64
|
|
Inclusive *zfs.NilBool // must not be nil
|
|
}
|
|
|
|
// A non-empty range of CreateTXGs
|
|
//
|
|
// If both Since and Until are nil, any CreateTXG is acceptable
|
|
type CreateTXGRange struct {
|
|
// if not nil: The hold's snapshot or the bookmark's createtxg must be greater than (or equal) Since
|
|
// else: CreateTXG of the hold or bookmark can be any value accepted by Until
|
|
Since *CreateTXGRangeBound
|
|
// if not nil: The hold's snapshot or the bookmark's createtxg must be less than (or equal) Until
|
|
// else: CreateTXG of the hold or bookmark can be any value accepted by Since
|
|
Until *CreateTXGRangeBound
|
|
}
|
|
|
|
// FS == nil XOR Filter == nil
|
|
type ListZFSHoldsAndBookmarksQueryFilesystemFilter struct {
|
|
FS *string
|
|
Filter zfs.DatasetFilter
|
|
}
|
|
|
|
func (q *ListZFSHoldsAndBookmarksQuery) Validate() error {
|
|
if err := q.FS.Validate(); err != nil {
|
|
return errors.Wrap(err, "FS")
|
|
}
|
|
if q.JobID != nil {
|
|
q.JobID.MustValidate() // FIXME
|
|
}
|
|
if err := q.CreateTXG.Validate(); err != nil {
|
|
return errors.Wrap(err, "CreateTXGRange")
|
|
}
|
|
if err := q.What.Validate(); err != nil {
|
|
return err
|
|
}
|
|
if q.Concurrency < 1 {
|
|
return errors.New("Concurrency must be >= 1")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var createTXGRangeBoundAllowCreateTXG0 = envconst.Bool("ZREPL_ENDPOINT_LIST_ABSTRACTIONS_QUERY_CREATETXG_RANGE_BOUND_ALLOW_0", false)
|
|
|
|
func (i *CreateTXGRangeBound) Validate() error {
|
|
if err := i.Inclusive.Validate(); err != nil {
|
|
return errors.Wrap(err, "Inclusive")
|
|
}
|
|
if i.CreateTXG == 0 && !createTXGRangeBoundAllowCreateTXG0 {
|
|
return errors.New("CreateTXG must be non-zero")
|
|
}
|
|
return nil
|
|
|
|
}
|
|
|
|
func (f *ListZFSHoldsAndBookmarksQueryFilesystemFilter) Validate() error {
|
|
if f == nil {
|
|
return nil
|
|
}
|
|
fsSet := f.FS != nil
|
|
filterSet := f.Filter != nil
|
|
if fsSet && filterSet || !fsSet && !filterSet {
|
|
return fmt.Errorf("must set FS or Filter field, but fsIsSet=%v and filterIsSet=%v", fsSet, filterSet)
|
|
}
|
|
if fsSet {
|
|
if err := zfs.EntityNamecheck(*f.FS, zfs.EntityTypeFilesystem); err != nil {
|
|
return errors.Wrap(err, "FS invalid")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (f *ListZFSHoldsAndBookmarksQueryFilesystemFilter) Filesystems(ctx context.Context) ([]string, error) {
|
|
if err := f.Validate(); err != nil {
|
|
panic(err)
|
|
}
|
|
if f.FS != nil {
|
|
return []string{*f.FS}, nil
|
|
}
|
|
if f.Filter != nil {
|
|
dps, err := zfs.ZFSListMapping(ctx, f.Filter)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
fss := make([]string, len(dps))
|
|
for i, dp := range dps {
|
|
fss[i] = dp.ToString()
|
|
}
|
|
return fss, nil
|
|
}
|
|
panic("unreachable")
|
|
}
|
|
|
|
func (r *CreateTXGRange) Validate() error {
|
|
if r.Since != nil {
|
|
if err := r.Since.Validate(); err != nil {
|
|
return errors.Wrap(err, "Since")
|
|
}
|
|
}
|
|
if r.Until != nil {
|
|
if err := r.Until.Validate(); err != nil {
|
|
return errors.Wrap(err, "Until")
|
|
}
|
|
}
|
|
if _, err := r.effectiveBounds(); err != nil {
|
|
return errors.Wrapf(err, "specified range %s is semantically invalid", r)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// inclusive-inclusive bounds
|
|
type effectiveBounds struct {
|
|
sinceInclusive uint64
|
|
sinceUnbounded bool
|
|
untilInclusive uint64
|
|
untilUnbounded bool
|
|
}
|
|
|
|
// callers must have validated r.Since and r.Until before calling this method
|
|
func (r *CreateTXGRange) effectiveBounds() (bounds effectiveBounds, err error) {
|
|
|
|
bounds.sinceUnbounded = r.Since == nil
|
|
bounds.untilUnbounded = r.Until == nil
|
|
|
|
if r.Since == nil && r.Until == nil {
|
|
return bounds, nil
|
|
}
|
|
|
|
if r.Since != nil {
|
|
bounds.sinceInclusive = r.Since.CreateTXG
|
|
if !r.Since.Inclusive.B {
|
|
if r.Since.CreateTXG == math.MaxUint64 {
|
|
return bounds, errors.Errorf("Since-exclusive (%v) must be less than math.MaxUint64 (%v)",
|
|
r.Since.CreateTXG, uint64(math.MaxUint64))
|
|
}
|
|
bounds.sinceInclusive++
|
|
}
|
|
}
|
|
|
|
if r.Until != nil {
|
|
bounds.untilInclusive = r.Until.CreateTXG
|
|
if !r.Until.Inclusive.B {
|
|
if r.Until.CreateTXG == 0 {
|
|
return bounds, errors.Errorf("Until-exclusive (%v) must be greater than 0", r.Until.CreateTXG)
|
|
}
|
|
bounds.untilInclusive--
|
|
}
|
|
}
|
|
|
|
if !bounds.sinceUnbounded && !bounds.untilUnbounded {
|
|
if bounds.sinceInclusive >= bounds.untilInclusive {
|
|
return bounds, errors.Errorf("effective range bounds are [%v,%v] which is empty or invalid", bounds.sinceInclusive, bounds.untilInclusive)
|
|
}
|
|
// fallthrough
|
|
}
|
|
|
|
return bounds, nil
|
|
}
|
|
|
|
func (r *CreateTXGRange) String() string {
|
|
var buf strings.Builder
|
|
if r.Since == nil {
|
|
fmt.Fprintf(&buf, "~")
|
|
} else {
|
|
if err := r.Since.Inclusive.Validate(); err != nil {
|
|
fmt.Fprintf(&buf, "?")
|
|
} else if r.Since.Inclusive.B {
|
|
fmt.Fprintf(&buf, "[")
|
|
} else {
|
|
fmt.Fprintf(&buf, "(")
|
|
}
|
|
fmt.Fprintf(&buf, "%d", r.Since.CreateTXG)
|
|
}
|
|
|
|
fmt.Fprintf(&buf, ",")
|
|
|
|
if r.Until == nil {
|
|
fmt.Fprintf(&buf, "~")
|
|
} else {
|
|
fmt.Fprintf(&buf, "%d", r.Until.CreateTXG)
|
|
if err := r.Until.Inclusive.Validate(); err != nil {
|
|
fmt.Fprintf(&buf, "?")
|
|
} else if r.Until.Inclusive.B {
|
|
fmt.Fprintf(&buf, "]")
|
|
} else {
|
|
fmt.Fprintf(&buf, ")")
|
|
}
|
|
}
|
|
|
|
return buf.String()
|
|
}
|
|
|
|
// panics if not .Validate()
|
|
func (r *CreateTXGRange) IsUnbounded() bool {
|
|
if err := r.Validate(); err != nil {
|
|
panic(err)
|
|
}
|
|
bounds, err := r.effectiveBounds()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return bounds.sinceUnbounded && bounds.untilUnbounded
|
|
}
|
|
|
|
// panics if not .Validate()
|
|
func (r *CreateTXGRange) Contains(qCreateTxg uint64) bool {
|
|
if err := r.Validate(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
bounds, err := r.effectiveBounds()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
sinceMatches := bounds.sinceUnbounded || bounds.sinceInclusive <= qCreateTxg
|
|
untilMatches := bounds.untilUnbounded || qCreateTxg <= bounds.untilInclusive
|
|
|
|
return sinceMatches && untilMatches
|
|
}
|
|
|
|
type ListAbstractionsError struct {
|
|
FS string
|
|
Snap string
|
|
What string
|
|
Err error
|
|
}
|
|
|
|
func (e ListAbstractionsError) Error() string {
|
|
if e.FS == "" {
|
|
return fmt.Sprintf("list endpoint abstractions: %s: %s", e.What, e.Err)
|
|
} else {
|
|
v := e.FS
|
|
if e.Snap != "" {
|
|
v = fmt.Sprintf("%s@%s", e.FS, e.Snap)
|
|
}
|
|
return fmt.Sprintf("list endpoint abstractions on %q: %s: %s", v, e.What, e.Err)
|
|
}
|
|
}
|
|
|
|
type putListAbstractionErr func(err error, fs string, what string)
|
|
type putListAbstraction func(a Abstraction)
|
|
|
|
type ListAbstractionsErrors []ListAbstractionsError
|
|
|
|
func (e ListAbstractionsErrors) Error() string {
|
|
if len(e) == 0 {
|
|
panic(e)
|
|
}
|
|
if len(e) == 1 {
|
|
return fmt.Sprintf("list endpoint abstractions: %s", e[0])
|
|
}
|
|
msgs := make([]string, len(e))
|
|
for i := range e {
|
|
msgs[i] = e.Error()
|
|
}
|
|
return fmt.Sprintf("list endpoint abstractions: multiple errors:\n%s", strings.Join(msgs, "\n"))
|
|
}
|
|
|
|
func ListAbstractions(ctx context.Context, query ListZFSHoldsAndBookmarksQuery) (out []Abstraction, outErrs []ListAbstractionsError, err error) {
|
|
outChan, outErrsChan, err := ListAbstractionsStreamed(ctx, query)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
var wg sync.WaitGroup
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
for a := range outChan {
|
|
out = append(out, a)
|
|
}
|
|
}()
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
for err := range outErrsChan {
|
|
outErrs = append(outErrs, err)
|
|
}
|
|
}()
|
|
wg.Wait()
|
|
return out, outErrs, nil
|
|
}
|
|
|
|
// if err != nil, the returned channels are both nil
|
|
// if err == nil, both channels must be fully drained by the caller to avoid leaking goroutines
|
|
func ListAbstractionsStreamed(ctx context.Context, query ListZFSHoldsAndBookmarksQuery) (<-chan Abstraction, <-chan ListAbstractionsError, error) {
|
|
|
|
// impl note: structure the query processing in such a way that
|
|
// a minimum amount of zfs shell-outs needs to be done
|
|
|
|
if err := query.Validate(); err != nil {
|
|
return nil, nil, errors.Wrap(err, "validate query")
|
|
}
|
|
|
|
fss, err := query.FS.Filesystems(ctx)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrap(err, "list filesystems")
|
|
}
|
|
|
|
outErrs := make(chan ListAbstractionsError)
|
|
out := make(chan Abstraction)
|
|
|
|
errCb := func(err error, fs string, what string) {
|
|
outErrs <- ListAbstractionsError{Err: err, FS: fs, What: what}
|
|
}
|
|
emitAbstraction := func(a Abstraction) {
|
|
jobIdMatches := query.JobID == nil || a.GetJobID() == nil || *a.GetJobID() == *query.JobID
|
|
|
|
createTXGMatches := query.CreateTXG.Contains(a.GetCreateTXG())
|
|
|
|
if jobIdMatches && createTXGMatches {
|
|
out <- a
|
|
}
|
|
}
|
|
|
|
sem := semaphore.New(int64(query.Concurrency))
|
|
go func() {
|
|
defer close(out)
|
|
defer close(outErrs)
|
|
var wg sync.WaitGroup
|
|
defer wg.Wait()
|
|
for i := range fss {
|
|
wg.Add(1)
|
|
go func(i int) {
|
|
defer wg.Done()
|
|
g, err := sem.Acquire(ctx)
|
|
if err != nil {
|
|
errCb(err, fss[i], err.Error())
|
|
return
|
|
}
|
|
func() {
|
|
defer g.Release()
|
|
listAbstractionsImplFS(ctx, fss[i], &query, emitAbstraction, errCb)
|
|
}()
|
|
}(i)
|
|
}
|
|
}()
|
|
|
|
return out, outErrs, nil
|
|
}
|
|
|
|
func listAbstractionsImplFS(ctx context.Context, fs string, query *ListZFSHoldsAndBookmarksQuery, emitCandidate putListAbstraction, errCb putListAbstractionErr) {
|
|
fsp, err := zfs.NewDatasetPath(fs)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if len(query.What) == 0 {
|
|
return
|
|
}
|
|
|
|
whatTypes := zfs.VersionTypeSet{}
|
|
for what := range query.What {
|
|
if e := what.BookmarkExtractor(); e != nil {
|
|
whatTypes[zfs.Bookmark] = true
|
|
}
|
|
if e := what.HoldExtractor(); e != nil {
|
|
whatTypes[zfs.Snapshot] = true
|
|
}
|
|
}
|
|
fsvs, err := zfs.ZFSListFilesystemVersions(ctx, fsp, zfs.ListFilesystemVersionsOptions{
|
|
Types: whatTypes,
|
|
})
|
|
if err != nil {
|
|
errCb(err, fs, "list filesystem versions")
|
|
return
|
|
}
|
|
|
|
for at := range query.What {
|
|
bmE := at.BookmarkExtractor()
|
|
holdE := at.HoldExtractor()
|
|
if bmE == nil && holdE == nil || bmE != nil && holdE != nil {
|
|
panic("implementation error: extractors misconfigured for " + at)
|
|
}
|
|
for _, v := range fsvs {
|
|
var a Abstraction
|
|
if v.Type == zfs.Bookmark && bmE != nil {
|
|
a = bmE(fsp, v)
|
|
}
|
|
if v.Type == zfs.Snapshot && holdE != nil && query.CreateTXG.Contains(v.GetCreateTXG()) && (!v.UserRefs.Valid || v.UserRefs.Value > 0) {
|
|
holds, err := zfs.ZFSHolds(ctx, fsp.ToString(), v.Name)
|
|
if err != nil {
|
|
errCb(err, v.ToAbsPath(fsp), "get hold on snap")
|
|
continue
|
|
}
|
|
for _, tag := range holds {
|
|
a = holdE(fsp, v, tag)
|
|
}
|
|
}
|
|
if a != nil {
|
|
emitCandidate(a)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
type BatchDestroyResult struct {
|
|
Abstraction
|
|
DestroyErr error
|
|
}
|
|
|
|
var _ json.Marshaler = (*BatchDestroyResult)(nil)
|
|
|
|
func (r BatchDestroyResult) MarshalJSON() ([]byte, error) {
|
|
err := ""
|
|
if r.DestroyErr != nil {
|
|
err = r.DestroyErr.Error()
|
|
}
|
|
s := struct {
|
|
Abstraction AbstractionJSON
|
|
DestroyErr string
|
|
}{
|
|
AbstractionJSON{r.Abstraction},
|
|
err,
|
|
}
|
|
return json.Marshal(s)
|
|
}
|
|
|
|
func BatchDestroy(ctx context.Context, abs []Abstraction) <-chan BatchDestroyResult {
|
|
// hold-based batching: per snapshot
|
|
// bookmark-based batching: none possible via CLI
|
|
// => not worth the trouble for now, will be worth it once we start using channel programs
|
|
// => TODO: actual batching using channel programs
|
|
res := make(chan BatchDestroyResult, len(abs))
|
|
go func() {
|
|
for _, a := range abs {
|
|
res <- BatchDestroyResult{
|
|
a,
|
|
a.Destroy(ctx),
|
|
}
|
|
}
|
|
close(res)
|
|
}()
|
|
return res
|
|
}
|
|
|
|
type StalenessInfo struct {
|
|
ConstructedWithQuery ListZFSHoldsAndBookmarksQuery
|
|
Live []Abstraction
|
|
Stale []Abstraction
|
|
}
|
|
|
|
type fsAndJobId struct {
|
|
fs string
|
|
jobId JobID
|
|
}
|
|
|
|
type ListStaleQueryError struct {
|
|
error
|
|
}
|
|
|
|
// returns *ListStaleQueryError if the given query cannot be used for determining staleness info
|
|
func ListStale(ctx context.Context, q ListZFSHoldsAndBookmarksQuery) (*StalenessInfo, error) {
|
|
if !q.CreateTXG.IsUnbounded() {
|
|
// we must determine the most recent step per FS, can't allow that
|
|
return nil, &ListStaleQueryError{errors.New("ListStale cannot have Until != nil set on query")}
|
|
}
|
|
|
|
// if asking for step holds, must also as for step bookmarks (same kind of abstraction)
|
|
// as well as replication cursor bookmarks (for firstNotStale)
|
|
ifAnyThenAll := AbstractionTypeSet{
|
|
AbstractionStepHold: true,
|
|
AbstractionStepBookmark: true,
|
|
AbstractionReplicationCursorBookmarkV2: true,
|
|
}
|
|
if q.What.ContainsAnyOf(ifAnyThenAll) && !q.What.ContainsAll(ifAnyThenAll) {
|
|
return nil, &ListStaleQueryError{errors.Errorf("ListStale requires query to ask for all of %s", ifAnyThenAll.String())}
|
|
}
|
|
|
|
// ----------------- done validating query for listStaleFiltering -----------------------
|
|
|
|
qAbs, absErr, err := ListAbstractions(ctx, q)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(absErr) > 0 {
|
|
// can't go on here because we can't determine the most recent step
|
|
return nil, ListAbstractionsErrors(absErr)
|
|
}
|
|
|
|
si := listStaleFiltering(qAbs, q.CreateTXG.Since)
|
|
si.ConstructedWithQuery = q
|
|
return si, nil
|
|
}
|
|
|
|
type fsAjobAtype struct {
|
|
fsAndJobId
|
|
Type AbstractionType
|
|
}
|
|
|
|
// For step holds and bookmarks, only those older than the most recent replication cursor
|
|
// of their (filesystem,job) is considered because younger ones cannot be stale by definition
|
|
// (if we destroy them, we might actually lose the hold on the `To` for an ongoing incremental replication)
|
|
//
|
|
// For replication cursors and last-received-holds, only the most recent one is kept.
|
|
//
|
|
// the returned StalenessInfo.ConstructedWithQuery is not set
|
|
func listStaleFiltering(abs []Abstraction, sinceBound *CreateTXGRangeBound) *StalenessInfo {
|
|
|
|
var noJobId []Abstraction
|
|
by := make(map[fsAjobAtype][]Abstraction)
|
|
for _, a := range abs {
|
|
if a.GetJobID() == nil {
|
|
noJobId = append(noJobId, a)
|
|
continue
|
|
}
|
|
faj := fsAjobAtype{fsAndJobId{a.GetFS(), *a.GetJobID()}, a.GetType()}
|
|
l := by[faj]
|
|
l = append(l, a)
|
|
by[faj] = l
|
|
}
|
|
|
|
type stepFirstNotStaleCandidate struct {
|
|
cursor *Abstraction
|
|
step *Abstraction
|
|
}
|
|
stepFirstNotStaleCandidates := make(map[fsAndJobId]stepFirstNotStaleCandidate) // empty map => will always return nil
|
|
for _, a := range abs {
|
|
if a.GetJobID() == nil {
|
|
continue // already put those into always-live list noJobId in above loop
|
|
}
|
|
key := fsAndJobId{a.GetFS(), *a.GetJobID()}
|
|
c := stepFirstNotStaleCandidates[key]
|
|
|
|
switch a.GetType() {
|
|
// stepFirstNotStaleCandidate.cursor
|
|
case AbstractionReplicationCursorBookmarkV2:
|
|
if c.cursor == nil || (*c.cursor).GetCreateTXG() < a.GetCreateTXG() {
|
|
a := a
|
|
c.cursor = &a
|
|
}
|
|
|
|
// stepFirstNotStaleCandidate.step
|
|
case AbstractionStepBookmark:
|
|
fallthrough
|
|
case AbstractionStepHold:
|
|
if c.step == nil || (*c.step).GetCreateTXG() < a.GetCreateTXG() {
|
|
a := a
|
|
c.step = &a
|
|
}
|
|
|
|
// not interested in the others
|
|
default:
|
|
continue // not relevant
|
|
|
|
}
|
|
|
|
stepFirstNotStaleCandidates[key] = c
|
|
}
|
|
|
|
ret := &StalenessInfo{
|
|
Live: noJobId,
|
|
Stale: []Abstraction{},
|
|
}
|
|
|
|
for k := range by {
|
|
l := by[k]
|
|
|
|
if k.Type == AbstractionStepHold || k.Type == AbstractionStepBookmark {
|
|
// all older than the most recent cursor are stale, others are always live
|
|
|
|
// if we don't have a replication cursor yet, use untilBound = nil
|
|
// to consider all steps stale (...at first)
|
|
var untilBound *CreateTXGRangeBound
|
|
{
|
|
sfnsc := stepFirstNotStaleCandidates[k.fsAndJobId]
|
|
|
|
// if there's a replication cursor, use it as a cutoff between live and stale
|
|
// if there's none, we are in initial replication and only need to keep
|
|
// the most recent step hold live, since that's what our initial replication strategy
|
|
// uses (both initially and on resume)
|
|
// (FIXME hardcoded replication strategy)
|
|
if sfnsc.cursor != nil {
|
|
untilBound = &CreateTXGRangeBound{
|
|
CreateTXG: (*sfnsc.cursor).GetCreateTXG(),
|
|
// if we have a cursor, can throw away step hold on both From and To
|
|
Inclusive: &zfs.NilBool{B: true},
|
|
}
|
|
} else if sfnsc.step != nil {
|
|
untilBound = &CreateTXGRangeBound{
|
|
CreateTXG: (*sfnsc.step).GetCreateTXG(),
|
|
// if we don't have a cursor, the step most recent step hold is our
|
|
// initial replication cursor and it's possibly still live (interrupted initial replication)
|
|
Inclusive: &zfs.NilBool{B: false},
|
|
}
|
|
} else {
|
|
untilBound = nil // consider everything stale
|
|
}
|
|
}
|
|
staleRange := CreateTXGRange{
|
|
Since: sinceBound,
|
|
Until: untilBound,
|
|
}
|
|
|
|
// partition by staleRange
|
|
for _, a := range l {
|
|
if staleRange.Contains(a.GetCreateTXG()) {
|
|
ret.Stale = append(ret.Stale, a)
|
|
} else {
|
|
ret.Live = append(ret.Live, a)
|
|
}
|
|
}
|
|
|
|
} else if k.Type == AbstractionReplicationCursorBookmarkV2 || k.Type == AbstractionLastReceivedHold {
|
|
// all but the most recent are stale by definition (we always _move_ them)
|
|
// NOTE: must not use firstNotStale in this branch, not computed for these types
|
|
|
|
// sort descending (highest createtxg first), then cut off
|
|
sort.Slice(l, func(i, j int) bool {
|
|
return l[i].GetCreateTXG() > l[j].GetCreateTXG()
|
|
})
|
|
if len(l) > 0 {
|
|
ret.Live = append(ret.Live, l[0])
|
|
ret.Stale = append(ret.Stale, l[1:]...)
|
|
}
|
|
} else {
|
|
ret.Live = append(ret.Live, l...)
|
|
}
|
|
}
|
|
|
|
return ret
|
|
|
|
}
|