mirror of https://github.com/knative/pkg.git
113 lines
3.2 KiB
Go
113 lines
3.2 KiB
Go
/*
|
|
Copyright 2020 The Knative Authors
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package reconciler
|
|
|
|
import metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
|
|
// AnnotationFilterFunc creates a FilterFunc only accepting objects with given annotation key and value
|
|
func AnnotationFilterFunc(key, value string, allowUnset bool) func(interface{}) bool {
|
|
return func(obj interface{}) bool {
|
|
if mo, ok := obj.(metav1.Object); ok {
|
|
return mapHasOrDefault(mo.GetAnnotations(), key, value, allowUnset)
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
// LabelExistsFilterFunc creates a FilterFunc only accepting objects which have a given label.
|
|
func LabelExistsFilterFunc(label string) func(obj interface{}) bool {
|
|
return func(obj interface{}) bool {
|
|
if mo, ok := obj.(metav1.Object); ok {
|
|
labels := mo.GetLabels()
|
|
_, ok := labels[label]
|
|
return ok
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
// LabelFilterFunc creates a FilterFunc only accepting objects where a label is set to a specific value.
|
|
func LabelFilterFunc(label, value string, allowUnset bool) func(interface{}) bool {
|
|
return func(obj interface{}) bool {
|
|
if mo, ok := obj.(metav1.Object); ok {
|
|
return mapHasOrDefault(mo.GetLabels(), label, value, allowUnset)
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
// NameFilterFunc creates a FilterFunc only accepting objects with the given name.
|
|
func NameFilterFunc(name string) func(interface{}) bool {
|
|
return func(obj interface{}) bool {
|
|
if mo, ok := obj.(metav1.Object); ok {
|
|
return mo.GetName() == name
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
// NamespaceFilterFunc creates a FilterFunc only accepting objects in the given namespace.
|
|
func NamespaceFilterFunc(namespace string) func(interface{}) bool {
|
|
return func(obj interface{}) bool {
|
|
if mo, ok := obj.(metav1.Object); ok {
|
|
return mo.GetNamespace() == namespace
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
// Not inverts the result of the predicate.
|
|
func Not(f func(interface{}) bool) func(interface{}) bool {
|
|
return func(obj interface{}) bool {
|
|
return !f(obj)
|
|
}
|
|
}
|
|
|
|
// Or creates a FilterFunc which performs an OR of the passed in FilterFuncs
|
|
func Or(funcs ...func(interface{}) bool) func(interface{}) bool {
|
|
return func(obj interface{}) bool {
|
|
for _, f := range funcs {
|
|
if f(obj) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
// ChainFilterFuncs creates a FilterFunc which performs an AND of the passed FilterFuncs.
|
|
func ChainFilterFuncs(funcs ...func(interface{}) bool) func(interface{}) bool {
|
|
return func(obj interface{}) bool {
|
|
for _, f := range funcs {
|
|
if !f(obj) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
}
|
|
|
|
// mapHasOrDefault returns true if the map has the key and its value is equal to value.
|
|
// If the key is not found, it returns defaultValue.
|
|
func mapHasOrDefault(m map[string]string, key string, value string, defaultValue bool) bool {
|
|
val, ok := m[key]
|
|
if !ok {
|
|
return defaultValue
|
|
}
|
|
return val == value
|
|
}
|