/
ingress.go
71 lines (61 loc) · 1.74 KB
/
ingress.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
package builder
import (
netv1 "k8s.io/api/networking/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"github.com/kong/kubernetes-ingress-controller/v3/internal/annotations"
)
type IngressBuilder struct {
ingress netv1.Ingress
}
// NewIngress builds an Ingress object with the given name and class, when "" is passed as class parameter
// the field .Spec.IngressClassName is not set.
func NewIngress(name string, class string) *IngressBuilder {
var classToSet *string
if class != "" {
classToSet = &class
}
return &IngressBuilder{
ingress: netv1.Ingress{
ObjectMeta: metav1.ObjectMeta{
Name: name,
Annotations: make(map[string]string),
},
TypeMeta: metav1.TypeMeta{
Kind: "Ingress",
APIVersion: "networking.k8s.io/v1",
},
Spec: netv1.IngressSpec{
IngressClassName: classToSet,
},
},
}
}
func (b *IngressBuilder) Build() *netv1.Ingress {
return &b.ingress
}
func (b *IngressBuilder) WithLegacyClassAnnotation(class string) *IngressBuilder {
b.ingress.Annotations[annotations.IngressClassKey] = class
return b
}
func (b *IngressBuilder) WithRules(rules ...netv1.IngressRule) *IngressBuilder {
b.ingress.Spec.Rules = append(b.ingress.Spec.Rules, rules...)
return b
}
func (b *IngressBuilder) WithNamespace(namespace string) *IngressBuilder {
b.ingress.ObjectMeta.Namespace = namespace
return b
}
func (b *IngressBuilder) WithAnnotations(annotations map[string]string) *IngressBuilder {
if b.ingress.Annotations == nil {
b.ingress.Annotations = annotations
return b
}
for k, v := range annotations {
b.ingress.Annotations[k] = v
}
return b
}
func (b *IngressBuilder) WithDefaultBackend(backend *netv1.IngressBackend) *IngressBuilder {
b.ingress.Spec.DefaultBackend = backend
return b
}