-
Notifications
You must be signed in to change notification settings - Fork 9
/
wire.go
100 lines (83 loc) · 3.08 KB
/
wire.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
// Copyright 2022 Namespace Labs Inc; All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
package controllers
import (
"context"
"flag"
"fmt"
"log"
corev1 "k8s.io/api/core/v1"
"namespacelabs.dev/foundation/framework/kubernetes/kubedef"
"namespacelabs.dev/foundation/std/go/server"
controllerruntime "sigs.k8s.io/controller-runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/healthz"
"sigs.k8s.io/controller-runtime/pkg/predicate"
)
var (
// Kubernetes controller settings.
controllerPort = flag.Int("controller_port", 18443, "Port that the Kubernetes controller binds to.")
metricsAddress = flag.String("controller_metrics_address", "0.0.0.0:18080",
"Address port pair that the Kubernetes controller metrics endpoint binds to.")
probeAddress = flag.String("controller_health_probe_bind_address", "0.0.0.0:18081",
"Address port pair that the Kubernetes controller health probe endpoint binds to.")
enableLeaderElection = flag.Bool("controller_enable_leader_election", false,
"Enable leader election for the Kubernetes controller manager, with true guaranteeing only one active controller manager.")
)
func WireService(context.Context, server.Registrar, ServiceDeps) {
if err := setupControllers(); err != nil {
log.Fatal(err)
}
}
func setupControllers() error {
mgr, err := controllerruntime.NewManager(controllerruntime.GetConfigOrDie(), controllerruntime.Options{
MetricsBindAddress: *metricsAddress,
Port: *controllerPort,
HealthProbeBindAddress: *probeAddress,
LeaderElection: *enableLeaderElection,
LeaderElectionID: "64367099.k8s.namespacelabs.dev",
})
if err := mgr.AddHealthzCheck("healthz", healthz.Ping); err != nil {
return fmt.Errorf("failed to set up healthz: %+v", err)
}
if err := mgr.AddReadyzCheck("readyz", healthz.Ping); err != nil {
return fmt.Errorf("failed to set up readyz: %+v", err)
}
if err != nil {
return fmt.Errorf("failed to start the controller manager: %+v", err)
}
if err := controllerruntime.NewControllerManagedBy(mgr).
For(&corev1.Pod{}).
Owns(&corev1.ConfigMap{}).
WithEventFilter(predicate.NewPredicateFuncs(managedByUs)).
Complete(&RuntimeConfigReconciler{
client: mgr.GetClient(),
}); err != nil {
return err
}
if err := controllerruntime.NewControllerManagedBy(mgr).
For(&corev1.ConfigMap{}).
Owns(&corev1.ConfigMap{}).
WithEventFilter(predicate.NewPredicateFuncs(managedByUs)).
Complete(&RuntimeConfigGC{
client: mgr.GetClient(),
recorder: mgr.GetEventRecorderFor("runtimeconfig-gc"),
}); err != nil {
return err
}
go func() {
// XXX we don't have a good way to model background work.
// Do not use the received context, as that has a timeout built-in.
if err := mgr.Start(context.Background()); err != nil {
log.Fatal(err)
}
}()
return nil
}
func managedByUs(obj client.Object) bool {
if mgr, ok := obj.GetLabels()[kubedef.AppKubernetesIoManagedBy]; ok {
return mgr == kubedef.ManagerId
}
return false
}