pikers
/
piker
3
0
Fork 0
Code Issues 4 Pull Requests 6 Packages Projects Releases Wiki Activity

Use new public lifetime-stack class attr

tractor_core_port
Tyler Goodlet 2022-08-29 15:18:26 -04:00
parent 4fa901dbcb
commit d1cc52dff5
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
piker/data/_sharedmem.py
View File

@ -528,9 +528,9 @@ def open_shm_array(
# "unlink" created shm on process teardown by # "unlink" created shm on process teardown by
# pushing teardown calls onto actor context stack # pushing teardown calls onto actor context stack
# TODO: make this a public API in ``tractor``.. stack = tractor.current_actor().lifetime_stack
tractor._runtime._lifetime_stack.callback(shmarr.close) stack.callback(shmarr.close)
tractor._runtime._lifetime_stack.callback(shmarr.destroy) stack.callback(shmarr.destroy)
return shmarr return shmarr
@ -614,8 +614,8 @@ def attach_shm_array(
if key not in _known_tokens: if key not in _known_tokens:
_known_tokens[key] = token _known_tokens[key] = token
# "close" attached shm on process teardown # "close" attached shm on actor teardown
tractor._runtime._lifetime_stack.callback(sha.close) tractor.current_actor().lifetime_stack.callback(sha.close)
return sha return sha