Set WorkScheduler in controllers

hobbyfarm · Jun 7, 2024 · 3daf235 · 3daf235
1 parent 31918e2
commit 3daf235
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/v3/services/sessionsvc/internal/controller.go b/v3/services/sessionsvc/internal/controller.go
@@ -55,6 +55,7 @@ func NewSessionController(
 		vmClaimClient:               vmClaimClient,
 	}
 	sessionController.SetReconciler(sessionController)
+	sessionController.SetWorkScheduler(sessionController)
 
 	return sessionController, nil
 }

diff --git a/v3/services/vmclaimsvc/internal/controller.go b/v3/services/vmclaimsvc/internal/controller.go
@@ -85,6 +85,7 @@ func NewVMClaimController(
 		vmTemplateClient:            vmTemplateClient,
 	}
 	vmClaimController.SetReconciler(vmClaimController)
+	vmClaimController.SetWorkScheduler(vmClaimController)
 
 	return vmClaimController, nil
 }

diff --git a/v3/services/vmsetsvc/internal/controller.go b/v3/services/vmsetsvc/internal/controller.go
@@ -66,6 +66,7 @@ func NewVMSetController(
 		vmTemplateClient:            vmTemplateClient,
 	}
 	vmSetController.SetReconciler(vmSetController)
+	vmSetController.SetWorkScheduler(vmSetController)
 
 	return vmSetController, nil
 }

diff --git a/v3/services/vmsvc/internal/controller.go b/v3/services/vmsvc/internal/controller.go
@@ -81,6 +81,7 @@ func NewVMController(
 		vmTemplateClient:            vmTemplateClient,
 	}
 	vmController.SetReconciler(vmController)
+	vmController.SetWorkScheduler(vmController)
 
 	return vmController, nil
 }