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: