Skip to content

Commit d2ff271

Browse files
committed
Add DRA node labeling via unified PodNodeLabel reconciler
Merge DevicePluginPodReconciler and the new DRA pod labeling logic into a single PodNodeLabelReconciler that handles both roles. The reconciler watches DaemonSet pods with a ModuleNameLabel, determines the role from the DaemonSetRole label, and manages the corresponding node label (device-plugin-ready or dra-ready) based on pod readiness.
1 parent 40f1f16 commit d2ff271

6 files changed

Lines changed: 550 additions & 278 deletions

File tree

cmd/manager/main.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,8 +166,8 @@ func main() {
166166
cmd.FatalError(setupLogger, err, "unable to create controller", "name", controllers.NodeModulesConfigReconcilerName)
167167
}
168168

169-
if err = controllers.NewDevicePluginPodReconciler(client).SetupWithManager(mgr); err != nil {
170-
cmd.FatalError(setupLogger, err, "unable to create controller", "name", controllers.DevicePluginPodReconcilerName)
169+
if err = controllers.NewPodNodeLabelReconciler(client).SetupWithManager(mgr); err != nil {
170+
cmd.FatalError(setupLogger, err, "unable to create controller", "name", controllers.PodNodeLabelReconcilerName)
171171
}
172172

173173
if err = controllers.NewDRAReconciler(client, nodeAPI, scheme).SetupWithManager(mgr); err != nil {

internal/controllers/device_plugin_pod_reconciler_test.go

Lines changed: 0 additions & 214 deletions
This file was deleted.

0 commit comments

Comments
 (0)