-
Notifications
You must be signed in to change notification settings - Fork 462
/
controller.go
70 lines (58 loc) · 2.38 KB
/
controller.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
// SPDX-FileCopyrightText: 2024 SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0
package heartbeat
import (
"context"
"k8s.io/utils/clock"
"sigs.k8s.io/controller-runtime/pkg/controller"
"sigs.k8s.io/controller-runtime/pkg/manager"
"github.com/gardener/gardener/pkg/controllerutils"
)
// ControllerName is the name of the controller.
const ControllerName = "heartbeat"
// DefaultAddOptions are the default AddOptions for AddToManager.
var DefaultAddOptions = AddOptions{
RenewIntervalSeconds: 30,
}
// AddOptions are options to apply when adding the heartbeat controller to the manager.
type AddOptions struct {
// ExtensionName is the name of the extension that this heartbeat controller is part of.
ExtensionName string
// Namespace is the namespace which will be used for the heartbeat lease resource.
Namespace string
// RenewIntervalSeconds defines how often the heartbeat lease is renewed.
RenewIntervalSeconds int32
}
// AddToManager adds the heartbeat controller with the default Options to the manager.
func AddToManager(_ context.Context, mgr manager.Manager) error {
return Add(mgr, AddArgs{
ExtensionName: DefaultAddOptions.ExtensionName,
Namespace: DefaultAddOptions.Namespace,
RenewIntervalSeconds: DefaultAddOptions.RenewIntervalSeconds,
Clock: clock.RealClock{},
})
}
// AddArgs are arguments for adding a heartbeat controller to a manager.
type AddArgs struct {
// ControllerOptions are the controller.Options.
ControllerOptions controller.Options
// ExtensionName is the name of the extension controller.
ExtensionName string
// Namespace is the namespace which will be used for the heartbeat lease resource.
Namespace string
// RenewIntervalSeconds defines how often the heartbeat lease is renewed.
RenewIntervalSeconds int32
// Clock is the clock to use when renewing the heartbeat lease resource.
Clock clock.Clock
}
// Add creates a new heartbeat controller and adds it to the given manager.
func Add(mgr manager.Manager, args AddArgs) error {
args.ControllerOptions.Reconciler = NewReconciler(mgr, args.ExtensionName, args.Namespace, args.RenewIntervalSeconds, args.Clock)
args.ControllerOptions.MaxConcurrentReconciles = 1
ctrl, err := controller.New(ControllerName, mgr, args.ControllerOptions)
if err != nil {
return err
}
return ctrl.Watch(controllerutils.EnqueueOnce, nil)
}