Patch in #15515

apache · Sep 28, 2021 · d60e52f · d60e52f
1 parent 1abd6ba
commit d60e52f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/...form/src/main/java/org/apache/beam/sdk/io/gcp/pubsublite/SubscriptionPartitionLoader.java b/...form/src/main/java/org/apache/beam/sdk/io/gcp/pubsublite/SubscriptionPartitionLoader.java
@@ -92,9 +92,7 @@ public PollResult<Partition> apply(TopicPath element, Context c) {
                     })
                 .withPollInterval(pollDuration)
                 .withTerminationPerInput(
-                    terminate
-                        ? Watch.Growth.afterTotalOf(pollDuration.multipliedBy(10))
-                        : Watch.Growth.never()));
+                    terminate ? Watch.Growth.afterIterations(10) : Watch.Growth.never()));
     return partitions.apply(
         MapElements.into(TypeDescriptor.of(SubscriptionPartition.class))
             .via(kv -> SubscriptionPartition.of(subscription, kv.getValue())));