fix: Add missing monitor from startSubscriber. (#142)

googleapis · Jun 16, 2020 · d2a90d8 · d2a90d8
1 parent bcfd538
commit d2a90d8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...e/src/main/java/com/google/cloud/pubsublite/cloudpubsub/internal/AssigningSubscriber.java b/...e/src/main/java/com/google/cloud/pubsublite/cloudpubsub/internal/AssigningSubscriber.java
@@ -107,7 +107,9 @@ public void failed(State from, Throwable failure) {
 
           @Override
           public void terminated(State from) {
-            stoppingSubscribers.remove(subscriber);
+            try (CloseableMonitor.Hold h = monitor.enter()) {
+              stoppingSubscribers.remove(subscriber);
+            }
           }
         },
         MoreExecutors.directExecutor());