-
Notifications
You must be signed in to change notification settings - Fork 0
/
ingressClassCreator.go
108 lines (89 loc) · 3.46 KB
/
ingressClassCreator.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
package controllers
import (
"context"
"fmt"
"k8s.io/client-go/tools/record"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
"sigs.k8s.io/controller-runtime/pkg/log"
"k8s.io/apimachinery/pkg/types"
"sigs.k8s.io/controller-runtime/pkg/client"
networking "k8s.io/api/networking/v1"
apierrors "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
const (
ingressClassCreationEventReason = "IngressClassCreation"
)
var k8sCesLabels = map[string]string{"app": "ces", "app.kubernetes.io/name": "k8s-service-discovery"}
type eventRecorder interface {
record.EventRecorder
}
// ingressClassCreator is responsible to create a cluster wide ingress class in the cluster.
type ingressClassCreator struct {
client client.Client
className string
namespace string
eventRecorder eventRecorder
}
// NewIngressClassCreator creates a new ingress class creator.
func NewIngressClassCreator(client client.Client, className string, namespace string, recorder eventRecorder) *ingressClassCreator {
return &ingressClassCreator{
client: client,
className: className,
namespace: namespace,
eventRecorder: recorder,
}
}
// CreateIngressClass check whether the ingress class for the generator exists. If not it will be created.
func (icc ingressClassCreator) CreateIngressClass(ctx context.Context) error {
log.FromContext(ctx).Info(fmt.Sprintf("checking for existing ingress class [%s]", icc.className))
ok, err := icc.isIngressClassAvailable(ctx)
if err != nil {
return fmt.Errorf("failed to check if ingress class [%s] exists: %w", icc.className, err)
}
deployment := &appsv1.Deployment{}
err = icc.client.Get(ctx, types.NamespacedName{Name: "k8s-service-discovery-controller-manager", Namespace: icc.namespace}, deployment)
if err != nil {
return fmt.Errorf("create ingress class: failed to get deployment [k8s-service-discovery-controller-manager]: %w", err)
}
if ok {
icc.eventRecorder.Eventf(deployment, corev1.EventTypeNormal, ingressClassCreationEventReason, "Ingress class [%s] already exists.", icc.className)
log.FromContext(ctx).Info(fmt.Sprintf("ingress class [%s] already exists -> skip creation", icc.className))
return nil
}
ingressClassResource := &networking.IngressClass{
TypeMeta: metav1.TypeMeta{},
ObjectMeta: metav1.ObjectMeta{
Name: icc.className,
Labels: k8sCesLabels,
},
Spec: networking.IngressClassSpec{
Controller: "k8s.io/nginx-ingress",
},
}
err = icc.client.Create(ctx, ingressClassResource)
if err != nil {
return fmt.Errorf("cannot create ingress class [%s] with clientset: %w", icc.className, err)
}
icc.eventRecorder.Eventf(deployment, corev1.EventTypeNormal, ingressClassCreationEventReason, "Ingress class [%s] created.", icc.className)
return nil
}
// isIngressClassAvailable check whether an ingress class with the given name exists in the current namespace.
func (icc ingressClassCreator) isIngressClassAvailable(ctx context.Context) (bool, error) {
ingressClassKey := types.NamespacedName{
Namespace: "",
Name: icc.className,
}
ingressClass := &networking.IngressClass{}
err := icc.client.Get(ctx, ingressClassKey, ingressClass)
if err != nil && apierrors.IsNotFound(err) {
return false, nil
} else if err != nil {
return false, fmt.Errorf("failed to get ingress class [%s] with clientset: %w", icc.className, err)
}
return true, nil
}
func (icc ingressClassCreator) Start(ctx context.Context) error {
return icc.CreateIngressClass(ctx)
}