Skip to content
This repository was archived by the owner on Jul 30, 2021. It is now read-only.

Commit 2d85601

Browse files
SataQiuSataQiu
committed
move control plane locker from CAPA
1 parent 7a638d0 commit 2d85601

File tree

3 files changed

+250
-0
lines changed

3 files changed

+250
-0
lines changed

‎controllers/control_plane_init_locker.go

Lines changed: 104 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,104 @@
1+
/*
2+
Copyright 2019 The Kubernetes Authors.
3+
4+
Licensed under the Apache License, Version 2.0 (the "License");
5+
you may not use this file except in compliance with the License.
6+
You may obtain a copy of the License at
7+
8+
http://www.apache.org/licenses/LICENSE-2.0
9+
10+
Unless required by applicable law or agreed to in writing, software
11+
distributed under the License is distributed on an "AS IS" BASIS,
12+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
See the License for the specific language governing permissions and
14+
limitations under the License.
15+
*/
16+
17+
package controllers
18+
19+
import (
20+
"fmt"
21+
22+
"github.com/go-logr/logr"
23+
apicorev1 "k8s.io/api/core/v1"
24+
apierrors "k8s.io/apimachinery/pkg/api/errors"
25+
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
26+
corev1 "k8s.io/client-go/kubernetes/typed/core/v1"
27+
clusterv2 "sigs.k8s.io/cluster-api/pkg/apis/cluster/v1alpha2"
28+
)
29+
30+
// ControlPlaneInitLocker provides a locking mechanism for cluster initialization.
31+
type ControlPlaneInitLocker interface {
32+
// Acquire returns true if it acquires the lock for the cluster.
33+
Acquire(cluster *clusterv2.Cluster) bool
34+
}
35+
36+
// controlPlaneInitLocker uses a ConfigMap to synchronize cluster initialization.
37+
type controlPlaneInitLocker struct {
38+
log logr.Logger
39+
configMapClient corev1.ConfigMapsGetter
40+
}
41+
42+
var _ ControlPlaneInitLocker = &controlPlaneInitLocker{}
43+
44+
func newControlPlaneInitLocker(log logr.Logger, configMapClient corev1.ConfigMapsGetter) *controlPlaneInitLocker {
45+
return &controlPlaneInitLocker{
46+
log: log,
47+
configMapClient: configMapClient,
48+
}
49+
}
50+
51+
func (l *controlPlaneInitLocker) Acquire(cluster *clusterv2.Cluster) bool {
52+
configMapName := fmt.Sprintf("%s-controlplane", cluster.UID)
53+
log := l.log.WithValues("namespace", cluster.Namespace, "cluster-name", cluster.Name, "configmap-name", configMapName)
54+
55+
exists, err := l.configMapExists(cluster.Namespace, configMapName)
56+
if err != nil {
57+
log.Error(err, "Error checking for control plane configmap lock existence")
58+
return false
59+
}
60+
if exists {
61+
return false
62+
}
63+
64+
controlPlaneConfigMap := &apicorev1.ConfigMap{
65+
ObjectMeta: metav1.ObjectMeta{
66+
Namespace: cluster.Namespace,
67+
Name: configMapName,
68+
OwnerReferences: []metav1.OwnerReference{
69+
{
70+
APIVersion: cluster.APIVersion,
71+
Kind: cluster.Kind,
72+
Name: cluster.Name,
73+
UID: cluster.UID,
74+
},
75+
},
76+
},
77+
}
78+
79+
log.Info("Attempting to create control plane configmap lock")
80+
_, err = l.configMapClient.ConfigMaps(cluster.Namespace).Create(controlPlaneConfigMap)
81+
if err != nil {
82+
if apierrors.IsAlreadyExists(err) {
83+
// Someone else beat us to it
84+
log.Info("Control plane configmap lock already exists")
85+
} else {
86+
log.Error(err, "Error creating control plane configmap lock")
87+
}
88+
89+
// Unable to acquire
90+
return false
91+
}
92+
93+
// Successfully acquired
94+
return true
95+
}
96+
97+
func (l *controlPlaneInitLocker) configMapExists(namespace, name string) (bool, error) {
98+
_, err := l.configMapClient.ConfigMaps(namespace).Get(name, metav1.GetOptions{})
99+
if apierrors.IsNotFound(err) {
100+
return false, nil
101+
}
102+
103+
return err == nil, err
104+
}

‎controllers/control_plane_init_locker_test.go

