Update bluesky_kafka/__init__.py

Prevent potential infinite recursion by checking page_len Co-authored-by: Thomas A Caswell <[email protected]>
bluesky · Mar 26, 2024 · bf89cfc · bf89cfc
1 parent 1bdc4b4
commit bf89cfc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bluesky_kafka/__init__.py b/bluesky_kafka/__init__.py
@@ -152,6 +152,8 @@ def __call__(self, name, doc):
             if KErr.code() == KafkaError.MSG_SIZE_TOO_LARGE:
                 if name == "event_page":
                     page_len = len(doc['seq_num'])
+                    if page_len == 1:
+                        raise
                     new_event_list = rechunk_event_pages([doc], (page_len + 1)//2)
                     for event in new_event_list:
                         self.__call__(name, event)