fix: include invocation-id for resumable PUTs (#2047)

BenWhitehead · web-flow · commit a590ae3856fe · 2023-06-05T14:46:55.000-04:00
Change BlobWriteChannel to us Retrying.run so it will receive an invocation id
diff --git a/google-cloud-storage/src/main/java/com/google/cloud/storage/BlobWriteChannel.java b/google-cloud-storage/src/main/java/com/google/cloud/storage/BlobWriteChannel.java
@@ -16,7 +16,6 @@
 
 package com.google.cloud.storage;
 
-import static com.google.cloud.RetryHelper.runWithRetries;
 import static java.util.Objects.requireNonNull;
 import static java.util.concurrent.Executors.callable;
 
@@ -28,6 +27,7 @@
 import com.google.cloud.RetryHelper;
 import com.google.cloud.WriteChannel;
 import java.math.BigInteger;
+import java.util.function.Function;
 import java.util.function.Supplier;
 import org.checkerframework.checker.nullness.qual.NonNull;
 
@@ -186,7 +186,9 @@ private static StorageException unrecoverableState(
   @Override
   protected void flushBuffer(final int length, final boolean lastChunk) {
     try {
-      runWithRetries(
+      Retrying.run(
+          getOptions(),
+          algorithmForWrite,
           callable(
               new Runnable() {
                 @Override
@@ -274,9 +276,7 @@ public void run() {
                   }
                 }
               }),
-          getOptions().getRetrySettings(),
-          algorithmForWrite,
-          getOptions().getClock());
+          Function.identity());
     } catch (RetryHelper.RetryHelperException e) {
       throw StorageException.translateAndThrow(e);
     }
