fix: batch at most 1,000 ack ids per request

googleapis · Oct 8, 2022 · be73324 · be73324
1 parent 09a409c
commit be73324
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 11 deletions.
diff --git a/google/cloud/pubsub_v1/subscriber/_protocol/dispatcher.py b/google/cloud/pubsub_v1/subscriber/_protocol/dispatcher.py
@@ -37,16 +37,8 @@
 """The maximum amount of time in seconds to wait for additional request items
 before processing the next batch of requests."""
 
-_ACK_IDS_BATCH_SIZE = 2500
+_ACK_IDS_BATCH_SIZE = 1000
 """The maximum number of ACK IDs to send in a single StreamingPullRequest.
-
-The backend imposes a maximum request size limit of 524288 bytes (512 KiB) per
-acknowledge / modifyAckDeadline request. ACK IDs have a maximum size of 164
-bytes, thus we cannot send more than o 524288/176 ~= 2979 ACK IDs in a single
-StreamingPullRequest message.
-
-Accounting for some overhead, we should thus only send a maximum of 2500 ACK
-IDs at a time.
 """
 
 

diff --git a/tests/unit/pubsub_v1/subscriber/test_dispatcher.py b/tests/unit/pubsub_v1/subscriber/test_dispatcher.py
@@ -120,7 +120,7 @@ def test_ack_splitting_large_payload():
     dispatcher_.ack(items)
 
     calls = manager.send.call_args_list
-    assert len(calls) == 3
+    assert len(calls) == 6
 
     all_ack_ids = {item.ack_id for item in items}
     sent_ack_ids = collections.Counter()
@@ -231,7 +231,7 @@ def test_modify_ack_deadline_splitting_large_payload():
     dispatcher_.modify_ack_deadline(items)
 
     calls = manager.send.call_args_list
-    assert len(calls) == 3
+    assert len(calls) == 6
 
     all_ack_ids = {item.ack_id for item in items}
     sent_ack_ids = collections.Counter()