-
Notifications
You must be signed in to change notification settings - Fork 462
/
options.go
69 lines (57 loc) · 2.07 KB
/
options.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
// SPDX-FileCopyrightText: 2024 SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0
package cmd
import (
"fmt"
"github.com/spf13/pflag"
"github.com/gardener/gardener/extensions/pkg/controller/heartbeat"
)
// Options are command line options that can be set for the heartbeat controller.
type Options struct {
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
config *Config
}
// AddFlags implements Flagger.AddFlags.
func (o *Options) AddFlags(fs *pflag.FlagSet) {
fs.StringVar(&o.Namespace, "namespace", o.Namespace, "The namespace to use for the heartbeat lease resource.")
fs.Int32Var(&o.RenewIntervalSeconds, "renew-interval-seconds", 30, "How often the heartbeat lease will be renewed. Default is 30 seconds.")
}
// Validate validates the options.
func (o *Options) Validate() error {
if o.RenewIntervalSeconds <= 0 {
return fmt.Errorf("--heartbeat-renew-interval-seconds must be greater than 0")
}
return nil
}
// Complete implements Completer.Complete.
func (o *Options) Complete() error {
o.config = &Config{
ExtensionName: o.ExtensionName,
Namespace: o.Namespace,
RenewIntervalSeconds: o.RenewIntervalSeconds,
}
return nil
}
// Completed returns the completed Config. Only call this if `Complete` was successful.
func (o *Options) Completed() *Config {
return o.config
}
// Config is a completed heartbeat controller configuration.
type Config struct {
ExtensionName string
// Namespace is the namespace which will be used for heartbeat lease resource.
Namespace string
// RenewIntervalSeconds defines how often the heartbeat lease is renewed.
RenewIntervalSeconds int32
}
// Apply sets the values of this Config in the given heartbeat.AddOptions.
func (c *Config) Apply(opts *heartbeat.AddOptions) {
opts.ExtensionName = c.ExtensionName
opts.Namespace = c.Namespace
opts.RenewIntervalSeconds = c.RenewIntervalSeconds
}