fix: Respect lock annotations in RetryingConnection (#669)

googleapis · Jun 8, 2021 · d583623 · d583623
1 parent 587c91c
commit d583623
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...blite/src/main/java/com/google/cloud/pubsublite/internal/wire/RetryingConnectionImpl.java b/...blite/src/main/java/com/google/cloud/pubsublite/internal/wire/RetryingConnectionImpl.java
@@ -84,10 +84,14 @@ class RetryingConnectionImpl<
 
   @Override
   protected void doStart() {
+    StreamRequestT initialInitialRequest;
+    try (CloseableMonitor.Hold h = connectionMonitor.enter()) {
+      initialInitialRequest = lastInitialRequest;
+    }
     SystemExecutors.getAlarmExecutor()
         .execute(
             () -> {
-              reinitialize(lastInitialRequest);
+              reinitialize(initialInitialRequest);
               notifyStarted();
             });
   }