@@ -22,26 +22,23 @@ import (
22
22
"os"
23
23
"strconv"
24
24
25
+ "github.com/kubernetes-sigs/kernel-module-management/internal/mbsc"
25
26
"github.com/kubernetes-sigs/kernel-module-management/internal/mic"
26
27
"github.com/kubernetes-sigs/kernel-module-management/internal/node"
27
28
"github.com/kubernetes-sigs/kernel-module-management/internal/pod"
28
29
29
30
"github.com/kubernetes-sigs/kernel-module-management/api/v1beta1"
30
31
"github.com/kubernetes-sigs/kernel-module-management/api/v1beta2"
31
- buildpod "github.com/kubernetes-sigs/kernel-module-management/internal/build/pod"
32
32
"github.com/kubernetes-sigs/kernel-module-management/internal/buildsign"
33
33
buildsignpod "github.com/kubernetes-sigs/kernel-module-management/internal/buildsign/pod"
34
34
"github.com/kubernetes-sigs/kernel-module-management/internal/cmd"
35
35
"github.com/kubernetes-sigs/kernel-module-management/internal/config"
36
- "github.com/kubernetes-sigs/kernel-module-management/internal/constants"
37
36
"github.com/kubernetes-sigs/kernel-module-management/internal/controllers"
38
37
"github.com/kubernetes-sigs/kernel-module-management/internal/filter"
39
38
"github.com/kubernetes-sigs/kernel-module-management/internal/metrics"
40
39
"github.com/kubernetes-sigs/kernel-module-management/internal/module"
41
40
"github.com/kubernetes-sigs/kernel-module-management/internal/nmc"
42
- "github.com/kubernetes-sigs/kernel-module-management/internal/preflight"
43
41
"github.com/kubernetes-sigs/kernel-module-management/internal/registry"
44
- signpod "github.com/kubernetes-sigs/kernel-module-management/internal/sign/pod"
45
42
46
43
"k8s.io/apimachinery/pkg/runtime"
47
44
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
@@ -124,6 +121,8 @@ func main() {
124
121
nodeAPI := node .NewNode (client )
125
122
kernelAPI := module .NewKernelMapper (buildSignHelperAPI )
126
123
micAPI := mic .New (client , scheme )
124
+ mbscAPI := mbsc .New (client , scheme )
125
+ imagePullerAPI := pod .NewImagePuller (client , scheme )
127
126
128
127
dpc := controllers .NewDevicePluginReconciler (
129
128
client ,
@@ -168,6 +167,10 @@ func main() {
168
167
cmd .FatalError (setupLogger , err , "unable to create controller" , "name" , controllers .NodeLabelModuleVersionReconcilerName )
169
168
}
170
169
170
+ if err = controllers .NewMICReconciler (client , micAPI , mbscAPI , imagePullerAPI , scheme ).SetupWithManager (mgr ); err != nil {
171
+ cmd .FatalError (setupLogger , err , "unable to create controller" , "name" , controllers .MICReconcilerName )
172
+ }
173
+
171
174
if managed {
172
175
setupLogger .Info ("Starting as managed" )
173
176
@@ -179,30 +182,11 @@ func main() {
179
182
cmd .FatalError (setupLogger , err , "unable to create controller" , "name" , controllers .NodeKernelClusterClaimReconcilerName )
180
183
}
181
184
} else {
182
- buildSignPodAPI := buildsignpod .NewBuildSignPodManager (client )
183
-
184
- buildAPI := buildpod .NewBuildManager (
185
- client ,
186
- buildpod .NewMaker (client , buildSignHelperAPI , buildSignPodAPI , scheme ),
187
- buildSignPodAPI ,
188
- registryAPI ,
189
- )
190
-
191
- signAPI := signpod .NewSignPodManager (
192
- client ,
193
- signpod .NewSigner (client , scheme , buildSignPodAPI ),
194
- buildSignPodAPI ,
195
- registryAPI ,
196
- )
197
- bsc := controllers .NewBuildSignReconciler (
198
- client ,
199
- buildAPI ,
200
- signAPI ,
201
- kernelAPI ,
202
- filterAPI ,
203
- nodeAPI )
204
- if err = bsc .SetupWithManager (mgr , constants .KernelLabel ); err != nil {
205
- cmd .FatalError (setupLogger , err , "unable to create controller" , "name" , controllers .BuildSignReconcilerName )
185
+ builSignAPI := buildsignpod .NewManager (client , buildSignHelperAPI , scheme )
186
+
187
+ mbscr := controllers .NewMBSCReconciler (client , builSignAPI , mbscAPI )
188
+ if err = mbscr .SetupWithManager (mgr ); err != nil {
189
+ cmd .FatalError (setupLogger , err , "unable to create controller" , "name" , controllers .MBSCReconcilerName )
206
190
}
207
191
208
192
helper := controllers .NewJobEventReconcilerHelper (client )
@@ -215,12 +199,15 @@ func main() {
215
199
cmd .FatalError (setupLogger , err , "unable to create controller" , "name" , controllers .JobGCReconcilerName )
216
200
}
217
201
218
- preflightStatusUpdaterAPI := preflight .NewStatusUpdater (client )
219
- preflightAPI := preflight .NewPreflightAPI (client , buildAPI , signAPI , registryAPI , preflightStatusUpdaterAPI , kernelAPI )
202
+ //[TODO] - update the preflight flow with the MIC/MBSC implementation and then uncomment the preflight conroller
203
+ /*
204
+ preflightStatusUpdaterAPI := preflight.NewStatusUpdater(client)
205
+ preflightAPI := preflight.NewPreflightAPI(client, buildAPI, signAPI, registryAPI, preflightStatusUpdaterAPI, kernelAPI)
220
206
221
- if err = controllers .NewPreflightValidationReconciler (client , filterAPI , metricsAPI , preflightStatusUpdaterAPI , preflightAPI ).SetupWithManager (mgr ); err != nil {
222
- cmd .FatalError (setupLogger , err , "unable to create controller" , "name" , controllers .PreflightValidationReconcilerName )
223
- }
207
+ if err = controllers.NewPreflightValidationReconciler(client, filterAPI, metricsAPI, preflightStatusUpdaterAPI, preflightAPI).SetupWithManager(mgr); err != nil {
208
+ cmd.FatalError(setupLogger, err, "unable to create controller", "name", controllers.PreflightValidationReconcilerName)
209
+ }
210
+ */
224
211
}
225
212
226
213
//+kubebuilder:scaffold:builder
0 commit comments