Add capture exception for CreateLogStream, to avoid flush stuck when …

…CreateLogStream throttle. (#192)
kislyuk · Mar 10, 2024 · 04edbfe · 04edbfe
1 parent f8e498e
commit 04edbfe
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/watchtower/__init__.py b/watchtower/__init__.py
@@ -370,6 +370,13 @@ def _submit_batch(self, batch, log_stream_name, max_retries=5):
                             # at this point, the first write to the new stream
                             # should not contain a sequence token at all.
                             kwargs.pop("sequenceToken", None)
+                        except Exception as e2:
+                            # Make sure exception in CreateLogStream not exit
+                            # this thread but conitnue to retry
+                            warnings.warn(
+                                "Failed to create log stream {} when delivering log: {}".format(log_stream_name, e2),
+                                WatchtowerWarning,
+                            )
                         finally:
                             self.creating_log_stream = False
                 else: