Skip to content

Commit 4298ef3

Browse files
committed
Add catch for stopped connection in producer
1 parent 03b6c8f commit 4298ef3

1 file changed

Lines changed: 10 additions & 1 deletion

File tree

varys/producer.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,17 @@ def publish_message(self, message, max_attempts=3):
4747

4848
attempt = 0
4949
while attempt < max_attempts:
50+
attempt += 1
51+
52+
if self._connection is None or self._connection.is_closed:
53+
self._log.warning(
54+
"Connection is closed, cannot publish message, attempting to reconnect..."
55+
)
56+
if self._reconnect_wait > 0:
57+
time.sleep(self._reconnect_wait)
58+
continue
59+
5060
try:
51-
attempt += 1
5261
self._log.info(
5362
f"Sending message (attempt {attempt}): {json.dumps(message)}"
5463
)

0 commit comments

Comments
 (0)