add timeout for resolveTarget

Signed-off-by: Xuhui zhang <[email protected]>
juicedata · Nov 1, 2024 · 22d3e81 · 22d3e81
1 parent ac303d2
commit 22d3e81
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/pkg/controller/pod_driver.go b/pkg/controller/pod_driver.go
@@ -593,12 +593,15 @@ func (p *PodDriver) recover(ctx context.Context, pod *corev1.Pod, mntPath string
 	}
 	for k, target := range pod.Annotations {
 		if k == util.GetReferenceKey(target) {
-			mi := p.mit.resolveTarget(ctx, target)
-			if mi == nil {
-				log.Info("pod target resolve fail", "target", target)
+			var mi *mountItem
+			err := util.DoWithTimeout(ctx, defaultCheckoutTimeout, func() error {
+				mi = p.mit.resolveTarget(ctx, target)
+				return nil
+			})
+			if err != nil || mi == nil {
+				log.Info("pod target resolve fail", "target", target, "err", err)
 				continue
 			}
-
 			if err := p.recoverTarget(ctx, pod.Name, mntPath, mi.baseTarget, mi); err != nil {
 				return err
 			}