Skip to content

Commit 6429c7f

Browse files
killianmuldoondpf-release
authored andcommitted
fix: watch dpunode
* fix: watch dpunode * fix: watch dpunode fix: watch dpunode Signed-off-by: Peng Wang <[email protected]> (cherry picked from commit 52f5de77e1b5f92d517b96e5c885b5516f419a22) af51c92c fix: watch dpunode Co-authored-by: Peng Wang (Networking SW) <[email protected]> Based on release-v25.7@f84f0d80d4bad3523897e2bc4344bdf2b2ab7285
1 parent 91072fc commit 6429c7f

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

internal/provisioning/controllers/dpuset/dpuset_controller.go

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,8 @@ func (r *DPUSetReconciler) Reconcile(ctx context.Context, req ctrl.Request) (_ c
102102
}
103103

104104
func (r *DPUSetReconciler) Handle(ctx context.Context, dpuSet *provisioningv1.DPUSet) (ctrl.Result, error) {
105+
logger := log.FromContext(ctx)
106+
105107
// Add finalizer if not set.
106108
if !controllerutil.ContainsFinalizer(dpuSet, provisioningv1.DPUSetFinalizer) {
107109
controllerutil.AddFinalizer(dpuSet, provisioningv1.DPUSetFinalizer)
@@ -113,6 +115,7 @@ func (r *DPUSetReconciler) Handle(ctx context.Context, dpuSet *provisioningv1.DP
113115
if err != nil {
114116
return ctrl.Result{}, fmt.Errorf("failed to get DPUDevice map %w", err)
115117
}
118+
logger.Info(fmt.Sprintf("DPUSet %s/%s selected %d DPUDevices", dpuSet.Namespace, dpuSet.Name, len(dpuDeviceMap)))
116119

117120
// Get dpu map which are owned by dpuset
118121
dpuMap, err := r.getDPUsMap(ctx, dpuSet)
@@ -158,13 +161,15 @@ func (r *DPUSetReconciler) SetupWithManager(mgr ctrl.Manager) error {
158161
For(&provisioningv1.DPUSet{}).
159162
Owns(&provisioningv1.DPU{}).
160163
Watches(&provisioningv1.DPUDevice{},
161-
handler.EnqueueRequestsFromMapFunc(r.dpuDeviceToDPUSetReq)).
164+
handler.EnqueueRequestsFromMapFunc(r.resourceToDPUSetReq)).
165+
Watches(&provisioningv1.DPUNode{},
166+
handler.EnqueueRequestsFromMapFunc(r.resourceToDPUSetReq)).
162167
Watches(&provisioningv1.DPUFlavor{},
163168
handler.EnqueueRequestsFromMapFunc(r.flavorToDPUSetReq)).
164169
Complete(r)
165170
}
166171

167-
func (r *DPUSetReconciler) dpuDeviceToDPUSetReq(ctx context.Context, resource client.Object) []reconcile.Request {
172+
func (r *DPUSetReconciler) resourceToDPUSetReq(ctx context.Context, resource client.Object) []reconcile.Request {
168173
requests := make([]reconcile.Request, 0)
169174
dpuSetList := &provisioningv1.DPUSetList{}
170175
if err := r.List(ctx, dpuSetList); err == nil {

0 commit comments

Comments
 (0)