fix: get disk stuck issue

kubernetes-sigs · Jan 3, 2025 · a009c9b · a009c9b
1 parent 0de3521
commit a009c9b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/azuredisk/azuredisk.go b/pkg/azuredisk/azuredisk.go
@@ -410,7 +410,12 @@ func (d *Driver) checkDiskExists(ctx context.Context, diskURI string) (*armcompu
 	if err != nil {
 		return nil, err
 	}
-	disk, err := diskClient.Get(ctx, resourceGroup, diskName)
+
+	// get disk operation should timeout within 1min if it takes too long time
+	newCtx, cancel := context.WithTimeout(ctx, time.Minute)
+	defer cancel()
+
+	disk, err := diskClient.Get(newCtx, resourceGroup, diskName)
 	if err != nil {
 		return nil, err
 	}