forked from DataDog/chaos-controller
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
126 lines (105 loc) · 4.03 KB
/
main.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
// Unless explicitly stated otherwise all files in this repository are licensed
// under the Apache License Version 2.0.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2021 Datadog, Inc.
/*
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package main
import (
"os"
chaosv1beta1 "github.com/DataDog/chaos-controller/api/v1beta1"
"github.com/DataDog/chaos-controller/controllers"
"github.com/DataDog/chaos-controller/log"
"github.com/DataDog/chaos-controller/metrics"
"github.com/DataDog/chaos-controller/metrics/types"
"github.com/spf13/pflag"
"k8s.io/apimachinery/pkg/runtime"
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
_ "k8s.io/client-go/plugin/pkg/client/auth/gcp"
ctrl "sigs.k8s.io/controller-runtime"
// +kubebuilder:scaffold:imports
)
var (
scheme = runtime.NewScheme()
setupLog = ctrl.Log.WithName("setup")
)
func init() {
// +kubebuilder:scaffold:scheme
_ = clientgoscheme.AddToScheme(scheme)
_ = chaosv1beta1.AddToScheme(scheme)
}
func main() {
var (
metricsAddr string
enableLeaderElection bool
deleteOnly bool
sink string
injectorAnnotations map[string]string
)
pflag.StringVar(&metricsAddr, "metrics-addr", ":8080", "The address the metric endpoint binds to.")
pflag.BoolVar(&enableLeaderElection, "enable-leader-election", false, "Enable leader election for controller manager. Enabling this will ensure there is only one active controller manager.")
pflag.BoolVar(&deleteOnly, "delete-only", false,
"Enable delete only mode which will not allow new disruption to start and will only continue to clean up and remove existing disruptions.")
pflag.StringToStringVar(&injectorAnnotations, "injector-annotations", map[string]string{}, "Annotations added to the generated injector pods")
pflag.StringVar(&sink, "metrics-sink", "noop", "Metrics sink (datadog, or noop)")
pflag.Parse()
logger, err := log.NewZapLogger()
if err != nil {
setupLog.Error(err, "error creating controller logger")
os.Exit(1)
}
mgr, err := ctrl.NewManager(ctrl.GetConfigOrDie(), ctrl.Options{
Scheme: scheme,
MetricsBindAddress: metricsAddr,
LeaderElection: enableLeaderElection,
LeaderElectionID: "75ec2fa4.datadoghq.com",
Port: 9443,
})
if err != nil {
setupLog.Error(err, "unable to start manager")
os.Exit(1)
}
// metrics sink
ms, err := metrics.GetSink(types.SinkDriver(sink), types.SinkAppController)
if err != nil {
ctrl.Log.Error(err, "error while creating metric sink")
}
// handle metrics sink client close on exit
defer func() {
ctrl.Log.Info("closing metrics sink client before exiting", "sink", ms.GetSinkName())
if err := ms.Close(); err != nil {
ctrl.Log.Error(err, "error closing metrics sink client", "sink", ms.GetSinkName())
}
}()
// create reconciler
r := &controllers.DisruptionReconciler{
Client: mgr.GetClient(),
BaseLog: logger,
Scheme: mgr.GetScheme(),
Recorder: mgr.GetEventRecorderFor("disruption-controller"),
MetricsSink: ms,
TargetSelector: controllers.RunningTargetSelector{},
DeleteOnly: deleteOnly,
InjectorAnnotations: injectorAnnotations,
}
if err := r.SetupWithManager(mgr); err != nil {
setupLog.Error(err, "unable to create controller", "controller", "Disruption")
os.Exit(1)
}
go r.ReportMetrics()
// +kubebuilder:scaffold:builder
setupLog.Info("starting manager")
if err := mgr.Start(ctrl.SetupSignalHandler()); err != nil {
setupLog.Error(err, "problem running manager")
os.Exit(1)
}
}