goodboy
/
tractor
1
0
Fork 1
Code Issues 1 Pull Requests 4 Packages Projects Releases Wiki Activity

Tolerate one decode error; may have been a registry ping

optional_msgspec_support
Tyler Goodlet 2021-09-30 11:32:28 -04:00
parent ef75883b62
commit 135459ca25
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tractor/_ipc.py
View File

@ -177,17 +177,19 @@ class MsgspecTCPStream(MsgpackTCPStream):
log.transport(f"received {msg_bytes}") # type: ignore log.transport(f"received {msg_bytes}") # type: ignore
try: try:
assert not last_decode_failed
yield self.decode(msg_bytes) yield self.decode(msg_bytes)
except ( except (
msgspec.DecodingError, msgspec.DecodingError,
UnicodeDecodeError, UnicodeDecodeError,
): ):
if not last_decode_failed:
# ignore decoding errors for now and assume they have to # ignore decoding errors for now and assume they have to
# do with a channel drop - hope that receiving from the # do with a channel drop - hope that receiving from the
# channel will raise an expected error and bubble up. # channel will raise an expected error and bubble up.
log.error('`msgspec` failed to decode!?') log.error('`msgspec` failed to decode!?')
last_decode_failed = True last_decode_failed = True
else:
raise
async def send(self, data: Any) -> None: async def send(self, data: Any) -> None:
async with self._send_lock: async with self._send_lock: