-
Notifications
You must be signed in to change notification settings - Fork 26
/
main.go
148 lines (122 loc) · 4.31 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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2020-2023 Intel Corporation
package main
import (
"flag"
"os"
"syscall"
"github.com/go-logr/logr"
"github.com/google/uuid"
"github.com/smart-edge-open/sriov-fec-operator/pkg/common/drainhelper"
"github.com/smart-edge-open/sriov-fec-operator/pkg/common/utils"
"k8s.io/apimachinery/pkg/types"
sriovv2 "github.com/smart-edge-open/sriov-fec-operator/api/sriovfec/v2"
vrbv1 "github.com/smart-edge-open/sriov-fec-operator/api/sriovvrb/v1"
"github.com/smart-edge-open/sriov-fec-operator/pkg/daemon"
"k8s.io/apimachinery/pkg/runtime"
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
clientset "k8s.io/client-go/kubernetes"
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
ctrl "sigs.k8s.io/controller-runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
)
var (
scheme = runtime.NewScheme()
setupLog = utils.NewLogger()
)
func init() {
utilruntime.Must(clientgoscheme.AddToScheme(scheme))
utilruntime.Must(sriovv2.AddToScheme(scheme))
utilruntime.Must(vrbv1.AddToScheme(scheme))
}
func main() {
syscall.Umask(0077)
ctrl.SetLogger(logr.New(utils.NewLogWrapper()))
nodeName := getNodeNameFromEnvOrDie()
ns := getSriovFecNameSpaceFromEnvOrDie()
config := ctrl.GetConfigOrDie()
directClient, err := client.New(config, client.Options{Scheme: scheme})
if err != nil {
setupLog.WithError(err).Error("failed to create direct client")
os.Exit(1)
}
pfBbConfigCliCmd := flag.String("C", "", "CLI command string")
flag.Usage = func() {
daemon.ShowHelp()
}
flag.Parse()
if *pfBbConfigCliCmd != "" {
// Get the additional arguments after CLI command
args := flag.Args()
daemon.StartPfBbConfigCli(nodeName, ns, directClient, *pfBbConfigCliCmd, args, setupLog)
return
}
cset, err := clientset.NewForConfig(config)
if err != nil {
setupLog.WithError(err).Error("failed to create clientset")
os.Exit(1)
}
mgr, err := daemon.CreateManager(config, scheme, ns, 8080, 8081, setupLog)
if err != nil {
setupLog.WithError(err).Error("unable to start manager")
os.Exit(1)
}
daemon.StartTelemetryDaemon(mgr, nodeName, ns, directClient, setupLog)
vfioTokenBytes, err := os.ReadFile("/sriov_config/vfiotoken")
if err != nil {
setupLog.Error(err)
os.Exit(1)
}
vfioToken, err := uuid.ParseBytes(vfioTokenBytes)
if err != nil {
setupLog.Errorf("provided vfioToken(%s) is not in UUID format: %s", vfioTokenBytes, err)
os.Exit(1)
}
isSingleNodeCluster, err := utils.IsSingleNodeCluster(directClient)
if err != nil {
setupLog.WithError(err).Errorf("failed to determine cluster type")
os.Exit(1)
}
nodeNameRef := types.NamespacedName{Namespace: ns, Name: nodeName}
drainHelper := drainhelper.NewDrainHelper(utils.NewLogger(), cset, nodeName, ns, isSingleNodeCluster)
pfBBConfigController := daemon.NewPfBBConfigController(utils.NewLogger(), vfioToken.String())
nodeConfigurer := daemon.NewNodeConfigurator(utils.NewLogger(), pfBBConfigController, mgr.GetClient(), nodeNameRef)
devicePluginController := daemon.NewDevicePluginController(mgr.GetClient(), utils.NewLogger(), nodeNameRef)
reconciler, err := daemon.NewNodeConfigReconciler(mgr.GetClient(), drainHelper.Run, nodeNameRef, nodeConfigurer, nodeConfigurer, devicePluginController.RestartDevicePlugin)
if err != nil {
setupLog.WithError(err).Error("unable to create reconciler")
os.Exit(1)
}
if err := reconciler.SetupWithManager(mgr); err != nil {
setupLog.WithError(err).Error("unable to create controller for SrionvFecNodeConfig")
os.Exit(1)
}
if err := reconciler.VrbSetupWithManager(mgr); err != nil {
setupLog.WithError(err).Error("unable to create controller for SriovVrbNodeConfig")
os.Exit(1)
}
if err := reconciler.CreateEmptyNodeConfigIfNeeded(directClient); err != nil {
setupLog.WithError(err).Error("failed to create initial NodeConfig CR")
os.Exit(1)
}
if err := mgr.Start(ctrl.SetupSignalHandler()); err != nil {
setupLog.WithError(err).Error("problem running manager")
os.Exit(1)
}
}
func getSriovFecNameSpaceFromEnvOrDie() string {
ns := os.Getenv("SRIOV_FEC_NAMESPACE")
if ns == "" {
setupLog.Error("SRIOV_FEC_NAMESPACE environment variable is empty")
os.Exit(1)
}
return ns
}
func getNodeNameFromEnvOrDie() string {
nodeName := os.Getenv("NODENAME")
if nodeName == "" {
setupLog.Error("NODENAME environment variable is empty")
os.Exit(1)
}
return nodeName
}