/
reconcile.go
116 lines (107 loc) · 3.86 KB
/
reconcile.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
package certmanagerpki
import (
"context"
"fmt"
"reflect"
certv1 "github.com/cert-manager/cert-manager/pkg/apis/certmanager/v1"
"go.uber.org/zap"
corev1 "k8s.io/api/core/v1"
apierrors "k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/types"
"sigs.k8s.io/controller-runtime/pkg/client"
v1 "github.com/konpyutaika/nifikop/api/v1"
"github.com/konpyutaika/nifikop/pkg/k8sutil"
)
// reconcile ensures the given kubernetes object.
func reconcile(ctx context.Context, log zap.Logger, client client.Client, object runtime.Object, cluster *v1.NifiCluster) (err error) {
switch object := object.(type) {
case *certv1.Issuer:
return reconcileIssuer(ctx, log, client, object, cluster)
case *certv1.ClusterIssuer:
return reconcileClusterIssuer(ctx, log, client, object, cluster)
case *certv1.Certificate:
return reconcileCertificate(ctx, log, client, object, cluster)
case *corev1.Secret:
return reconcileSecret(ctx, log, client, object, cluster)
case *v1.NifiUser:
return reconcileUser(ctx, log, client, object, cluster)
default:
panic(fmt.Sprintf("Invalid object type: %v", reflect.TypeOf(object)))
}
}
// reconcileClusterIssuer ensures a cert-manager ClusterIssuer.
func reconcileClusterIssuer(ctx context.Context, log zap.Logger, client client.Client, issuer *certv1.ClusterIssuer, cluster *v1.NifiCluster) error {
obj := &certv1.ClusterIssuer{}
var err error
if err = client.Get(ctx, types.NamespacedName{Name: issuer.Name, Namespace: issuer.Namespace}, obj); err != nil {
if !apierrors.IsNotFound(err) {
return err
}
return client.Create(ctx, issuer)
}
if err = k8sutil.Reconcile(log, client, issuer, cluster, &cluster.Status); err != nil {
return err
}
return nil
}
// reconcileIssuer ensures a cert-manager Issuer.
func reconcileIssuer(ctx context.Context, log zap.Logger, client client.Client, issuer *certv1.Issuer, cluster *v1.NifiCluster) error {
obj := &certv1.Issuer{}
var err error
if err = client.Get(ctx, types.NamespacedName{Name: issuer.Name, Namespace: issuer.Namespace}, obj); err != nil {
if !apierrors.IsNotFound(err) {
return err
}
return client.Create(ctx, issuer)
}
if err = k8sutil.Reconcile(log, client, issuer, cluster, &cluster.Status); err != nil {
return err
}
return nil
}
// reconcileCertificate ensures a cert-manager certificate.
func reconcileCertificate(ctx context.Context, log zap.Logger, client client.Client, cert *certv1.Certificate, cluster *v1.NifiCluster) error {
obj := &certv1.Certificate{}
var err error
if err = client.Get(ctx, types.NamespacedName{Name: cert.Name, Namespace: cert.Namespace}, obj); err != nil {
if !apierrors.IsNotFound(err) {
return err
}
return client.Create(ctx, cert)
}
if err = k8sutil.Reconcile(log, client, cert, cluster, &cluster.Status); err != nil {
return err
}
return nil
}
// reconcileSecret ensures a Kubernetes secret.
func reconcileSecret(ctx context.Context, log zap.Logger, client client.Client, secret *corev1.Secret, cluster *v1.NifiCluster) error {
obj := &corev1.Secret{}
var err error
if err = client.Get(ctx, types.NamespacedName{Name: secret.Name, Namespace: secret.Namespace}, obj); err != nil {
if !apierrors.IsNotFound(err) {
return err
}
return client.Create(ctx, secret)
}
if err = k8sutil.Reconcile(log, client, secret, cluster, &cluster.Status); err != nil {
return err
}
return nil
}
// reconcileUser ensures a v1.NifiUser.
func reconcileUser(ctx context.Context, log zap.Logger, client client.Client, user *v1.NifiUser, cluster *v1.NifiCluster) error {
obj := &v1.NifiUser{}
var err error
if err = client.Get(ctx, types.NamespacedName{Name: user.Name, Namespace: user.Namespace}, obj); err != nil {
if !apierrors.IsNotFound(err) {
return err
}
return client.Create(ctx, user)
}
if err = k8sutil.Reconcile(log, client, user, cluster, &cluster.Status); err != nil {
return err
}
return nil
}