/
generic_client.go
92 lines (75 loc) · 3.09 KB
/
generic_client.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
package controller
import (
"context"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/tools/cache"
operatorapiv1 "github.com/openshift/api/operator/v1"
operatorclientv1 "github.com/openshift/client-go/operator/clientset/versioned/typed/operator/v1"
operatorinformers "github.com/openshift/client-go/operator/informers/externalversions"
)
type genericClient struct {
informers operatorinformers.SharedInformerFactory
client operatorclientv1.OperatorV1Interface
}
func (p *genericClient) Informer() cache.SharedIndexInformer {
return p.informers.Operator().V1().InsightsOperators().Informer()
}
func (p *genericClient) CurrentStatus() (operatorapiv1.OperatorStatus, error) {
instance, err := p.informers.Operator().V1().InsightsOperators().Lister().Get("cluster")
if err != nil {
return operatorapiv1.OperatorStatus{}, err
}
return instance.Status.OperatorStatus, nil
}
func (p *genericClient) GetOperatorState() (*operatorapiv1.OperatorSpec, *operatorapiv1.OperatorStatus, string, error) {
instance, err := p.informers.Operator().V1().InsightsOperators().Lister().Get("cluster")
if err != nil {
return nil, nil, "", err
}
return &instance.Spec.OperatorSpec, &instance.Status.OperatorStatus, instance.ResourceVersion, nil
}
func (p *genericClient) GetObjectMeta() (*metav1.ObjectMeta, error) {
resource, err := p.informers.Operator().V1().InsightsOperators().Lister().Get("cluster")
if err != nil {
return nil, err
}
return &resource.ObjectMeta, nil
}
func (p *genericClient) UpdateOperatorSpec(ctx context.Context, resourceVersion string,
spec *operatorapiv1.OperatorSpec) (*operatorapiv1.OperatorSpec, string, error) {
resource, err := p.informers.Operator().V1().InsightsOperators().Lister().Get("cluster")
if err != nil {
return nil, "", err
}
resourceCopy := resource.DeepCopy()
resourceCopy.ResourceVersion = resourceVersion
resourceCopy.Spec.OperatorSpec = *spec
ret, err := p.client.InsightsOperators().Update(ctx, resourceCopy, metav1.UpdateOptions{})
if err != nil {
return nil, "", err
}
return &ret.Spec.OperatorSpec, ret.ResourceVersion, nil
}
func (p *genericClient) UpdateOperatorStatus(ctx context.Context, resourceVersion string,
status *operatorapiv1.OperatorStatus) (*operatorapiv1.OperatorStatus, error) {
resource, err := p.informers.Operator().V1().InsightsOperators().Lister().Get("cluster")
if err != nil {
return nil, err
}
resourceCopy := resource.DeepCopy()
resourceCopy.ResourceVersion = resourceVersion
resourceCopy.Status.OperatorStatus = *status
ret, err := p.client.InsightsOperators().UpdateStatus(ctx, resourceCopy, metav1.UpdateOptions{})
if err != nil {
return nil, err
}
return &ret.Status.OperatorStatus, nil
}
func (p *genericClient) GetOperatorStateWithQuorum(_ context.Context) (spec *operatorapiv1.OperatorSpec,
status *operatorapiv1.OperatorStatus, resourceVersion string, err error) {
resource, err := p.informers.Operator().V1().InsightsOperators().Lister().Get("cluster")
if err != nil {
return nil, nil, "", err
}
return &resource.Spec.OperatorSpec, &resource.Status.OperatorStatus, resource.ResourceVersion, nil
}