Skip to content

Commit cb93dc3

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 cb93dc3

8 files changed

Lines changed: 602 additions & 282 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/constants/constants.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ const (
1313

1414
WorkerPodVersionLabelPrefix = "beta.kmm.node.kubernetes.io/version-worker-pod"
1515
DevicePluginVersionLabelPrefix = "beta.kmm.node.kubernetes.io/version-device-plugin"
16+
DRAVersionLabelPrefix = "beta.kmm.node.kubernetes.io/version-dra"
1617
ModuleVersionLabelPrefix = "kmm.node.kubernetes.io/version-module"
1718

1819
GCDelayFinalizer = "kmm.node.kubernetes.io/gc-delay"

internal/controllers/device_plugin_pod_reconciler_test.go

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

0 commit comments

Comments
 (0)