Skip to content

Commit 67708d2

Browse files
committed
MOD: Improve Live client error reporting
1 parent 0ca4bc3 commit 67708d2

File tree

2 files changed

+17
-13
lines changed

2 files changed

+17
-13
lines changed

CHANGELOG.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,10 @@
11
# Changelog
22

3+
## 0.44.1 - TBD
4+
5+
#### Enhancements
6+
- Improved exception messages emitted by the `Live` client to always include contents of any `ErrorMsg` sent by the gateway
7+
38
## 0.44.0 - 2024-10-22
49

510
#### Enhancements

databento/live/protocol.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -195,21 +195,20 @@ def connection_lost(self, exc: Exception | None) -> None:
195195
"""
196196
super().connection_lost(exc)
197197
if not self.disconnected.done():
198-
if exc is None:
199-
logger.info("connection closed")
200-
if self._error_msgs:
201-
error_msg = ", ".join(self._error_msgs)
202-
if len(self._error_msgs) > 1:
203-
error_msg = f"The following errors occurred: {error_msg}"
204-
self._error_msgs.clear()
205-
self.disconnected.set_exception(
206-
BentoError(error_msg),
207-
)
208-
else:
209-
self.disconnected.set_result(None)
210-
else:
198+
if self._error_msgs:
199+
error_msg = ", ".join(self._error_msgs)
200+
if len(self._error_msgs) > 1:
201+
error_msg = f"The following errors occurred: {error_msg}"
202+
self._error_msgs.clear()
203+
204+
logger.error("gateway error: %s", exc)
205+
self.disconnected.set_exception(BentoError(error_msg))
206+
elif exc is not None:
211207
logger.error("connection lost: %s", exc)
212208
self.disconnected.set_exception(exc)
209+
else:
210+
logger.info("connection closed")
211+
self.disconnected.set_result(None)
213212

214213
def eof_received(self) -> bool | None:
215214
"""

0 commit comments

Comments
 (0)