Lines changed: 145 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,145 @@
1+
/*
2+
Copyright 2019 The Kubernetes Authors.
3+
4+
Licensed under the Apache License, Version 2.0 (the "License");
5+
you may not use this file except in compliance with the License.
6+
You may obtain a copy of the License at
7+
8+
http://www.apache.org/licenses/LICENSE-2.0
9+
10+
Unless required by applicable law or agreed to in writing, software
11+
distributed under the License is distributed on an "AS IS" BASIS,
12+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
See the License for the specific language governing permissions and
14+
limitations under the License.
15+
*/
16+
17+
package controllers
18+
19+
import (
20+
"testing"
21+
22+
"github.com/pkg/errors"
23+
v1 "k8s.io/api/core/v1"
24+
apierrors "k8s.io/apimachinery/pkg/api/errors"
25+
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
26+
"k8s.io/apimachinery/pkg/runtime/schema"
27+
"k8s.io/apimachinery/pkg/types"
28+
"k8s.io/apimachinery/pkg/watch"
29+
corev1client "k8s.io/client-go/kubernetes/typed/core/v1"
30+
clusterv2 "sigs.k8s.io/cluster-api/pkg/apis/cluster/v1alpha2"
31+
"sigs.k8s.io/controller-runtime/pkg/runtime/log"
32+
)
33+
34+
func TestControlPlaneInitLockerAcquire(t *testing.T) {
35+
tests := []struct {
36+
name string
37+
configMap *v1.ConfigMap
38+
getError error
39+
createError error
40+
expectAcquire bool
41+
}{
42+
{
43+
name: "configmap already exists",
44+
configMap: &v1.ConfigMap{},
45+
expectAcquire: false,
46+
},
47+
{
48+
name: "error getting configmap",
49+
getError: errors.New("get error"),
50+
expectAcquire: false,
51+
},
52+
{
53+
name: "create succeeds",
54+
getError: apierrors.NewNotFound(schema.GroupResource{Group: "", Resource: "configmaps"}, "uid1-configmap"),
55+
expectAcquire: true,
56+
},
57+
{
58+
name: "create fails",
59+
getError: apierrors.NewNotFound(schema.GroupResource{Group: "", Resource: "configmaps"}, "uid1-configmap"),
60+
createError: errors.New("create error"),
61+
expectAcquire: false,
62+
},
63+
}
64+
65+
for _, tc := range tests {
66+
t.Run(tc.name, func(t *testing.T) {
67+
l := &controlPlaneInitLocker{
68+
log: log.ZapLogger(true),
69+
configMapClient: &configMapsGetter{
70+
configMap: tc.configMap,
71+
getError: tc.getError,
72+
createError: tc.createError,
73+
},
74+
}
75+
76+
cluster := &clusterv2.Cluster{
77+
ObjectMeta: metav1.ObjectMeta{
78+
Namespace: "ns1",
79+
Name: "name1",
80+
UID: types.UID("uid1"),
81+
},
82+
}
83+
84+
acquired := l.Acquire(cluster)
85+
if tc.expectAcquire != acquired {
86+
t.Errorf("expected %t, got %t", tc.expectAcquire, acquired)
87+
}
88+
})
89+
}
90+
}
91+
92+
type configMapsGetter struct {
93+
configMap *v1.ConfigMap
94+
getError error
95+
createError error
96+
}
97+
98+
func (c *configMapsGetter) ConfigMaps(namespace string) corev1client.ConfigMapInterface {
99+
return &configMapClient{
100+
configMap: c.configMap,
101+
getError: c.getError,
102+
createError: c.createError,
103+
}
104+
}
105+
106+
type configMapClient struct {
107+
configMap *v1.ConfigMap
108+
getError error
109+
createError error
110+
}
111+
112+
func (c *configMapClient) Create(configMap *v1.ConfigMap) (*v1.ConfigMap, error) {
113+
return c.configMap, c.createError
114+
}
115+
116+
func (c *configMapClient) Get(name string, getOptions metav1.GetOptions) (*v1.ConfigMap, error) {
117+
if c.getError != nil {
118+
return nil, c.getError
119+
}
120+
return c.configMap, nil
121+
}
122+
123+
func (c *configMapClient) Update(*v1.ConfigMap) (*v1.ConfigMap, error) {
124+
panic("not implemented")
125+
}
126+
127+
func (c *configMapClient) Delete(name string, options *metav1.DeleteOptions) error {
128+
panic("not implemented")
129+
}
130+
131+
func (c *configMapClient) DeleteCollection(options *metav1.DeleteOptions, listOptions metav1.ListOptions) error {
132+
panic("not implemented")
133+
}
134+
135+
func (c *configMapClient) List(opts metav1.ListOptions) (*v1.ConfigMapList, error) {
136+
panic("not implemented")
137+
}
138+
139+
func (c *configMapClient) Watch(opts metav1.ListOptions) (watch.Interface, error) {
140+
panic("not implemented")
141+
}
142+
143+
func (c *configMapClient) Patch(name string, pt types.PatchType, data []byte, subresources ...string) (result *v1.ConfigMap, err error) {
144+
panic("not implemented")
145+
}

‎go.mod

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ require (
1212
k8s.io/apimachinery v0.0.0-20190704094733-8f6ac2502e51
1313
k8s.io/client-go v11.0.1-0.20190704100234-640d9f240853+incompatible
1414
k8s.io/cluster-bootstrap v0.0.0-20190703212826-5ad085674a4f
15+
k8s.io/klog v0.3.1
1516
k8s.io/utils v0.0.0-20190607212802-c55fbcfc754a // indirect
1617
sigs.k8s.io/cluster-api v0.0.0-20190711133056-09e491e49d7c
1718
sigs.k8s.io/controller-runtime v0.2.0-beta.4

0 commit comments

Comments
 (0)