-
Notifications
You must be signed in to change notification settings - Fork 33
/
app.go
130 lines (106 loc) · 4.03 KB
/
app.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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
// SPDX-FileCopyrightText: 2024 SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0
package app
import (
"context"
"fmt"
"os"
"github.com/gardener/gardener/extensions/pkg/controller"
controllercmd "github.com/gardener/gardener/extensions/pkg/controller/cmd"
"github.com/gardener/gardener/extensions/pkg/controller/heartbeat"
heartbeatcmd "github.com/gardener/gardener/extensions/pkg/controller/heartbeat/cmd"
"github.com/spf13/cobra"
corev1 "k8s.io/api/core/v1"
"k8s.io/component-base/version/verflag"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/manager"
gvisorcontroller "github.com/gardener/gardener-extension-runtime-gvisor/pkg/controller"
"github.com/gardener/gardener-extension-runtime-gvisor/pkg/gvisor"
"github.com/gardener/gardener-extension-runtime-gvisor/pkg/healthcheck"
)
// NewControllerManagerCommand creates a new command that is used to start the Container runtime gvisor controller.
func NewControllerManagerCommand(ctx context.Context) *cobra.Command {
var (
generalOpts = &controllercmd.GeneralOptions{}
restOpts = &controllercmd.RESTOptions{}
mgrOpts = &controllercmd.ManagerOptions{
LeaderElection: true,
LeaderElectionID: controllercmd.LeaderElectionNameID(gvisor.Name),
LeaderElectionNamespace: os.Getenv("LEADER_ELECTION_NAMESPACE"),
}
reconcileOpts = &controllercmd.ReconcilerOptions{
IgnoreOperationAnnotation: true,
}
// options for the runtime-gvisor controller
gvisorCtrlOpts = &controllercmd.ControllerOptions{
MaxConcurrentReconciles: 5,
}
// options for the health care controller
healthCheckCtrlOpts = &controllercmd.ControllerOptions{
MaxConcurrentReconciles: 5,
}
// options for the heartbeat controller
heartbeatCtrlOpts = &heartbeatcmd.Options{
ExtensionName: gvisor.Name,
RenewIntervalSeconds: 30,
Namespace: os.Getenv("LEADER_ELECTION_NAMESPACE"),
}
aggOption = controllercmd.NewOptionAggregator(
generalOpts,
restOpts,
mgrOpts,
gvisorCtrlOpts,
controllercmd.PrefixOption("healthcheck-", healthCheckCtrlOpts),
controllercmd.PrefixOption("heartbeat-", heartbeatCtrlOpts),
reconcileOpts,
)
)
cmd := &cobra.Command{
Use: fmt.Sprintf("%s-controller-manager", gvisor.Name),
RunE: func(cmd *cobra.Command, args []string) error {
// Act on version flag, if one was specified
verflag.PrintAndExitIfRequested()
if err := aggOption.Complete(); err != nil {
return fmt.Errorf("error completing options: %w", err)
}
if err := heartbeatCtrlOpts.Validate(); err != nil {
return err
}
completedMgrOpts := mgrOpts.Completed().Options()
completedMgrOpts.Client = client.Options{
Cache: &client.CacheOptions{
DisableFor: []client.Object{
&corev1.Secret{}, // applied for ManagedResources
},
},
}
mgr, err := manager.New(restOpts.Completed().Config, completedMgrOpts)
if err != nil {
return fmt.Errorf("could not instantiate manager: %w", err)
}
if err := controller.AddToScheme(mgr.GetScheme()); err != nil {
return fmt.Errorf("could not update manager scheme: %w", err)
}
reconcileOpts.Completed().Apply(&gvisorcontroller.DefaultAddOptions.IgnoreOperationAnnotation)
gvisorCtrlOpts.Completed().Apply(&gvisorcontroller.DefaultAddOptions.Controller)
heartbeatCtrlOpts.Completed().Apply(&heartbeat.DefaultAddOptions)
if err := gvisorcontroller.AddToManager(ctx, mgr); err != nil {
return fmt.Errorf("could not add controllers to manager: %w", err)
}
if err := healthcheck.AddToManager(ctx, mgr); err != nil {
return fmt.Errorf("could not add health check controller to manager: %w", err)
}
if err := heartbeat.AddToManager(ctx, mgr); err != nil {
return fmt.Errorf("could not add heartbeat controller to manager: %w", err)
}
if err := mgr.Start(ctx); err != nil {
return fmt.Errorf("error running manager: %w", err)
}
return nil
},
}
verflag.AddFlags(cmd.Flags())
aggOption.AddFlags(cmd.Flags())
return cmd
}