athenad: explicitly delete socket in getMessage (#34098)

explicitly delete socket in getMessage
This commit is contained in:
Dean Lee
2024-11-27 16:35:36 +08:00
committed by GitHub
parent 8f71d53eb0
commit b737e8472f
+8 -5
View File
@@ -309,13 +309,16 @@ def getMessage(service: str, timeout: int = 1000) -> dict:
raise Exception("invalid service")
socket = messaging.sub_sock(service, timeout=timeout)
ret = messaging.recv_one(socket)
try:
ret = messaging.recv_one(socket)
if ret is None:
raise TimeoutError
if ret is None:
raise TimeoutError
# this is because capnp._DynamicStructReader doesn't have typing information
return cast(dict, ret.to_dict())
# this is because capnp._DynamicStructReader doesn't have typing information
return cast(dict, ret.to_dict())
finally:
del socket
@dispatcher.add_method