jc211
/
tractor
forked from goodboy/tractor
1
0
Fork 0
Code Pull Requests Activity

Only close recv chan if we get a ref

transport_hardening
Tyler Goodlet 2021-06-10 13:58:06 -04:00
parent b92ed701bd
commit 9b70f5d312
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tractor/_portal.py
View File

@ -359,6 +359,7 @@ class Portal:
fn_mod_path, fn_name = func_deats(func) fn_mod_path, fn_name = func_deats(func)
recv_chan: trio.ReceiveMemoryChannel = None
try: try:
cid, recv_chan, functype, first_msg = await self._submit( cid, recv_chan, functype, first_msg = await self._submit(
fn_mod_path, fn_name, kwargs) fn_mod_path, fn_name, kwargs)
@ -390,6 +391,7 @@ class Portal:
await ctx.cancel() await ctx.cancel()
finally: finally:
if recv_chan is not None:
await recv_chan.aclose() await recv_chan.aclose()
@dataclass @dataclass