1 files changed, 8 insertions, 11 deletions
diff --git a/vendor/cloud.google.com/go/storage/invoke.go b/vendor/cloud.google.com/go/storage/invoke.go
index 03b98f4..e8fc924 100644
--- a/vendor/cloud.google.com/go/storage/invoke.go
+++ b/vendor/cloud.google.com/go/storage/invoke.go
@@ -15,6 +15,7 @@
 package storage
 
 import (
+	"cloud.google.com/go/internal"
 	gax "github.com/googleapis/gax-go"
 	"golang.org/x/net/context"
 	"google.golang.org/api/googleapi"
@@ -23,24 +24,20 @@ import (
 // runWithRetry calls the function until it returns nil or a non-retryable error, or
 // the context is done.
 func runWithRetry(ctx context.Context, call func() error) error {
-	var backoff gax.Backoff // use defaults for gax exponential backoff
-	for {
-		err := call()
+	return internal.Retry(ctx, gax.Backoff{}, func() (stop bool, err error) {
+		err = call()
 		if err == nil {
-			return nil
+			return true, nil
 		}
 		e, ok := err.(*googleapi.Error)
 		if !ok {
-			return err
+			return true, err
 		}
 		// Retry on 429 and 5xx, according to
 		// https://cloud.google.com/storage/docs/exponential-backoff.
 		if e.Code == 429 || (e.Code >= 500 && e.Code < 600) {
-			if err := gax.Sleep(ctx, backoff.Pause()); err != nil {
-				return err
-			}
-			continue
+			return false, nil
 		}
-		return err
-	}
+		return true, err
+	})
 }