-
Notifications
You must be signed in to change notification settings - Fork 114
/
labels.go
57 lines (52 loc) · 1.75 KB
/
labels.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package controller
import (
"github.com/pkg/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/labels"
"k8s.io/apimachinery/pkg/selection"
"sigs.k8s.io/controller-runtime/pkg/predicate"
kargoapi "github.com/akuity/kargo/api/v1alpha1"
)
// GetShardPredicate constructs a predicate used as an event filter for various
// reconcilers. If a non-empty shard name is passed to this function, it returns
// a predicate that matches ONLY resources labeled for that shard. If an empty
// shard name is passed to this function, it returns a predicate that matches
// ONLY resources that are NOT labeled for ANY shard.
func GetShardPredicate(shard string) (predicate.Predicate, error) {
if shard == "" {
pred, err := predicate.LabelSelectorPredicate(
metav1.LabelSelector{
MatchExpressions: []metav1.LabelSelectorRequirement{
{
Key: kargoapi.ShardLabelKey,
Operator: metav1.LabelSelectorOpDoesNotExist,
},
},
},
)
return pred, errors.Wrap(err, "error creating default selector predicate")
}
pred, err := predicate.LabelSelectorPredicate(
*metav1.SetAsLabelSelector(
labels.Set(
map[string]string{
kargoapi.ShardLabelKey: shard,
},
),
),
)
return pred, errors.Wrap(err, "error creating shard selector predicate")
}
func GetShardRequirement(shard string) (*labels.Requirement, error) {
req, err := labels.NewRequirement(kargoapi.ShardLabelKey, selection.Equals, []string{shard})
if err != nil {
return nil, errors.Wrap(err, "error creating shard label selector")
}
if shard == "" {
req, err = labels.NewRequirement(kargoapi.ShardLabelKey, selection.DoesNotExist, nil)
if err != nil {
return nil, errors.Wrap(err, "error creating default label selector")
}
}
return req, nil
}