/
controller.go
62 lines (53 loc) · 1.97 KB
/
controller.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
package serviceaccount
import (
"fmt"
"k8s.io/client-go/kubernetes"
listersv1 "k8s.io/client-go/listers/core/v1"
"k8s.io/client-go/tools/record"
v1alpha1 "github.com/jetstack/navigator/pkg/apis/navigator/v1alpha1"
"github.com/jetstack/navigator/pkg/controllers/elasticsearch/util"
)
type defaultElasticsearchClusterServiceAccountControl struct {
kubeClient kubernetes.Interface
serviceAccountLister listersv1.ServiceAccountLister
recorder record.EventRecorder
}
type Interface interface {
Sync(*v1alpha1.ElasticsearchCluster) (v1alpha1.ElasticsearchClusterStatus, error)
}
var _ Interface = &defaultElasticsearchClusterServiceAccountControl{}
func NewController(
kubeClient kubernetes.Interface,
serviceAccountLister listersv1.ServiceAccountLister,
recorder record.EventRecorder,
) Interface {
return &defaultElasticsearchClusterServiceAccountControl{
kubeClient: kubeClient,
serviceAccountLister: serviceAccountLister,
recorder: recorder,
}
}
func (e *defaultElasticsearchClusterServiceAccountControl) Sync(c *v1alpha1.ElasticsearchCluster) (v1alpha1.ElasticsearchClusterStatus, error) {
// lookup existing ServiceAccount with appropriate labels for np in cluster c
// if multiple exist, exit with an error
// if one exists, return
// if none exist, create one
sel, err := util.SelectorForCluster(c.Name)
if err != nil {
return c.Status, fmt.Errorf("error creating label selector for cluster '%s': %s", c.Name, err.Error())
}
svcAccts, err := e.serviceAccountLister.ServiceAccounts(c.Namespace).List(sel)
if err != nil {
return c.Status, err
}
// if more than one serviceaccount matches the labels, exit here to be safe
if len(svcAccts) > 1 {
return c.Status, fmt.Errorf("multiple service accounts match label selector (%s)", sel.String())
}
expected := clusterServiceAccount(c)
if len(svcAccts) == 0 {
_, err := e.kubeClient.CoreV1().ServiceAccounts(c.Namespace).Create(expected)
return c.Status, err
}
return c.Status, nil
}