@@ -321,7 +321,7 @@ func setupTopologyMutationHookHandlers(runtimeExtensionWebhookServer *server.Ser
321321func setupLifecycleHookHandlers (mgr ctrl.Manager , runtimeExtensionWebhookServer * server.Server ) {
322322 // Create the ExtensionHandlers for the lifecycle hooks
323323 // NOTE: it is not mandatory to group all the ExtensionHandlers using a struct, what is important
324- // is to have HandlerFunc with the signature defined in sigs.k8s.io/cluster-api/exp /runtime/hooks/api /v1alpha1.
324+ // is to have HandlerFunc with the signature defined in sigs.k8s.io/cluster-api/api /runtime/hooks/v1alpha1.
325325 lifecycleExtensionHandlers := lifecycle .NewExtensionHandlers (mgr .GetClient ())
326326
327327 if err := runtimeExtensionWebhookServer .AddExtensionHandler (server.ExtensionHandler {
@@ -351,6 +351,15 @@ func setupLifecycleHookHandlers(mgr ctrl.Manager, runtimeExtensionWebhookServer
351351 os .Exit (1 )
352352 }
353353
354+ if err := runtimeExtensionWebhookServer .AddExtensionHandler (server.ExtensionHandler {
355+ Hook : runtimehooksv1 .BeforeControlPlaneUpgrade ,
356+ Name : "before-control-plane-upgrade" ,
357+ HandlerFunc : lifecycleExtensionHandlers .DoBeforeControlPlaneUpgrade ,
358+ }); err != nil {
359+ setupLog .Error (err , "Error adding handler" )
360+ os .Exit (1 )
361+ }
362+
354363 if err := runtimeExtensionWebhookServer .AddExtensionHandler (server.ExtensionHandler {
355364 Hook : runtimehooksv1 .AfterControlPlaneUpgrade ,
356365 Name : "after-control-plane-upgrade" ,
@@ -360,6 +369,24 @@ func setupLifecycleHookHandlers(mgr ctrl.Manager, runtimeExtensionWebhookServer
360369 os .Exit (1 )
361370 }
362371
372+ if err := runtimeExtensionWebhookServer .AddExtensionHandler (server.ExtensionHandler {
373+ Hook : runtimehooksv1 .BeforeWorkersUpgrade ,
374+ Name : "before-workers-upgrade" ,
375+ HandlerFunc : lifecycleExtensionHandlers .DoBeforeWorkersUpgrade ,
376+ }); err != nil {
377+ setupLog .Error (err , "Error adding handler" )
378+ os .Exit (1 )
379+ }
380+
381+ if err := runtimeExtensionWebhookServer .AddExtensionHandler (server.ExtensionHandler {
382+ Hook : runtimehooksv1 .AfterWorkersUpgrade ,
383+ Name : "after-workers-upgrade" ,
384+ HandlerFunc : lifecycleExtensionHandlers .DoAfterWorkersUpgrade ,
385+ }); err != nil {
386+ setupLog .Error (err , "Error adding handler" )
387+ os .Exit (1 )
388+ }
389+
363390 if err := runtimeExtensionWebhookServer .AddExtensionHandler (server.ExtensionHandler {
364391 Hook : runtimehooksv1 .AfterClusterUpgrade ,
365392 Name : "after-cluster-upgrade" ,
0 commit comments