32787c5258
fix test error formatting updates from comments update gofmt simplify tests add to new sync controller add tests remove configmap changes due to rebase updates from review refactor tests to be based on operations improvements from review updates from rebase rebase to #45374 updates from review refactor SendToCluster for tests fix import order rebase to upstream
87 lines
2.6 KiB
Go
87 lines
2.6 KiB
Go
/*
|
|
Copyright 2017 The Kubernetes 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 clusterselector
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/apimachinery/pkg/selection"
|
|
federation_v1beta1 "k8s.io/kubernetes/federation/apis/federation/v1beta1"
|
|
)
|
|
|
|
// Parses the cluster selector annotation to find out if the object with that annotation should be forwarded to a cluster with the given clusterLabels.
|
|
func SendToCluster(clusterLabels map[string]string, annotations map[string]string) (bool, error) {
|
|
// Check if a ClusterSelector annotation exists and send to all clusters when it does not exist
|
|
val, ok := annotations[federation_v1beta1.FederationClusterSelectorAnnotation]
|
|
if !ok {
|
|
return true, nil
|
|
}
|
|
|
|
selector, err := getSelector(val)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return selector.Matches(labels.Set(clusterLabels)), nil
|
|
}
|
|
|
|
func getSelector(annotation string) (labels.Selector, error) {
|
|
selector := labels.NewSelector()
|
|
requirements := make([]federation_v1beta1.ClusterSelectorRequirement, 0)
|
|
err := json.Unmarshal([]byte(annotation), &requirements)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, requirement := range requirements {
|
|
r, err := labels.NewRequirement(requirement.Key, ConvertOperator(requirement.Operator), requirement.Values)
|
|
if err != nil {
|
|
// Stop processing and assume failure since we have no way of knowing the end users intent for this or any other clusters.
|
|
return nil, err
|
|
}
|
|
selector = selector.Add(*r)
|
|
}
|
|
|
|
return selector, nil
|
|
}
|
|
|
|
// ConvertOperator converts a string operator into selection.Operator type
|
|
func ConvertOperator(source string) selection.Operator {
|
|
var op selection.Operator
|
|
switch source {
|
|
case "!", "DoesNotExist":
|
|
op = selection.DoesNotExist
|
|
case "=":
|
|
op = selection.Equals
|
|
case "==":
|
|
op = selection.DoubleEquals
|
|
case "in", "In":
|
|
op = selection.In
|
|
case "!=":
|
|
op = selection.NotEquals
|
|
case "notin", "NotIn":
|
|
op = selection.NotIn
|
|
case "exists", "Exists":
|
|
op = selection.Exists
|
|
case "gt", "Gt", ">":
|
|
op = selection.GreaterThan
|
|
case "lt", "Lt", "<":
|
|
op = selection.LessThan
|
|
}
|
|
return op
|
|
}
|