-
Notifications
You must be signed in to change notification settings - Fork 830
/
main.go
85 lines (72 loc) · 3.07 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
/*
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 (
"flag"
"fmt"
"github.com/awslabs/karpenter/pkg/apis"
"github.com/awslabs/karpenter/pkg/cloudprovider"
"github.com/awslabs/karpenter/pkg/cloudprovider/registry"
"github.com/awslabs/karpenter/pkg/controllers"
"github.com/awslabs/karpenter/pkg/controllers/allocation"
"github.com/awslabs/karpenter/pkg/controllers/reallocation"
"github.com/awslabs/karpenter/pkg/controllers/termination"
"github.com/awslabs/karpenter/pkg/utils/log"
"go.uber.org/zap/zapcore"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/client-go/kubernetes"
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
_ "k8s.io/client-go/plugin/pkg/client/auth/gcp"
controllerruntime "sigs.k8s.io/controller-runtime"
controllerruntimezap "sigs.k8s.io/controller-runtime/pkg/log/zap"
)
var (
scheme = runtime.NewScheme()
options = Options{}
)
func init() {
_ = clientgoscheme.AddToScheme(scheme)
_ = apis.AddToScheme(scheme)
}
// Options for running this binary
type Options struct {
EnableVerboseLogging bool
MetricsPort int
HealthProbePort int
}
func main() {
flag.BoolVar(&options.EnableVerboseLogging, "verbose", false, "Enable verbose logging")
flag.IntVar(&options.MetricsPort, "metrics-port", 8080, "The port the metric endpoint binds to for operating metrics about the controller itself")
flag.IntVar(&options.HealthProbePort, "health-probe-port", 8081, "The port the health probe endpoint binds to for reporting controller health")
flag.Parse()
log.Setup(
controllerruntimezap.UseDevMode(options.EnableVerboseLogging),
controllerruntimezap.ConsoleEncoder(),
controllerruntimezap.StacktraceLevel(zapcore.DPanicLevel),
)
manager := controllers.NewManagerOrDie(controllerruntime.GetConfigOrDie(), controllerruntime.Options{
LeaderElection: true,
LeaderElectionID: "karpenter-leader-election",
Scheme: scheme,
MetricsBindAddress: fmt.Sprintf(":%d", options.MetricsPort),
HealthProbeBindAddress: fmt.Sprintf(":%d", options.HealthProbePort),
})
clientSet := kubernetes.NewForConfigOrDie(manager.GetConfig())
cloudProvider := registry.NewCloudProvider(cloudprovider.Options{ClientSet: clientSet})
if err := manager.RegisterControllers(
allocation.NewController(manager.GetClient(), clientSet.CoreV1(), cloudProvider),
reallocation.NewController(manager.GetClient(), clientSet.CoreV1(), cloudProvider),
termination.NewController(manager.GetClient(), clientSet.CoreV1(), cloudProvider),
).Start(controllerruntime.SetupSignalHandler()); err != nil {
panic(fmt.Sprintf("Unable to start manager, %s", err.Error()))
}
}