mirror of
https://github.com/zrepl/zrepl.git
synced 2025-04-10 19:38:16 +02:00
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package zfs
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/zrepl/zrepl/internal/zfs/zfscmd"
|
|
)
|
|
|
|
type DatasetFilter interface {
|
|
Filter(p *DatasetPath) (pass bool, err error)
|
|
// The caller owns the returned set.
|
|
// Implementations should return a copy.
|
|
UserSpecifiedDatasets() UserSpecifiedDatasetsSet
|
|
}
|
|
|
|
// A set of dataset names that the user specified in the configuration file.
|
|
type UserSpecifiedDatasetsSet map[string]bool
|
|
|
|
// Returns a DatasetFilter that does not filter (passes all paths)
|
|
func NoFilter() DatasetFilter {
|
|
return noFilter{}
|
|
}
|
|
|
|
type noFilter struct{}
|
|
|
|
var _ DatasetFilter = noFilter{}
|
|
|
|
func (noFilter) Filter(p *DatasetPath) (pass bool, err error) { return true, nil }
|
|
func (noFilter) UserSpecifiedDatasets() UserSpecifiedDatasetsSet { return nil }
|
|
|
|
func ZFSListMapping(ctx context.Context, filter DatasetFilter) (datasets []*DatasetPath, err error) {
|
|
res, err := ZFSListMappingProperties(ctx, filter, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
datasets = make([]*DatasetPath, len(res))
|
|
for i, r := range res {
|
|
datasets[i] = r.Path
|
|
}
|
|
return datasets, nil
|
|
}
|
|
|
|
type ZFSListMappingPropertiesResult struct {
|
|
Path *DatasetPath
|
|
// Guaranteed to have the same length as properties in the originating call
|
|
Fields []string
|
|
}
|
|
|
|
// properties must not contain 'name'
|
|
func ZFSListMappingProperties(ctx context.Context, filter DatasetFilter, properties []string) (datasets []ZFSListMappingPropertiesResult, err error) {
|
|
|
|
if filter == nil {
|
|
panic("filter must not be nil")
|
|
}
|
|
|
|
for _, p := range properties {
|
|
if p == "name" {
|
|
panic("properties must not contain 'name'")
|
|
}
|
|
}
|
|
newProps := make([]string, len(properties)+1)
|
|
newProps[0] = "name"
|
|
copy(newProps[1:], properties)
|
|
properties = newProps
|
|
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
defer cancel()
|
|
rchan := make(chan ZFSListResult)
|
|
|
|
args := []string{"-r", "-t", "filesystem,volume"}
|
|
unmatchedUserSpecifiedDatasets := filter.UserSpecifiedDatasets()
|
|
for ds, _ := range unmatchedUserSpecifiedDatasets {
|
|
var path *DatasetPath
|
|
if path, err = NewDatasetPath(ds); err != nil {
|
|
return
|
|
}
|
|
if ph, err := ZFSGetFilesystemPlaceholderState(ctx, path); err == nil && ph.FSExists {
|
|
args = append(args, ds)
|
|
delete(unmatchedUserSpecifiedDatasets, path.ToString())
|
|
}
|
|
}
|
|
err = nil
|
|
|
|
go ZFSListChan(ctx, rchan, properties, nil, args...)
|
|
datasets = make([]ZFSListMappingPropertiesResult, 0)
|
|
for r := range rchan {
|
|
|
|
if r.Err != nil {
|
|
err = r.Err
|
|
return
|
|
}
|
|
|
|
var path *DatasetPath
|
|
if path, err = NewDatasetPath(r.Fields[0]); err != nil {
|
|
return
|
|
}
|
|
|
|
pass, filterErr := filter.Filter(path)
|
|
if filterErr != nil {
|
|
return nil, fmt.Errorf("error calling filter: %s", filterErr)
|
|
}
|
|
if pass {
|
|
datasets = append(datasets, ZFSListMappingPropertiesResult{
|
|
Path: path,
|
|
Fields: r.Fields[1:],
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
jobid := zfscmd.GetJobIDOrDefault(ctx, "__nojobid")
|
|
metric := prom.ZFSListUnmatchedUserSpecifiedDatasetCount.WithLabelValues(jobid)
|
|
metric.Add(float64(len(unmatchedUserSpecifiedDatasets)))
|
|
|
|
return
|
|
}
|