-
Notifications
You must be signed in to change notification settings - Fork 252
/
operator_deployment.go
124 lines (108 loc) · 3.6 KB
/
operator_deployment.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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/*
Copyright The CloudNativePG Contributors
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 certs
import (
"context"
"fmt"
v1 "k8s.io/api/apps/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/labels"
"sigs.k8s.io/controller-runtime/pkg/client"
"github.com/cloudnative-pg/cloudnative-pg/pkg/management/log"
"github.com/cloudnative-pg/cloudnative-pg/pkg/utils"
)
// SetAsOwnedByOperatorDeployment sets the controlled object as owned by the operator deployment.
//
// IMPORTANT: The controlled resource must reside in the same namespace as the operator as described by:
// https://kubernetes.io/docs/concepts/overview/working-with-objects/owners-dependents/
func SetAsOwnedByOperatorDeployment(ctx context.Context,
kubeClient client.Client,
controlled *metav1.ObjectMeta,
operatorLabelSelector string,
) error {
deployment, err := GetOperatorDeployment(ctx, kubeClient, controlled.Namespace, operatorLabelSelector)
if err != nil {
return err
}
// The deployment typeMeta is empty (kubernetes bug), so we need to explicitly populate it.
typeMeta := metav1.TypeMeta{
Kind: "Deployment",
APIVersion: "apps/v1",
}
utils.SetAsOwnedBy(controlled, deployment.ObjectMeta, typeMeta)
return nil
}
// GetOperatorDeployment find the operator deployment using labels
// and then return the deployment object, in case we can't find a deployment
// or we find more than one, we just return an error.
func GetOperatorDeployment(
ctx context.Context,
kubeClient client.Client,
namespace, operatorLabelSelector string,
) (*v1.Deployment, error) {
labelMap, err := labels.ConvertSelectorToLabelsMap(operatorLabelSelector)
if err != nil {
return nil, err
}
deployment, err := findOperatorDeploymentByFilter(ctx,
kubeClient,
namespace,
client.MatchingLabelsSelector{Selector: labelMap.AsSelector()})
if err != nil {
return nil, err
}
if deployment != nil {
return deployment, nil
}
deployment, err = findOperatorDeploymentByFilter(ctx,
kubeClient,
namespace,
client.HasLabels{"operators.coreos.com/cloudnative-pg.openshift-operators="})
if err != nil {
return nil, err
}
if deployment != nil {
return deployment, nil
}
return nil, fmt.Errorf("no deployment detected")
}
// findOperatorDeploymentByFilter search in a defined namespace
// looking for a deployment with the defined filter
func findOperatorDeploymentByFilter(ctx context.Context,
kubeClient client.Client,
namespace string,
filter client.ListOption,
) (*v1.Deployment, error) {
logger := log.FromContext(ctx)
deploymentList := &v1.DeploymentList{}
err := kubeClient.List(
ctx,
deploymentList,
client.InNamespace(namespace),
filter,
)
if err != nil {
return nil, err
}
switch {
case len(deploymentList.Items) == 1:
return &deploymentList.Items[0], nil
case len(deploymentList.Items) > 1:
err = fmt.Errorf("more than one operator deployment running")
logger.Error(err, "more than one operator deployment found with the filter", "filter", filter)
return nil, err
}
err = fmt.Errorf("no operator deployment found")
logger.Error(err, "no operator deployment found with the filter", "filter", filter)
return nil, err
}