Compare commits
454 Commits
main
...
uv_migrati
| Author | SHA1 | Date |
|---|---|---|
|
|
a163769ea6 | |
|
|
bd3668e2bf | |
|
|
d74dbab1be | |
|
|
9be457fcf3 | |
|
|
e6f3f187b6 | |
|
|
924eff2985 | |
|
|
89fc072ca0 | |
|
|
7b8a8dcc7c | |
|
|
c63b94f61f | |
|
|
0e39b3902f | |
|
|
bf9689e10a | |
|
|
350a94f39e | |
|
|
0945631629 | |
|
|
0a0d30d108 | |
|
|
dcb6706489 | |
|
|
170e198683 | |
|
|
840c328f19 | |
|
|
46dbe6d2fc | |
|
|
f08e888138 | |
|
|
441cf0962d | |
|
|
fb04f74605 | |
|
|
aa1f6fa4b5 | |
|
|
9002f608ee | |
|
|
8ebc022535 | |
|
|
e26fa8330f | |
|
|
a2659069c5 | |
|
|
54699d7a0b | |
|
|
b91ab9e3a8 | |
|
|
cd14c4fe72 | |
|
|
ad40fcd2bc | |
|
|
508ba510a5 | |
|
|
b875b35b98 | |
|
|
46ddc214cd | |
|
|
b3ee20d3b9 | |
|
|
cf3e6c1218 | |
|
|
8af9b0201d | |
|
|
5c2e972315 | |
|
|
59f4024242 | |
|
|
7859e743cc | |
|
|
f7f738638d | |
|
|
ae95e0c83e | |
|
|
3b39cce741 | |
|
|
110a023a03 | |
|
|
89127614d5 | |
|
|
1f1a3f19d5 | |
|
|
f7469442e3 | |
|
|
8363317e11 | |
|
|
a628eabb30 | |
|
|
d216068713 | |
|
|
131e3e8157 | |
|
|
fc95c6719f | |
|
|
bef3dd9e97 | |
|
|
e6ccfce751 | |
|
|
31207f92ee | |
|
|
5f8f8e98ba | |
|
|
b56352b0e4 | |
|
|
9be821a5cf | |
|
|
b46400a86f | |
|
|
02812b9f51 | |
|
|
3c5816c977 | |
|
|
af3745684c | |
|
|
3907cba68e | |
|
|
e3d59964af | |
|
|
ba83bab776 | |
|
|
18d440c207 | |
|
|
edac717613 | |
|
|
7e93b81a83 | |
|
|
4fbd469c33 | |
|
|
cb90f3e6ba | |
|
|
5e009a8229 | |
|
|
b72a025d0f | |
|
|
5739e79645 | |
|
|
2ac999cc3c | |
|
|
9f9b0b17dc | |
|
|
9133f42b07 | |
|
|
268bd0d8ec | |
|
|
4f1db1ff52 | |
|
|
a870df68c0 | |
|
|
3d12a7e005 | |
|
|
9292d73b40 | |
|
|
83d69fe395 | |
|
|
72df312e71 | |
|
|
711f639fc5 | |
|
|
8477919fc9 | |
|
|
872feef24b | |
|
|
affc210033 | |
|
|
04bd111037 | |
|
|
a0ee0cc713 | |
|
|
5449bd5673 | |
|
|
e6d4ec43b9 | |
|
|
418c6907fd | |
|
|
d528e7ab4d | |
|
|
7a89b59a3f | |
|
|
7d4cd8944c | |
|
|
a6058d14ae | |
|
|
43a8cf4be1 | |
|
|
6534a363a5 | |
|
|
30d60379c1 | |
|
|
408a74784e | |
|
|
f0342d6ae3 | |
|
|
21f633a900 | |
|
|
4a270f85ca | |
|
|
d802c8aa90 | |
|
|
8ea0f08386 | |
|
|
13ea500a44 | |
|
|
2f854a3e86 | |
|
|
cdb1311e40 | |
|
|
fcd089c08f | |
|
|
993281882b | |
|
|
bbb4d4e52c | |
|
|
0e8c60ee4a | |
|
|
1db5d4def2 | |
|
|
6e54abc56d | |
|
|
28af4749cc | |
|
|
02a7c7c276 | |
|
|
4fa71cc01c | |
|
|
6a4ee461f5 | |
|
|
2db03444f7 | |
|
|
a1b124b62b | |
|
|
59ca256183 | |
|
|
6c2efc96dc | |
|
|
f7fd8278af | |
|
|
7ac730e326 | |
|
|
582144830f | |
|
|
8b860f4245 | |
|
|
27fd96729a | |
|
|
eee4c61b51 | |
|
|
42ba855d1b | |
|
|
c2cc12e14f | |
|
|
e4ec6b7b0c | |
|
|
9ce958cb4a | |
|
|
ce4d64ed2f | |
|
|
c6f599b1be | |
|
|
9eb74560ad | |
|
|
702dfe47d5 | |
|
|
d15e73557a | |
|
|
74d4b5280a | |
|
|
3538ccd799 | |
|
|
b22f7dcae0 | |
|
|
fde62c72be | |
|
|
4ef77bb64f | |
|
|
e78fdf2f69 | |
|
|
13bc3c308d | |
|
|
60fc43e530 | |
|
|
30afcd2b6b | |
|
|
c80f020ebc | |
|
|
262a0e36c6 | |
|
|
d93135acd8 | |
|
|
b23780c102 | |
|
|
31de5f6648 | |
|
|
236083b6e4 | |
|
|
d2dee87b36 | |
|
|
5cb0cc0f0b | |
|
|
fc075e96c6 | |
|
|
d6ca4771ce | |
|
|
c5a0cfc639 | |
|
|
f85314ecab | |
|
|
c929bc15c9 | |
|
|
6690968236 | |
|
|
343b7c9712 | |
|
|
45f37870af | |
|
|
4d528b76a0 | |
|
|
05b143d9ef | |
|
|
a354732a9e | |
|
|
fbc21a1dec | |
|
|
b278164f83 | |
|
|
8ffa6a5e68 | |
|
|
7707e0e75a | |
|
|
523c24eb72 | |
|
|
544ff5ab4c | |
|
|
63c23d6b82 | |
|
|
cca3206fd6 | |
|
|
54530dcf94 | |
|
|
338395346d | |
|
|
30c5896d26 | |
|
|
88a0e90f82 | |
|
|
40c972f0ec | |
|
|
f139adddca | |
|
|
979af79588 | |
|
|
a3429268ea | |
|
|
d285a3479a | |
|
|
61db040702 | |
|
|
a5a0e6854b | |
|
|
c383978402 | |
|
|
08fcd3fb03 | |
|
|
adba454d1d | |
|
|
4bab998ff9 | |
|
|
c25c77c573 | |
|
|
188ff0e0e5 | |
|
|
6b30c86eca | |
|
|
6aa52417ef | |
|
|
18e97a8f9a | |
|
|
5eb9144921 | |
|
|
a51632ffa6 | |
|
|
0df7d557db | |
|
|
7b020c42cc | |
|
|
d18cf32e28 | |
|
|
dd6a4d49d8 | |
|
|
d51be2a36a | |
|
|
3018187228 | |
|
|
e5f0b450cf | |
|
|
4aa24f8518 | |
|
|
d2f6428e46 | |
|
|
5439060cd3 | |
|
|
7372404d76 | |
|
|
77a15ebf19 | |
|
|
d0e7610073 | |
|
|
a73b24cf4a | |
|
|
5dfff3f75a | |
|
|
d4155396bf | |
|
|
3869e91b19 | |
|
|
829dfa7520 | |
|
|
b209990d04 | |
|
|
60aa16adf6 | |
|
|
eca2c02f8b | |
|
|
921f72f7fe | |
|
|
38a6483859 | |
|
|
f72b972348 | |
|
|
2edfed75eb | |
|
|
2d22713806 | |
|
|
df548257ad | |
|
|
3fb3608879 | |
|
|
faa7194daf | |
|
|
eec240a70a | |
|
|
322e015d32 | |
|
|
dbc445ff9d | |
|
|
7aaa2a61ec | |
|
|
0dcaf5f3b2 | |
|
|
af013912ac | |
|
|
8839bb06a3 | |
|
|
a35c1d40ab | |
|
|
15549f7c26 | |
|
|
cf48fdecfe | |
|
|
b341146bd1 | |
|
|
2f451ab9a3 | |
|
|
8e83455a78 | |
|
|
38111e8d53 | |
|
|
aea5abdd70 | |
|
|
aca6503fcd | |
|
|
b9a61ded0a | |
|
|
4cfe4979ff | |
|
|
97bfbdbc1c | |
|
|
b1fd8b2ec3 | |
|
|
5c1401bf81 | |
|
|
7f1c2b8ecf | |
|
|
10c98946bd | |
|
|
5b551dd9fa | |
|
|
0fcd424d57 | |
|
|
70ab60ce7c | |
|
|
a65e1e7a88 | |
|
|
40cba51909 | |
|
|
e153cc0187 | |
|
|
f2ce4a3469 | |
|
|
3aa964315a | |
|
|
f3ca8608d5 | |
|
|
25ffdedc06 | |
|
|
3ba46362a9 | |
|
|
fb8196e354 | |
|
|
b6ed26589a | |
|
|
8ff18739be | |
|
|
456979dd12 | |
|
|
995af130cf | |
|
|
d55266f4a2 | |
|
|
79211eab9a | |
|
|
336db8425e | |
|
|
2eaef26547 | |
|
|
0a69829ec5 | |
|
|
496dce57a8 | |
|
|
72b4dc1461 | |
|
|
90bfdaf58c | |
|
|
507cd96904 | |
|
|
2588e54867 | |
|
|
0055c1d954 | |
|
|
4f863a6989 | |
|
|
c04d77a3c9 | |
|
|
8e66f45e23 | |
|
|
290b0a86b1 | |
|
|
d5e5174d97 | |
|
|
8ab5e08830 | |
|
|
668016d37b | |
|
|
9221c57234 | |
|
|
78434f6317 | |
|
|
5fb5682269 | |
|
|
71de56b09a | |
|
|
e5cb39804c | |
|
|
d28c7e17c6 | |
|
|
d23d8c1779 | |
|
|
58cc57a422 | |
|
|
da913ef2bb | |
|
|
96992bcbb9 | |
|
|
6533285d7d | |
|
|
8c39b8b124 | |
|
|
ededa2e88f | |
|
|
dd168184c3 | |
|
|
37ee477aee | |
|
|
f067cf48a7 | |
|
|
c56d4b0a79 | |
|
|
7cafb59ab7 | |
|
|
7458f99733 | |
|
|
4c3c3e4b56 | |
|
|
b29d33d603 | |
|
|
1617e0ff2c | |
|
|
c025761f15 | |
|
|
2e797ef7ee | |
|
|
c36deb1f4d | |
|
|
fa7e37d6ed | |
|
|
364ea91983 | |
|
|
7ae9b5319b | |
|
|
6156ff95f8 | |
|
|
9e3f41a5b1 | |
|
|
7c22f76274 | |
|
|
04c99c2749 | |
|
|
e536057fea | |
|
|
c6b4da5788 | |
|
|
1f7f84fdfa | |
|
|
a5bdc6db66 | |
|
|
9a18b57d38 | |
|
|
ed10632d97 | |
|
|
299429a278 | |
|
|
28fefe4ffe | |
|
|
08a6a51cb8 | |
|
|
50465d4b34 | |
|
|
4f69af872c | |
|
|
9bc6a61c93 | |
|
|
23aa97692e | |
|
|
1e5810e56c | |
|
|
b54cb6682c | |
|
|
3ed309f019 | |
|
|
d08aeaeafe | |
|
|
c6ee4e5dc1 | |
|
|
ad5eee5666 | |
|
|
fc72d75061 | |
|
|
de1843dc84 | |
|
|
930d498841 | |
|
|
5ea112699d | |
|
|
e244747bc3 | |
|
|
5a09ccf459 | |
|
|
ce1bcf6d36 | |
|
|
28ba5e5435 | |
|
|
10adf34be5 | |
|
|
82dcaff8db | |
|
|
621b252b0c | |
|
|
20a089c331 | |
|
|
df50d78042 | |
|
|
114ec36436 | |
|
|
179d7d2b04 | |
|
|
f568fca98f | |
|
|
6c9bc627d8 | |
|
|
1d7cf7d1dd | |
|
|
54a0a0000d | |
|
|
0268b2ce91 | |
|
|
81f8e2d4ac | |
|
|
bf0739c194 | |
|
|
5fe3f58ea9 | |
|
|
3e1d033708 | |
|
|
c35576e196 | |
|
|
8ce26d692f | |
|
|
7f29fd8dcf | |
|
|
7fbada8a15 | |
|
|
286e75d342 | |
|
|
df641d9d31 | |
|
|
35b0c4bef0 | |
|
|
c4496f21fc | |
|
|
7e0e627921 | |
|
|
28ea8e787a | |
|
|
0294455c5e | |
|
|
734bc09b67 | |
|
|
0bcdea28a0 | |
|
|
fdf3a1b01b | |
|
|
ce7b8a5e18 | |
|
|
00024181cd | |
|
|
814384848d | |
|
|
bea31f6d19 | |
|
|
250275d98d | |
|
|
f415fc43ce | |
|
|
3f15923537 | |
|
|
87cd725adb | |
|
|
48accbd28f | |
|
|
227c9ea173 | |
|
|
d651f3d8e9 | |
|
|
ef0cfc4b20 | |
|
|
ecb525a2bc | |
|
|
b77d123edd | |
|
|
f4e63465de | |
|
|
df31047ecb | |
|
|
131674eabd | |
|
|
5a94e8fb5b | |
|
|
0518b3ab04 | |
|
|
2f0bed3018 | |
|
|
9da3b63644 | |
|
|
1d6f55543d | |
|
|
a3ed30e62b | |
|
|
42d621bba7 | |
|
|
2e81ccf5b4 | |
|
|
022bf8ce75 | |
|
|
0e9457299c | |
|
|
6b1ceee19f | |
|
|
1e689ee701 | |
|
|
190845ce1d | |
|
|
0c74b04c83 | |
|
|
215fec1d41 | |
|
|
fcc8cee9d3 | |
|
|
ca3f7a1b6b | |
|
|
87c1113de4 | |
|
|
43b659dbe4 | |
|
|
63b1488ab6 | |
|
|
7eb31f3fea | |
|
|
534e5d150d | |
|
|
e4a6223256 | |
|
|
ab2664da70 | |
|
|
ae326cbb9a | |
|
|
07cec02303 | |
|
|
2fdb8fc25a | |
|
|
6d951c526a | |
|
|
575a24adf1 | |
|
|
919e462f88 | |
|
|
a09b8560bb | |
|
|
c4cd573b26 | |
|
|
d24a9e158f | |
|
|
18a1634025 | |
|
|
78c0d2b234 | |
|
|
4314a59327 | |
|
|
e94f1261b5 | |
|
|
86da79a854 | |
|
|
de89e3a9c4 | |
|
|
7bed470f5c | |
|
|
fa9a9cfb1d | |
|
|
3d0e95513c | |
|
|
ee151b00af | |
|
|
22c14e235e | |
|
|
1102843087 | |
|
|
e03bec5efc | |
|
|
bee2c36072 | |
|
|
b36b3d522f | |
|
|
4ace8f6037 | |
|
|
98a7326c85 | |
|
|
46972df041 | |
|
|
565d7c3ee5 | |
|
|
ac695a05bf | |
|
|
fc56971a2d | |
|
|
ee87cf0e29 | |
|
|
ebcb275cd8 | |
|
|
f745da9fb2 | |
|
|
4f442efbd7 | |
|
|
f9a84f0732 | |
|
|
e0bf964ff0 | |
|
|
a9fc4c1b91 | |
|
|
b52ff270c5 | |
|
|
1713ecd9f8 | |
|
|
edb82fdd78 | |
|
|
339d787cf8 | |
|
|
c32b21b4b1 | |
|
|
71477290fc | |
|
|
9716d86825 |
|
|
@ -8,70 +8,46 @@ on:
|
||||||
workflow_dispatch:
|
workflow_dispatch:
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
# ------ sdist ------
|
|
||||||
|
mypy:
|
||||||
|
name: 'MyPy'
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Checkout
|
||||||
|
uses: actions/checkout@v2
|
||||||
|
|
||||||
|
- name: Setup python
|
||||||
|
uses: actions/setup-python@v2
|
||||||
|
with:
|
||||||
|
python-version: '3.10'
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
run: pip install -U . --upgrade-strategy eager -r requirements-test.txt
|
||||||
|
|
||||||
|
- name: Run MyPy check
|
||||||
|
run: mypy tractor/ --ignore-missing-imports --show-traceback
|
||||||
|
|
||||||
# test that we can generate a software distribution and install it
|
# test that we can generate a software distribution and install it
|
||||||
# thus avoid missing file issues after packaging.
|
# thus avoid missing file issues after packaging.
|
||||||
#
|
|
||||||
# -[x] produce sdist with uv
|
|
||||||
# ------ - ------
|
|
||||||
sdist-linux:
|
sdist-linux:
|
||||||
name: 'sdist'
|
name: 'sdist'
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v2
|
||||||
|
|
||||||
- name: Install latest uv
|
- name: Setup python
|
||||||
uses: astral-sh/setup-uv@v6
|
uses: actions/setup-python@v2
|
||||||
|
with:
|
||||||
|
python-version: '3.10'
|
||||||
|
|
||||||
- name: Build sdist as tar.gz
|
- name: Build sdist
|
||||||
run: uv build --sdist --python=3.13
|
run: python setup.py sdist --formats=zip
|
||||||
|
|
||||||
- name: Install sdist from .tar.gz
|
- name: Install sdist from .zips
|
||||||
run: python -m pip install dist/*.tar.gz
|
run: python -m pip install dist/*.zip
|
||||||
|
|
||||||
# ------ type-check ------
|
|
||||||
# mypy:
|
|
||||||
# name: 'MyPy'
|
|
||||||
# runs-on: ubuntu-latest
|
|
||||||
|
|
||||||
# steps:
|
|
||||||
# - name: Checkout
|
|
||||||
# uses: actions/checkout@v4
|
|
||||||
|
|
||||||
# - name: Install latest uv
|
|
||||||
# uses: astral-sh/setup-uv@v6
|
|
||||||
|
|
||||||
# # faster due to server caching?
|
|
||||||
# # https://docs.astral.sh/uv/guides/integration/github/#setting-up-python
|
|
||||||
# - name: "Set up Python"
|
|
||||||
# uses: actions/setup-python@v6
|
|
||||||
# with:
|
|
||||||
# python-version-file: "pyproject.toml"
|
|
||||||
|
|
||||||
# # w uv
|
|
||||||
# # - name: Set up Python
|
|
||||||
# # run: uv python install
|
|
||||||
|
|
||||||
# - name: Setup uv venv
|
|
||||||
# run: uv venv .venv --python=3.13
|
|
||||||
|
|
||||||
# - name: Install
|
|
||||||
# run: uv sync --dev
|
|
||||||
|
|
||||||
# # TODO, ty cmd over repo
|
|
||||||
# # - name: type check with ty
|
|
||||||
# # run: ty ./tractor/
|
|
||||||
|
|
||||||
# # - uses: actions/cache@v3
|
|
||||||
# # name: Cache uv virtenv as default .venv
|
|
||||||
# # with:
|
|
||||||
# # path: ./.venv
|
|
||||||
# # key: venv-${{ hashFiles('uv.lock') }}
|
|
||||||
|
|
||||||
# - name: Run MyPy check
|
|
||||||
# run: mypy tractor/ --ignore-missing-imports --show-traceback
|
|
||||||
|
|
||||||
|
|
||||||
testing-linux:
|
testing-linux:
|
||||||
|
|
@ -83,45 +59,32 @@ jobs:
|
||||||
fail-fast: false
|
fail-fast: false
|
||||||
matrix:
|
matrix:
|
||||||
os: [ubuntu-latest]
|
os: [ubuntu-latest]
|
||||||
python-version: ['3.13']
|
python: ['3.10']
|
||||||
spawn_backend: [
|
spawn_backend: [
|
||||||
'trio',
|
'trio',
|
||||||
# 'mp_spawn',
|
'mp_spawn',
|
||||||
# 'mp_forkserver',
|
'mp_forkserver',
|
||||||
]
|
]
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
|
|
||||||
- uses: actions/checkout@v4
|
- name: Checkout
|
||||||
|
uses: actions/checkout@v2
|
||||||
|
|
||||||
- name: 'Install uv + py-${{ matrix.python-version }}'
|
- name: Setup python
|
||||||
uses: astral-sh/setup-uv@v6
|
uses: actions/setup-python@v2
|
||||||
with:
|
with:
|
||||||
python-version: ${{ matrix.python-version }}
|
python-version: '${{ matrix.python }}'
|
||||||
|
|
||||||
# GH way.. faster?
|
- name: Install dependencies
|
||||||
# - name: setup-python@v6
|
run: pip install -U . -r requirements-test.txt -r requirements-docs.txt --upgrade-strategy eager
|
||||||
# uses: actions/setup-python@v6
|
|
||||||
# with:
|
|
||||||
# python-version: '${{ matrix.python-version }}'
|
|
||||||
|
|
||||||
# consider caching for speedups?
|
- name: List dependencies
|
||||||
# https://docs.astral.sh/uv/guides/integration/github/#caching
|
run: pip list
|
||||||
|
|
||||||
- name: Install the project w uv
|
|
||||||
run: uv sync --all-extras --dev
|
|
||||||
|
|
||||||
# - name: Install dependencies
|
|
||||||
# run: pip install -U . -r requirements-test.txt -r requirements-docs.txt --upgrade-strategy eager
|
|
||||||
|
|
||||||
- name: List deps tree
|
|
||||||
run: uv tree
|
|
||||||
|
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: uv run pytest tests/ --spawn-backend=${{ matrix.spawn_backend }} -rsx
|
run: pytest tests/ --spawn-backend=${{ matrix.spawn_backend }} -rsx
|
||||||
|
|
||||||
# XXX legacy NOTE XXX
|
|
||||||
#
|
|
||||||
# We skip 3.10 on windows for now due to not having any collabs to
|
# We skip 3.10 on windows for now due to not having any collabs to
|
||||||
# debug the CI failures. Anyone wanting to hack and solve them is very
|
# debug the CI failures. Anyone wanting to hack and solve them is very
|
||||||
# welcome, but our primary user base is not using that OS.
|
# welcome, but our primary user base is not using that OS.
|
||||||
|
|
|
||||||
19
default.nix
19
default.nix
|
|
@ -1,19 +0,0 @@
|
||||||
{ pkgs ? import <nixpkgs> {} }:
|
|
||||||
let
|
|
||||||
nativeBuildInputs = with pkgs; [
|
|
||||||
stdenv.cc.cc.lib
|
|
||||||
uv
|
|
||||||
];
|
|
||||||
|
|
||||||
in
|
|
||||||
pkgs.mkShell {
|
|
||||||
inherit nativeBuildInputs;
|
|
||||||
|
|
||||||
LD_LIBRARY_PATH = pkgs.lib.makeLibraryPath nativeBuildInputs;
|
|
||||||
TMPDIR = "/tmp";
|
|
||||||
|
|
||||||
shellHook = ''
|
|
||||||
set -e
|
|
||||||
uv venv .venv --python=3.12
|
|
||||||
'';
|
|
||||||
}
|
|
||||||
167
docs/README.rst
167
docs/README.rst
|
|
@ -1,20 +1,20 @@
|
||||||
|logo| ``tractor``: distributed structurred concurrency
|
|logo| ``tractor``: next-gen Python parallelism
|
||||||
|
|
||||||
``tractor`` is a `structured concurrency`_ (SC), multi-processing_ runtime built on trio_.
|
|gh_actions|
|
||||||
|
|docs|
|
||||||
|
|
||||||
Fundamentally, ``tractor`` provides parallelism via
|
``tractor`` is a `structured concurrent`_, (optionally
|
||||||
``trio``-"*actors*": independent Python **processes** (i.e.
|
distributed_) multi-processing_ runtime built on trio_.
|
||||||
*non-shared-memory threads*) which can schedule ``trio`` tasks whilst
|
|
||||||
maintaining *end-to-end SC* inside a *distributed supervision tree*.
|
Fundamentally, ``tractor`` gives you parallelism via
|
||||||
|
``trio``-"*actors*": independent Python processes (aka
|
||||||
|
non-shared-memory threads) which maintain structured
|
||||||
|
concurrency (SC) *end-to-end* inside a *supervision tree*.
|
||||||
|
|
||||||
Cross-process (and thus cross-host) SC is accomplished through the
|
Cross-process (and thus cross-host) SC is accomplished through the
|
||||||
combined use of our,
|
combined use of our "actor nurseries_" and an "SC-transitive IPC
|
||||||
|
protocol" constructed on top of multiple Pythons each running a ``trio``
|
||||||
- "actor nurseries_" which provide for spawning multiple, and
|
scheduled runtime - a call to ``trio.run()``.
|
||||||
possibly nested, Python processes each running a ``trio`` scheduled
|
|
||||||
runtime - a call to ``trio.run()``,
|
|
||||||
- an "SC-transitive supervision protocol" enforced as an
|
|
||||||
IPC-message-spec encapsulating all RPC-dialogs.
|
|
||||||
|
|
||||||
We believe the system adheres to the `3 axioms`_ of an "`actor model`_"
|
We believe the system adheres to the `3 axioms`_ of an "`actor model`_"
|
||||||
but likely **does not** look like what **you** probably *think* an "actor
|
but likely **does not** look like what **you** probably *think* an "actor
|
||||||
|
|
@ -27,7 +27,6 @@ The first step to grok ``tractor`` is to get an intermediate
|
||||||
knowledge of ``trio`` and **structured concurrency** B)
|
knowledge of ``trio`` and **structured concurrency** B)
|
||||||
|
|
||||||
Some great places to start are,
|
Some great places to start are,
|
||||||
|
|
||||||
- the seminal `blog post`_
|
- the seminal `blog post`_
|
||||||
- obviously the `trio docs`_
|
- obviously the `trio docs`_
|
||||||
- wikipedia's nascent SC_ page
|
- wikipedia's nascent SC_ page
|
||||||
|
|
@ -36,91 +35,22 @@ Some great places to start are,
|
||||||
|
|
||||||
Features
|
Features
|
||||||
--------
|
--------
|
||||||
- **It's just** a ``trio`` API!
|
- **It's just** a ``trio`` API
|
||||||
- *Infinitely nesteable* process trees running embedded ``trio`` tasks.
|
- *Infinitely nesteable* process trees
|
||||||
- Swappable, OS-specific, process spawning via multiple backends.
|
- Builtin IPC streaming APIs with task fan-out broadcasting
|
||||||
- Modular IPC stack, allowing for custom interchange formats (eg.
|
- A "native" multi-core debugger REPL using `pdbp`_ (a fork & fix of
|
||||||
as offered from `msgspec`_), varied transport protocols (TCP, RUDP,
|
`pdb++`_ thanks to @mdmintz!)
|
||||||
QUIC, wireguard), and OS-env specific higher-perf primitives (UDS,
|
- Support for a swappable, OS specific, process spawning layer
|
||||||
shm-ring-buffers).
|
- A modular transport stack, allowing for custom serialization (eg. with
|
||||||
- Optionally distributed_: all IPC and RPC APIs work over multi-host
|
`msgspec`_), communications protocols, and environment specific IPC
|
||||||
transports the same as local.
|
primitives
|
||||||
- Builtin high-level streaming API that enables your app to easily
|
- Support for spawning process-level-SC, inter-loop one-to-one-task oriented
|
||||||
leverage the benefits of a "`cheap or nasty`_" `(un)protocol`_.
|
``asyncio`` actors via "infected ``asyncio``" mode
|
||||||
- A "native UX" around a multi-process safe debugger REPL using
|
- `structured chadcurrency`_ from the ground up
|
||||||
`pdbp`_ (a fork & fix of `pdb++`_)
|
|
||||||
- "Infected ``asyncio``" mode: support for starting an actor's
|
|
||||||
runtime as a `guest`_ on the ``asyncio`` loop allowing us to
|
|
||||||
provide stringent SC-style ``trio.Task``-supervision around any
|
|
||||||
``asyncio.Task`` spawned via our ``tractor.to_asyncio`` APIs.
|
|
||||||
- A **very naive** and still very much work-in-progress inter-actor
|
|
||||||
`discovery`_ sys with plans to support multiple `modern protocol`_
|
|
||||||
approaches.
|
|
||||||
- Various ``trio`` extension APIs via ``tractor.trionics`` such as,
|
|
||||||
- task fan-out `broadcasting`_,
|
|
||||||
- multi-task-single-resource-caching and fan-out-to-multi
|
|
||||||
``__aenter__()`` APIs for ``@acm`` functions,
|
|
||||||
- (WIP) a ``TaskMngr``: one-cancels-one style nursery supervisor.
|
|
||||||
|
|
||||||
|
|
||||||
Status of `main` / infra
|
|
||||||
------------------------
|
|
||||||
|
|
||||||
- |gh_actions|
|
|
||||||
- |docs|
|
|
||||||
|
|
||||||
|
|
||||||
Install
|
|
||||||
-------
|
|
||||||
``tractor`` is still in a *alpha-near-beta-stage* for many
|
|
||||||
of its subsystems, however we are very close to having a stable
|
|
||||||
lowlevel runtime and API.
|
|
||||||
|
|
||||||
As such, it's currently recommended that you clone and install the
|
|
||||||
repo from source::
|
|
||||||
|
|
||||||
pip install git+git://github.com/goodboy/tractor.git
|
|
||||||
|
|
||||||
|
|
||||||
We use the very hip `uv`_ for project mgmt::
|
|
||||||
|
|
||||||
git clone https://github.com/goodboy/tractor.git
|
|
||||||
cd tractor
|
|
||||||
uv sync --dev
|
|
||||||
uv run python examples/rpc_bidir_streaming.py
|
|
||||||
|
|
||||||
Consider activating a virtual/project-env before starting to hack on
|
|
||||||
the code base::
|
|
||||||
|
|
||||||
# you could use plain ol' venvs
|
|
||||||
# https://docs.astral.sh/uv/pip/environments/
|
|
||||||
uv venv tractor_py313 --python 3.13
|
|
||||||
|
|
||||||
# but @goodboy prefers the more explicit (and shell agnostic)
|
|
||||||
# https://docs.astral.sh/uv/configuration/environment/#uv_project_environment
|
|
||||||
UV_PROJECT_ENVIRONMENT="tractor_py313
|
|
||||||
|
|
||||||
# hint hint, enter @goodboy's fave shell B)
|
|
||||||
uv run --dev xonsh
|
|
||||||
|
|
||||||
Alongside all this we ofc offer "releases" on PyPi::
|
|
||||||
|
|
||||||
pip install tractor
|
|
||||||
|
|
||||||
Just note that YMMV since the main git branch is often much further
|
|
||||||
ahead then any latest release.
|
|
||||||
|
|
||||||
|
|
||||||
Example codez
|
|
||||||
-------------
|
|
||||||
In ``tractor``'s (very lacking) documention we prefer to point to
|
|
||||||
example scripts in the repo over duplicating them in docs, but with
|
|
||||||
that in mind here are some definitive snippets to try and hook you
|
|
||||||
into digging deeper.
|
|
||||||
|
|
||||||
|
|
||||||
Run a func in a process
|
Run a func in a process
|
||||||
***********************
|
-----------------------
|
||||||
Use ``trio``'s style of focussing on *tasks as functions*:
|
Use ``trio``'s style of focussing on *tasks as functions*:
|
||||||
|
|
||||||
.. code:: python
|
.. code:: python
|
||||||
|
|
@ -178,7 +108,7 @@ might want to check out `trio-parallel`_.
|
||||||
|
|
||||||
|
|
||||||
Zombie safe: self-destruct a process tree
|
Zombie safe: self-destruct a process tree
|
||||||
*****************************************
|
-----------------------------------------
|
||||||
``tractor`` tries to protect you from zombies, no matter what.
|
``tractor`` tries to protect you from zombies, no matter what.
|
||||||
|
|
||||||
.. code:: python
|
.. code:: python
|
||||||
|
|
@ -234,7 +164,7 @@ it **is a bug**.
|
||||||
|
|
||||||
|
|
||||||
"Native" multi-process debugging
|
"Native" multi-process debugging
|
||||||
********************************
|
--------------------------------
|
||||||
Using the magic of `pdbp`_ and our internal IPC, we've
|
Using the magic of `pdbp`_ and our internal IPC, we've
|
||||||
been able to create a native feeling debugging experience for
|
been able to create a native feeling debugging experience for
|
||||||
any (sub-)process in your ``tractor`` tree.
|
any (sub-)process in your ``tractor`` tree.
|
||||||
|
|
@ -289,7 +219,7 @@ We're hoping to add a respawn-from-repl system soon!
|
||||||
|
|
||||||
|
|
||||||
SC compatible bi-directional streaming
|
SC compatible bi-directional streaming
|
||||||
**************************************
|
--------------------------------------
|
||||||
Yes, you saw it here first; we provide 2-way streams
|
Yes, you saw it here first; we provide 2-way streams
|
||||||
with reliable, transitive setup/teardown semantics.
|
with reliable, transitive setup/teardown semantics.
|
||||||
|
|
||||||
|
|
@ -381,7 +311,7 @@ hear your thoughts on!
|
||||||
|
|
||||||
|
|
||||||
Worker poolz are easy peasy
|
Worker poolz are easy peasy
|
||||||
***************************
|
---------------------------
|
||||||
The initial ask from most new users is *"how do I make a worker
|
The initial ask from most new users is *"how do I make a worker
|
||||||
pool thing?"*.
|
pool thing?"*.
|
||||||
|
|
||||||
|
|
@ -403,10 +333,10 @@ This uses no extra threads, fancy semaphores or futures; all we need
|
||||||
is ``tractor``'s IPC!
|
is ``tractor``'s IPC!
|
||||||
|
|
||||||
"Infected ``asyncio``" mode
|
"Infected ``asyncio``" mode
|
||||||
***************************
|
---------------------------
|
||||||
Have a bunch of ``asyncio`` code you want to force to be SC at the process level?
|
Have a bunch of ``asyncio`` code you want to force to be SC at the process level?
|
||||||
|
|
||||||
Check out our experimental system for `guest`_-mode controlled
|
Check out our experimental system for `guest-mode`_ controlled
|
||||||
``asyncio`` actors:
|
``asyncio`` actors:
|
||||||
|
|
||||||
.. code:: python
|
.. code:: python
|
||||||
|
|
@ -512,7 +442,7 @@ We need help refining the `asyncio`-side channel API to be more
|
||||||
|
|
||||||
|
|
||||||
Higher level "cluster" APIs
|
Higher level "cluster" APIs
|
||||||
***************************
|
---------------------------
|
||||||
To be extra terse the ``tractor`` devs have started hacking some "higher
|
To be extra terse the ``tractor`` devs have started hacking some "higher
|
||||||
level" APIs for managing actor trees/clusters. These interfaces should
|
level" APIs for managing actor trees/clusters. These interfaces should
|
||||||
generally be condsidered provisional for now but we encourage you to try
|
generally be condsidered provisional for now but we encourage you to try
|
||||||
|
|
@ -569,6 +499,18 @@ spawn a flat cluster:
|
||||||
.. _full worker pool re-implementation: https://github.com/goodboy/tractor/blob/master/examples/parallelism/concurrent_actors_primes.py
|
.. _full worker pool re-implementation: https://github.com/goodboy/tractor/blob/master/examples/parallelism/concurrent_actors_primes.py
|
||||||
|
|
||||||
|
|
||||||
|
Install
|
||||||
|
-------
|
||||||
|
From PyPi::
|
||||||
|
|
||||||
|
pip install tractor
|
||||||
|
|
||||||
|
|
||||||
|
From git::
|
||||||
|
|
||||||
|
pip install git+git://github.com/goodboy/tractor.git
|
||||||
|
|
||||||
|
|
||||||
Under the hood
|
Under the hood
|
||||||
--------------
|
--------------
|
||||||
``tractor`` is an attempt to pair trionic_ `structured concurrency`_ with
|
``tractor`` is an attempt to pair trionic_ `structured concurrency`_ with
|
||||||
|
|
@ -672,32 +614,25 @@ channel`_!
|
||||||
.. _adherance to: https://www.youtube.com/watch?v=7erJ1DV_Tlo&t=1821s
|
.. _adherance to: https://www.youtube.com/watch?v=7erJ1DV_Tlo&t=1821s
|
||||||
.. _trio gitter channel: https://gitter.im/python-trio/general
|
.. _trio gitter channel: https://gitter.im/python-trio/general
|
||||||
.. _matrix channel: https://matrix.to/#/!tractor:matrix.org
|
.. _matrix channel: https://matrix.to/#/!tractor:matrix.org
|
||||||
.. _broadcasting: https://github.com/goodboy/tractor/pull/229
|
|
||||||
.. _modern procotol: https://en.wikipedia.org/wiki/Rendezvous_protocol
|
|
||||||
.. _pdbp: https://github.com/mdmintz/pdbp
|
.. _pdbp: https://github.com/mdmintz/pdbp
|
||||||
.. _pdb++: https://github.com/pdbpp/pdbpp
|
.. _pdb++: https://github.com/pdbpp/pdbpp
|
||||||
.. _cheap or nasty: https://zguide.zeromq.org/docs/chapter7/#The-Cheap-or-Nasty-Pattern
|
.. _guest mode: https://trio.readthedocs.io/en/stable/reference-lowlevel.html?highlight=guest%20mode#using-guest-mode-to-run-trio-on-top-of-other-event-loops
|
||||||
.. _(un)protocol: https://zguide.zeromq.org/docs/chapter7/#Unprotocols
|
|
||||||
.. _discovery: https://zguide.zeromq.org/docs/chapter8/#Discovery
|
|
||||||
.. _modern protocol: https://en.wikipedia.org/wiki/Rendezvous_protocol
|
|
||||||
.. _messages: https://en.wikipedia.org/wiki/Message_passing
|
.. _messages: https://en.wikipedia.org/wiki/Message_passing
|
||||||
.. _trio docs: https://trio.readthedocs.io/en/latest/
|
.. _trio docs: https://trio.readthedocs.io/en/latest/
|
||||||
.. _blog post: https://vorpus.org/blog/notes-on-structured-concurrency-or-go-statement-considered-harmful/
|
.. _blog post: https://vorpus.org/blog/notes-on-structured-concurrency-or-go-statement-considered-harmful/
|
||||||
.. _structured concurrency: https://en.wikipedia.org/wiki/Structured_concurrency
|
.. _structured concurrency: https://en.wikipedia.org/wiki/Structured_concurrency
|
||||||
.. _SC: https://en.wikipedia.org/wiki/Structured_concurrency
|
.. _SC: https://en.wikipedia.org/wiki/Structured_concurrency
|
||||||
.. _libdill-docs: https://sustrik.github.io/libdill/structured-concurrency.html
|
.. _libdill-docs: https://sustrik.github.io/libdill/structured-concurrency.html
|
||||||
|
.. _structured chadcurrency: https://en.wikipedia.org/wiki/Structured_concurrency
|
||||||
.. _unrequirements: https://en.wikipedia.org/wiki/Actor_model#Direct_communication_and_asynchrony
|
.. _unrequirements: https://en.wikipedia.org/wiki/Actor_model#Direct_communication_and_asynchrony
|
||||||
.. _async generators: https://www.python.org/dev/peps/pep-0525/
|
.. _async generators: https://www.python.org/dev/peps/pep-0525/
|
||||||
.. _trio-parallel: https://github.com/richardsheridan/trio-parallel
|
.. _trio-parallel: https://github.com/richardsheridan/trio-parallel
|
||||||
.. _uv: https://docs.astral.sh/uv/
|
|
||||||
.. _msgspec: https://jcristharif.com/msgspec/
|
.. _msgspec: https://jcristharif.com/msgspec/
|
||||||
.. _guest: https://trio.readthedocs.io/en/stable/reference-lowlevel.html?highlight=guest%20mode#using-guest-mode-to-run-trio-on-top-of-other-event-loops
|
.. _guest-mode: https://trio.readthedocs.io/en/stable/reference-lowlevel.html?highlight=guest%20mode#using-guest-mode-to-run-trio-on-top-of-other-event-loops
|
||||||
|
|
||||||
..
|
|
||||||
NOTE, on generating badge links from the UI
|
.. |gh_actions| image:: https://img.shields.io/endpoint.svg?url=https%3A%2F%2Factions-badge.atrox.dev%2Fgoodboy%2Ftractor%2Fbadge&style=popout-square
|
||||||
https://docs.github.com/en/actions/how-tos/monitoring-and-troubleshooting-workflows/monitoring-workflows/adding-a-workflow-status-badge?ref=gitguardian-blog-automated-secrets-detection#using-the-ui
|
:target: https://actions-badge.atrox.dev/goodboy/tractor/goto
|
||||||
.. |gh_actions| image:: https://github.com/goodboy/tractor/actions/workflows/ci.yml/badge.svg?branch=main
|
|
||||||
:target: https://github.com/goodboy/tractor/actions/workflows/ci.yml
|
|
||||||
|
|
||||||
.. |docs| image:: https://readthedocs.org/projects/tractor/badge/?version=latest
|
.. |docs| image:: https://readthedocs.org/projects/tractor/badge/?version=latest
|
||||||
:target: https://tractor.readthedocs.io/en/latest/?badge=latest
|
:target: https://tractor.readthedocs.io/en/latest/?badge=latest
|
||||||
|
|
|
||||||
|
|
@ -16,7 +16,6 @@ from tractor import (
|
||||||
ContextCancelled,
|
ContextCancelled,
|
||||||
MsgStream,
|
MsgStream,
|
||||||
_testing,
|
_testing,
|
||||||
trionics,
|
|
||||||
)
|
)
|
||||||
import trio
|
import trio
|
||||||
import pytest
|
import pytest
|
||||||
|
|
@ -63,8 +62,7 @@ async def recv_and_spawn_net_killers(
|
||||||
await ctx.started()
|
await ctx.started()
|
||||||
async with (
|
async with (
|
||||||
ctx.open_stream() as stream,
|
ctx.open_stream() as stream,
|
||||||
trionics.collapse_eg(),
|
trio.open_nursery() as n,
|
||||||
trio.open_nursery() as tn,
|
|
||||||
):
|
):
|
||||||
async for i in stream:
|
async for i in stream:
|
||||||
print(f'child echoing {i}')
|
print(f'child echoing {i}')
|
||||||
|
|
@ -79,11 +77,11 @@ async def recv_and_spawn_net_killers(
|
||||||
i >= break_ipc_after
|
i >= break_ipc_after
|
||||||
):
|
):
|
||||||
broke_ipc = True
|
broke_ipc = True
|
||||||
tn.start_soon(
|
n.start_soon(
|
||||||
iter_ipc_stream,
|
iter_ipc_stream,
|
||||||
stream,
|
stream,
|
||||||
)
|
)
|
||||||
tn.start_soon(
|
n.start_soon(
|
||||||
partial(
|
partial(
|
||||||
break_ipc_then_error,
|
break_ipc_then_error,
|
||||||
stream=stream,
|
stream=stream,
|
||||||
|
|
@ -120,7 +118,6 @@ async def main(
|
||||||
break_parent_ipc_after: int|bool = False,
|
break_parent_ipc_after: int|bool = False,
|
||||||
break_child_ipc_after: int|bool = False,
|
break_child_ipc_after: int|bool = False,
|
||||||
pre_close: bool = False,
|
pre_close: bool = False,
|
||||||
tpt_proto: str = 'tcp',
|
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
|
|
@ -132,7 +129,6 @@ async def main(
|
||||||
# a hang since it never engages due to broken IPC
|
# a hang since it never engages due to broken IPC
|
||||||
debug_mode=debug_mode,
|
debug_mode=debug_mode,
|
||||||
loglevel=loglevel,
|
loglevel=loglevel,
|
||||||
enable_transports=[tpt_proto],
|
|
||||||
|
|
||||||
) as an,
|
) as an,
|
||||||
):
|
):
|
||||||
|
|
@ -147,8 +143,7 @@ async def main(
|
||||||
_testing.expect_ctxc(
|
_testing.expect_ctxc(
|
||||||
yay=(
|
yay=(
|
||||||
break_parent_ipc_after
|
break_parent_ipc_after
|
||||||
or
|
or break_child_ipc_after
|
||||||
break_child_ipc_after
|
|
||||||
),
|
),
|
||||||
# TODO: we CAN'T remove this right?
|
# TODO: we CAN'T remove this right?
|
||||||
# since we need the ctxc to bubble up from either
|
# since we need the ctxc to bubble up from either
|
||||||
|
|
|
||||||
|
|
@ -25,15 +25,16 @@ async def bp_then_error(
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
# sync with `trio`-side (caller) task
|
# sync with ``trio``-side (caller) task
|
||||||
to_trio.send_nowait('start')
|
to_trio.send_nowait('start')
|
||||||
|
|
||||||
# NOTE: what happens here inside the hook needs some refinement..
|
# NOTE: what happens here inside the hook needs some refinement..
|
||||||
# => seems like it's still `.debug._set_trace()` but
|
# => seems like it's still `._debug._set_trace()` but
|
||||||
# we set `Lock.local_task_in_debug = 'sync'`, we probably want
|
# we set `Lock.local_task_in_debug = 'sync'`, we probably want
|
||||||
# some further, at least, meta-data about the task/actor in debug
|
# some further, at least, meta-data about the task/actor in debug
|
||||||
# in terms of making it clear it's `asyncio` mucking about.
|
# in terms of making it clear it's `asyncio` mucking about.
|
||||||
breakpoint() # asyncio-side
|
breakpoint()
|
||||||
|
|
||||||
|
|
||||||
# short checkpoint / delay
|
# short checkpoint / delay
|
||||||
await asyncio.sleep(0.5) # asyncio-side
|
await asyncio.sleep(0.5) # asyncio-side
|
||||||
|
|
@ -57,6 +58,7 @@ async def trio_ctx(
|
||||||
# this will block until the ``asyncio`` task sends a "first"
|
# this will block until the ``asyncio`` task sends a "first"
|
||||||
# message, see first line in above func.
|
# message, see first line in above func.
|
||||||
async with (
|
async with (
|
||||||
|
|
||||||
to_asyncio.open_channel_from(
|
to_asyncio.open_channel_from(
|
||||||
bp_then_error,
|
bp_then_error,
|
||||||
# raise_after_bp=not bp_before_started,
|
# raise_after_bp=not bp_before_started,
|
||||||
|
|
@ -67,7 +69,7 @@ async def trio_ctx(
|
||||||
assert first == 'start'
|
assert first == 'start'
|
||||||
|
|
||||||
if bp_before_started:
|
if bp_before_started:
|
||||||
await tractor.pause() # trio-side
|
await tractor.pause()
|
||||||
|
|
||||||
await ctx.started(first) # trio-side
|
await ctx.started(first) # trio-side
|
||||||
|
|
||||||
|
|
@ -109,7 +111,7 @@ async def main(
|
||||||
|
|
||||||
# pause in parent to ensure no cross-actor
|
# pause in parent to ensure no cross-actor
|
||||||
# locking problems exist!
|
# locking problems exist!
|
||||||
await tractor.pause() # trio-root
|
await tractor.pause()
|
||||||
|
|
||||||
if cancel_from_root:
|
if cancel_from_root:
|
||||||
await ctx.cancel()
|
await ctx.cancel()
|
||||||
|
|
|
||||||
|
|
@ -21,13 +21,11 @@ async def name_error():
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
'''
|
"""Test breakpoint in a streaming actor.
|
||||||
Test breakpoint in a streaming actor.
|
"""
|
||||||
|
|
||||||
'''
|
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
loglevel='cancel',
|
# loglevel='cancel',
|
||||||
# loglevel='devx',
|
# loglevel='devx',
|
||||||
) as n:
|
) as n:
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -40,7 +40,7 @@ async def main():
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
loglevel='devx',
|
# loglevel='cancel',
|
||||||
) as n:
|
) as n:
|
||||||
|
|
||||||
# spawn both actors
|
# spawn both actors
|
||||||
|
|
|
||||||
|
|
@ -4,11 +4,6 @@ import sys
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
# ensure mod-path is correct!
|
|
||||||
from tractor.devx.debug import (
|
|
||||||
_sync_pause_from_builtin as _sync_pause_from_builtin,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def main() -> None:
|
async def main() -> None:
|
||||||
|
|
||||||
|
|
@ -18,30 +13,27 @@ async def main() -> None:
|
||||||
|
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
loglevel='devx',
|
) as an:
|
||||||
maybe_enable_greenback=True,
|
assert an
|
||||||
# ^XXX REQUIRED to enable `breakpoint()` support (from sync
|
|
||||||
# fns) and thus required here to avoid an assertion err
|
|
||||||
# on the next line
|
|
||||||
):
|
|
||||||
assert (
|
assert (
|
||||||
(pybp_var := os.environ['PYTHONBREAKPOINT'])
|
(pybp_var := os.environ['PYTHONBREAKPOINT'])
|
||||||
==
|
==
|
||||||
'tractor.devx.debug._sync_pause_from_builtin'
|
'tractor.devx._debug._sync_pause_from_builtin'
|
||||||
)
|
)
|
||||||
|
|
||||||
# TODO: an assert that verifies the hook has indeed been, hooked
|
# TODO: an assert that verifies the hook has indeed been, hooked
|
||||||
# XD
|
# XD
|
||||||
assert (
|
assert (
|
||||||
(pybp_hook := sys.breakpointhook)
|
(pybp_hook := sys.breakpointhook)
|
||||||
is not tractor.devx.debug._set_trace
|
is not tractor.devx._debug._set_trace
|
||||||
)
|
)
|
||||||
|
|
||||||
print(
|
print(
|
||||||
f'$PYTHONOBREAKPOINT: {pybp_var!r}\n'
|
f'$PYTHONOBREAKPOINT: {pybp_var!r}\n'
|
||||||
f'`sys.breakpointhook`: {pybp_hook!r}\n'
|
f'`sys.breakpointhook`: {pybp_hook!r}\n'
|
||||||
)
|
)
|
||||||
breakpoint() # first bp, tractor hook set.
|
breakpoint()
|
||||||
|
pass # first bp, tractor hook set.
|
||||||
|
|
||||||
# XXX AFTER EXIT (of actor-runtime) verify the hook is unset..
|
# XXX AFTER EXIT (of actor-runtime) verify the hook is unset..
|
||||||
#
|
#
|
||||||
|
|
@ -51,7 +43,8 @@ async def main() -> None:
|
||||||
assert sys.breakpointhook
|
assert sys.breakpointhook
|
||||||
|
|
||||||
# now ensure a regular builtin pause still works
|
# now ensure a regular builtin pause still works
|
||||||
breakpoint() # last bp, stdlib hook restored
|
breakpoint()
|
||||||
|
pass # last bp, stdlib hook restored
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
|
||||||
|
|
@ -24,9 +24,10 @@ async def spawn_until(depth=0):
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
'''
|
"""The main ``tractor`` routine.
|
||||||
The process tree should look as approximately as follows when the
|
|
||||||
debugger first engages:
|
The process tree should look as approximately as follows when the debugger
|
||||||
|
first engages:
|
||||||
|
|
||||||
python examples/debugging/multi_nested_subactors_bp_forever.py
|
python examples/debugging/multi_nested_subactors_bp_forever.py
|
||||||
├─ python -m tractor._child --uid ('spawner1', '7eab8462 ...)
|
├─ python -m tractor._child --uid ('spawner1', '7eab8462 ...)
|
||||||
|
|
@ -36,11 +37,10 @@ async def main():
|
||||||
└─ python -m tractor._child --uid ('spawner0', '1d42012b ...)
|
└─ python -m tractor._child --uid ('spawner0', '1d42012b ...)
|
||||||
└─ python -m tractor._child --uid ('name_error', '6c2733b8 ...)
|
└─ python -m tractor._child --uid ('name_error', '6c2733b8 ...)
|
||||||
|
|
||||||
'''
|
"""
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
loglevel='devx',
|
loglevel='warning'
|
||||||
enable_transports=['uds'],
|
|
||||||
) as n:
|
) as n:
|
||||||
|
|
||||||
# spawn both actors
|
# spawn both actors
|
||||||
|
|
|
||||||
|
|
@ -1,35 +0,0 @@
|
||||||
import trio
|
|
||||||
import tractor
|
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
|
||||||
async with tractor.open_root_actor(
|
|
||||||
debug_mode=True,
|
|
||||||
loglevel='cancel',
|
|
||||||
) as _root:
|
|
||||||
|
|
||||||
# manually trigger self-cancellation and wait
|
|
||||||
# for it to fully trigger.
|
|
||||||
_root.cancel_soon()
|
|
||||||
await _root._cancel_complete.wait()
|
|
||||||
print('root cancelled')
|
|
||||||
|
|
||||||
# now ensure we can still use the REPL
|
|
||||||
try:
|
|
||||||
await tractor.pause()
|
|
||||||
except trio.Cancelled as _taskc:
|
|
||||||
assert (root_cs := _root._root_tn.cancel_scope).cancel_called
|
|
||||||
# NOTE^^ above logic but inside `open_root_actor()` and
|
|
||||||
# passed to the `shield=` expression is effectively what
|
|
||||||
# we're testing here!
|
|
||||||
await tractor.pause(shield=root_cs.cancel_called)
|
|
||||||
|
|
||||||
# XXX, if shield logic *is wrong* inside `open_root_actor()`'s
|
|
||||||
# crash-handler block this should never be interacted,
|
|
||||||
# instead `trio.Cancelled` would be bubbled up: the original
|
|
||||||
# BUG.
|
|
||||||
assert 0
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
|
||||||
trio.run(main)
|
|
||||||
|
|
@ -37,9 +37,9 @@ async def main(
|
||||||
enable_stack_on_sig=True,
|
enable_stack_on_sig=True,
|
||||||
# maybe_enable_greenback=False,
|
# maybe_enable_greenback=False,
|
||||||
loglevel='devx',
|
loglevel='devx',
|
||||||
enable_transports=['uds'],
|
|
||||||
) as an,
|
) as an,
|
||||||
):
|
):
|
||||||
|
|
||||||
ptl: tractor.Portal = await an.start_actor(
|
ptl: tractor.Portal = await an.start_actor(
|
||||||
'hanger',
|
'hanger',
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
|
|
@ -54,16 +54,13 @@ async def main(
|
||||||
|
|
||||||
print(
|
print(
|
||||||
'Yo my child hanging..?\n'
|
'Yo my child hanging..?\n'
|
||||||
# "i'm a user who wants to see a `stackscope` tree!\n"
|
'Sending SIGUSR1 to see a tree-trace!\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
# XXX simulate the wrapping test's "user actions"
|
# XXX simulate the wrapping test's "user actions"
|
||||||
# (i.e. if a human didn't run this manually but wants to
|
# (i.e. if a human didn't run this manually but wants to
|
||||||
# know what they should do to reproduce test behaviour)
|
# know what they should do to reproduce test behaviour)
|
||||||
if from_test:
|
if from_test:
|
||||||
print(
|
|
||||||
f'Sending SIGUSR1 to {cpid!r}!\n'
|
|
||||||
)
|
|
||||||
os.kill(
|
os.kill(
|
||||||
cpid,
|
cpid,
|
||||||
signal.SIGUSR1,
|
signal.SIGUSR1,
|
||||||
|
|
|
||||||
|
|
@ -33,11 +33,8 @@ async def just_bp(
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=True,
|
debug_mode=True,
|
||||||
enable_transports=['uds'],
|
|
||||||
loglevel='devx',
|
|
||||||
) as n:
|
) as n:
|
||||||
p = await n.start_actor(
|
p = await n.start_actor(
|
||||||
'bp_boi',
|
'bp_boi',
|
||||||
|
|
|
||||||
|
|
@ -6,7 +6,7 @@ import tractor
|
||||||
|
|
||||||
# TODO: only import these when not running from test harness?
|
# TODO: only import these when not running from test harness?
|
||||||
# can we detect `pexpect` usage maybe?
|
# can we detect `pexpect` usage maybe?
|
||||||
# from tractor.devx.debug import (
|
# from tractor.devx._debug import (
|
||||||
# get_lock,
|
# get_lock,
|
||||||
# get_debug_req,
|
# get_debug_req,
|
||||||
# )
|
# )
|
||||||
|
|
|
||||||
|
|
@ -91,7 +91,7 @@ async def main() -> list[int]:
|
||||||
an: ActorNursery
|
an: ActorNursery
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
loglevel='cancel',
|
loglevel='cancel',
|
||||||
# debug_mode=True,
|
debug_mode=True,
|
||||||
) as an:
|
) as an:
|
||||||
|
|
||||||
seed = int(1e3)
|
seed = int(1e3)
|
||||||
|
|
|
||||||
|
|
@ -3,18 +3,20 @@ import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
async def sleepy_jane() -> None:
|
async def sleepy_jane():
|
||||||
uid: tuple = tractor.current_actor().uid
|
uid = tractor.current_actor().uid
|
||||||
print(f'Yo i am actor {uid}')
|
print(f'Yo i am actor {uid}')
|
||||||
await trio.sleep_forever()
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
'''
|
'''
|
||||||
Spawn a flat actor cluster, with one process per detected core.
|
Spawn a flat actor cluster, with one process per
|
||||||
|
detected core.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
portal_map: dict[str, tractor.Portal]
|
portal_map: dict[str, tractor.Portal]
|
||||||
|
results: dict[str, str]
|
||||||
|
|
||||||
# look at this hip new syntax!
|
# look at this hip new syntax!
|
||||||
async with (
|
async with (
|
||||||
|
|
@ -23,15 +25,11 @@ async def main():
|
||||||
modules=[__name__]
|
modules=[__name__]
|
||||||
) as portal_map,
|
) as portal_map,
|
||||||
|
|
||||||
tractor.trionics.collapse_eg(),
|
trio.open_nursery() as n,
|
||||||
trio.open_nursery() as tn,
|
|
||||||
):
|
):
|
||||||
|
|
||||||
for (name, portal) in portal_map.items():
|
for (name, portal) in portal_map.items():
|
||||||
tn.start_soon(
|
n.start_soon(portal.run, sleepy_jane)
|
||||||
portal.run,
|
|
||||||
sleepy_jane,
|
|
||||||
)
|
|
||||||
|
|
||||||
await trio.sleep(0.5)
|
await trio.sleep(0.5)
|
||||||
|
|
||||||
|
|
@ -43,4 +41,4 @@ if __name__ == '__main__':
|
||||||
try:
|
try:
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
print('trio cancelled by KBI')
|
pass
|
||||||
|
|
|
||||||
|
|
@ -9,7 +9,7 @@ async def main(service_name):
|
||||||
async with tractor.open_nursery() as an:
|
async with tractor.open_nursery() as an:
|
||||||
await an.start_actor(service_name)
|
await an.start_actor(service_name)
|
||||||
|
|
||||||
async with tractor.get_registry() as portal:
|
async with tractor.get_registry('127.0.0.1', 1616) as portal:
|
||||||
print(f"Arbiter is listening on {portal.channel}")
|
print(f"Arbiter is listening on {portal.channel}")
|
||||||
|
|
||||||
async with tractor.wait_for_actor(service_name) as sockaddr:
|
async with tractor.wait_for_actor(service_name) as sockaddr:
|
||||||
|
|
|
||||||
|
|
@ -1,85 +0,0 @@
|
||||||
from contextlib import (
|
|
||||||
asynccontextmanager as acm,
|
|
||||||
)
|
|
||||||
from functools import partial
|
|
||||||
|
|
||||||
import tractor
|
|
||||||
import trio
|
|
||||||
|
|
||||||
|
|
||||||
log = tractor.log.get_logger(
|
|
||||||
name=__name__
|
|
||||||
)
|
|
||||||
|
|
||||||
_lock: trio.Lock|None = None
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
|
||||||
async def acquire_singleton_lock(
|
|
||||||
) -> None:
|
|
||||||
global _lock
|
|
||||||
if _lock is None:
|
|
||||||
log.info('Allocating LOCK')
|
|
||||||
_lock = trio.Lock()
|
|
||||||
|
|
||||||
log.info('TRYING TO LOCK ACQUIRE')
|
|
||||||
async with _lock:
|
|
||||||
log.info('ACQUIRED')
|
|
||||||
yield _lock
|
|
||||||
|
|
||||||
log.info('RELEASED')
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def hold_lock_forever(
|
|
||||||
task_status=trio.TASK_STATUS_IGNORED
|
|
||||||
):
|
|
||||||
async with (
|
|
||||||
tractor.trionics.maybe_raise_from_masking_exc(),
|
|
||||||
acquire_singleton_lock() as lock,
|
|
||||||
):
|
|
||||||
task_status.started(lock)
|
|
||||||
await trio.sleep_forever()
|
|
||||||
|
|
||||||
|
|
||||||
async def main(
|
|
||||||
ignore_special_cases: bool,
|
|
||||||
loglevel: str = 'info',
|
|
||||||
debug_mode: bool = True,
|
|
||||||
):
|
|
||||||
async with (
|
|
||||||
trio.open_nursery() as tn,
|
|
||||||
|
|
||||||
# tractor.trionics.maybe_raise_from_masking_exc()
|
|
||||||
# ^^^ XXX NOTE, interestingly putting the unmasker
|
|
||||||
# here does not exhibit the same behaviour ??
|
|
||||||
):
|
|
||||||
if not ignore_special_cases:
|
|
||||||
from tractor.trionics import _taskc
|
|
||||||
_taskc._mask_cases.clear()
|
|
||||||
|
|
||||||
_lock = await tn.start(
|
|
||||||
hold_lock_forever,
|
|
||||||
)
|
|
||||||
with trio.move_on_after(0.2):
|
|
||||||
await tn.start(
|
|
||||||
hold_lock_forever,
|
|
||||||
)
|
|
||||||
|
|
||||||
tn.cancel_scope.cancel()
|
|
||||||
|
|
||||||
|
|
||||||
# XXX, manual test as script
|
|
||||||
if __name__ == '__main__':
|
|
||||||
tractor.log.get_console_log(level='info')
|
|
||||||
for case in [True, False]:
|
|
||||||
log.info(
|
|
||||||
f'\n'
|
|
||||||
f'------ RUNNING SCRIPT TRIAL ------\n'
|
|
||||||
f'ignore_special_cases: {case!r}\n'
|
|
||||||
)
|
|
||||||
trio.run(partial(
|
|
||||||
main,
|
|
||||||
ignore_special_cases=case,
|
|
||||||
loglevel='info',
|
|
||||||
))
|
|
||||||
|
|
@ -1,195 +0,0 @@
|
||||||
from contextlib import (
|
|
||||||
contextmanager as cm,
|
|
||||||
# TODO, any diff in async case(s)??
|
|
||||||
# asynccontextmanager as acm,
|
|
||||||
)
|
|
||||||
from functools import partial
|
|
||||||
|
|
||||||
import tractor
|
|
||||||
import trio
|
|
||||||
|
|
||||||
|
|
||||||
log = tractor.log.get_logger(
|
|
||||||
name=__name__
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@cm
|
|
||||||
def teardown_on_exc(
|
|
||||||
raise_from_handler: bool = False,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
You could also have a teardown handler which catches any exc and
|
|
||||||
does some required teardown. In this case the problem is
|
|
||||||
compounded UNLESS you ensure the handler's scope is OUTSIDE the
|
|
||||||
`ux.aclose()`.. that is in the caller's enclosing scope.
|
|
||||||
|
|
||||||
'''
|
|
||||||
try:
|
|
||||||
yield
|
|
||||||
except BaseException as _berr:
|
|
||||||
berr = _berr
|
|
||||||
log.exception(
|
|
||||||
f'Handling termination teardown in child due to,\n'
|
|
||||||
f'{berr!r}\n'
|
|
||||||
)
|
|
||||||
if raise_from_handler:
|
|
||||||
# XXX teardown ops XXX
|
|
||||||
# on termination these steps say need to be run to
|
|
||||||
# ensure wider system consistency (like the state of
|
|
||||||
# remote connections/services).
|
|
||||||
#
|
|
||||||
# HOWEVER, any bug in this teardown code is also
|
|
||||||
# masked by the `tx.aclose()`!
|
|
||||||
# this is also true if `_tn.cancel_scope` is
|
|
||||||
# `.cancel_called` by the parent in a graceful
|
|
||||||
# request case..
|
|
||||||
|
|
||||||
# simulate a bug in teardown handler.
|
|
||||||
raise RuntimeError(
|
|
||||||
'woopsie teardown bug!'
|
|
||||||
)
|
|
||||||
|
|
||||||
raise # no teardown bug.
|
|
||||||
|
|
||||||
|
|
||||||
async def finite_stream_to_rent(
|
|
||||||
tx: trio.abc.SendChannel,
|
|
||||||
child_errors_mid_stream: bool,
|
|
||||||
raise_unmasked: bool,
|
|
||||||
|
|
||||||
task_status: trio.TaskStatus[
|
|
||||||
trio.CancelScope,
|
|
||||||
] = trio.TASK_STATUS_IGNORED,
|
|
||||||
):
|
|
||||||
async with (
|
|
||||||
# XXX without this unmasker the mid-streaming RTE is never
|
|
||||||
# reported since it is masked by the `tx.aclose()`
|
|
||||||
# call which in turn raises `Cancelled`!
|
|
||||||
#
|
|
||||||
# NOTE, this is WITHOUT doing any exception handling
|
|
||||||
# inside the child task!
|
|
||||||
#
|
|
||||||
# TODO, uncomment next LoC to see the supprsessed beg[RTE]!
|
|
||||||
tractor.trionics.maybe_raise_from_masking_exc(
|
|
||||||
raise_unmasked=raise_unmasked,
|
|
||||||
),
|
|
||||||
|
|
||||||
tx as tx, # .aclose() is the guilty masker chkpt!
|
|
||||||
|
|
||||||
# XXX, this ONLY matters in the
|
|
||||||
# `child_errors_mid_stream=False` case oddly!?
|
|
||||||
# THAT IS, if no tn is opened in that case then the
|
|
||||||
# test will not fail; it raises the RTE correctly?
|
|
||||||
#
|
|
||||||
# -> so it seems this new scope somehow affects the form of
|
|
||||||
# eventual in the parent EG?
|
|
||||||
tractor.trionics.maybe_open_nursery(
|
|
||||||
nursery=(
|
|
||||||
None
|
|
||||||
if not child_errors_mid_stream
|
|
||||||
else True
|
|
||||||
),
|
|
||||||
) as _tn,
|
|
||||||
):
|
|
||||||
# pass our scope back to parent for supervision\
|
|
||||||
# control.
|
|
||||||
cs: trio.CancelScope|None = (
|
|
||||||
None
|
|
||||||
if _tn is True
|
|
||||||
else _tn.cancel_scope
|
|
||||||
)
|
|
||||||
task_status.started(cs)
|
|
||||||
|
|
||||||
with teardown_on_exc(
|
|
||||||
raise_from_handler=not child_errors_mid_stream,
|
|
||||||
):
|
|
||||||
for i in range(100):
|
|
||||||
log.debug(
|
|
||||||
f'Child tx {i!r}\n'
|
|
||||||
)
|
|
||||||
if (
|
|
||||||
child_errors_mid_stream
|
|
||||||
and
|
|
||||||
i == 66
|
|
||||||
):
|
|
||||||
# oh wait but WOOPS there's a bug
|
|
||||||
# in that teardown code!?
|
|
||||||
raise RuntimeError(
|
|
||||||
'woopsie, a mid-streaming bug!?'
|
|
||||||
)
|
|
||||||
|
|
||||||
await tx.send(i)
|
|
||||||
|
|
||||||
|
|
||||||
async def main(
|
|
||||||
# TODO! toggle this for the 2 cases!
|
|
||||||
# 1. child errors mid-stream while parent is also requesting
|
|
||||||
# (graceful) cancel of that child streamer.
|
|
||||||
#
|
|
||||||
# 2. child contains a teardown handler which contains a
|
|
||||||
# bug and raises.
|
|
||||||
#
|
|
||||||
child_errors_mid_stream: bool,
|
|
||||||
|
|
||||||
raise_unmasked: bool = False,
|
|
||||||
loglevel: str = 'info',
|
|
||||||
):
|
|
||||||
tractor.log.get_console_log(level=loglevel)
|
|
||||||
|
|
||||||
# the `.aclose()` being checkpoints on these
|
|
||||||
# is the source of the problem..
|
|
||||||
tx, rx = trio.open_memory_channel(1)
|
|
||||||
|
|
||||||
async with (
|
|
||||||
tractor.trionics.collapse_eg(),
|
|
||||||
trio.open_nursery() as tn,
|
|
||||||
rx as rx,
|
|
||||||
):
|
|
||||||
_child_cs = await tn.start(
|
|
||||||
partial(
|
|
||||||
finite_stream_to_rent,
|
|
||||||
child_errors_mid_stream=child_errors_mid_stream,
|
|
||||||
raise_unmasked=raise_unmasked,
|
|
||||||
tx=tx,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
async for msg in rx:
|
|
||||||
log.debug(
|
|
||||||
f'Rent rx {msg!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# simulate some external cancellation
|
|
||||||
# request **JUST BEFORE** the child errors.
|
|
||||||
if msg == 65:
|
|
||||||
log.cancel(
|
|
||||||
f'Cancelling parent on,\n'
|
|
||||||
f'msg={msg}\n'
|
|
||||||
f'\n'
|
|
||||||
f'Simulates OOB cancel request!\n'
|
|
||||||
)
|
|
||||||
tn.cancel_scope.cancel()
|
|
||||||
|
|
||||||
|
|
||||||
# XXX, manual test as script
|
|
||||||
if __name__ == '__main__':
|
|
||||||
tractor.log.get_console_log(level='info')
|
|
||||||
for case in [True, False]:
|
|
||||||
log.info(
|
|
||||||
f'\n'
|
|
||||||
f'------ RUNNING SCRIPT TRIAL ------\n'
|
|
||||||
f'child_errors_midstream: {case!r}\n'
|
|
||||||
)
|
|
||||||
try:
|
|
||||||
trio.run(partial(
|
|
||||||
main,
|
|
||||||
child_errors_mid_stream=case,
|
|
||||||
# raise_unmasked=True,
|
|
||||||
loglevel='info',
|
|
||||||
))
|
|
||||||
except Exception as _exc:
|
|
||||||
exc = _exc
|
|
||||||
log.exception(
|
|
||||||
'Should have raised an RTE or Cancelled?\n'
|
|
||||||
)
|
|
||||||
breakpoint()
|
|
||||||
|
|
@ -2,14 +2,12 @@
|
||||||
requires = ["hatchling"]
|
requires = ["hatchling"]
|
||||||
build-backend = "hatchling.build"
|
build-backend = "hatchling.build"
|
||||||
|
|
||||||
# ------ build-system ------
|
|
||||||
|
|
||||||
[project]
|
[project]
|
||||||
name = "tractor"
|
name = "tractor"
|
||||||
version = "0.1.0a6dev0"
|
version = "0.1.0a6dev0"
|
||||||
description = 'structured concurrent `trio`-"actors"'
|
description = 'structured concurrent `trio`-"actors"'
|
||||||
authors = [{ name = "Tyler Goodlet", email = "goodboy_foss@protonmail.com" }]
|
authors = [{ name = "Tyler Goodlet", email = "goodboy_foss@protonmail.com" }]
|
||||||
requires-python = ">= 3.11"
|
requires-python = "~=3.11"
|
||||||
readme = "docs/README.rst"
|
readme = "docs/README.rst"
|
||||||
license = "AGPL-3.0-or-later"
|
license = "AGPL-3.0-or-later"
|
||||||
keywords = [
|
keywords = [
|
||||||
|
|
@ -32,77 +30,25 @@ classifiers = [
|
||||||
"Topic :: System :: Distributed Computing",
|
"Topic :: System :: Distributed Computing",
|
||||||
]
|
]
|
||||||
dependencies = [
|
dependencies = [
|
||||||
# trio runtime and friends
|
"trio>=0.24,<0.25",
|
||||||
# (poetry) proper range specs,
|
|
||||||
# https://packaging.python.org/en/latest/discussions/install-requires-vs-requirements/#id5
|
|
||||||
# TODO, for 3.13 we must go go `0.27` which means we have to
|
|
||||||
# disable strict egs or port to handling them internally!
|
|
||||||
"trio>0.27",
|
|
||||||
"tricycle>=0.4.1,<0.5",
|
"tricycle>=0.4.1,<0.5",
|
||||||
|
"trio-typing>=0.10.0,<0.11",
|
||||||
"wrapt>=1.16.0,<2",
|
"wrapt>=1.16.0,<2",
|
||||||
"colorlog>=6.8.2,<7",
|
"colorlog>=6.8.2,<7",
|
||||||
# built-in multi-actor `pdb` REPL
|
"pdbp>=1.5.0,<2",
|
||||||
"pdbp>=1.6,<2", # windows only (from `pdbp`)
|
"msgspec",
|
||||||
# typed IPC msging
|
|
||||||
"msgspec>=0.19.0",
|
|
||||||
"cffi>=1.17.1",
|
|
||||||
"bidict>=0.23.1",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
# ------ project ------
|
|
||||||
|
|
||||||
[dependency-groups]
|
[dependency-groups]
|
||||||
dev = [
|
dev = [
|
||||||
# test suite
|
"pytest>=8.2.0,<9",
|
||||||
# TODO: maybe some of these layout choices?
|
|
||||||
# https://docs.pytest.org/en/8.0.x/explanation/goodpractices.html#choosing-a-test-layout-import-rules
|
|
||||||
"pytest>=8.3.5",
|
|
||||||
"pexpect>=4.9.0,<5",
|
"pexpect>=4.9.0,<5",
|
||||||
# `tractor.devx` tooling
|
|
||||||
"greenback>=1.2.1,<2",
|
"greenback>=1.2.1,<2",
|
||||||
"stackscope>=0.2.2,<0.3",
|
"stackscope>=0.2.2,<0.3",
|
||||||
# ^ requires this?
|
"xontrib-vox>=0.0.1,<0.0.2",
|
||||||
"typing-extensions>=4.14.1",
|
"prompt-toolkit>=3.0.43,<4",
|
||||||
|
"xonsh-vox-tabcomplete>=0.5,<0.6",
|
||||||
"pyperclip>=1.9.0",
|
|
||||||
"prompt-toolkit>=3.0.50",
|
|
||||||
"xonsh>=0.19.2",
|
|
||||||
"psutil>=7.0.0",
|
|
||||||
]
|
]
|
||||||
# TODO, add these with sane versions; were originally in
|
|
||||||
# `requirements-docs.txt`..
|
|
||||||
# docs = [
|
|
||||||
# "sphinx>="
|
|
||||||
# "sphinx_book_theme>="
|
|
||||||
# ]
|
|
||||||
|
|
||||||
# ------ dependency-groups ------
|
|
||||||
|
|
||||||
# ------ dependency-groups ------
|
|
||||||
|
|
||||||
[tool.uv.sources]
|
|
||||||
# XXX NOTE, only for @goodboy's hacking on `pprint(sort_dicts=False)`
|
|
||||||
# for the `pp` alias..
|
|
||||||
# pdbp = { path = "../pdbp", editable = true }
|
|
||||||
|
|
||||||
# ------ tool.uv.sources ------
|
|
||||||
# TODO, distributed (multi-host) extensions
|
|
||||||
# linux kernel networking
|
|
||||||
# 'pyroute2
|
|
||||||
|
|
||||||
# ------ tool.uv.sources ------
|
|
||||||
|
|
||||||
[tool.uv]
|
|
||||||
# XXX NOTE, prefer the sys python bc apparently the distis from
|
|
||||||
# `astral` are built in a way that breaks `pdbp`+`tabcompleter`'s
|
|
||||||
# likely due to linking against `libedit` over `readline`..
|
|
||||||
# |_https://docs.astral.sh/uv/concepts/python-versions/#managed-python-distributions
|
|
||||||
# |_https://gregoryszorc.com/docs/python-build-standalone/main/quirks.html#use-of-libedit-on-linux
|
|
||||||
#
|
|
||||||
# https://docs.astral.sh/uv/reference/settings/#python-preference
|
|
||||||
python-preference = 'system'
|
|
||||||
|
|
||||||
# ------ tool.uv ------
|
|
||||||
|
|
||||||
[tool.hatch.build.targets.sdist]
|
[tool.hatch.build.targets.sdist]
|
||||||
include = ["tractor"]
|
include = ["tractor"]
|
||||||
|
|
@ -110,8 +56,6 @@ include = ["tractor"]
|
||||||
[tool.hatch.build.targets.wheel]
|
[tool.hatch.build.targets.wheel]
|
||||||
include = ["tractor"]
|
include = ["tractor"]
|
||||||
|
|
||||||
# ------ tool.hatch ------
|
|
||||||
|
|
||||||
[tool.towncrier]
|
[tool.towncrier]
|
||||||
package = "tractor"
|
package = "tractor"
|
||||||
filename = "NEWS.rst"
|
filename = "NEWS.rst"
|
||||||
|
|
@ -141,8 +85,6 @@ all_bullets = true
|
||||||
name = "Trivial/Internal Changes"
|
name = "Trivial/Internal Changes"
|
||||||
showcontent = true
|
showcontent = true
|
||||||
|
|
||||||
# ------ tool.towncrier ------
|
|
||||||
|
|
||||||
[tool.pytest.ini_options]
|
[tool.pytest.ini_options]
|
||||||
minversion = '6.0'
|
minversion = '6.0'
|
||||||
testpaths = [
|
testpaths = [
|
||||||
|
|
@ -157,8 +99,6 @@ addopts = [
|
||||||
'--show-capture=no',
|
'--show-capture=no',
|
||||||
]
|
]
|
||||||
log_cli = false
|
log_cli = false
|
||||||
# TODO: maybe some of these layout choices?
|
|
||||||
# https://docs.pytest.org/en/8.0.x/explanation/goodpractices.html#choosing-a-test-layout-import-rules
|
|
||||||
# pythonpath = "src"
|
|
||||||
|
|
||||||
# ------ tool.pytest ------
|
[tool.uv.sources]
|
||||||
|
msgspec = { git = "https://github.com/jcrist/msgspec.git" }
|
||||||
|
|
|
||||||
|
|
@ -1,8 +0,0 @@
|
||||||
# vim: ft=ini
|
|
||||||
# pytest.ini for tractor
|
|
||||||
|
|
||||||
[pytest]
|
|
||||||
# don't show frickin captured logs AGAIN in the report..
|
|
||||||
addopts = --show-capture='no'
|
|
||||||
log_cli = false
|
|
||||||
; minversion = 6.0
|
|
||||||
|
|
@ -0,0 +1,2 @@
|
||||||
|
sphinx
|
||||||
|
sphinx_book_theme
|
||||||
|
|
@ -0,0 +1,9 @@
|
||||||
|
pytest
|
||||||
|
pytest-trio
|
||||||
|
pytest-timeout
|
||||||
|
pdbp
|
||||||
|
mypy
|
||||||
|
trio_typing
|
||||||
|
pexpect
|
||||||
|
towncrier
|
||||||
|
numpy
|
||||||
82
ruff.toml
82
ruff.toml
|
|
@ -1,82 +0,0 @@
|
||||||
# from default `ruff.toml` @
|
|
||||||
# https://docs.astral.sh/ruff/configuration/
|
|
||||||
|
|
||||||
# Exclude a variety of commonly ignored directories.
|
|
||||||
exclude = [
|
|
||||||
".bzr",
|
|
||||||
".direnv",
|
|
||||||
".eggs",
|
|
||||||
".git",
|
|
||||||
".git-rewrite",
|
|
||||||
".hg",
|
|
||||||
".ipynb_checkpoints",
|
|
||||||
".mypy_cache",
|
|
||||||
".nox",
|
|
||||||
".pants.d",
|
|
||||||
".pyenv",
|
|
||||||
".pytest_cache",
|
|
||||||
".pytype",
|
|
||||||
".ruff_cache",
|
|
||||||
".svn",
|
|
||||||
".tox",
|
|
||||||
".venv",
|
|
||||||
".vscode",
|
|
||||||
"__pypackages__",
|
|
||||||
"_build",
|
|
||||||
"buck-out",
|
|
||||||
"build",
|
|
||||||
"dist",
|
|
||||||
"node_modules",
|
|
||||||
"site-packages",
|
|
||||||
"venv",
|
|
||||||
]
|
|
||||||
|
|
||||||
# Same as Black.
|
|
||||||
line-length = 88
|
|
||||||
indent-width = 4
|
|
||||||
|
|
||||||
# Assume Python 3.9
|
|
||||||
target-version = "py311"
|
|
||||||
|
|
||||||
[lint]
|
|
||||||
# Enable Pyflakes (`F`) and a subset of the pycodestyle (`E`) codes by default.
|
|
||||||
# Unlike Flake8, Ruff doesn't enable pycodestyle warnings (`W`) or
|
|
||||||
# McCabe complexity (`C901`) by default.
|
|
||||||
select = ["E4", "E7", "E9", "F"]
|
|
||||||
ignore = [
|
|
||||||
'E402', # https://docs.astral.sh/ruff/rules/module-import-not-at-top-of-file/
|
|
||||||
]
|
|
||||||
|
|
||||||
# Allow fix for all enabled rules (when `--fix`) is provided.
|
|
||||||
fixable = ["ALL"]
|
|
||||||
unfixable = []
|
|
||||||
|
|
||||||
# Allow unused variables when underscore-prefixed.
|
|
||||||
# dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"
|
|
||||||
|
|
||||||
[format]
|
|
||||||
# Use single quotes in `ruff format`.
|
|
||||||
quote-style = "single"
|
|
||||||
|
|
||||||
# Like Black, indent with spaces, rather than tabs.
|
|
||||||
indent-style = "space"
|
|
||||||
|
|
||||||
# Like Black, respect magic trailing commas.
|
|
||||||
skip-magic-trailing-comma = false
|
|
||||||
|
|
||||||
# Like Black, automatically detect the appropriate line ending.
|
|
||||||
line-ending = "auto"
|
|
||||||
|
|
||||||
# Enable auto-formatting of code examples in docstrings. Markdown,
|
|
||||||
# reStructuredText code/literal blocks and doctests are all supported.
|
|
||||||
#
|
|
||||||
# This is currently disabled by default, but it is planned for this
|
|
||||||
# to be opt-out in the future.
|
|
||||||
docstring-code-format = false
|
|
||||||
|
|
||||||
# Set the line length limit used when formatting code snippets in
|
|
||||||
# docstrings.
|
|
||||||
#
|
|
||||||
# This only has an effect when the `docstring-code-format` setting is
|
|
||||||
# enabled.
|
|
||||||
docstring-code-line-length = "dynamic"
|
|
||||||
|
|
@ -0,0 +1,102 @@
|
||||||
|
#!/usr/bin/env python
|
||||||
|
#
|
||||||
|
# tractor: structured concurrent "actors".
|
||||||
|
#
|
||||||
|
# Copyright 2018-eternity Tyler Goodlet.
|
||||||
|
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU Affero General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU Affero General Public License for more details.
|
||||||
|
|
||||||
|
# You should have received a copy of the GNU Affero General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
from setuptools import setup
|
||||||
|
|
||||||
|
with open('docs/README.rst', encoding='utf-8') as f:
|
||||||
|
readme = f.read()
|
||||||
|
|
||||||
|
|
||||||
|
setup(
|
||||||
|
name="tractor",
|
||||||
|
version='0.1.0a6dev0', # alpha zone
|
||||||
|
description='structured concurrent `trio`-"actors"',
|
||||||
|
long_description=readme,
|
||||||
|
license='AGPLv3',
|
||||||
|
author='Tyler Goodlet',
|
||||||
|
maintainer='Tyler Goodlet',
|
||||||
|
maintainer_email='goodboy_foss@protonmail.com',
|
||||||
|
url='https://github.com/goodboy/tractor',
|
||||||
|
platforms=['linux', 'windows'],
|
||||||
|
packages=[
|
||||||
|
'tractor',
|
||||||
|
'tractor.experimental', # wacky ideas
|
||||||
|
'tractor.trionics', # trio extensions
|
||||||
|
'tractor.msg', # lowlevel data types
|
||||||
|
'tractor.devx', # "dev-experience"
|
||||||
|
],
|
||||||
|
install_requires=[
|
||||||
|
|
||||||
|
# trio related
|
||||||
|
# proper range spec:
|
||||||
|
# https://packaging.python.org/en/latest/discussions/install-requires-vs-requirements/#id5
|
||||||
|
'trio >= 0.24',
|
||||||
|
|
||||||
|
# 'async_generator', # in stdlib mostly!
|
||||||
|
# 'trio_typing', # trio==0.23.0 has type hints!
|
||||||
|
# 'exceptiongroup', # in stdlib as of 3.11!
|
||||||
|
|
||||||
|
# tooling
|
||||||
|
'stackscope',
|
||||||
|
'tricycle',
|
||||||
|
'trio_typing',
|
||||||
|
'colorlog',
|
||||||
|
'wrapt',
|
||||||
|
|
||||||
|
# IPC serialization
|
||||||
|
'msgspec>=0.18.5',
|
||||||
|
|
||||||
|
# debug mode REPL
|
||||||
|
'pdbp',
|
||||||
|
|
||||||
|
# TODO: distributed transport using
|
||||||
|
# linux kernel networking
|
||||||
|
# 'pyroute2',
|
||||||
|
|
||||||
|
# pip ref docs on these specs:
|
||||||
|
# https://pip.pypa.io/en/stable/reference/requirement-specifiers/#examples
|
||||||
|
# and pep:
|
||||||
|
# https://peps.python.org/pep-0440/#version-specifiers
|
||||||
|
|
||||||
|
],
|
||||||
|
tests_require=['pytest'],
|
||||||
|
python_requires=">=3.10",
|
||||||
|
keywords=[
|
||||||
|
'trio',
|
||||||
|
'async',
|
||||||
|
'concurrency',
|
||||||
|
'structured concurrency',
|
||||||
|
'actor model',
|
||||||
|
'distributed',
|
||||||
|
'multiprocessing'
|
||||||
|
],
|
||||||
|
classifiers=[
|
||||||
|
"Development Status :: 3 - Alpha",
|
||||||
|
"Operating System :: POSIX :: Linux",
|
||||||
|
"Operating System :: Microsoft :: Windows",
|
||||||
|
"Framework :: Trio",
|
||||||
|
"License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)",
|
||||||
|
"Programming Language :: Python :: Implementation :: CPython",
|
||||||
|
"Programming Language :: Python :: 3 :: Only",
|
||||||
|
"Programming Language :: Python :: 3.10",
|
||||||
|
"Intended Audience :: Science/Research",
|
||||||
|
"Intended Audience :: Developers",
|
||||||
|
"Topic :: System :: Distributed Computing",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
@ -1,27 +1,24 @@
|
||||||
"""
|
"""
|
||||||
Top level of the testing suites!
|
``tractor`` testing!!
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from __future__ import annotations
|
|
||||||
import sys
|
import sys
|
||||||
import subprocess
|
import subprocess
|
||||||
import os
|
import os
|
||||||
|
import random
|
||||||
import signal
|
import signal
|
||||||
import platform
|
import platform
|
||||||
import time
|
import time
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
import tractor
|
||||||
from tractor._testing import (
|
from tractor._testing import (
|
||||||
examples_dir as examples_dir,
|
examples_dir as examples_dir,
|
||||||
tractor_test as tractor_test,
|
tractor_test as tractor_test,
|
||||||
expect_ctxc as expect_ctxc,
|
expect_ctxc as expect_ctxc,
|
||||||
)
|
)
|
||||||
|
|
||||||
pytest_plugins: list[str] = [
|
# TODO: include wtv plugin(s) we build in `._testing.pytest`?
|
||||||
'pytester',
|
pytest_plugins = ['pytester']
|
||||||
'tractor._testing.pytest',
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
# Sending signal.SIGINT on subprocess fails on windows. Use CTRL_* alternatives
|
# Sending signal.SIGINT on subprocess fails on windows. Use CTRL_* alternatives
|
||||||
if platform.system() == 'Windows':
|
if platform.system() == 'Windows':
|
||||||
|
|
@ -33,11 +30,7 @@ else:
|
||||||
_KILL_SIGNAL = signal.SIGKILL
|
_KILL_SIGNAL = signal.SIGKILL
|
||||||
_INT_SIGNAL = signal.SIGINT
|
_INT_SIGNAL = signal.SIGINT
|
||||||
_INT_RETURN_CODE = 1 if sys.version_info < (3, 8) else -signal.SIGINT.value
|
_INT_RETURN_CODE = 1 if sys.version_info < (3, 8) else -signal.SIGINT.value
|
||||||
_PROC_SPAWN_WAIT = (
|
_PROC_SPAWN_WAIT = 0.6 if sys.version_info < (3, 7) else 0.4
|
||||||
0.6
|
|
||||||
if sys.version_info < (3, 7)
|
|
||||||
else 0.4
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
no_windows = pytest.mark.skipif(
|
no_windows = pytest.mark.skipif(
|
||||||
|
|
@ -46,12 +39,7 @@ no_windows = pytest.mark.skipif(
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def pytest_addoption(
|
def pytest_addoption(parser):
|
||||||
parser: pytest.Parser,
|
|
||||||
):
|
|
||||||
# ?TODO? should this be exposed from our `._testing.pytest`
|
|
||||||
# plugin or should we make it more explicit with `--tl` for
|
|
||||||
# tractor logging like we do in other client projects?
|
|
||||||
parser.addoption(
|
parser.addoption(
|
||||||
"--ll",
|
"--ll",
|
||||||
action="store",
|
action="store",
|
||||||
|
|
@ -59,10 +47,39 @@ def pytest_addoption(
|
||||||
default='ERROR', help="logging level to set when testing"
|
default='ERROR', help="logging level to set when testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
parser.addoption(
|
||||||
|
"--spawn-backend",
|
||||||
|
action="store",
|
||||||
|
dest='spawn_backend',
|
||||||
|
default='trio',
|
||||||
|
help="Processing spawning backend to use for test run",
|
||||||
|
)
|
||||||
|
|
||||||
|
parser.addoption(
|
||||||
|
"--tpdb", "--debug-mode",
|
||||||
|
action="store_true",
|
||||||
|
dest='tractor_debug_mode',
|
||||||
|
# default=False,
|
||||||
|
help=(
|
||||||
|
'Enable a flag that can be used by tests to to set the '
|
||||||
|
'`debug_mode: bool` for engaging the internal '
|
||||||
|
'multi-proc debugger sys.'
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def pytest_configure(config):
|
||||||
|
backend = config.option.spawn_backend
|
||||||
|
tractor._spawn.try_set_start_method(backend)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope='session')
|
||||||
|
def debug_mode(request):
|
||||||
|
return request.config.option.tractor_debug_mode
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='session', autouse=True)
|
@pytest.fixture(scope='session', autouse=True)
|
||||||
def loglevel(request):
|
def loglevel(request):
|
||||||
import tractor
|
|
||||||
orig = tractor.log._default_loglevel
|
orig = tractor.log._default_loglevel
|
||||||
level = tractor.log._default_loglevel = request.config.option.loglevel
|
level = tractor.log._default_loglevel = request.config.option.loglevel
|
||||||
tractor.log.get_console_log(level)
|
tractor.log.get_console_log(level)
|
||||||
|
|
@ -70,44 +87,88 @@ def loglevel(request):
|
||||||
tractor.log._default_loglevel = orig
|
tractor.log._default_loglevel = orig
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope='session')
|
||||||
|
def spawn_backend(request) -> str:
|
||||||
|
return request.config.option.spawn_backend
|
||||||
|
|
||||||
|
|
||||||
_ci_env: bool = os.environ.get('CI', False)
|
_ci_env: bool = os.environ.get('CI', False)
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='session')
|
@pytest.fixture(scope='session')
|
||||||
def ci_env() -> bool:
|
def ci_env() -> bool:
|
||||||
'''
|
'''
|
||||||
Detect CI environment.
|
Detect CI envoirment.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
return _ci_env
|
return _ci_env
|
||||||
|
|
||||||
|
|
||||||
def sig_prog(
|
# TODO: also move this to `._testing` for now?
|
||||||
proc: subprocess.Popen,
|
# -[ ] possibly generalize and re-use for multi-tree spawning
|
||||||
sig: int,
|
# along with the new stuff for multi-addrs in distribute_dis
|
||||||
canc_timeout: float = 0.1,
|
# branch?
|
||||||
) -> int:
|
#
|
||||||
|
# choose randomly at import time
|
||||||
|
_reg_addr: tuple[str, int] = (
|
||||||
|
'127.0.0.1',
|
||||||
|
random.randint(1000, 9999),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope='session')
|
||||||
|
def reg_addr() -> tuple[str, int]:
|
||||||
|
|
||||||
|
# globally override the runtime to the per-test-session-dynamic
|
||||||
|
# addr so that all tests never conflict with any other actor
|
||||||
|
# tree using the default.
|
||||||
|
from tractor import _root
|
||||||
|
_root._default_lo_addrs = [_reg_addr]
|
||||||
|
|
||||||
|
return _reg_addr
|
||||||
|
|
||||||
|
|
||||||
|
def pytest_generate_tests(metafunc):
|
||||||
|
spawn_backend = metafunc.config.option.spawn_backend
|
||||||
|
|
||||||
|
if not spawn_backend:
|
||||||
|
# XXX some weird windows bug with `pytest`?
|
||||||
|
spawn_backend = 'trio'
|
||||||
|
|
||||||
|
# TODO: maybe just use the literal `._spawn.SpawnMethodKey`?
|
||||||
|
assert spawn_backend in (
|
||||||
|
'mp_spawn',
|
||||||
|
'mp_forkserver',
|
||||||
|
'trio',
|
||||||
|
)
|
||||||
|
|
||||||
|
# NOTE: used to be used to dyanmically parametrize tests for when
|
||||||
|
# you just passed --spawn-backend=`mp` on the cli, but now we expect
|
||||||
|
# that cli input to be manually specified, BUT, maybe we'll do
|
||||||
|
# something like this again in the future?
|
||||||
|
if 'start_method' in metafunc.fixturenames:
|
||||||
|
metafunc.parametrize("start_method", [spawn_backend], scope='module')
|
||||||
|
|
||||||
|
|
||||||
|
def sig_prog(proc, sig):
|
||||||
"Kill the actor-process with ``sig``."
|
"Kill the actor-process with ``sig``."
|
||||||
proc.send_signal(sig)
|
proc.send_signal(sig)
|
||||||
time.sleep(canc_timeout)
|
time.sleep(0.1)
|
||||||
if not proc.poll():
|
if not proc.poll():
|
||||||
# TODO: why sometimes does SIGINT not work on teardown?
|
# TODO: why sometimes does SIGINT not work on teardown?
|
||||||
# seems to happen only when trace logging enabled?
|
# seems to happen only when trace logging enabled?
|
||||||
proc.send_signal(_KILL_SIGNAL)
|
proc.send_signal(_KILL_SIGNAL)
|
||||||
ret: int = proc.wait()
|
ret = proc.wait()
|
||||||
assert ret
|
assert ret
|
||||||
|
|
||||||
|
|
||||||
# TODO: factor into @cm and move to `._testing`?
|
# TODO: factor into @cm and move to `._testing`?
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def daemon(
|
def daemon(
|
||||||
debug_mode: bool,
|
|
||||||
loglevel: str,
|
loglevel: str,
|
||||||
testdir: pytest.Pytester,
|
testdir,
|
||||||
reg_addr: tuple[str, int],
|
reg_addr: tuple[str, int],
|
||||||
tpt_proto: str,
|
):
|
||||||
|
|
||||||
) -> subprocess.Popen:
|
|
||||||
'''
|
'''
|
||||||
Run a daemon root actor as a separate actor-process tree and
|
Run a daemon root actor as a separate actor-process tree and
|
||||||
"remote registrar" for discovery-protocol related tests.
|
"remote registrar" for discovery-protocol related tests.
|
||||||
|
|
@ -119,99 +180,27 @@ def daemon(
|
||||||
|
|
||||||
code: str = (
|
code: str = (
|
||||||
"import tractor; "
|
"import tractor; "
|
||||||
"tractor.run_daemon([], "
|
"tractor.run_daemon([], registry_addrs={reg_addrs}, loglevel={ll})"
|
||||||
"registry_addrs={reg_addrs}, "
|
|
||||||
"debug_mode={debug_mode}, "
|
|
||||||
"loglevel={ll})"
|
|
||||||
).format(
|
).format(
|
||||||
reg_addrs=str([reg_addr]),
|
reg_addrs=str([reg_addr]),
|
||||||
ll="'{}'".format(loglevel) if loglevel else None,
|
ll="'{}'".format(loglevel) if loglevel else None,
|
||||||
debug_mode=debug_mode,
|
|
||||||
)
|
)
|
||||||
cmd: list[str] = [
|
cmd: list[str] = [
|
||||||
sys.executable,
|
sys.executable,
|
||||||
'-c', code,
|
'-c', code,
|
||||||
]
|
]
|
||||||
# breakpoint()
|
|
||||||
kwargs = {}
|
kwargs = {}
|
||||||
if platform.system() == 'Windows':
|
if platform.system() == 'Windows':
|
||||||
# without this, tests hang on windows forever
|
# without this, tests hang on windows forever
|
||||||
kwargs['creationflags'] = subprocess.CREATE_NEW_PROCESS_GROUP
|
kwargs['creationflags'] = subprocess.CREATE_NEW_PROCESS_GROUP
|
||||||
|
|
||||||
proc: subprocess.Popen = testdir.popen(
|
proc = testdir.popen(
|
||||||
cmd,
|
cmd,
|
||||||
|
stdout=subprocess.PIPE,
|
||||||
|
stderr=subprocess.PIPE,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
)
|
)
|
||||||
|
|
||||||
# UDS sockets are **really** fast to bind()/listen()/connect()
|
|
||||||
# so it's often required that we delay a bit more starting
|
|
||||||
# the first actor-tree..
|
|
||||||
if tpt_proto == 'uds':
|
|
||||||
global _PROC_SPAWN_WAIT
|
|
||||||
_PROC_SPAWN_WAIT = 0.6
|
|
||||||
|
|
||||||
time.sleep(_PROC_SPAWN_WAIT)
|
|
||||||
|
|
||||||
assert not proc.returncode
|
assert not proc.returncode
|
||||||
|
time.sleep(_PROC_SPAWN_WAIT)
|
||||||
yield proc
|
yield proc
|
||||||
sig_prog(proc, _INT_SIGNAL)
|
sig_prog(proc, _INT_SIGNAL)
|
||||||
|
|
||||||
# XXX! yeah.. just be reaaal careful with this bc sometimes it
|
|
||||||
# can lock up on the `_io.BufferedReader` and hang..
|
|
||||||
stderr: str = proc.stderr.read().decode()
|
|
||||||
if stderr:
|
|
||||||
print(
|
|
||||||
f'Daemon actor tree produced STDERR:\n'
|
|
||||||
f'{proc.args}\n'
|
|
||||||
f'\n'
|
|
||||||
f'{stderr}\n'
|
|
||||||
)
|
|
||||||
if proc.returncode != -2:
|
|
||||||
raise RuntimeError(
|
|
||||||
'Daemon actor tree failed !?\n'
|
|
||||||
f'{proc.args}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# @pytest.fixture(autouse=True)
|
|
||||||
# def shared_last_failed(pytestconfig):
|
|
||||||
# val = pytestconfig.cache.get("example/value", None)
|
|
||||||
# breakpoint()
|
|
||||||
# if val is None:
|
|
||||||
# pytestconfig.cache.set("example/value", val)
|
|
||||||
# return val
|
|
||||||
|
|
||||||
|
|
||||||
# TODO: a way to let test scripts (like from `examples/`)
|
|
||||||
# guarantee they won't `registry_addrs` collide!
|
|
||||||
# -[ ] maybe use some kinda standard `def main()` arg-spec that
|
|
||||||
# we can introspect from a fixture that is called from the test
|
|
||||||
# body?
|
|
||||||
# -[ ] test and figure out typing for below prototype! Bp
|
|
||||||
#
|
|
||||||
# @pytest.fixture
|
|
||||||
# def set_script_runtime_args(
|
|
||||||
# reg_addr: tuple,
|
|
||||||
# ) -> Callable[[...], None]:
|
|
||||||
|
|
||||||
# def import_n_partial_in_args_n_triorun(
|
|
||||||
# script: Path, # under examples?
|
|
||||||
# **runtime_args,
|
|
||||||
# ) -> Callable[[], Any]: # a `partial`-ed equiv of `trio.run()`
|
|
||||||
|
|
||||||
# # NOTE, below is taken from
|
|
||||||
# # `.test_advanced_faults.test_ipc_channel_break_during_stream`
|
|
||||||
# mod: ModuleType = import_path(
|
|
||||||
# examples_dir() / 'advanced_faults'
|
|
||||||
# / 'ipc_failure_during_stream.py',
|
|
||||||
# root=examples_dir(),
|
|
||||||
# consider_namespace_packages=False,
|
|
||||||
# )
|
|
||||||
# return partial(
|
|
||||||
# trio.run,
|
|
||||||
# partial(
|
|
||||||
# mod.main,
|
|
||||||
# **runtime_args,
|
|
||||||
# )
|
|
||||||
# )
|
|
||||||
# return import_n_partial_in_args_n_triorun
|
|
||||||
|
|
|
||||||
|
|
@ -2,11 +2,9 @@
|
||||||
`tractor.devx.*` tooling sub-pkg test space.
|
`tractor.devx.*` tooling sub-pkg test space.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from __future__ import annotations
|
|
||||||
import time
|
import time
|
||||||
from typing import (
|
from typing import (
|
||||||
Callable,
|
Callable,
|
||||||
TYPE_CHECKING,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
@ -18,32 +16,24 @@ from pexpect.spawnbase import SpawnBase
|
||||||
from tractor._testing import (
|
from tractor._testing import (
|
||||||
mk_cmd,
|
mk_cmd,
|
||||||
)
|
)
|
||||||
from tractor.devx.debug import (
|
from tractor.devx._debug import (
|
||||||
_pause_msg as _pause_msg,
|
_pause_msg as _pause_msg,
|
||||||
_crash_msg as _crash_msg,
|
_crash_msg as _crash_msg,
|
||||||
_repl_fail_msg as _repl_fail_msg,
|
_repl_fail_msg as _repl_fail_msg,
|
||||||
_ctlc_ignore_header as _ctlc_ignore_header,
|
_ctlc_ignore_header as _ctlc_ignore_header,
|
||||||
)
|
)
|
||||||
from ..conftest import (
|
from conftest import (
|
||||||
_ci_env,
|
_ci_env,
|
||||||
)
|
)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from pexpect import pty_spawn
|
|
||||||
|
|
||||||
|
|
||||||
# a fn that sub-instantiates a `pexpect.spawn()`
|
|
||||||
# and returns it.
|
|
||||||
type PexpectSpawner = Callable[[str], pty_spawn.spawn]
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def spawn(
|
def spawn(
|
||||||
start_method: str,
|
start_method,
|
||||||
testdir: pytest.Pytester,
|
testdir: pytest.Testdir,
|
||||||
reg_addr: tuple[str, int],
|
reg_addr: tuple[str, int],
|
||||||
|
|
||||||
) -> PexpectSpawner:
|
) -> Callable[[str], None]:
|
||||||
'''
|
'''
|
||||||
Use the `pexpect` module shipped via `testdir.spawn()` to
|
Use the `pexpect` module shipped via `testdir.spawn()` to
|
||||||
run an `./examples/..` script by name.
|
run an `./examples/..` script by name.
|
||||||
|
|
@ -54,36 +44,20 @@ def spawn(
|
||||||
'`pexpect` based tests only supported on `trio` backend'
|
'`pexpect` based tests only supported on `trio` backend'
|
||||||
)
|
)
|
||||||
|
|
||||||
def unset_colors():
|
|
||||||
'''
|
|
||||||
Python 3.13 introduced colored tracebacks that break patt
|
|
||||||
matching,
|
|
||||||
|
|
||||||
https://docs.python.org/3/using/cmdline.html#envvar-PYTHON_COLORS
|
|
||||||
https://docs.python.org/3/using/cmdline.html#using-on-controlling-color
|
|
||||||
|
|
||||||
'''
|
|
||||||
import os
|
|
||||||
os.environ['PYTHON_COLORS'] = '0'
|
|
||||||
|
|
||||||
def _spawn(
|
def _spawn(
|
||||||
cmd: str,
|
cmd: str,
|
||||||
**mkcmd_kwargs,
|
**mkcmd_kwargs,
|
||||||
) -> pty_spawn.spawn:
|
):
|
||||||
unset_colors()
|
|
||||||
return testdir.spawn(
|
return testdir.spawn(
|
||||||
cmd=mk_cmd(
|
cmd=mk_cmd(
|
||||||
cmd,
|
cmd,
|
||||||
**mkcmd_kwargs,
|
**mkcmd_kwargs,
|
||||||
),
|
),
|
||||||
expect_timeout=3,
|
expect_timeout=3,
|
||||||
# preexec_fn=unset_colors,
|
|
||||||
# ^TODO? get `pytest` core to expose underlying
|
|
||||||
# `pexpect.spawn()` stuff?
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# such that test-dep can pass input script name.
|
# such that test-dep can pass input script name.
|
||||||
return _spawn # the `PexpectSpawner`, type alias.
|
return _spawn
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(
|
@pytest.fixture(
|
||||||
|
|
@ -109,19 +83,11 @@ def ctlc(
|
||||||
'https://github.com/goodboy/tractor/issues/320'
|
'https://github.com/goodboy/tractor/issues/320'
|
||||||
)
|
)
|
||||||
|
|
||||||
if mark.name == 'ctlcs_bish':
|
|
||||||
pytest.skip(
|
|
||||||
f'Test {node} prolly uses something from the stdlib (namely `asyncio`..)\n'
|
|
||||||
f'The test and/or underlying example script can *sometimes* run fine '
|
|
||||||
f'locally but more then likely until the cpython peeps get their sh#$ together, '
|
|
||||||
f'this test will definitely not behave like `trio` under SIGINT..\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
if use_ctlc:
|
if use_ctlc:
|
||||||
# XXX: disable pygments highlighting for auto-tests
|
# XXX: disable pygments highlighting for auto-tests
|
||||||
# since some envs (like actions CI) will struggle
|
# since some envs (like actions CI) will struggle
|
||||||
# the the added color-char encoding..
|
# the the added color-char encoding..
|
||||||
from tractor.devx.debug import TractorConfig
|
from tractor.devx._debug import TractorConfig
|
||||||
TractorConfig.use_pygements = False
|
TractorConfig.use_pygements = False
|
||||||
|
|
||||||
yield use_ctlc
|
yield use_ctlc
|
||||||
|
|
|
||||||
|
|
@ -1,8 +1,8 @@
|
||||||
"""
|
"""
|
||||||
That "native" debug mode better work!
|
That "native" debug mode better work!
|
||||||
|
|
||||||
All these tests can be understood (somewhat) by running the
|
All these tests can be understood (somewhat) by running the equivalent
|
||||||
equivalent `examples/debugging/` scripts manually.
|
`examples/debugging/` scripts manually.
|
||||||
|
|
||||||
TODO:
|
TODO:
|
||||||
- none of these tests have been run successfully on windows yet but
|
- none of these tests have been run successfully on windows yet but
|
||||||
|
|
@ -10,14 +10,10 @@ TODO:
|
||||||
- wonder if any of it'll work on OS X?
|
- wonder if any of it'll work on OS X?
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from __future__ import annotations
|
|
||||||
from functools import partial
|
from functools import partial
|
||||||
import itertools
|
import itertools
|
||||||
import platform
|
import platform
|
||||||
import time
|
import time
|
||||||
from typing import (
|
|
||||||
TYPE_CHECKING,
|
|
||||||
)
|
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from pexpect.exceptions import (
|
from pexpect.exceptions import (
|
||||||
|
|
@ -38,9 +34,6 @@ from .conftest import (
|
||||||
assert_before,
|
assert_before,
|
||||||
)
|
)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from ..conftest import PexpectSpawner
|
|
||||||
|
|
||||||
# TODO: The next great debugger audit could be done by you!
|
# TODO: The next great debugger audit could be done by you!
|
||||||
# - recurrent entry to breakpoint() from single actor *after* and an
|
# - recurrent entry to breakpoint() from single actor *after* and an
|
||||||
# error in another task?
|
# error in another task?
|
||||||
|
|
@ -316,12 +309,10 @@ def test_subactor_breakpoint(
|
||||||
child.expect(EOF)
|
child.expect(EOF)
|
||||||
|
|
||||||
assert in_prompt_msg(
|
assert in_prompt_msg(
|
||||||
child, [
|
child,
|
||||||
'RemoteActorError:',
|
['RemoteActorError:',
|
||||||
"('breakpoint_forever'",
|
"('breakpoint_forever'",
|
||||||
'bdb.BdbQuit',
|
'bdb.BdbQuit',]
|
||||||
],
|
|
||||||
pause_on_false=True,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -534,7 +525,7 @@ def test_multi_daemon_subactors(
|
||||||
# now the root actor won't clobber the bp_forever child
|
# now the root actor won't clobber the bp_forever child
|
||||||
# during it's first access to the debug lock, but will instead
|
# during it's first access to the debug lock, but will instead
|
||||||
# wait for the lock to release, by the edge triggered
|
# wait for the lock to release, by the edge triggered
|
||||||
# ``devx.debug.Lock.no_remote_has_tty`` event before sending cancel messages
|
# ``devx._debug.Lock.no_remote_has_tty`` event before sending cancel messages
|
||||||
# (via portals) to its underlings B)
|
# (via portals) to its underlings B)
|
||||||
|
|
||||||
# at some point here there should have been some warning msg from
|
# at some point here there should have been some warning msg from
|
||||||
|
|
@ -925,7 +916,6 @@ def test_post_mortem_api(
|
||||||
"<Task 'name_error'",
|
"<Task 'name_error'",
|
||||||
"NameError",
|
"NameError",
|
||||||
"('child'",
|
"('child'",
|
||||||
'getattr(doggypants)', # exc-LoC
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
if ctlc:
|
if ctlc:
|
||||||
|
|
@ -942,8 +932,8 @@ def test_post_mortem_api(
|
||||||
"<Task '__main__.main'",
|
"<Task '__main__.main'",
|
||||||
"('root'",
|
"('root'",
|
||||||
"NameError",
|
"NameError",
|
||||||
|
"tractor.post_mortem()",
|
||||||
"src_uid=('child'",
|
"src_uid=('child'",
|
||||||
"tractor.post_mortem()", # in `main()`-LoC
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
if ctlc:
|
if ctlc:
|
||||||
|
|
@ -961,10 +951,6 @@ def test_post_mortem_api(
|
||||||
"('root'",
|
"('root'",
|
||||||
"NameError",
|
"NameError",
|
||||||
"src_uid=('child'",
|
"src_uid=('child'",
|
||||||
|
|
||||||
# raising line in `main()` but from crash-handling
|
|
||||||
# in `tractor.open_nursery()`.
|
|
||||||
'async with p.open_context(name_error) as (ctx, first):',
|
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
if ctlc:
|
if ctlc:
|
||||||
|
|
@ -1074,136 +1060,6 @@ def test_shield_pause(
|
||||||
child.expect(EOF)
|
child.expect(EOF)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'quit_early', [False, True]
|
|
||||||
)
|
|
||||||
def test_ctxep_pauses_n_maybe_ipc_breaks(
|
|
||||||
spawn: PexpectSpawner,
|
|
||||||
quit_early: bool,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Audit generator embedded `.pause()`es from within a `@context`
|
|
||||||
endpoint with a chan close at the end, requiring that ctl-c is
|
|
||||||
mashed and zombie reaper kills sub with no hangs.
|
|
||||||
|
|
||||||
'''
|
|
||||||
child = spawn('subactor_bp_in_ctx')
|
|
||||||
child.expect(PROMPT)
|
|
||||||
|
|
||||||
# 3 iters for the `gen()` pause-points
|
|
||||||
for i in range(3):
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
[
|
|
||||||
_pause_msg,
|
|
||||||
"('bp_boi'", # actor name
|
|
||||||
"<Task 'just_bp'", # task name
|
|
||||||
]
|
|
||||||
)
|
|
||||||
if (
|
|
||||||
i == 1
|
|
||||||
and
|
|
||||||
quit_early
|
|
||||||
):
|
|
||||||
child.sendline('q')
|
|
||||||
child.expect(PROMPT)
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
["tractor._exceptions.RemoteActorError: remote task raised a 'BdbQuit'",
|
|
||||||
"bdb.BdbQuit",
|
|
||||||
"('bp_boi'",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
child.sendline('c')
|
|
||||||
child.expect(EOF)
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
["tractor._exceptions.RemoteActorError: remote task raised a 'BdbQuit'",
|
|
||||||
"bdb.BdbQuit",
|
|
||||||
"('bp_boi'",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
break # end-of-test
|
|
||||||
|
|
||||||
child.sendline('c')
|
|
||||||
try:
|
|
||||||
child.expect(PROMPT)
|
|
||||||
except TIMEOUT:
|
|
||||||
# no prompt since we hang due to IPC chan purposely
|
|
||||||
# closed so verify we see error reporting as well as
|
|
||||||
# a failed crash-REPL request msg and can CTL-c our way
|
|
||||||
# out.
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
['peer IPC channel closed abruptly?',
|
|
||||||
'another task closed this fd',
|
|
||||||
'Debug lock request was CANCELLED?',
|
|
||||||
"TransportClosed: 'MsgpackUDSStream' was already closed locally ?",]
|
|
||||||
|
|
||||||
# XXX races on whether these show/hit?
|
|
||||||
# 'Failed to REPl via `_pause()` You called `tractor.pause()` from an already cancelled scope!',
|
|
||||||
# 'AssertionError',
|
|
||||||
)
|
|
||||||
# OSc(ancel) the hanging tree
|
|
||||||
do_ctlc(
|
|
||||||
child=child,
|
|
||||||
expect_prompt=False,
|
|
||||||
)
|
|
||||||
child.expect(EOF)
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
['KeyboardInterrupt'],
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def test_crash_handling_within_cancelled_root_actor(
|
|
||||||
spawn: PexpectSpawner,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Ensure that when only a root-actor is started via `open_root_actor()`
|
|
||||||
we can crash-handle in debug-mode despite self-cancellation.
|
|
||||||
|
|
||||||
More-or-less ensures we conditionally shield the pause in
|
|
||||||
`._root.open_root_actor()`'s `await debug._maybe_enter_pm()`
|
|
||||||
call.
|
|
||||||
|
|
||||||
'''
|
|
||||||
child = spawn('root_self_cancelled_w_error')
|
|
||||||
child.expect(PROMPT)
|
|
||||||
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
[
|
|
||||||
"Actor.cancel_soon()` was called!",
|
|
||||||
"root cancelled",
|
|
||||||
_pause_msg,
|
|
||||||
"('root'", # actor name
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
child.sendline('c')
|
|
||||||
child.expect(PROMPT)
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
[
|
|
||||||
_crash_msg,
|
|
||||||
"('root'", # actor name
|
|
||||||
"AssertionError",
|
|
||||||
"assert 0",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
child.sendline('c')
|
|
||||||
child.expect(EOF)
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
[
|
|
||||||
"AssertionError",
|
|
||||||
"assert 0",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# TODO: better error for "non-ideal" usage from the root actor.
|
# TODO: better error for "non-ideal" usage from the root actor.
|
||||||
# -[ ] if called from an async scope emit a message that suggests
|
# -[ ] if called from an async scope emit a message that suggests
|
||||||
# using `await tractor.pause()` instead since it's less overhead
|
# using `await tractor.pause()` instead since it's less overhead
|
||||||
|
|
|
||||||
|
|
@ -6,9 +6,6 @@ All these tests can be understood (somewhat) by running the
|
||||||
equivalent `examples/debugging/` scripts manually.
|
equivalent `examples/debugging/` scripts manually.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from contextlib import (
|
|
||||||
contextmanager as cm,
|
|
||||||
)
|
|
||||||
# from functools import partial
|
# from functools import partial
|
||||||
# import itertools
|
# import itertools
|
||||||
import time
|
import time
|
||||||
|
|
@ -18,7 +15,7 @@ import time
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from pexpect.exceptions import (
|
from pexpect.exceptions import (
|
||||||
TIMEOUT,
|
# TIMEOUT,
|
||||||
EOF,
|
EOF,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -35,23 +32,7 @@ from .conftest import (
|
||||||
# _repl_fail_msg,
|
# _repl_fail_msg,
|
||||||
)
|
)
|
||||||
|
|
||||||
@cm
|
|
||||||
def maybe_expect_timeout(
|
|
||||||
ctlc: bool = False,
|
|
||||||
) -> None:
|
|
||||||
try:
|
|
||||||
yield
|
|
||||||
except TIMEOUT:
|
|
||||||
# breakpoint()
|
|
||||||
if ctlc:
|
|
||||||
pytest.xfail(
|
|
||||||
'Some kinda redic threading SIGINT bug i think?\n'
|
|
||||||
'See the notes in `examples/debugging/sync_bp.py`..\n'
|
|
||||||
)
|
|
||||||
raise
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.ctlcs_bish
|
|
||||||
def test_pause_from_sync(
|
def test_pause_from_sync(
|
||||||
spawn,
|
spawn,
|
||||||
ctlc: bool,
|
ctlc: bool,
|
||||||
|
|
@ -86,10 +67,10 @@ def test_pause_from_sync(
|
||||||
child.expect(PROMPT)
|
child.expect(PROMPT)
|
||||||
|
|
||||||
# XXX shouldn't see gb loaded message with PDB loglevel!
|
# XXX shouldn't see gb loaded message with PDB loglevel!
|
||||||
# assert not in_prompt_msg(
|
assert not in_prompt_msg(
|
||||||
# child,
|
child,
|
||||||
# ['`greenback` portal opened!'],
|
['`greenback` portal opened!'],
|
||||||
# )
|
)
|
||||||
# should be same root task
|
# should be same root task
|
||||||
assert_before(
|
assert_before(
|
||||||
child,
|
child,
|
||||||
|
|
@ -181,13 +162,6 @@ def test_pause_from_sync(
|
||||||
)
|
)
|
||||||
|
|
||||||
child.sendline('c')
|
child.sendline('c')
|
||||||
|
|
||||||
# XXX TODO, weird threading bug it seems despite the
|
|
||||||
# `abandon_on_cancel: bool` setting to
|
|
||||||
# `trio.to_thread.run_sync()`..
|
|
||||||
with maybe_expect_timeout(
|
|
||||||
ctlc=ctlc,
|
|
||||||
):
|
|
||||||
child.expect(EOF)
|
child.expect(EOF)
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -244,12 +218,11 @@ def expect_any_of(
|
||||||
)
|
)
|
||||||
|
|
||||||
return expected_patts
|
return expected_patts
|
||||||
|
# yield child
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.ctlcs_bish
|
def test_pause_from_asyncio_task(
|
||||||
def test_sync_pause_from_aio_task(
|
|
||||||
spawn,
|
spawn,
|
||||||
|
|
||||||
ctlc: bool
|
ctlc: bool
|
||||||
# ^TODO, fix for `asyncio`!!
|
# ^TODO, fix for `asyncio`!!
|
||||||
):
|
):
|
||||||
|
|
@ -298,12 +271,10 @@ def test_sync_pause_from_aio_task(
|
||||||
# error raised in `asyncio.Task`
|
# error raised in `asyncio.Task`
|
||||||
"raise ValueError('asyncio side error!')": [
|
"raise ValueError('asyncio side error!')": [
|
||||||
_crash_msg,
|
_crash_msg,
|
||||||
|
'return await chan.receive()', # `.to_asyncio` impl internals in tb
|
||||||
"<Task 'trio_ctx'",
|
"<Task 'trio_ctx'",
|
||||||
"@ ('aio_daemon'",
|
"@ ('aio_daemon'",
|
||||||
"ValueError: asyncio side error!",
|
"ValueError: asyncio side error!",
|
||||||
|
|
||||||
# XXX, we no longer show this frame by default!
|
|
||||||
# 'return await chan.receive()', # `.to_asyncio` impl internals in tb
|
|
||||||
],
|
],
|
||||||
|
|
||||||
# parent-side propagation via actor-nursery/portal
|
# parent-side propagation via actor-nursery/portal
|
||||||
|
|
@ -355,27 +326,4 @@ def test_sync_pause_from_aio_task(
|
||||||
)
|
)
|
||||||
|
|
||||||
child.sendline('c')
|
child.sendline('c')
|
||||||
# with maybe_expect_timeout():
|
|
||||||
child.expect(EOF)
|
child.expect(EOF)
|
||||||
|
|
||||||
|
|
||||||
def test_sync_pause_from_non_greenbacked_aio_task():
|
|
||||||
'''
|
|
||||||
Where the `breakpoint()` caller task is NOT spawned by
|
|
||||||
`tractor.to_asyncio` and thus never activates
|
|
||||||
a `greenback.ensure_portal()` beforehand, presumably bc the task
|
|
||||||
was started by some lib/dep as in often seen in the field.
|
|
||||||
|
|
||||||
Ensure sync pausing works when the pause is in,
|
|
||||||
|
|
||||||
- the root actor running in infected-mode?
|
|
||||||
|_ since we don't need any IPC to acquire the debug lock?
|
|
||||||
|_ is there some way to handle this like the non-main-thread case?
|
|
||||||
|
|
||||||
All other cases need to error out appropriately right?
|
|
||||||
|
|
||||||
- for any subactor we can't avoid needing the repl lock..
|
|
||||||
|_ is there a way to hook into `asyncio.ensure_future(obj)`?
|
|
||||||
|
|
||||||
'''
|
|
||||||
pass
|
|
||||||
|
|
|
||||||
|
|
@ -13,37 +13,18 @@ TODO:
|
||||||
when debugging a problem inside the stack vs. in their app.
|
when debugging a problem inside the stack vs. in their app.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from __future__ import annotations
|
|
||||||
from contextlib import (
|
|
||||||
contextmanager as cm,
|
|
||||||
)
|
|
||||||
import os
|
import os
|
||||||
import signal
|
import signal
|
||||||
import time
|
|
||||||
from typing import (
|
|
||||||
TYPE_CHECKING,
|
|
||||||
)
|
|
||||||
|
|
||||||
from .conftest import (
|
from .conftest import (
|
||||||
expect,
|
expect,
|
||||||
assert_before,
|
assert_before,
|
||||||
in_prompt_msg,
|
# in_prompt_msg,
|
||||||
PROMPT,
|
|
||||||
_pause_msg,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
import pytest
|
|
||||||
from pexpect.exceptions import (
|
|
||||||
# TIMEOUT,
|
|
||||||
EOF,
|
|
||||||
)
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from ..conftest import PexpectSpawner
|
|
||||||
|
|
||||||
|
|
||||||
def test_shield_pause(
|
def test_shield_pause(
|
||||||
spawn: PexpectSpawner,
|
spawn,
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Verify the `tractor.pause()/.post_mortem()` API works inside an
|
Verify the `tractor.pause()/.post_mortem()` API works inside an
|
||||||
|
|
@ -66,39 +47,41 @@ def test_shield_pause(
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
script_pid: int = child.pid
|
|
||||||
print(
|
print(
|
||||||
f'Sending SIGUSR1 to {script_pid}\n'
|
'Sending SIGUSR1 to see a tree-trace!',
|
||||||
f'(kill -s SIGUSR1 {script_pid})\n'
|
|
||||||
)
|
)
|
||||||
os.kill(
|
os.kill(
|
||||||
script_pid,
|
child.pid,
|
||||||
signal.SIGUSR1,
|
signal.SIGUSR1,
|
||||||
)
|
)
|
||||||
time.sleep(0.2)
|
|
||||||
expect(
|
expect(
|
||||||
child,
|
child,
|
||||||
# end-of-tree delimiter
|
# end-of-tree delimiter
|
||||||
"end-of-\('root'",
|
"------ \('root', ",
|
||||||
)
|
)
|
||||||
|
|
||||||
assert_before(
|
assert_before(
|
||||||
child,
|
child,
|
||||||
[
|
[
|
||||||
# 'Srying to dump `stackscope` tree..',
|
'Trying to dump `stackscope` tree..',
|
||||||
# 'Dumping `stackscope` tree for actor',
|
'Dumping `stackscope` tree for actor',
|
||||||
"('root'", # uid line
|
"('root'", # uid line
|
||||||
|
|
||||||
# TODO!? this used to show?
|
|
||||||
# -[ ] mk reproducable for @oremanj?
|
|
||||||
#
|
|
||||||
# parent block point (non-shielded)
|
# parent block point (non-shielded)
|
||||||
# 'await trio.sleep_forever() # in root',
|
'await trio.sleep_forever() # in root',
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# expect(
|
||||||
|
# child,
|
||||||
|
# # relay to the sub should be reported
|
||||||
|
# 'Relaying `SIGUSR1`[10] to sub-actor',
|
||||||
|
# )
|
||||||
|
|
||||||
expect(
|
expect(
|
||||||
child,
|
child,
|
||||||
# end-of-tree delimiter
|
# end-of-tree delimiter
|
||||||
"end-of-\('hanger'",
|
"------ \('hanger', ",
|
||||||
)
|
)
|
||||||
assert_before(
|
assert_before(
|
||||||
child,
|
child,
|
||||||
|
|
@ -108,11 +91,11 @@ def test_shield_pause(
|
||||||
|
|
||||||
"('hanger'", # uid line
|
"('hanger'", # uid line
|
||||||
|
|
||||||
# TODO!? SEE ABOVE
|
|
||||||
# hanger LOC where it's shield-halted
|
# hanger LOC where it's shield-halted
|
||||||
# 'await trio.sleep_forever() # in subactor',
|
'await trio.sleep_forever() # in subactor',
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
# breakpoint()
|
||||||
|
|
||||||
# simulate the user sending a ctl-c to the hanging program.
|
# simulate the user sending a ctl-c to the hanging program.
|
||||||
# this should result in the terminator kicking in since
|
# this should result in the terminator kicking in since
|
||||||
|
|
@ -121,11 +104,9 @@ def test_shield_pause(
|
||||||
child.pid,
|
child.pid,
|
||||||
signal.SIGINT,
|
signal.SIGINT,
|
||||||
)
|
)
|
||||||
from tractor._supervise import _shutdown_msg
|
|
||||||
expect(
|
expect(
|
||||||
child,
|
child,
|
||||||
# 'Shutting down actor runtime',
|
'Shutting down actor runtime',
|
||||||
_shutdown_msg,
|
|
||||||
timeout=6,
|
timeout=6,
|
||||||
)
|
)
|
||||||
assert_before(
|
assert_before(
|
||||||
|
|
@ -137,170 +118,3 @@ def test_shield_pause(
|
||||||
"'--uid', \"('hanger',",
|
"'--uid', \"('hanger',",
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_breakpoint_hook_restored(
|
|
||||||
spawn: PexpectSpawner,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Ensures our actor runtime sets a custom `breakpoint()` hook
|
|
||||||
on open then restores the stdlib's default on close.
|
|
||||||
|
|
||||||
The hook state validation is done via `assert`s inside the
|
|
||||||
invoked script with only `breakpoint()` (not `tractor.pause()`)
|
|
||||||
calls used.
|
|
||||||
|
|
||||||
'''
|
|
||||||
child = spawn('restore_builtin_breakpoint')
|
|
||||||
|
|
||||||
child.expect(PROMPT)
|
|
||||||
try:
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
[
|
|
||||||
_pause_msg,
|
|
||||||
"<Task '__main__.main'",
|
|
||||||
"('root'",
|
|
||||||
"first bp, tractor hook set",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
# XXX if the above raises `AssertionError`, without sending
|
|
||||||
# the final 'continue' cmd to the REPL-active sub-process,
|
|
||||||
# we'll hang waiting for that pexpect instance to terminate..
|
|
||||||
finally:
|
|
||||||
child.sendline('c')
|
|
||||||
|
|
||||||
child.expect(PROMPT)
|
|
||||||
assert_before(
|
|
||||||
child,
|
|
||||||
[
|
|
||||||
"last bp, stdlib hook restored",
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
# since the stdlib hook was already restored there should be NO
|
|
||||||
# `tractor` `log.pdb()` content from console!
|
|
||||||
assert not in_prompt_msg(
|
|
||||||
child,
|
|
||||||
[
|
|
||||||
_pause_msg,
|
|
||||||
"<Task '__main__.main'",
|
|
||||||
"('root'",
|
|
||||||
],
|
|
||||||
)
|
|
||||||
child.sendline('c')
|
|
||||||
child.expect(EOF)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
_to_raise = Exception('Triggering a crash')
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'to_raise',
|
|
||||||
[
|
|
||||||
None,
|
|
||||||
_to_raise,
|
|
||||||
RuntimeError('Never crash handle this!'),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'raise_on_exit',
|
|
||||||
[
|
|
||||||
True,
|
|
||||||
[type(_to_raise)],
|
|
||||||
False,
|
|
||||||
]
|
|
||||||
)
|
|
||||||
def test_crash_handler_cms(
|
|
||||||
debug_mode: bool,
|
|
||||||
to_raise: Exception,
|
|
||||||
raise_on_exit: bool|list[Exception],
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Verify the `.devx.open_crash_handler()` API(s) by also
|
|
||||||
(conveniently enough) tesing its `repl_fixture: ContextManager`
|
|
||||||
param support which for this suite allows use to avoid use of
|
|
||||||
a `pexpect`-style-test since we use the fixture to avoid actually
|
|
||||||
entering `PdbpREPL.iteract()` :smirk:
|
|
||||||
|
|
||||||
'''
|
|
||||||
import tractor
|
|
||||||
# import trio
|
|
||||||
|
|
||||||
# state flags
|
|
||||||
repl_acquired: bool = False
|
|
||||||
repl_released: bool = False
|
|
||||||
|
|
||||||
@cm
|
|
||||||
def block_repl_ux(
|
|
||||||
repl: tractor.devx.debug.PdbREPL,
|
|
||||||
maybe_bxerr: (
|
|
||||||
tractor.devx._debug.BoxedMaybeException
|
|
||||||
|None
|
|
||||||
) = None,
|
|
||||||
enter_repl: bool = True,
|
|
||||||
|
|
||||||
) -> bool:
|
|
||||||
'''
|
|
||||||
Set pre/post-REPL state vars and bypass actual conole
|
|
||||||
interaction.
|
|
||||||
|
|
||||||
'''
|
|
||||||
nonlocal repl_acquired, repl_released
|
|
||||||
|
|
||||||
# task: trio.Task = trio.lowlevel.current_task()
|
|
||||||
# print(f'pre-REPL active_task={task.name}')
|
|
||||||
|
|
||||||
print('pre-REPL')
|
|
||||||
repl_acquired = True
|
|
||||||
yield False # never actually .interact()
|
|
||||||
print('post-REPL')
|
|
||||||
repl_released = True
|
|
||||||
|
|
||||||
try:
|
|
||||||
# TODO, with runtime's `debug_mode` setting
|
|
||||||
# -[ ] need to open runtime tho obvi..
|
|
||||||
#
|
|
||||||
# with tractor.devx.maybe_open_crash_handler(
|
|
||||||
# pdb=True,
|
|
||||||
|
|
||||||
with tractor.devx.open_crash_handler(
|
|
||||||
raise_on_exit=raise_on_exit,
|
|
||||||
repl_fixture=block_repl_ux
|
|
||||||
) as bxerr:
|
|
||||||
if to_raise is not None:
|
|
||||||
raise to_raise
|
|
||||||
|
|
||||||
except Exception as _exc:
|
|
||||||
exc = _exc
|
|
||||||
if (
|
|
||||||
raise_on_exit is True
|
|
||||||
or
|
|
||||||
type(to_raise) in raise_on_exit
|
|
||||||
):
|
|
||||||
assert (
|
|
||||||
exc
|
|
||||||
is
|
|
||||||
to_raise
|
|
||||||
is
|
|
||||||
bxerr.value
|
|
||||||
)
|
|
||||||
|
|
||||||
else:
|
|
||||||
raise
|
|
||||||
else:
|
|
||||||
assert (
|
|
||||||
to_raise is None
|
|
||||||
or
|
|
||||||
not raise_on_exit
|
|
||||||
or
|
|
||||||
type(to_raise) not in raise_on_exit
|
|
||||||
)
|
|
||||||
assert bxerr.value is to_raise
|
|
||||||
|
|
||||||
assert bxerr.raise_on_exit == raise_on_exit
|
|
||||||
|
|
||||||
if to_raise is not None:
|
|
||||||
assert repl_acquired
|
|
||||||
assert repl_released
|
|
||||||
|
|
|
||||||
|
|
@ -1,4 +0,0 @@
|
||||||
'''
|
|
||||||
`tractor.ipc` subsystem(s)/unit testing suites.
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
@ -1,114 +0,0 @@
|
||||||
'''
|
|
||||||
Unit-ish tests for specific IPC transport protocol backends.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
import trio
|
|
||||||
import tractor
|
|
||||||
from tractor import (
|
|
||||||
Actor,
|
|
||||||
_state,
|
|
||||||
_addr,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
|
||||||
def bindspace_dir_str() -> str:
|
|
||||||
|
|
||||||
rt_dir: Path = tractor._state.get_rt_dir()
|
|
||||||
bs_dir: Path = rt_dir / 'doggy'
|
|
||||||
bs_dir_str: str = str(bs_dir)
|
|
||||||
assert not bs_dir.is_dir()
|
|
||||||
|
|
||||||
yield bs_dir_str
|
|
||||||
|
|
||||||
# delete it on suite teardown.
|
|
||||||
# ?TODO? should we support this internally
|
|
||||||
# or is leaking it ok?
|
|
||||||
if bs_dir.is_dir():
|
|
||||||
bs_dir.rmdir()
|
|
||||||
|
|
||||||
|
|
||||||
def test_uds_bindspace_created_implicitly(
|
|
||||||
debug_mode: bool,
|
|
||||||
bindspace_dir_str: str,
|
|
||||||
):
|
|
||||||
registry_addr: tuple = (
|
|
||||||
f'{bindspace_dir_str}',
|
|
||||||
'registry@doggy.sock',
|
|
||||||
)
|
|
||||||
bs_dir_str: str = registry_addr[0]
|
|
||||||
|
|
||||||
# XXX, ensure bindspace-dir DNE beforehand!
|
|
||||||
assert not Path(bs_dir_str).is_dir()
|
|
||||||
|
|
||||||
async def main():
|
|
||||||
async with tractor.open_nursery(
|
|
||||||
enable_transports=['uds'],
|
|
||||||
registry_addrs=[registry_addr],
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
) as _an:
|
|
||||||
|
|
||||||
# XXX MUST be created implicitly by
|
|
||||||
# `.ipc._uds.start_listener()`!
|
|
||||||
assert Path(bs_dir_str).is_dir()
|
|
||||||
|
|
||||||
root: Actor = tractor.current_actor()
|
|
||||||
assert root.is_registrar
|
|
||||||
|
|
||||||
assert registry_addr in root.reg_addrs
|
|
||||||
assert (
|
|
||||||
registry_addr
|
|
||||||
in
|
|
||||||
_state._runtime_vars['_registry_addrs']
|
|
||||||
)
|
|
||||||
assert (
|
|
||||||
_addr.wrap_address(registry_addr)
|
|
||||||
in
|
|
||||||
root.registry_addrs
|
|
||||||
)
|
|
||||||
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
|
|
||||||
def test_uds_double_listen_raises_connerr(
|
|
||||||
debug_mode: bool,
|
|
||||||
bindspace_dir_str: str,
|
|
||||||
):
|
|
||||||
registry_addr: tuple = (
|
|
||||||
f'{bindspace_dir_str}',
|
|
||||||
'registry@doggy.sock',
|
|
||||||
)
|
|
||||||
|
|
||||||
async def main():
|
|
||||||
async with tractor.open_nursery(
|
|
||||||
enable_transports=['uds'],
|
|
||||||
registry_addrs=[registry_addr],
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
) as _an:
|
|
||||||
|
|
||||||
# runtime up
|
|
||||||
root: Actor = tractor.current_actor()
|
|
||||||
|
|
||||||
from tractor.ipc._uds import (
|
|
||||||
start_listener,
|
|
||||||
UDSAddress,
|
|
||||||
)
|
|
||||||
ya_bound_addr: UDSAddress = root.registry_addrs[0]
|
|
||||||
try:
|
|
||||||
await start_listener(
|
|
||||||
addr=ya_bound_addr,
|
|
||||||
)
|
|
||||||
except ConnectionError as connerr:
|
|
||||||
assert type(src_exc := connerr.__context__) is OSError
|
|
||||||
assert 'Address already in use' in src_exc.args
|
|
||||||
# complete, exit test.
|
|
||||||
|
|
||||||
else:
|
|
||||||
pytest.fail('It dint raise a connerr !?')
|
|
||||||
|
|
||||||
|
|
||||||
trio.run(main)
|
|
||||||
|
|
@ -1,95 +0,0 @@
|
||||||
'''
|
|
||||||
Verify the `enable_transports` param drives various
|
|
||||||
per-root/sub-actor IPC endpoint/server settings.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
import trio
|
|
||||||
import tractor
|
|
||||||
from tractor import (
|
|
||||||
Actor,
|
|
||||||
Portal,
|
|
||||||
ipc,
|
|
||||||
msg,
|
|
||||||
_state,
|
|
||||||
_addr,
|
|
||||||
)
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def chk_tpts(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
tpt_proto_key: str,
|
|
||||||
):
|
|
||||||
rtvars = _state._runtime_vars
|
|
||||||
assert (
|
|
||||||
tpt_proto_key
|
|
||||||
in
|
|
||||||
rtvars['_enable_tpts']
|
|
||||||
)
|
|
||||||
actor: Actor = tractor.current_actor()
|
|
||||||
spec: msg.types.SpawnSpec = actor._spawn_spec
|
|
||||||
assert spec._runtime_vars == rtvars
|
|
||||||
|
|
||||||
# ensure individual IPC ep-addr types
|
|
||||||
serv: ipc._server.Server = actor.ipc_server
|
|
||||||
addr: ipc._types.Address
|
|
||||||
for addr in serv.addrs:
|
|
||||||
assert addr.proto_key == tpt_proto_key
|
|
||||||
|
|
||||||
# Actor delegate-props enforcement
|
|
||||||
assert (
|
|
||||||
actor.accept_addrs
|
|
||||||
==
|
|
||||||
serv.accept_addrs
|
|
||||||
)
|
|
||||||
|
|
||||||
await ctx.started(serv.accept_addrs)
|
|
||||||
|
|
||||||
|
|
||||||
# TODO, parametrize over mis-matched-proto-typed `registry_addrs`
|
|
||||||
# since i seems to work in `piker` but not exactly sure if both tcp
|
|
||||||
# & uds are being deployed then?
|
|
||||||
#
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'tpt_proto_key',
|
|
||||||
['tcp', 'uds'],
|
|
||||||
ids=lambda item: f'ipc_tpt={item!r}'
|
|
||||||
)
|
|
||||||
def test_root_passes_tpt_to_sub(
|
|
||||||
tpt_proto_key: str,
|
|
||||||
reg_addr: tuple,
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
async def main():
|
|
||||||
async with tractor.open_nursery(
|
|
||||||
enable_transports=[tpt_proto_key],
|
|
||||||
registry_addrs=[reg_addr],
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
) as an:
|
|
||||||
|
|
||||||
assert (
|
|
||||||
tpt_proto_key
|
|
||||||
in
|
|
||||||
_state._runtime_vars['_enable_tpts']
|
|
||||||
)
|
|
||||||
|
|
||||||
ptl: Portal = await an.start_actor(
|
|
||||||
name='sub',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
)
|
|
||||||
async with ptl.open_context(
|
|
||||||
chk_tpts,
|
|
||||||
tpt_proto_key=tpt_proto_key,
|
|
||||||
) as (ctx, accept_addrs):
|
|
||||||
|
|
||||||
uw_addr: tuple
|
|
||||||
for uw_addr in accept_addrs:
|
|
||||||
addr = _addr.wrap_address(uw_addr)
|
|
||||||
assert addr.is_valid
|
|
||||||
|
|
||||||
# shudown sub-actor(s)
|
|
||||||
await an.cancel()
|
|
||||||
|
|
||||||
trio.run(main)
|
|
||||||
|
|
@ -1,72 +0,0 @@
|
||||||
'''
|
|
||||||
High-level `.ipc._server` unit tests.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
import trio
|
|
||||||
from tractor import (
|
|
||||||
devx,
|
|
||||||
ipc,
|
|
||||||
log,
|
|
||||||
)
|
|
||||||
from tractor._testing.addr import (
|
|
||||||
get_rando_addr,
|
|
||||||
)
|
|
||||||
# TODO, use/check-roundtripping with some of these wrapper types?
|
|
||||||
#
|
|
||||||
# from .._addr import Address
|
|
||||||
# from ._chan import Channel
|
|
||||||
# from ._transport import MsgTransport
|
|
||||||
# from ._uds import UDSAddress
|
|
||||||
# from ._tcp import TCPAddress
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'_tpt_proto',
|
|
||||||
['uds', 'tcp']
|
|
||||||
)
|
|
||||||
def test_basic_ipc_server(
|
|
||||||
_tpt_proto: str,
|
|
||||||
debug_mode: bool,
|
|
||||||
loglevel: str,
|
|
||||||
):
|
|
||||||
|
|
||||||
# so we see the socket-listener reporting on console
|
|
||||||
log.get_console_log("INFO")
|
|
||||||
|
|
||||||
rando_addr: tuple = get_rando_addr(
|
|
||||||
tpt_proto=_tpt_proto,
|
|
||||||
)
|
|
||||||
async def main():
|
|
||||||
async with ipc._server.open_ipc_server() as server:
|
|
||||||
|
|
||||||
assert (
|
|
||||||
server._parent_tn
|
|
||||||
and
|
|
||||||
server._parent_tn is server._stream_handler_tn
|
|
||||||
)
|
|
||||||
assert server._no_more_peers.is_set()
|
|
||||||
|
|
||||||
eps: list[ipc._server.Endpoint] = await server.listen_on(
|
|
||||||
accept_addrs=[rando_addr],
|
|
||||||
stream_handler_nursery=None,
|
|
||||||
)
|
|
||||||
assert (
|
|
||||||
len(eps) == 1
|
|
||||||
and
|
|
||||||
(ep := eps[0])._listener
|
|
||||||
and
|
|
||||||
not ep.peer_tpts
|
|
||||||
)
|
|
||||||
|
|
||||||
server._parent_tn.cancel_scope.cancel()
|
|
||||||
|
|
||||||
# !TODO! actually make a bg-task connection from a client
|
|
||||||
# using `ipc._chan._connect_chan()`
|
|
||||||
|
|
||||||
with devx.maybe_open_crash_handler(
|
|
||||||
pdb=debug_mode,
|
|
||||||
):
|
|
||||||
trio.run(main)
|
|
||||||
|
|
@ -3,6 +3,7 @@ Sketchy network blackoutz, ugly byzantine gens, puedes eschuchar la
|
||||||
cancelacion?..
|
cancelacion?..
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
import itertools
|
||||||
from functools import partial
|
from functools import partial
|
||||||
from types import ModuleType
|
from types import ModuleType
|
||||||
|
|
||||||
|
|
@ -10,9 +11,6 @@ import pytest
|
||||||
from _pytest.pathlib import import_path
|
from _pytest.pathlib import import_path
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
from tractor import (
|
|
||||||
TransportClosed,
|
|
||||||
)
|
|
||||||
from tractor._testing import (
|
from tractor._testing import (
|
||||||
examples_dir,
|
examples_dir,
|
||||||
break_ipc,
|
break_ipc,
|
||||||
|
|
@ -77,7 +75,6 @@ def test_ipc_channel_break_during_stream(
|
||||||
spawn_backend: str,
|
spawn_backend: str,
|
||||||
ipc_break: dict|None,
|
ipc_break: dict|None,
|
||||||
pre_aclose_msgstream: bool,
|
pre_aclose_msgstream: bool,
|
||||||
tpt_proto: str,
|
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Ensure we can have an IPC channel break its connection during
|
Ensure we can have an IPC channel break its connection during
|
||||||
|
|
@ -95,7 +92,7 @@ def test_ipc_channel_break_during_stream(
|
||||||
# non-`trio` spawners should never hit the hang condition that
|
# non-`trio` spawners should never hit the hang condition that
|
||||||
# requires the user to do ctl-c to cancel the actor tree.
|
# requires the user to do ctl-c to cancel the actor tree.
|
||||||
# expect_final_exc = trio.ClosedResourceError
|
# expect_final_exc = trio.ClosedResourceError
|
||||||
expect_final_exc = TransportClosed
|
expect_final_exc = tractor.TransportClosed
|
||||||
|
|
||||||
mod: ModuleType = import_path(
|
mod: ModuleType = import_path(
|
||||||
examples_dir() / 'advanced_faults'
|
examples_dir() / 'advanced_faults'
|
||||||
|
|
@ -108,8 +105,6 @@ def test_ipc_channel_break_during_stream(
|
||||||
# period" wherein the user eventually hits ctl-c to kill the
|
# period" wherein the user eventually hits ctl-c to kill the
|
||||||
# root-actor tree.
|
# root-actor tree.
|
||||||
expect_final_exc: BaseException = KeyboardInterrupt
|
expect_final_exc: BaseException = KeyboardInterrupt
|
||||||
expect_final_cause: BaseException|None = None
|
|
||||||
|
|
||||||
if (
|
if (
|
||||||
# only expect EoC if trans is broken on the child side,
|
# only expect EoC if trans is broken on the child side,
|
||||||
ipc_break['break_child_ipc_after'] is not False
|
ipc_break['break_child_ipc_after'] is not False
|
||||||
|
|
@ -144,9 +139,6 @@ def test_ipc_channel_break_during_stream(
|
||||||
# a user sending ctl-c by raising a KBI.
|
# a user sending ctl-c by raising a KBI.
|
||||||
if pre_aclose_msgstream:
|
if pre_aclose_msgstream:
|
||||||
expect_final_exc = KeyboardInterrupt
|
expect_final_exc = KeyboardInterrupt
|
||||||
if tpt_proto == 'uds':
|
|
||||||
expect_final_exc = TransportClosed
|
|
||||||
expect_final_cause = trio.BrokenResourceError
|
|
||||||
|
|
||||||
# XXX OLD XXX
|
# XXX OLD XXX
|
||||||
# if child calls `MsgStream.aclose()` then expect EoC.
|
# if child calls `MsgStream.aclose()` then expect EoC.
|
||||||
|
|
@ -166,10 +158,6 @@ def test_ipc_channel_break_during_stream(
|
||||||
if pre_aclose_msgstream:
|
if pre_aclose_msgstream:
|
||||||
expect_final_exc = KeyboardInterrupt
|
expect_final_exc = KeyboardInterrupt
|
||||||
|
|
||||||
if tpt_proto == 'uds':
|
|
||||||
expect_final_exc = TransportClosed
|
|
||||||
expect_final_cause = trio.BrokenResourceError
|
|
||||||
|
|
||||||
# NOTE when the parent IPC side dies (even if the child does as well
|
# NOTE when the parent IPC side dies (even if the child does as well
|
||||||
# but the child fails BEFORE the parent) we always expect the
|
# but the child fails BEFORE the parent) we always expect the
|
||||||
# IPC layer to raise a closed-resource, NEVER do we expect
|
# IPC layer to raise a closed-resource, NEVER do we expect
|
||||||
|
|
@ -182,8 +170,8 @@ def test_ipc_channel_break_during_stream(
|
||||||
and
|
and
|
||||||
ipc_break['break_child_ipc_after'] is False
|
ipc_break['break_child_ipc_after'] is False
|
||||||
):
|
):
|
||||||
|
# expect_final_exc = trio.ClosedResourceError
|
||||||
expect_final_exc = tractor.TransportClosed
|
expect_final_exc = tractor.TransportClosed
|
||||||
expect_final_cause = trio.ClosedResourceError
|
|
||||||
|
|
||||||
# BOTH but, PARENT breaks FIRST
|
# BOTH but, PARENT breaks FIRST
|
||||||
elif (
|
elif (
|
||||||
|
|
@ -194,8 +182,8 @@ def test_ipc_channel_break_during_stream(
|
||||||
ipc_break['break_parent_ipc_after']
|
ipc_break['break_parent_ipc_after']
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
|
# expect_final_exc = trio.ClosedResourceError
|
||||||
expect_final_exc = tractor.TransportClosed
|
expect_final_exc = tractor.TransportClosed
|
||||||
expect_final_cause = trio.ClosedResourceError
|
|
||||||
|
|
||||||
with pytest.raises(
|
with pytest.raises(
|
||||||
expected_exception=(
|
expected_exception=(
|
||||||
|
|
@ -211,7 +199,6 @@ def test_ipc_channel_break_during_stream(
|
||||||
start_method=spawn_backend,
|
start_method=spawn_backend,
|
||||||
loglevel=loglevel,
|
loglevel=loglevel,
|
||||||
pre_close=pre_aclose_msgstream,
|
pre_close=pre_aclose_msgstream,
|
||||||
tpt_proto=tpt_proto,
|
|
||||||
**ipc_break,
|
**ipc_break,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
@ -234,24 +221,22 @@ def test_ipc_channel_break_during_stream(
|
||||||
)
|
)
|
||||||
cause: Exception = tc.__cause__
|
cause: Exception = tc.__cause__
|
||||||
assert (
|
assert (
|
||||||
# type(cause) is trio.ClosedResourceError
|
type(cause) is trio.ClosedResourceError
|
||||||
type(cause) is expect_final_cause
|
and
|
||||||
|
cause.args[0] == 'another task closed this fd'
|
||||||
# TODO, should we expect a certain exc-message (per
|
|
||||||
# tpt) as well??
|
|
||||||
# and
|
|
||||||
# cause.args[0] == 'another task closed this fd'
|
|
||||||
)
|
)
|
||||||
|
|
||||||
raise
|
raise
|
||||||
|
|
||||||
# get raw instance from pytest wrapper
|
# get raw instance from pytest wrapper
|
||||||
value = excinfo.value
|
value = excinfo.value
|
||||||
if isinstance(value, ExceptionGroup):
|
if isinstance(value, ExceptionGroup):
|
||||||
excs = value.exceptions
|
value = next(
|
||||||
assert len(excs) == 1
|
itertools.dropwhile(
|
||||||
final_exc = excs[0]
|
lambda exc: not isinstance(exc, expect_final_exc),
|
||||||
assert isinstance(final_exc, expect_final_exc)
|
value.exceptions,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
assert value
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
|
|
@ -274,16 +259,15 @@ async def break_ipc_after_started(
|
||||||
|
|
||||||
def test_stream_closed_right_after_ipc_break_and_zombie_lord_engages():
|
def test_stream_closed_right_after_ipc_break_and_zombie_lord_engages():
|
||||||
'''
|
'''
|
||||||
Verify that is a subactor's IPC goes down just after bringing up
|
Verify that is a subactor's IPC goes down just after bringing up a stream
|
||||||
a stream the parent can trigger a SIGINT and the child will be
|
the parent can trigger a SIGINT and the child will be reaped out-of-IPC by
|
||||||
reaped out-of-IPC by the localhost process supervision machinery:
|
the localhost process supervision machinery: aka "zombie lord".
|
||||||
aka "zombie lord".
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
with trio.fail_after(3):
|
with trio.fail_after(3):
|
||||||
async with tractor.open_nursery() as an:
|
async with tractor.open_nursery() as n:
|
||||||
portal = await an.start_actor(
|
portal = await n.start_actor(
|
||||||
'ipc_breaker',
|
'ipc_breaker',
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
|
||||||
|
|
@ -307,13 +307,6 @@ async def inf_streamer(
|
||||||
|
|
||||||
async with (
|
async with (
|
||||||
ctx.open_stream() as stream,
|
ctx.open_stream() as stream,
|
||||||
|
|
||||||
# XXX TODO, INTERESTING CASE!!
|
|
||||||
# - if we don't collapse the eg then the embedded
|
|
||||||
# `trio.EndOfChannel` doesn't propagate directly to the above
|
|
||||||
# .open_stream() parent, resulting in it also raising instead
|
|
||||||
# of gracefully absorbing as normal.. so how to handle?
|
|
||||||
tractor.trionics.collapse_eg(),
|
|
||||||
trio.open_nursery() as tn,
|
trio.open_nursery() as tn,
|
||||||
):
|
):
|
||||||
async def close_stream_on_sentinel():
|
async def close_stream_on_sentinel():
|
||||||
|
|
|
||||||
|
|
@ -14,7 +14,7 @@ import tractor
|
||||||
from tractor._testing import (
|
from tractor._testing import (
|
||||||
tractor_test,
|
tractor_test,
|
||||||
)
|
)
|
||||||
from .conftest import no_windows
|
from conftest import no_windows
|
||||||
|
|
||||||
|
|
||||||
def is_win():
|
def is_win():
|
||||||
|
|
@ -130,7 +130,7 @@ def test_multierror(
|
||||||
try:
|
try:
|
||||||
await portal2.result()
|
await portal2.result()
|
||||||
except tractor.RemoteActorError as err:
|
except tractor.RemoteActorError as err:
|
||||||
assert err.boxed_type is AssertionError
|
assert err.boxed_type == AssertionError
|
||||||
print("Look Maa that first actor failed hard, hehh")
|
print("Look Maa that first actor failed hard, hehh")
|
||||||
raise
|
raise
|
||||||
|
|
||||||
|
|
@ -182,7 +182,7 @@ def test_multierror_fast_nursery(reg_addr, start_method, num_subactors, delay):
|
||||||
|
|
||||||
for exc in exceptions:
|
for exc in exceptions:
|
||||||
assert isinstance(exc, tractor.RemoteActorError)
|
assert isinstance(exc, tractor.RemoteActorError)
|
||||||
assert exc.boxed_type is AssertionError
|
assert exc.boxed_type == AssertionError
|
||||||
|
|
||||||
|
|
||||||
async def do_nothing():
|
async def do_nothing():
|
||||||
|
|
@ -236,10 +236,7 @@ async def stream_forever():
|
||||||
async def test_cancel_infinite_streamer(start_method):
|
async def test_cancel_infinite_streamer(start_method):
|
||||||
|
|
||||||
# stream for at most 1 seconds
|
# stream for at most 1 seconds
|
||||||
with (
|
with trio.move_on_after(1) as cancel_scope:
|
||||||
trio.fail_after(4),
|
|
||||||
trio.move_on_after(1) as cancel_scope
|
|
||||||
):
|
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery() as n:
|
||||||
portal = await n.start_actor(
|
portal = await n.start_actor(
|
||||||
'donny',
|
'donny',
|
||||||
|
|
@ -287,32 +284,20 @@ async def test_cancel_infinite_streamer(start_method):
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_some_cancels_all(
|
async def test_some_cancels_all(num_actors_and_errs, start_method, loglevel):
|
||||||
num_actors_and_errs: tuple,
|
"""Verify a subset of failed subactors causes all others in
|
||||||
start_method: str,
|
|
||||||
loglevel: str,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Verify a subset of failed subactors causes all others in
|
|
||||||
the nursery to be cancelled just like the strategy in trio.
|
the nursery to be cancelled just like the strategy in trio.
|
||||||
|
|
||||||
This is the first and only supervisory strategy at the moment.
|
This is the first and only supervisory strategy at the moment.
|
||||||
|
"""
|
||||||
'''
|
num_actors, first_err, err_type, ria_func, da_func = num_actors_and_errs
|
||||||
(
|
|
||||||
num_actors,
|
|
||||||
first_err,
|
|
||||||
err_type,
|
|
||||||
ria_func,
|
|
||||||
da_func,
|
|
||||||
) = num_actors_and_errs
|
|
||||||
try:
|
try:
|
||||||
async with tractor.open_nursery() as an:
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
# spawn the same number of deamon actors which should be cancelled
|
# spawn the same number of deamon actors which should be cancelled
|
||||||
dactor_portals = []
|
dactor_portals = []
|
||||||
for i in range(num_actors):
|
for i in range(num_actors):
|
||||||
dactor_portals.append(await an.start_actor(
|
dactor_portals.append(await n.start_actor(
|
||||||
f'deamon_{i}',
|
f'deamon_{i}',
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
))
|
))
|
||||||
|
|
@ -322,7 +307,7 @@ async def test_some_cancels_all(
|
||||||
for i in range(num_actors):
|
for i in range(num_actors):
|
||||||
# start actor(s) that will fail immediately
|
# start actor(s) that will fail immediately
|
||||||
riactor_portals.append(
|
riactor_portals.append(
|
||||||
await an.run_in_actor(
|
await n.run_in_actor(
|
||||||
func,
|
func,
|
||||||
name=f'actor_{i}',
|
name=f'actor_{i}',
|
||||||
**kwargs
|
**kwargs
|
||||||
|
|
@ -352,8 +337,7 @@ async def test_some_cancels_all(
|
||||||
|
|
||||||
# should error here with a ``RemoteActorError`` or ``MultiError``
|
# should error here with a ``RemoteActorError`` or ``MultiError``
|
||||||
|
|
||||||
except first_err as _err:
|
except first_err as err:
|
||||||
err = _err
|
|
||||||
if isinstance(err, BaseExceptionGroup):
|
if isinstance(err, BaseExceptionGroup):
|
||||||
assert len(err.exceptions) == num_actors
|
assert len(err.exceptions) == num_actors
|
||||||
for exc in err.exceptions:
|
for exc in err.exceptions:
|
||||||
|
|
@ -364,8 +348,8 @@ async def test_some_cancels_all(
|
||||||
elif isinstance(err, tractor.RemoteActorError):
|
elif isinstance(err, tractor.RemoteActorError):
|
||||||
assert err.boxed_type == err_type
|
assert err.boxed_type == err_type
|
||||||
|
|
||||||
assert an.cancelled is True
|
assert n.cancelled is True
|
||||||
assert not an._children
|
assert not n._children
|
||||||
else:
|
else:
|
||||||
pytest.fail("Should have gotten a remote assertion error?")
|
pytest.fail("Should have gotten a remote assertion error?")
|
||||||
|
|
||||||
|
|
@ -520,9 +504,7 @@ def test_cancel_via_SIGINT_other_task(
|
||||||
if is_win(): # smh
|
if is_win(): # smh
|
||||||
timeout += 1
|
timeout += 1
|
||||||
|
|
||||||
async def spawn_and_sleep_forever(
|
async def spawn_and_sleep_forever(task_status=trio.TASK_STATUS_IGNORED):
|
||||||
task_status=trio.TASK_STATUS_IGNORED
|
|
||||||
):
|
|
||||||
async with tractor.open_nursery() as tn:
|
async with tractor.open_nursery() as tn:
|
||||||
for i in range(3):
|
for i in range(3):
|
||||||
await tn.run_in_actor(
|
await tn.run_in_actor(
|
||||||
|
|
@ -535,15 +517,8 @@ def test_cancel_via_SIGINT_other_task(
|
||||||
async def main():
|
async def main():
|
||||||
# should never timeout since SIGINT should cancel the current program
|
# should never timeout since SIGINT should cancel the current program
|
||||||
with trio.fail_after(timeout):
|
with trio.fail_after(timeout):
|
||||||
async with (
|
async with trio.open_nursery() as n:
|
||||||
|
await n.start(spawn_and_sleep_forever)
|
||||||
# XXX ?TODO? why no work!?
|
|
||||||
# tractor.trionics.collapse_eg(),
|
|
||||||
trio.open_nursery(
|
|
||||||
strict_exception_groups=False,
|
|
||||||
) as tn,
|
|
||||||
):
|
|
||||||
await tn.start(spawn_and_sleep_forever)
|
|
||||||
if 'mp' in spawn_backend:
|
if 'mp' in spawn_backend:
|
||||||
time.sleep(0.1)
|
time.sleep(0.1)
|
||||||
os.kill(pid, signal.SIGINT)
|
os.kill(pid, signal.SIGINT)
|
||||||
|
|
@ -554,123 +529,38 @@ def test_cancel_via_SIGINT_other_task(
|
||||||
|
|
||||||
async def spin_for(period=3):
|
async def spin_for(period=3):
|
||||||
"Sync sleep."
|
"Sync sleep."
|
||||||
print(f'sync sleeping in sub-sub for {period}\n')
|
|
||||||
time.sleep(period)
|
time.sleep(period)
|
||||||
|
|
||||||
|
|
||||||
async def spawn_sub_with_sync_blocking_task():
|
async def spawn():
|
||||||
async with tractor.open_nursery() as an:
|
async with tractor.open_nursery() as tn:
|
||||||
print('starting sync blocking subactor..\n')
|
await tn.run_in_actor(
|
||||||
await an.run_in_actor(
|
|
||||||
spin_for,
|
spin_for,
|
||||||
name='sleeper',
|
name='sleeper',
|
||||||
)
|
)
|
||||||
print('exiting first subactor layer..\n')
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'man_cancel_outer',
|
|
||||||
[
|
|
||||||
False, # passes if delay != 2
|
|
||||||
|
|
||||||
# always causes an unexpected eg-w-embedded-assert-err?
|
|
||||||
pytest.param(True,
|
|
||||||
marks=pytest.mark.xfail(
|
|
||||||
reason=(
|
|
||||||
'always causes an unexpected eg-w-embedded-assert-err?'
|
|
||||||
)
|
|
||||||
),
|
|
||||||
),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
@no_windows
|
@no_windows
|
||||||
def test_cancel_while_childs_child_in_sync_sleep(
|
def test_cancel_while_childs_child_in_sync_sleep(
|
||||||
loglevel: str,
|
loglevel,
|
||||||
start_method: str,
|
start_method,
|
||||||
spawn_backend: str,
|
spawn_backend,
|
||||||
debug_mode: bool,
|
|
||||||
reg_addr: tuple,
|
|
||||||
man_cancel_outer: bool,
|
|
||||||
):
|
):
|
||||||
'''
|
"""Verify that a child cancelled while executing sync code is torn
|
||||||
Verify that a child cancelled while executing sync code is torn
|
|
||||||
down even when that cancellation is triggered by the parent
|
down even when that cancellation is triggered by the parent
|
||||||
2 nurseries "up".
|
2 nurseries "up".
|
||||||
|
"""
|
||||||
Though the grandchild should stay blocking its actor runtime, its
|
|
||||||
parent should issue a "zombie reaper" to hard kill it after
|
|
||||||
sufficient timeout.
|
|
||||||
|
|
||||||
'''
|
|
||||||
if start_method == 'forkserver':
|
if start_method == 'forkserver':
|
||||||
pytest.skip("Forksever sux hard at resuming from sync sleep...")
|
pytest.skip("Forksever sux hard at resuming from sync sleep...")
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
#
|
with trio.fail_after(2):
|
||||||
# XXX BIG TODO NOTE XXX
|
async with tractor.open_nursery() as tn:
|
||||||
#
|
await tn.run_in_actor(
|
||||||
# it seems there's a strange race that can happen
|
spawn,
|
||||||
# where where the fail-after will trigger outer scope
|
name='spawn',
|
||||||
# .cancel() which then causes the inner scope to raise,
|
|
||||||
#
|
|
||||||
# BaseExceptionGroup('Exceptions from Trio nursery', [
|
|
||||||
# BaseExceptionGroup('Exceptions from Trio nursery',
|
|
||||||
# [
|
|
||||||
# Cancelled(),
|
|
||||||
# Cancelled(),
|
|
||||||
# ]
|
|
||||||
# ),
|
|
||||||
# AssertionError('assert 0')
|
|
||||||
# ])
|
|
||||||
#
|
|
||||||
# WHY THIS DOESN'T MAKE SENSE:
|
|
||||||
# ---------------------------
|
|
||||||
# - it should raise too-slow-error when too slow..
|
|
||||||
# * verified that using simple-cs and manually cancelling
|
|
||||||
# you get same outcome -> indicates that the fail-after
|
|
||||||
# can have its TooSlowError overriden!
|
|
||||||
# |_ to check this it's easy, simplly decrease the timeout
|
|
||||||
# as per the var below.
|
|
||||||
#
|
|
||||||
# - when using the manual simple-cs the outcome is different
|
|
||||||
# DESPITE the `assert 0` which means regardless of the
|
|
||||||
# inner scope effectively failing in the same way, the
|
|
||||||
# bubbling up **is NOT the same**.
|
|
||||||
#
|
|
||||||
# delays trigger diff outcomes..
|
|
||||||
# ---------------------------
|
|
||||||
# as seen by uncommenting various lines below there is from
|
|
||||||
# my POV an unexpected outcome due to the delay=2 case.
|
|
||||||
#
|
|
||||||
# delay = 1 # no AssertionError in eg, TooSlowError raised.
|
|
||||||
# delay = 2 # is AssertionError in eg AND no TooSlowError !?
|
|
||||||
delay = 4 # is AssertionError in eg AND no _cs cancellation.
|
|
||||||
|
|
||||||
with trio.fail_after(delay) as _cs:
|
|
||||||
# with trio.CancelScope() as cs:
|
|
||||||
# ^XXX^ can be used instead to see same outcome.
|
|
||||||
|
|
||||||
async with (
|
|
||||||
# tractor.trionics.collapse_eg(), # doesn't help
|
|
||||||
tractor.open_nursery(
|
|
||||||
hide_tb=False,
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
registry_addrs=[reg_addr],
|
|
||||||
) as an,
|
|
||||||
):
|
|
||||||
await an.run_in_actor(
|
|
||||||
spawn_sub_with_sync_blocking_task,
|
|
||||||
name='sync_blocking_sub',
|
|
||||||
)
|
)
|
||||||
await trio.sleep(1)
|
await trio.sleep(1)
|
||||||
|
|
||||||
if man_cancel_outer:
|
|
||||||
print('Cancelling manually in root')
|
|
||||||
_cs.cancel()
|
|
||||||
|
|
||||||
# trigger exc-srced taskc down
|
|
||||||
# the actor tree.
|
|
||||||
print('RAISING IN ROOT')
|
|
||||||
assert 0
|
assert 0
|
||||||
|
|
||||||
with pytest.raises(AssertionError):
|
with pytest.raises(AssertionError):
|
||||||
|
|
@ -720,12 +610,6 @@ def test_fast_graceful_cancel_when_spawn_task_in_soft_proc_wait_for_daemon(
|
||||||
nurse.start_soon(delayed_kbi)
|
nurse.start_soon(delayed_kbi)
|
||||||
|
|
||||||
await p.run(do_nuthin)
|
await p.run(do_nuthin)
|
||||||
|
|
||||||
# need to explicitly re-raise the lone kbi..now
|
|
||||||
except* KeyboardInterrupt as kbi_eg:
|
|
||||||
assert (len(excs := kbi_eg.exceptions) == 1)
|
|
||||||
raise excs[0]
|
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
duration = time.time() - start
|
duration = time.time() - start
|
||||||
if duration > timeout:
|
if duration > timeout:
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,917 @@
|
||||||
|
'''
|
||||||
|
Low-level functional audits for our
|
||||||
|
"capability based messaging"-spec feats.
|
||||||
|
|
||||||
|
B~)
|
||||||
|
|
||||||
|
'''
|
||||||
|
import typing
|
||||||
|
from typing import (
|
||||||
|
Any,
|
||||||
|
Type,
|
||||||
|
Union,
|
||||||
|
)
|
||||||
|
|
||||||
|
from msgspec import (
|
||||||
|
structs,
|
||||||
|
msgpack,
|
||||||
|
Struct,
|
||||||
|
ValidationError,
|
||||||
|
)
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
import tractor
|
||||||
|
from tractor import (
|
||||||
|
_state,
|
||||||
|
MsgTypeError,
|
||||||
|
Context,
|
||||||
|
)
|
||||||
|
from tractor.msg import (
|
||||||
|
_codec,
|
||||||
|
_ctxvar_MsgCodec,
|
||||||
|
|
||||||
|
NamespacePath,
|
||||||
|
MsgCodec,
|
||||||
|
mk_codec,
|
||||||
|
apply_codec,
|
||||||
|
current_codec,
|
||||||
|
)
|
||||||
|
from tractor.msg.types import (
|
||||||
|
_payload_msgs,
|
||||||
|
log,
|
||||||
|
PayloadMsg,
|
||||||
|
Started,
|
||||||
|
mk_msg_spec,
|
||||||
|
)
|
||||||
|
import trio
|
||||||
|
|
||||||
|
|
||||||
|
def mk_custom_codec(
|
||||||
|
pld_spec: Union[Type]|Any,
|
||||||
|
add_hooks: bool,
|
||||||
|
|
||||||
|
) -> MsgCodec:
|
||||||
|
'''
|
||||||
|
Create custom `msgpack` enc/dec-hooks and set a `Decoder`
|
||||||
|
which only loads `pld_spec` (like `NamespacePath`) types.
|
||||||
|
|
||||||
|
'''
|
||||||
|
uid: tuple[str, str] = tractor.current_actor().uid
|
||||||
|
|
||||||
|
# XXX NOTE XXX: despite defining `NamespacePath` as a type
|
||||||
|
# field on our `PayloadMsg.pld`, we still need a enc/dec_hook() pair
|
||||||
|
# to cast to/from that type on the wire. See the docs:
|
||||||
|
# https://jcristharif.com/msgspec/extending.html#mapping-to-from-native-types
|
||||||
|
|
||||||
|
def enc_nsp(obj: Any) -> Any:
|
||||||
|
print(f'{uid} ENC HOOK')
|
||||||
|
match obj:
|
||||||
|
case NamespacePath():
|
||||||
|
print(
|
||||||
|
f'{uid}: `NamespacePath`-Only ENCODE?\n'
|
||||||
|
f'obj-> `{obj}`: {type(obj)}\n'
|
||||||
|
)
|
||||||
|
# if type(obj) != NamespacePath:
|
||||||
|
# breakpoint()
|
||||||
|
return str(obj)
|
||||||
|
|
||||||
|
print(
|
||||||
|
f'{uid}\n'
|
||||||
|
'CUSTOM ENCODE\n'
|
||||||
|
f'obj-arg-> `{obj}`: {type(obj)}\n'
|
||||||
|
)
|
||||||
|
logmsg: str = (
|
||||||
|
f'{uid}\n'
|
||||||
|
'FAILED ENCODE\n'
|
||||||
|
f'obj-> `{obj}: {type(obj)}`\n'
|
||||||
|
)
|
||||||
|
raise NotImplementedError(logmsg)
|
||||||
|
|
||||||
|
def dec_nsp(
|
||||||
|
obj_type: Type,
|
||||||
|
obj: Any,
|
||||||
|
|
||||||
|
) -> Any:
|
||||||
|
print(
|
||||||
|
f'{uid}\n'
|
||||||
|
'CUSTOM DECODE\n'
|
||||||
|
f'type-arg-> {obj_type}\n'
|
||||||
|
f'obj-arg-> `{obj}`: {type(obj)}\n'
|
||||||
|
)
|
||||||
|
nsp = None
|
||||||
|
|
||||||
|
if (
|
||||||
|
obj_type is NamespacePath
|
||||||
|
and isinstance(obj, str)
|
||||||
|
and ':' in obj
|
||||||
|
):
|
||||||
|
nsp = NamespacePath(obj)
|
||||||
|
# TODO: we could built a generic handler using
|
||||||
|
# JUST matching the obj_type part?
|
||||||
|
# nsp = obj_type(obj)
|
||||||
|
|
||||||
|
if nsp:
|
||||||
|
print(f'Returning NSP instance: {nsp}')
|
||||||
|
return nsp
|
||||||
|
|
||||||
|
logmsg: str = (
|
||||||
|
f'{uid}\n'
|
||||||
|
'FAILED DECODE\n'
|
||||||
|
f'type-> {obj_type}\n'
|
||||||
|
f'obj-arg-> `{obj}`: {type(obj)}\n\n'
|
||||||
|
f'current codec:\n'
|
||||||
|
f'{current_codec()}\n'
|
||||||
|
)
|
||||||
|
# TODO: figure out the ignore subsys for this!
|
||||||
|
# -[ ] option whether to defense-relay backc the msg
|
||||||
|
# inside an `Invalid`/`Ignore`
|
||||||
|
# -[ ] how to make this handling pluggable such that a
|
||||||
|
# `Channel`/`MsgTransport` can intercept and process
|
||||||
|
# back msgs either via exception handling or some other
|
||||||
|
# signal?
|
||||||
|
log.warning(logmsg)
|
||||||
|
# NOTE: this delivers the invalid
|
||||||
|
# value up to `msgspec`'s decoding
|
||||||
|
# machinery for error raising.
|
||||||
|
return obj
|
||||||
|
# raise NotImplementedError(logmsg)
|
||||||
|
|
||||||
|
nsp_codec: MsgCodec = mk_codec(
|
||||||
|
ipc_pld_spec=pld_spec,
|
||||||
|
|
||||||
|
# NOTE XXX: the encode hook MUST be used no matter what since
|
||||||
|
# our `NamespacePath` is not any of a `Any` native type nor
|
||||||
|
# a `msgspec.Struct` subtype - so `msgspec` has no way to know
|
||||||
|
# how to encode it unless we provide the custom hook.
|
||||||
|
#
|
||||||
|
# AGAIN that is, regardless of whether we spec an
|
||||||
|
# `Any`-decoded-pld the enc has no knowledge (by default)
|
||||||
|
# how to enc `NamespacePath` (nsp), so we add a custom
|
||||||
|
# hook to do that ALWAYS.
|
||||||
|
enc_hook=enc_nsp if add_hooks else None,
|
||||||
|
|
||||||
|
# XXX NOTE: pretty sure this is mutex with the `type=` to
|
||||||
|
# `Decoder`? so it won't work in tandem with the
|
||||||
|
# `ipc_pld_spec` passed above?
|
||||||
|
dec_hook=dec_nsp if add_hooks else None,
|
||||||
|
)
|
||||||
|
return nsp_codec
|
||||||
|
|
||||||
|
|
||||||
|
def chk_codec_applied(
|
||||||
|
expect_codec: MsgCodec,
|
||||||
|
enter_value: MsgCodec|None = None,
|
||||||
|
|
||||||
|
) -> MsgCodec:
|
||||||
|
'''
|
||||||
|
buncha sanity checks ensuring that the IPC channel's
|
||||||
|
context-vars are set to the expected codec and that are
|
||||||
|
ctx-var wrapper APIs match the same.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# TODO: play with tricyle again, bc this is supposed to work
|
||||||
|
# the way we want?
|
||||||
|
#
|
||||||
|
# TreeVar
|
||||||
|
# task: trio.Task = trio.lowlevel.current_task()
|
||||||
|
# curr_codec = _ctxvar_MsgCodec.get_in(task)
|
||||||
|
|
||||||
|
# ContextVar
|
||||||
|
# task_ctx: Context = task.context
|
||||||
|
# assert _ctxvar_MsgCodec in task_ctx
|
||||||
|
# curr_codec: MsgCodec = task.context[_ctxvar_MsgCodec]
|
||||||
|
|
||||||
|
# NOTE: currently we use this!
|
||||||
|
# RunVar
|
||||||
|
curr_codec: MsgCodec = current_codec()
|
||||||
|
last_read_codec = _ctxvar_MsgCodec.get()
|
||||||
|
# assert curr_codec is last_read_codec
|
||||||
|
|
||||||
|
assert (
|
||||||
|
(same_codec := expect_codec) is
|
||||||
|
# returned from `mk_codec()`
|
||||||
|
|
||||||
|
# yielded value from `apply_codec()`
|
||||||
|
|
||||||
|
# read from current task's `contextvars.Context`
|
||||||
|
curr_codec is
|
||||||
|
last_read_codec
|
||||||
|
|
||||||
|
# the default `msgspec` settings
|
||||||
|
is not _codec._def_msgspec_codec
|
||||||
|
is not _codec._def_tractor_codec
|
||||||
|
)
|
||||||
|
|
||||||
|
if enter_value:
|
||||||
|
enter_value is same_codec
|
||||||
|
|
||||||
|
|
||||||
|
def iter_maybe_sends(
|
||||||
|
send_items: dict[Union[Type], Any] | list[tuple],
|
||||||
|
ipc_pld_spec: Union[Type] | Any,
|
||||||
|
add_codec_hooks: bool,
|
||||||
|
|
||||||
|
codec: MsgCodec|None = None,
|
||||||
|
|
||||||
|
) -> tuple[Any, bool]:
|
||||||
|
|
||||||
|
if isinstance(send_items, dict):
|
||||||
|
send_items = send_items.items()
|
||||||
|
|
||||||
|
for (
|
||||||
|
send_type_spec,
|
||||||
|
send_value,
|
||||||
|
) in send_items:
|
||||||
|
|
||||||
|
expect_roundtrip: bool = False
|
||||||
|
|
||||||
|
# values-to-typespec santiy
|
||||||
|
send_type = type(send_value)
|
||||||
|
assert send_type == send_type_spec or (
|
||||||
|
(subtypes := getattr(send_type_spec, '__args__', None))
|
||||||
|
and send_type in subtypes
|
||||||
|
)
|
||||||
|
|
||||||
|
spec_subtypes: set[Union[Type]] = (
|
||||||
|
getattr(
|
||||||
|
ipc_pld_spec,
|
||||||
|
'__args__',
|
||||||
|
{ipc_pld_spec,},
|
||||||
|
)
|
||||||
|
)
|
||||||
|
send_in_spec: bool = (
|
||||||
|
send_type == ipc_pld_spec
|
||||||
|
or (
|
||||||
|
ipc_pld_spec != Any
|
||||||
|
and # presume `Union` of types
|
||||||
|
send_type in spec_subtypes
|
||||||
|
)
|
||||||
|
or (
|
||||||
|
ipc_pld_spec == Any
|
||||||
|
and
|
||||||
|
send_type != NamespacePath
|
||||||
|
)
|
||||||
|
)
|
||||||
|
expect_roundtrip = (
|
||||||
|
send_in_spec
|
||||||
|
# any spec should support all other
|
||||||
|
# builtin py values that we send
|
||||||
|
# except our custom nsp type which
|
||||||
|
# we should be able to send as long
|
||||||
|
# as we provide the custom codec hooks.
|
||||||
|
or (
|
||||||
|
ipc_pld_spec == Any
|
||||||
|
and
|
||||||
|
send_type == NamespacePath
|
||||||
|
and
|
||||||
|
add_codec_hooks
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
if codec is not None:
|
||||||
|
# XXX FIRST XXX ensure roundtripping works
|
||||||
|
# before touching any IPC primitives/APIs.
|
||||||
|
wire_bytes: bytes = codec.encode(
|
||||||
|
Started(
|
||||||
|
cid='blahblah',
|
||||||
|
pld=send_value,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
# NOTE: demonstrates the decoder loading
|
||||||
|
# to via our native SCIPP msg-spec
|
||||||
|
# (structurred-conc-inter-proc-protocol)
|
||||||
|
# implemented as per,
|
||||||
|
try:
|
||||||
|
msg: Started = codec.decode(wire_bytes)
|
||||||
|
if not expect_roundtrip:
|
||||||
|
pytest.fail(
|
||||||
|
f'NOT-EXPECTED able to roundtrip value given spec:\n'
|
||||||
|
f'ipc_pld_spec -> {ipc_pld_spec}\n'
|
||||||
|
f'value -> {send_value}: {send_type}\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
pld = msg.pld
|
||||||
|
assert pld == send_value
|
||||||
|
|
||||||
|
except ValidationError:
|
||||||
|
if expect_roundtrip:
|
||||||
|
pytest.fail(
|
||||||
|
f'EXPECTED to roundtrip value given spec:\n'
|
||||||
|
f'ipc_pld_spec -> {ipc_pld_spec}\n'
|
||||||
|
f'value -> {send_value}: {send_type}\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
yield (
|
||||||
|
str(send_type),
|
||||||
|
send_value,
|
||||||
|
expect_roundtrip,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def dec_type_union(
|
||||||
|
type_names: list[str],
|
||||||
|
) -> Type:
|
||||||
|
'''
|
||||||
|
Look up types by name, compile into a list and then create and
|
||||||
|
return a `typing.Union` from the full set.
|
||||||
|
|
||||||
|
'''
|
||||||
|
import importlib
|
||||||
|
types: list[Type] = []
|
||||||
|
for type_name in type_names:
|
||||||
|
for mod in [
|
||||||
|
typing,
|
||||||
|
importlib.import_module(__name__),
|
||||||
|
]:
|
||||||
|
if type_ref := getattr(
|
||||||
|
mod,
|
||||||
|
type_name,
|
||||||
|
False,
|
||||||
|
):
|
||||||
|
types.append(type_ref)
|
||||||
|
|
||||||
|
# special case handling only..
|
||||||
|
# ipc_pld_spec: Union[Type] = eval(
|
||||||
|
# pld_spec_str,
|
||||||
|
# {}, # globals
|
||||||
|
# {'typing': typing}, # locals
|
||||||
|
# )
|
||||||
|
|
||||||
|
return Union[*types]
|
||||||
|
|
||||||
|
|
||||||
|
def enc_type_union(
|
||||||
|
union_or_type: Union[Type]|Type,
|
||||||
|
) -> list[str]:
|
||||||
|
'''
|
||||||
|
Encode a type-union or single type to a list of type-name-strings
|
||||||
|
ready for IPC interchange.
|
||||||
|
|
||||||
|
'''
|
||||||
|
type_strs: list[str] = []
|
||||||
|
for typ in getattr(
|
||||||
|
union_or_type,
|
||||||
|
'__args__',
|
||||||
|
{union_or_type,},
|
||||||
|
):
|
||||||
|
type_strs.append(typ.__qualname__)
|
||||||
|
|
||||||
|
return type_strs
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def send_back_values(
|
||||||
|
ctx: Context,
|
||||||
|
expect_debug: bool,
|
||||||
|
pld_spec_type_strs: list[str],
|
||||||
|
add_hooks: bool,
|
||||||
|
started_msg_bytes: bytes,
|
||||||
|
expect_ipc_send: dict[str, tuple[Any, bool]],
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Setup up a custom codec to load instances of `NamespacePath`
|
||||||
|
and ensure we can round trip a func ref with our parent.
|
||||||
|
|
||||||
|
'''
|
||||||
|
uid: tuple = tractor.current_actor().uid
|
||||||
|
|
||||||
|
# debug mode sanity check (prolly superfluous but, meh)
|
||||||
|
assert expect_debug == _state.debug_mode()
|
||||||
|
|
||||||
|
# init state in sub-actor should be default
|
||||||
|
chk_codec_applied(
|
||||||
|
expect_codec=_codec._def_tractor_codec,
|
||||||
|
)
|
||||||
|
|
||||||
|
# load pld spec from input str
|
||||||
|
ipc_pld_spec = dec_type_union(
|
||||||
|
pld_spec_type_strs,
|
||||||
|
)
|
||||||
|
pld_spec_str = str(ipc_pld_spec)
|
||||||
|
|
||||||
|
# same as on parent side config.
|
||||||
|
nsp_codec: MsgCodec = mk_custom_codec(
|
||||||
|
pld_spec=ipc_pld_spec,
|
||||||
|
add_hooks=add_hooks,
|
||||||
|
)
|
||||||
|
with (
|
||||||
|
apply_codec(nsp_codec) as codec,
|
||||||
|
):
|
||||||
|
chk_codec_applied(
|
||||||
|
expect_codec=nsp_codec,
|
||||||
|
enter_value=codec,
|
||||||
|
)
|
||||||
|
|
||||||
|
print(
|
||||||
|
f'{uid}: attempting `Started`-bytes DECODE..\n'
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
msg: Started = nsp_codec.decode(started_msg_bytes)
|
||||||
|
expected_pld_spec_str: str = msg.pld
|
||||||
|
assert pld_spec_str == expected_pld_spec_str
|
||||||
|
|
||||||
|
# TODO: maybe we should add our own wrapper error so as to
|
||||||
|
# be interchange-lib agnostic?
|
||||||
|
# -[ ] the error type is wtv is raised from the hook so we
|
||||||
|
# could also require a type-class of errors for
|
||||||
|
# indicating whether the hook-failure can be handled by
|
||||||
|
# a nasty-dialog-unprot sub-sys?
|
||||||
|
except ValidationError:
|
||||||
|
|
||||||
|
# NOTE: only in the `Any` spec case do we expect this to
|
||||||
|
# work since otherwise no spec covers a plain-ol'
|
||||||
|
# `.pld: str`
|
||||||
|
if pld_spec_str == 'Any':
|
||||||
|
raise
|
||||||
|
else:
|
||||||
|
print(
|
||||||
|
f'{uid}: (correctly) unable to DECODE `Started`-bytes\n'
|
||||||
|
f'{started_msg_bytes}\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
iter_send_val_items = iter(expect_ipc_send.values())
|
||||||
|
sent: list[Any] = []
|
||||||
|
for send_value, expect_send in iter_send_val_items:
|
||||||
|
try:
|
||||||
|
print(
|
||||||
|
f'{uid}: attempting to `.started({send_value})`\n'
|
||||||
|
f'=> expect_send: {expect_send}\n'
|
||||||
|
f'SINCE, ipc_pld_spec: {ipc_pld_spec}\n'
|
||||||
|
f'AND, codec: {codec}\n'
|
||||||
|
)
|
||||||
|
await ctx.started(send_value)
|
||||||
|
sent.append(send_value)
|
||||||
|
if not expect_send:
|
||||||
|
|
||||||
|
# XXX NOTE XXX THIS WON'T WORK WITHOUT SPECIAL
|
||||||
|
# `str` handling! or special debug mode IPC
|
||||||
|
# msgs!
|
||||||
|
await tractor.pause()
|
||||||
|
|
||||||
|
raise RuntimeError(
|
||||||
|
f'NOT-EXPECTED able to roundtrip value given spec:\n'
|
||||||
|
f'ipc_pld_spec -> {ipc_pld_spec}\n'
|
||||||
|
f'value -> {send_value}: {type(send_value)}\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
break # move on to streaming block..
|
||||||
|
|
||||||
|
except tractor.MsgTypeError:
|
||||||
|
await tractor.pause()
|
||||||
|
|
||||||
|
if expect_send:
|
||||||
|
raise RuntimeError(
|
||||||
|
f'EXPECTED to `.started()` value given spec:\n'
|
||||||
|
f'ipc_pld_spec -> {ipc_pld_spec}\n'
|
||||||
|
f'value -> {send_value}: {type(send_value)}\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
async with ctx.open_stream() as ipc:
|
||||||
|
print(
|
||||||
|
f'{uid}: Entering streaming block to send remaining values..'
|
||||||
|
)
|
||||||
|
|
||||||
|
for send_value, expect_send in iter_send_val_items:
|
||||||
|
send_type: Type = type(send_value)
|
||||||
|
print(
|
||||||
|
'------ - ------\n'
|
||||||
|
f'{uid}: SENDING NEXT VALUE\n'
|
||||||
|
f'ipc_pld_spec: {ipc_pld_spec}\n'
|
||||||
|
f'expect_send: {expect_send}\n'
|
||||||
|
f'val: {send_value}\n'
|
||||||
|
'------ - ------\n'
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
await ipc.send(send_value)
|
||||||
|
print(f'***\n{uid}-CHILD sent {send_value!r}\n***\n')
|
||||||
|
sent.append(send_value)
|
||||||
|
|
||||||
|
# NOTE: should only raise above on
|
||||||
|
# `.started()` or a `Return`
|
||||||
|
# if not expect_send:
|
||||||
|
# raise RuntimeError(
|
||||||
|
# f'NOT-EXPECTED able to roundtrip value given spec:\n'
|
||||||
|
# f'ipc_pld_spec -> {ipc_pld_spec}\n'
|
||||||
|
# f'value -> {send_value}: {send_type}\n'
|
||||||
|
# )
|
||||||
|
|
||||||
|
except ValidationError:
|
||||||
|
print(f'{uid} FAILED TO SEND {send_value}!')
|
||||||
|
|
||||||
|
# await tractor.pause()
|
||||||
|
if expect_send:
|
||||||
|
raise RuntimeError(
|
||||||
|
f'EXPECTED to roundtrip value given spec:\n'
|
||||||
|
f'ipc_pld_spec -> {ipc_pld_spec}\n'
|
||||||
|
f'value -> {send_value}: {send_type}\n'
|
||||||
|
)
|
||||||
|
# continue
|
||||||
|
|
||||||
|
else:
|
||||||
|
print(
|
||||||
|
f'{uid}: finished sending all values\n'
|
||||||
|
'Should be exiting stream block!\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
print(f'{uid}: exited streaming block!')
|
||||||
|
|
||||||
|
# TODO: this won't be true bc in streaming phase we DO NOT
|
||||||
|
# msgspec check outbound msgs!
|
||||||
|
# -[ ] once we implement the receiver side `InvalidMsg`
|
||||||
|
# then we can expect it here?
|
||||||
|
# assert (
|
||||||
|
# len(sent)
|
||||||
|
# ==
|
||||||
|
# len([val
|
||||||
|
# for val, expect in
|
||||||
|
# expect_ipc_send.values()
|
||||||
|
# if expect is True])
|
||||||
|
# )
|
||||||
|
|
||||||
|
|
||||||
|
def ex_func(*args):
|
||||||
|
print(f'ex_func({args})')
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'ipc_pld_spec',
|
||||||
|
[
|
||||||
|
Any,
|
||||||
|
NamespacePath,
|
||||||
|
NamespacePath|None, # the "maybe" spec Bo
|
||||||
|
],
|
||||||
|
ids=[
|
||||||
|
'any_type',
|
||||||
|
'nsp_type',
|
||||||
|
'maybe_nsp_type',
|
||||||
|
]
|
||||||
|
)
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'add_codec_hooks',
|
||||||
|
[
|
||||||
|
True,
|
||||||
|
False,
|
||||||
|
],
|
||||||
|
ids=['use_codec_hooks', 'no_codec_hooks'],
|
||||||
|
)
|
||||||
|
def test_codec_hooks_mod(
|
||||||
|
debug_mode: bool,
|
||||||
|
ipc_pld_spec: Union[Type]|Any,
|
||||||
|
# send_value: None|str|NamespacePath,
|
||||||
|
add_codec_hooks: bool,
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Audit the `.msg.MsgCodec` override apis details given our impl
|
||||||
|
uses `contextvars` to accomplish per `trio` task codec
|
||||||
|
application around an inter-proc-task-comms context.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
nsp = NamespacePath.from_ref(ex_func)
|
||||||
|
send_items: dict[Union, Any] = {
|
||||||
|
Union[None]: None,
|
||||||
|
Union[NamespacePath]: nsp,
|
||||||
|
Union[str]: str(nsp),
|
||||||
|
}
|
||||||
|
|
||||||
|
# init default state for actor
|
||||||
|
chk_codec_applied(
|
||||||
|
expect_codec=_codec._def_tractor_codec,
|
||||||
|
)
|
||||||
|
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=debug_mode,
|
||||||
|
) as an:
|
||||||
|
p: tractor.Portal = await an.start_actor(
|
||||||
|
'sub',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
# TODO: 2 cases:
|
||||||
|
# - codec not modified -> decode nsp as `str`
|
||||||
|
# - codec modified with hooks -> decode nsp as
|
||||||
|
# `NamespacePath`
|
||||||
|
nsp_codec: MsgCodec = mk_custom_codec(
|
||||||
|
pld_spec=ipc_pld_spec,
|
||||||
|
add_hooks=add_codec_hooks,
|
||||||
|
)
|
||||||
|
with apply_codec(nsp_codec) as codec:
|
||||||
|
chk_codec_applied(
|
||||||
|
expect_codec=nsp_codec,
|
||||||
|
enter_value=codec,
|
||||||
|
)
|
||||||
|
|
||||||
|
expect_ipc_send: dict[str, tuple[Any, bool]] = {}
|
||||||
|
|
||||||
|
report: str = (
|
||||||
|
'Parent report on send values with\n'
|
||||||
|
f'ipc_pld_spec: {ipc_pld_spec}\n'
|
||||||
|
' ------ - ------\n'
|
||||||
|
)
|
||||||
|
for val_type_str, val, expect_send in iter_maybe_sends(
|
||||||
|
send_items,
|
||||||
|
ipc_pld_spec,
|
||||||
|
add_codec_hooks=add_codec_hooks,
|
||||||
|
):
|
||||||
|
report += (
|
||||||
|
f'send_value: {val}: {type(val)} '
|
||||||
|
f'=> expect_send: {expect_send}\n'
|
||||||
|
)
|
||||||
|
expect_ipc_send[val_type_str] = (val, expect_send)
|
||||||
|
|
||||||
|
print(
|
||||||
|
report +
|
||||||
|
' ------ - ------\n'
|
||||||
|
)
|
||||||
|
assert len(expect_ipc_send) == len(send_items)
|
||||||
|
# now try over real IPC with a the subactor
|
||||||
|
# expect_ipc_rountrip: bool = True
|
||||||
|
expected_started = Started(
|
||||||
|
cid='cid',
|
||||||
|
pld=str(ipc_pld_spec),
|
||||||
|
)
|
||||||
|
# build list of values we expect to receive from
|
||||||
|
# the subactor.
|
||||||
|
expect_to_send: list[Any] = [
|
||||||
|
val
|
||||||
|
for val, expect_send in expect_ipc_send.values()
|
||||||
|
if expect_send
|
||||||
|
]
|
||||||
|
|
||||||
|
pld_spec_type_strs: list[str] = enc_type_union(ipc_pld_spec)
|
||||||
|
|
||||||
|
# XXX should raise an mte (`MsgTypeError`)
|
||||||
|
# when `add_codec_hooks == False` bc the input
|
||||||
|
# `expect_ipc_send` kwarg has a nsp which can't be
|
||||||
|
# serialized!
|
||||||
|
#
|
||||||
|
# TODO:can we ensure this happens from the
|
||||||
|
# `Return`-side (aka the sub) as well?
|
||||||
|
if not add_codec_hooks:
|
||||||
|
try:
|
||||||
|
async with p.open_context(
|
||||||
|
send_back_values,
|
||||||
|
expect_debug=debug_mode,
|
||||||
|
pld_spec_type_strs=pld_spec_type_strs,
|
||||||
|
add_hooks=add_codec_hooks,
|
||||||
|
started_msg_bytes=nsp_codec.encode(expected_started),
|
||||||
|
|
||||||
|
# XXX NOTE bc we send a `NamespacePath` in this kwarg
|
||||||
|
expect_ipc_send=expect_ipc_send,
|
||||||
|
|
||||||
|
) as (ctx, first):
|
||||||
|
pytest.fail('ctx should fail to open without custom enc_hook!?')
|
||||||
|
|
||||||
|
# this test passes bc we can go no further!
|
||||||
|
except MsgTypeError:
|
||||||
|
# teardown nursery
|
||||||
|
await p.cancel_actor()
|
||||||
|
return
|
||||||
|
|
||||||
|
# TODO: send the original nsp here and
|
||||||
|
# test with `limit_msg_spec()` above?
|
||||||
|
# await tractor.pause()
|
||||||
|
print('PARENT opening IPC ctx!\n')
|
||||||
|
async with (
|
||||||
|
|
||||||
|
# XXX should raise an mte (`MsgTypeError`)
|
||||||
|
# when `add_codec_hooks == False`..
|
||||||
|
p.open_context(
|
||||||
|
send_back_values,
|
||||||
|
expect_debug=debug_mode,
|
||||||
|
pld_spec_type_strs=pld_spec_type_strs,
|
||||||
|
add_hooks=add_codec_hooks,
|
||||||
|
started_msg_bytes=nsp_codec.encode(expected_started),
|
||||||
|
expect_ipc_send=expect_ipc_send,
|
||||||
|
) as (ctx, first),
|
||||||
|
|
||||||
|
ctx.open_stream() as ipc,
|
||||||
|
):
|
||||||
|
# ensure codec is still applied across
|
||||||
|
# `tractor.Context` + its embedded nursery.
|
||||||
|
chk_codec_applied(
|
||||||
|
expect_codec=nsp_codec,
|
||||||
|
enter_value=codec,
|
||||||
|
)
|
||||||
|
print(
|
||||||
|
'root: ENTERING CONTEXT BLOCK\n'
|
||||||
|
f'type(first): {type(first)}\n'
|
||||||
|
f'first: {first}\n'
|
||||||
|
)
|
||||||
|
expect_to_send.remove(first)
|
||||||
|
|
||||||
|
# TODO: explicit values we expect depending on
|
||||||
|
# codec config!
|
||||||
|
# assert first == first_val
|
||||||
|
# assert first == f'{__name__}:ex_func'
|
||||||
|
|
||||||
|
async for next_sent in ipc:
|
||||||
|
print(
|
||||||
|
'Parent: child sent next value\n'
|
||||||
|
f'{next_sent}: {type(next_sent)}\n'
|
||||||
|
)
|
||||||
|
if expect_to_send:
|
||||||
|
expect_to_send.remove(next_sent)
|
||||||
|
else:
|
||||||
|
print('PARENT should terminate stream loop + block!')
|
||||||
|
|
||||||
|
# all sent values should have arrived!
|
||||||
|
assert not expect_to_send
|
||||||
|
|
||||||
|
await p.cancel_actor()
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def chk_pld_type(
|
||||||
|
payload_spec: Type[Struct]|Any,
|
||||||
|
pld: Any,
|
||||||
|
|
||||||
|
expect_roundtrip: bool|None = None,
|
||||||
|
|
||||||
|
) -> bool:
|
||||||
|
|
||||||
|
pld_val_type: Type = type(pld)
|
||||||
|
|
||||||
|
# TODO: verify that the overridden subtypes
|
||||||
|
# DO NOT have modified type-annots from original!
|
||||||
|
# 'Start', .pld: FuncSpec
|
||||||
|
# 'StartAck', .pld: IpcCtxSpec
|
||||||
|
# 'Stop', .pld: UNSEt
|
||||||
|
# 'Error', .pld: ErrorData
|
||||||
|
|
||||||
|
codec: MsgCodec = mk_codec(
|
||||||
|
# NOTE: this ONLY accepts `PayloadMsg.pld` fields of a specified
|
||||||
|
# type union.
|
||||||
|
ipc_pld_spec=payload_spec,
|
||||||
|
)
|
||||||
|
|
||||||
|
# make a one-off dec to compare with our `MsgCodec` instance
|
||||||
|
# which does the below `mk_msg_spec()` call internally
|
||||||
|
ipc_msg_spec: Union[Type[Struct]]
|
||||||
|
msg_types: list[PayloadMsg[payload_spec]]
|
||||||
|
(
|
||||||
|
ipc_msg_spec,
|
||||||
|
msg_types,
|
||||||
|
) = mk_msg_spec(
|
||||||
|
payload_type_union=payload_spec,
|
||||||
|
)
|
||||||
|
_enc = msgpack.Encoder()
|
||||||
|
_dec = msgpack.Decoder(
|
||||||
|
type=ipc_msg_spec or Any, # like `PayloadMsg[Any]`
|
||||||
|
)
|
||||||
|
|
||||||
|
assert (
|
||||||
|
payload_spec
|
||||||
|
==
|
||||||
|
codec.pld_spec
|
||||||
|
)
|
||||||
|
|
||||||
|
# assert codec.dec == dec
|
||||||
|
#
|
||||||
|
# ^-XXX-^ not sure why these aren't "equal" but when cast
|
||||||
|
# to `str` they seem to match ?? .. kk
|
||||||
|
|
||||||
|
assert (
|
||||||
|
str(ipc_msg_spec)
|
||||||
|
==
|
||||||
|
str(codec.msg_spec)
|
||||||
|
==
|
||||||
|
str(_dec.type)
|
||||||
|
==
|
||||||
|
str(codec.dec.type)
|
||||||
|
)
|
||||||
|
|
||||||
|
# verify the boxed-type for all variable payload-type msgs.
|
||||||
|
if not msg_types:
|
||||||
|
breakpoint()
|
||||||
|
|
||||||
|
roundtrip: bool|None = None
|
||||||
|
pld_spec_msg_names: list[str] = [
|
||||||
|
td.__name__ for td in _payload_msgs
|
||||||
|
]
|
||||||
|
for typedef in msg_types:
|
||||||
|
|
||||||
|
skip_runtime_msg: bool = typedef.__name__ not in pld_spec_msg_names
|
||||||
|
if skip_runtime_msg:
|
||||||
|
continue
|
||||||
|
|
||||||
|
pld_field = structs.fields(typedef)[1]
|
||||||
|
assert pld_field.type is payload_spec # TODO-^ does this need to work to get all subtypes to adhere?
|
||||||
|
|
||||||
|
kwargs: dict[str, Any] = {
|
||||||
|
'cid': '666',
|
||||||
|
'pld': pld,
|
||||||
|
}
|
||||||
|
enc_msg: PayloadMsg = typedef(**kwargs)
|
||||||
|
|
||||||
|
_wire_bytes: bytes = _enc.encode(enc_msg)
|
||||||
|
wire_bytes: bytes = codec.enc.encode(enc_msg)
|
||||||
|
assert _wire_bytes == wire_bytes
|
||||||
|
|
||||||
|
ve: ValidationError|None = None
|
||||||
|
try:
|
||||||
|
dec_msg = codec.dec.decode(wire_bytes)
|
||||||
|
_dec_msg = _dec.decode(wire_bytes)
|
||||||
|
|
||||||
|
# decoded msg and thus payload should be exactly same!
|
||||||
|
assert (roundtrip := (
|
||||||
|
_dec_msg
|
||||||
|
==
|
||||||
|
dec_msg
|
||||||
|
==
|
||||||
|
enc_msg
|
||||||
|
))
|
||||||
|
|
||||||
|
if (
|
||||||
|
expect_roundtrip is not None
|
||||||
|
and expect_roundtrip != roundtrip
|
||||||
|
):
|
||||||
|
breakpoint()
|
||||||
|
|
||||||
|
assert (
|
||||||
|
pld
|
||||||
|
==
|
||||||
|
dec_msg.pld
|
||||||
|
==
|
||||||
|
enc_msg.pld
|
||||||
|
)
|
||||||
|
# assert (roundtrip := (_dec_msg == enc_msg))
|
||||||
|
|
||||||
|
except ValidationError as _ve:
|
||||||
|
ve = _ve
|
||||||
|
roundtrip: bool = False
|
||||||
|
if pld_val_type is payload_spec:
|
||||||
|
raise ValueError(
|
||||||
|
'Got `ValidationError` despite type-var match!?\n'
|
||||||
|
f'pld_val_type: {pld_val_type}\n'
|
||||||
|
f'payload_type: {payload_spec}\n'
|
||||||
|
) from ve
|
||||||
|
|
||||||
|
else:
|
||||||
|
# ow we good cuz the pld spec mismatched.
|
||||||
|
print(
|
||||||
|
'Got expected `ValidationError` since,\n'
|
||||||
|
f'{pld_val_type} is not {payload_spec}\n'
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
if (
|
||||||
|
payload_spec is not Any
|
||||||
|
and
|
||||||
|
pld_val_type is not payload_spec
|
||||||
|
):
|
||||||
|
raise ValueError(
|
||||||
|
'DID NOT `ValidationError` despite expected type match!?\n'
|
||||||
|
f'pld_val_type: {pld_val_type}\n'
|
||||||
|
f'payload_type: {payload_spec}\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
# full code decode should always be attempted!
|
||||||
|
if roundtrip is None:
|
||||||
|
breakpoint()
|
||||||
|
|
||||||
|
return roundtrip
|
||||||
|
|
||||||
|
|
||||||
|
def test_limit_msgspec():
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
debug_mode=True
|
||||||
|
):
|
||||||
|
|
||||||
|
# ensure we can round-trip a boxing `PayloadMsg`
|
||||||
|
assert chk_pld_type(
|
||||||
|
payload_spec=Any,
|
||||||
|
pld=None,
|
||||||
|
expect_roundtrip=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
# verify that a mis-typed payload value won't decode
|
||||||
|
assert not chk_pld_type(
|
||||||
|
payload_spec=int,
|
||||||
|
pld='doggy',
|
||||||
|
)
|
||||||
|
|
||||||
|
# parametrize the boxed `.pld` type as a custom-struct
|
||||||
|
# and ensure that parametrization propagates
|
||||||
|
# to all payload-msg-spec-able subtypes!
|
||||||
|
class CustomPayload(Struct):
|
||||||
|
name: str
|
||||||
|
value: Any
|
||||||
|
|
||||||
|
assert not chk_pld_type(
|
||||||
|
payload_spec=CustomPayload,
|
||||||
|
pld='doggy',
|
||||||
|
)
|
||||||
|
|
||||||
|
assert chk_pld_type(
|
||||||
|
payload_spec=CustomPayload,
|
||||||
|
pld=CustomPayload(name='doggy', value='urmom')
|
||||||
|
)
|
||||||
|
|
||||||
|
# yah, we can `.pause_from_sync()` now!
|
||||||
|
# breakpoint()
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
@ -95,8 +95,8 @@ async def trio_main(
|
||||||
|
|
||||||
# stash a "service nursery" as "actor local" (aka a Python global)
|
# stash a "service nursery" as "actor local" (aka a Python global)
|
||||||
global _nursery
|
global _nursery
|
||||||
tn = _nursery
|
n = _nursery
|
||||||
assert tn
|
assert n
|
||||||
|
|
||||||
async def consume_stream():
|
async def consume_stream():
|
||||||
async with wrapper_mngr() as stream:
|
async with wrapper_mngr() as stream:
|
||||||
|
|
@ -104,10 +104,10 @@ async def trio_main(
|
||||||
print(msg)
|
print(msg)
|
||||||
|
|
||||||
# run 2 tasks to ensure broadcaster chan use
|
# run 2 tasks to ensure broadcaster chan use
|
||||||
tn.start_soon(consume_stream)
|
n.start_soon(consume_stream)
|
||||||
tn.start_soon(consume_stream)
|
n.start_soon(consume_stream)
|
||||||
|
|
||||||
tn.start_soon(trio_sleep_and_err)
|
n.start_soon(trio_sleep_and_err)
|
||||||
|
|
||||||
await trio.sleep_forever()
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
@ -117,11 +117,8 @@ async def open_actor_local_nursery(
|
||||||
ctx: tractor.Context,
|
ctx: tractor.Context,
|
||||||
):
|
):
|
||||||
global _nursery
|
global _nursery
|
||||||
async with (
|
async with trio.open_nursery() as n:
|
||||||
tractor.trionics.collapse_eg(),
|
_nursery = n
|
||||||
trio.open_nursery() as tn
|
|
||||||
):
|
|
||||||
_nursery = tn
|
|
||||||
await ctx.started()
|
await ctx.started()
|
||||||
await trio.sleep(10)
|
await trio.sleep(10)
|
||||||
# await trio.sleep(1)
|
# await trio.sleep(1)
|
||||||
|
|
@ -135,7 +132,7 @@ async def open_actor_local_nursery(
|
||||||
# never yields back.. aka a scenario where the
|
# never yields back.. aka a scenario where the
|
||||||
# ``tractor.context`` task IS NOT in the service n's cancel
|
# ``tractor.context`` task IS NOT in the service n's cancel
|
||||||
# scope.
|
# scope.
|
||||||
tn.cancel_scope.cancel()
|
n.cancel_scope.cancel()
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
|
|
@ -160,7 +157,7 @@ def test_actor_managed_trio_nursery_task_error_cancels_aio(
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery() as n:
|
||||||
p = await n.start_actor(
|
p = await n.start_actor(
|
||||||
'nursery_mngr',
|
'nursery_mngr',
|
||||||
infect_asyncio=asyncio_mode, # TODO, is this enabling debug mode?
|
infect_asyncio=asyncio_mode,
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
async with (
|
async with (
|
||||||
|
|
|
||||||
|
|
@ -13,24 +13,26 @@ MESSAGE = 'tractoring at full speed'
|
||||||
def test_empty_mngrs_input_raises() -> None:
|
def test_empty_mngrs_input_raises() -> None:
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
with trio.fail_after(3):
|
with trio.fail_after(1):
|
||||||
async with (
|
async with (
|
||||||
open_actor_cluster(
|
open_actor_cluster(
|
||||||
modules=[__name__],
|
modules=[__name__],
|
||||||
|
|
||||||
# NOTE: ensure we can passthrough runtime opts
|
# NOTE: ensure we can passthrough runtime opts
|
||||||
loglevel='cancel',
|
loglevel='info',
|
||||||
debug_mode=False,
|
# debug_mode=True,
|
||||||
|
|
||||||
) as portals,
|
) as portals,
|
||||||
|
|
||||||
gather_contexts(mngrs=()),
|
gather_contexts(
|
||||||
|
# NOTE: it's the use of inline-generator syntax
|
||||||
|
# here that causes the empty input.
|
||||||
|
mngrs=(
|
||||||
|
p.open_context(worker) for p in portals.values()
|
||||||
|
),
|
||||||
|
),
|
||||||
):
|
):
|
||||||
# should fail before this?
|
assert 0
|
||||||
assert portals
|
|
||||||
|
|
||||||
# test should fail if we mk it here!
|
|
||||||
assert 0, 'Should have raised val-err !?'
|
|
||||||
|
|
||||||
with pytest.raises(ValueError):
|
with pytest.raises(ValueError):
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
|
||||||
|
|
@ -38,9 +38,9 @@ from tractor._testing import (
|
||||||
# - standard setup/teardown:
|
# - standard setup/teardown:
|
||||||
# ``Portal.open_context()`` starts a new
|
# ``Portal.open_context()`` starts a new
|
||||||
# remote task context in another actor. The target actor's task must
|
# remote task context in another actor. The target actor's task must
|
||||||
# call ``Context.started()`` to unblock this entry on the parent side.
|
# call ``Context.started()`` to unblock this entry on the caller side.
|
||||||
# the child task executes until complete and returns a final value
|
# the callee task executes until complete and returns a final value
|
||||||
# which is delivered to the parent side and retreived via
|
# which is delivered to the caller side and retreived via
|
||||||
# ``Context.result()``.
|
# ``Context.result()``.
|
||||||
|
|
||||||
# - cancel termination:
|
# - cancel termination:
|
||||||
|
|
@ -170,9 +170,9 @@ async def assert_state(value: bool):
|
||||||
[False, ValueError, KeyboardInterrupt],
|
[False, ValueError, KeyboardInterrupt],
|
||||||
)
|
)
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'child_blocks_forever',
|
'callee_blocks_forever',
|
||||||
[False, True],
|
[False, True],
|
||||||
ids=lambda item: f'child_blocks_forever={item}'
|
ids=lambda item: f'callee_blocks_forever={item}'
|
||||||
)
|
)
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'pointlessly_open_stream',
|
'pointlessly_open_stream',
|
||||||
|
|
@ -181,7 +181,7 @@ async def assert_state(value: bool):
|
||||||
)
|
)
|
||||||
def test_simple_context(
|
def test_simple_context(
|
||||||
error_parent,
|
error_parent,
|
||||||
child_blocks_forever,
|
callee_blocks_forever,
|
||||||
pointlessly_open_stream,
|
pointlessly_open_stream,
|
||||||
debug_mode: bool,
|
debug_mode: bool,
|
||||||
):
|
):
|
||||||
|
|
@ -204,13 +204,13 @@ def test_simple_context(
|
||||||
portal.open_context(
|
portal.open_context(
|
||||||
simple_setup_teardown,
|
simple_setup_teardown,
|
||||||
data=10,
|
data=10,
|
||||||
block_forever=child_blocks_forever,
|
block_forever=callee_blocks_forever,
|
||||||
) as (ctx, sent),
|
) as (ctx, sent),
|
||||||
):
|
):
|
||||||
assert current_ipc_ctx() is ctx
|
assert current_ipc_ctx() is ctx
|
||||||
assert sent == 11
|
assert sent == 11
|
||||||
|
|
||||||
if child_blocks_forever:
|
if callee_blocks_forever:
|
||||||
await portal.run(assert_state, value=True)
|
await portal.run(assert_state, value=True)
|
||||||
else:
|
else:
|
||||||
assert await ctx.result() == 'yo'
|
assert await ctx.result() == 'yo'
|
||||||
|
|
@ -220,7 +220,7 @@ def test_simple_context(
|
||||||
if error_parent:
|
if error_parent:
|
||||||
raise error_parent
|
raise error_parent
|
||||||
|
|
||||||
if child_blocks_forever:
|
if callee_blocks_forever:
|
||||||
await ctx.cancel()
|
await ctx.cancel()
|
||||||
else:
|
else:
|
||||||
# in this case the stream will send a
|
# in this case the stream will send a
|
||||||
|
|
@ -252,16 +252,16 @@ def test_simple_context(
|
||||||
pass
|
pass
|
||||||
except BaseExceptionGroup as beg:
|
except BaseExceptionGroup as beg:
|
||||||
# XXX: on windows it seems we may have to expect the group error
|
# XXX: on windows it seems we may have to expect the group error
|
||||||
from tractor.trionics import is_multi_cancelled
|
from tractor._exceptions import is_multi_cancelled
|
||||||
assert is_multi_cancelled(beg)
|
assert is_multi_cancelled(beg)
|
||||||
else:
|
else:
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'child_returns_early',
|
'callee_returns_early',
|
||||||
[True, False],
|
[True, False],
|
||||||
ids=lambda item: f'child_returns_early={item}'
|
ids=lambda item: f'callee_returns_early={item}'
|
||||||
)
|
)
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'cancel_method',
|
'cancel_method',
|
||||||
|
|
@ -273,14 +273,14 @@ def test_simple_context(
|
||||||
[True, False],
|
[True, False],
|
||||||
ids=lambda item: f'chk_ctx_result_before_exit={item}'
|
ids=lambda item: f'chk_ctx_result_before_exit={item}'
|
||||||
)
|
)
|
||||||
def test_parent_cancels(
|
def test_caller_cancels(
|
||||||
cancel_method: str,
|
cancel_method: str,
|
||||||
chk_ctx_result_before_exit: bool,
|
chk_ctx_result_before_exit: bool,
|
||||||
child_returns_early: bool,
|
callee_returns_early: bool,
|
||||||
debug_mode: bool,
|
debug_mode: bool,
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Verify that when the opening side of a context (aka the parent)
|
Verify that when the opening side of a context (aka the caller)
|
||||||
cancels that context, the ctx does not raise a cancelled when
|
cancels that context, the ctx does not raise a cancelled when
|
||||||
either calling `.result()` or on context exit.
|
either calling `.result()` or on context exit.
|
||||||
|
|
||||||
|
|
@ -294,7 +294,7 @@ def test_parent_cancels(
|
||||||
|
|
||||||
if (
|
if (
|
||||||
cancel_method == 'portal'
|
cancel_method == 'portal'
|
||||||
and not child_returns_early
|
and not callee_returns_early
|
||||||
):
|
):
|
||||||
try:
|
try:
|
||||||
res = await ctx.result()
|
res = await ctx.result()
|
||||||
|
|
@ -318,7 +318,7 @@ def test_parent_cancels(
|
||||||
pytest.fail(f'should not have raised ctxc\n{ctxc}')
|
pytest.fail(f'should not have raised ctxc\n{ctxc}')
|
||||||
|
|
||||||
# we actually get a result
|
# we actually get a result
|
||||||
if child_returns_early:
|
if callee_returns_early:
|
||||||
assert res == 'yo'
|
assert res == 'yo'
|
||||||
assert ctx.outcome is res
|
assert ctx.outcome is res
|
||||||
assert ctx.maybe_error is None
|
assert ctx.maybe_error is None
|
||||||
|
|
@ -362,14 +362,14 @@ def test_parent_cancels(
|
||||||
)
|
)
|
||||||
timeout: float = (
|
timeout: float = (
|
||||||
0.5
|
0.5
|
||||||
if not child_returns_early
|
if not callee_returns_early
|
||||||
else 2
|
else 2
|
||||||
)
|
)
|
||||||
with trio.fail_after(timeout):
|
with trio.fail_after(timeout):
|
||||||
async with (
|
async with (
|
||||||
expect_ctxc(
|
expect_ctxc(
|
||||||
yay=(
|
yay=(
|
||||||
not child_returns_early
|
not callee_returns_early
|
||||||
and cancel_method == 'portal'
|
and cancel_method == 'portal'
|
||||||
)
|
)
|
||||||
),
|
),
|
||||||
|
|
@ -377,13 +377,13 @@ def test_parent_cancels(
|
||||||
portal.open_context(
|
portal.open_context(
|
||||||
simple_setup_teardown,
|
simple_setup_teardown,
|
||||||
data=10,
|
data=10,
|
||||||
block_forever=not child_returns_early,
|
block_forever=not callee_returns_early,
|
||||||
) as (ctx, sent),
|
) as (ctx, sent),
|
||||||
):
|
):
|
||||||
|
|
||||||
if child_returns_early:
|
if callee_returns_early:
|
||||||
# ensure we block long enough before sending
|
# ensure we block long enough before sending
|
||||||
# a cancel such that the child has already
|
# a cancel such that the callee has already
|
||||||
# returned it's result.
|
# returned it's result.
|
||||||
await trio.sleep(0.5)
|
await trio.sleep(0.5)
|
||||||
|
|
||||||
|
|
@ -421,7 +421,7 @@ def test_parent_cancels(
|
||||||
# which should in turn cause `ctx._scope` to
|
# which should in turn cause `ctx._scope` to
|
||||||
# catch any cancellation?
|
# catch any cancellation?
|
||||||
if (
|
if (
|
||||||
not child_returns_early
|
not callee_returns_early
|
||||||
and cancel_method != 'portal'
|
and cancel_method != 'portal'
|
||||||
):
|
):
|
||||||
assert not ctx._scope.cancelled_caught
|
assert not ctx._scope.cancelled_caught
|
||||||
|
|
@ -430,11 +430,11 @@ def test_parent_cancels(
|
||||||
|
|
||||||
|
|
||||||
# basic stream terminations:
|
# basic stream terminations:
|
||||||
# - child context closes without using stream
|
# - callee context closes without using stream
|
||||||
# - parent context closes without using stream
|
# - caller context closes without using stream
|
||||||
# - parent context calls `Context.cancel()` while streaming
|
# - caller context calls `Context.cancel()` while streaming
|
||||||
# is ongoing resulting in child being cancelled
|
# is ongoing resulting in callee being cancelled
|
||||||
# - child calls `Context.cancel()` while streaming and parent
|
# - callee calls `Context.cancel()` while streaming and caller
|
||||||
# sees stream terminated in `RemoteActorError`
|
# sees stream terminated in `RemoteActorError`
|
||||||
|
|
||||||
# TODO: future possible features
|
# TODO: future possible features
|
||||||
|
|
@ -443,6 +443,7 @@ def test_parent_cancels(
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
async def close_ctx_immediately(
|
async def close_ctx_immediately(
|
||||||
|
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
@ -453,24 +454,13 @@ async def close_ctx_immediately(
|
||||||
async with ctx.open_stream():
|
async with ctx.open_stream():
|
||||||
pass
|
pass
|
||||||
|
|
||||||
print('child returning!')
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'parent_send_before_receive',
|
|
||||||
[
|
|
||||||
False,
|
|
||||||
True,
|
|
||||||
],
|
|
||||||
ids=lambda item: f'child_send_before_receive={item}'
|
|
||||||
)
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_child_exits_ctx_after_stream_open(
|
async def test_callee_closes_ctx_after_stream_open(
|
||||||
debug_mode: bool,
|
debug_mode: bool,
|
||||||
parent_send_before_receive: bool,
|
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
child context closes without using stream.
|
callee context closes without using stream.
|
||||||
|
|
||||||
This should result in a msg sequence
|
This should result in a msg sequence
|
||||||
|_<root>_
|
|_<root>_
|
||||||
|
|
@ -484,9 +474,6 @@ async def test_child_exits_ctx_after_stream_open(
|
||||||
=> {'stop': True, 'cid': <str>}
|
=> {'stop': True, 'cid': <str>}
|
||||||
|
|
||||||
'''
|
'''
|
||||||
timeout: float = (
|
|
||||||
0.5 if not debug_mode else 999
|
|
||||||
)
|
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=debug_mode,
|
debug_mode=debug_mode,
|
||||||
) as an:
|
) as an:
|
||||||
|
|
@ -495,7 +482,7 @@ async def test_child_exits_ctx_after_stream_open(
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
with trio.fail_after(timeout):
|
with trio.fail_after(0.5):
|
||||||
async with portal.open_context(
|
async with portal.open_context(
|
||||||
close_ctx_immediately,
|
close_ctx_immediately,
|
||||||
|
|
||||||
|
|
@ -507,56 +494,41 @@ async def test_child_exits_ctx_after_stream_open(
|
||||||
|
|
||||||
with trio.fail_after(0.4):
|
with trio.fail_after(0.4):
|
||||||
async with ctx.open_stream() as stream:
|
async with ctx.open_stream() as stream:
|
||||||
if parent_send_before_receive:
|
|
||||||
print('sending first msg from parent!')
|
|
||||||
await stream.send('yo')
|
|
||||||
|
|
||||||
# should fall through since ``StopAsyncIteration``
|
# should fall through since ``StopAsyncIteration``
|
||||||
# should be raised through translation of
|
# should be raised through translation of
|
||||||
# a ``trio.EndOfChannel`` by
|
# a ``trio.EndOfChannel`` by
|
||||||
# ``trio.abc.ReceiveChannel.__anext__()``
|
# ``trio.abc.ReceiveChannel.__anext__()``
|
||||||
msg = 10
|
async for _ in stream:
|
||||||
async for msg in stream:
|
|
||||||
# trigger failure if we DO NOT
|
# trigger failure if we DO NOT
|
||||||
# get an EOC!
|
# get an EOC!
|
||||||
assert 0
|
assert 0
|
||||||
else:
|
else:
|
||||||
# never should get anythinig new from
|
|
||||||
# the underlying stream
|
|
||||||
assert msg == 10
|
|
||||||
|
|
||||||
# verify stream is now closed
|
# verify stream is now closed
|
||||||
try:
|
try:
|
||||||
with trio.fail_after(0.3):
|
with trio.fail_after(0.3):
|
||||||
print('parent trying to `.receive()` on EoC stream!')
|
|
||||||
await stream.receive()
|
await stream.receive()
|
||||||
assert 0, 'should have raised eoc!?'
|
|
||||||
except trio.EndOfChannel:
|
except trio.EndOfChannel:
|
||||||
print('parent got EoC as expected!')
|
|
||||||
pass
|
pass
|
||||||
# raise
|
|
||||||
|
|
||||||
# TODO: should be just raise the closed resource err
|
# TODO: should be just raise the closed resource err
|
||||||
# directly here to enforce not allowing a re-open
|
# directly here to enforce not allowing a re-open
|
||||||
# of a stream to the context (at least until a time of
|
# of a stream to the context (at least until a time of
|
||||||
# if/when we decide that's a good idea?)
|
# if/when we decide that's a good idea?)
|
||||||
try:
|
try:
|
||||||
with trio.fail_after(timeout):
|
with trio.fail_after(0.5):
|
||||||
async with ctx.open_stream() as stream:
|
async with ctx.open_stream() as stream:
|
||||||
pass
|
pass
|
||||||
except trio.ClosedResourceError:
|
except trio.ClosedResourceError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
# if ctx._rx_chan._state.data:
|
|
||||||
# await tractor.pause()
|
|
||||||
|
|
||||||
await portal.cancel_actor()
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
async def expect_cancelled(
|
async def expect_cancelled(
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
send_before_receive: bool = False,
|
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
global _state
|
global _state
|
||||||
|
|
@ -566,10 +538,6 @@ async def expect_cancelled(
|
||||||
|
|
||||||
try:
|
try:
|
||||||
async with ctx.open_stream() as stream:
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
if send_before_receive:
|
|
||||||
await stream.send('yo')
|
|
||||||
|
|
||||||
async for msg in stream:
|
async for msg in stream:
|
||||||
await stream.send(msg) # echo server
|
await stream.send(msg) # echo server
|
||||||
|
|
||||||
|
|
@ -596,49 +564,26 @@ async def expect_cancelled(
|
||||||
raise
|
raise
|
||||||
|
|
||||||
else:
|
else:
|
||||||
assert 0, "child wasn't cancelled !?"
|
assert 0, "callee wasn't cancelled !?"
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'child_send_before_receive',
|
|
||||||
[
|
|
||||||
False,
|
|
||||||
True,
|
|
||||||
],
|
|
||||||
ids=lambda item: f'child_send_before_receive={item}'
|
|
||||||
)
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'rent_wait_for_msg',
|
|
||||||
[
|
|
||||||
False,
|
|
||||||
True,
|
|
||||||
],
|
|
||||||
ids=lambda item: f'rent_wait_for_msg={item}'
|
|
||||||
)
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'use_ctx_cancel_method',
|
'use_ctx_cancel_method',
|
||||||
[
|
[False, True],
|
||||||
False,
|
|
||||||
'pre_stream',
|
|
||||||
'post_stream_open',
|
|
||||||
'post_stream_close',
|
|
||||||
],
|
|
||||||
ids=lambda item: f'use_ctx_cancel_method={item}'
|
|
||||||
)
|
)
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_parent_exits_ctx_after_child_enters_stream(
|
async def test_caller_closes_ctx_after_callee_opens_stream(
|
||||||
use_ctx_cancel_method: bool|str,
|
use_ctx_cancel_method: bool,
|
||||||
debug_mode: bool,
|
debug_mode: bool,
|
||||||
rent_wait_for_msg: bool,
|
|
||||||
child_send_before_receive: bool,
|
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Parent-side of IPC context closes without sending on `MsgStream`.
|
caller context closes without using/opening stream
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=debug_mode,
|
debug_mode=debug_mode,
|
||||||
) as an:
|
) as an:
|
||||||
|
|
||||||
root: Actor = current_actor()
|
root: Actor = current_actor()
|
||||||
portal = await an.start_actor(
|
portal = await an.start_actor(
|
||||||
'ctx_cancelled',
|
'ctx_cancelled',
|
||||||
|
|
@ -647,45 +592,26 @@ async def test_parent_exits_ctx_after_child_enters_stream(
|
||||||
|
|
||||||
async with portal.open_context(
|
async with portal.open_context(
|
||||||
expect_cancelled,
|
expect_cancelled,
|
||||||
send_before_receive=child_send_before_receive,
|
|
||||||
) as (ctx, sent):
|
) as (ctx, sent):
|
||||||
assert sent is None
|
assert sent is None
|
||||||
|
|
||||||
await portal.run(assert_state, value=True)
|
await portal.run(assert_state, value=True)
|
||||||
|
|
||||||
# call `ctx.cancel()` explicitly
|
# call `ctx.cancel()` explicitly
|
||||||
if use_ctx_cancel_method == 'pre_stream':
|
if use_ctx_cancel_method:
|
||||||
await ctx.cancel()
|
await ctx.cancel()
|
||||||
|
|
||||||
# NOTE: means the local side `ctx._scope` will
|
# NOTE: means the local side `ctx._scope` will
|
||||||
# have been cancelled by an ctxc ack and thus
|
# have been cancelled by an ctxc ack and thus
|
||||||
# `._scope.cancelled_caught` should be set.
|
# `._scope.cancelled_caught` should be set.
|
||||||
async with (
|
try:
|
||||||
expect_ctxc(
|
async with ctx.open_stream() as stream:
|
||||||
|
async for msg in stream:
|
||||||
|
pass
|
||||||
|
|
||||||
|
except tractor.ContextCancelled as ctxc:
|
||||||
# XXX: the cause is US since we call
|
# XXX: the cause is US since we call
|
||||||
# `Context.cancel()` just above!
|
# `Context.cancel()` just above!
|
||||||
yay=True,
|
|
||||||
|
|
||||||
# XXX: must be propagated to __aexit__
|
|
||||||
# and should be silently absorbed there
|
|
||||||
# since we called `.cancel()` just above ;)
|
|
||||||
reraise=True,
|
|
||||||
) as maybe_ctxc,
|
|
||||||
):
|
|
||||||
async with ctx.open_stream() as stream:
|
|
||||||
|
|
||||||
if rent_wait_for_msg:
|
|
||||||
async for msg in stream:
|
|
||||||
print(f'PARENT rx: {msg!r}\n')
|
|
||||||
break
|
|
||||||
|
|
||||||
if use_ctx_cancel_method == 'post_stream_open':
|
|
||||||
await ctx.cancel()
|
|
||||||
|
|
||||||
if use_ctx_cancel_method == 'post_stream_close':
|
|
||||||
await ctx.cancel()
|
|
||||||
|
|
||||||
ctxc: tractor.ContextCancelled = maybe_ctxc.value
|
|
||||||
assert (
|
assert (
|
||||||
ctxc.canceller
|
ctxc.canceller
|
||||||
==
|
==
|
||||||
|
|
@ -694,6 +620,14 @@ async def test_parent_exits_ctx_after_child_enters_stream(
|
||||||
root.uid
|
root.uid
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# XXX: must be propagated to __aexit__
|
||||||
|
# and should be silently absorbed there
|
||||||
|
# since we called `.cancel()` just above ;)
|
||||||
|
raise
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert 0, "Should have context cancelled?"
|
||||||
|
|
||||||
# channel should still be up
|
# channel should still be up
|
||||||
assert portal.channel.connected()
|
assert portal.channel.connected()
|
||||||
|
|
||||||
|
|
@ -703,20 +637,13 @@ async def test_parent_exits_ctx_after_child_enters_stream(
|
||||||
value=False,
|
value=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
# XXX CHILD-BLOCKS case, we SHOULD NOT exit from the
|
|
||||||
# `.open_context()` before the child has returned,
|
|
||||||
# errored or been cancelled!
|
|
||||||
else:
|
else:
|
||||||
try:
|
try:
|
||||||
with trio.fail_after(
|
with trio.fail_after(0.2):
|
||||||
0.5 # if not debug_mode else 999
|
await ctx.result()
|
||||||
):
|
|
||||||
res = await ctx.wait_for_result()
|
|
||||||
assert res is not tractor._context.Unresolved
|
|
||||||
assert 0, "Callee should have blocked!?"
|
assert 0, "Callee should have blocked!?"
|
||||||
except trio.TooSlowError:
|
except trio.TooSlowError:
|
||||||
# NO-OP -> since already triggered by
|
# NO-OP -> since already called above
|
||||||
# `trio.fail_after()` above!
|
|
||||||
await ctx.cancel()
|
await ctx.cancel()
|
||||||
|
|
||||||
# NOTE: local scope should have absorbed the cancellation since
|
# NOTE: local scope should have absorbed the cancellation since
|
||||||
|
|
@ -756,7 +683,7 @@ async def test_parent_exits_ctx_after_child_enters_stream(
|
||||||
|
|
||||||
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_multitask_parent_cancels_from_nonroot_task(
|
async def test_multitask_caller_cancels_from_nonroot_task(
|
||||||
debug_mode: bool,
|
debug_mode: bool,
|
||||||
):
|
):
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
|
|
@ -808,6 +735,7 @@ async def test_multitask_parent_cancels_from_nonroot_task(
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
async def cancel_self(
|
async def cancel_self(
|
||||||
|
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
@ -847,11 +775,11 @@ async def cancel_self(
|
||||||
|
|
||||||
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_child_cancels_before_started(
|
async def test_callee_cancels_before_started(
|
||||||
debug_mode: bool,
|
debug_mode: bool,
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Callee calls `Context.cancel()` while streaming and parent
|
Callee calls `Context.cancel()` while streaming and caller
|
||||||
sees stream terminated in `ContextCancelled`.
|
sees stream terminated in `ContextCancelled`.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
|
@ -898,13 +826,14 @@ async def never_open_stream(
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
async def keep_sending_from_child(
|
async def keep_sending_from_callee(
|
||||||
|
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
msg_buffer_size: int|None = None,
|
msg_buffer_size: int|None = None,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
Send endlessly on the child stream.
|
Send endlessly on the calleee stream.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
await ctx.started()
|
await ctx.started()
|
||||||
|
|
@ -912,7 +841,7 @@ async def keep_sending_from_child(
|
||||||
msg_buffer_size=msg_buffer_size,
|
msg_buffer_size=msg_buffer_size,
|
||||||
) as stream:
|
) as stream:
|
||||||
for msg in count():
|
for msg in count():
|
||||||
print(f'child sending {msg}')
|
print(f'callee sending {msg}')
|
||||||
await stream.send(msg)
|
await stream.send(msg)
|
||||||
await trio.sleep(0.01)
|
await trio.sleep(0.01)
|
||||||
|
|
||||||
|
|
@ -920,12 +849,12 @@ async def keep_sending_from_child(
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'overrun_by',
|
'overrun_by',
|
||||||
[
|
[
|
||||||
('parent', 1, never_open_stream),
|
('caller', 1, never_open_stream),
|
||||||
('child', 0, keep_sending_from_child),
|
('callee', 0, keep_sending_from_callee),
|
||||||
],
|
],
|
||||||
ids=[
|
ids=[
|
||||||
('parent_1buf_never_open_stream'),
|
('caller_1buf_never_open_stream'),
|
||||||
('child_0buf_keep_sending_from_child'),
|
('callee_0buf_keep_sending_from_callee'),
|
||||||
]
|
]
|
||||||
)
|
)
|
||||||
def test_one_end_stream_not_opened(
|
def test_one_end_stream_not_opened(
|
||||||
|
|
@ -956,7 +885,8 @@ def test_one_end_stream_not_opened(
|
||||||
) as (ctx, sent):
|
) as (ctx, sent):
|
||||||
assert sent is None
|
assert sent is None
|
||||||
|
|
||||||
if 'parent' in overrunner:
|
if 'caller' in overrunner:
|
||||||
|
|
||||||
async with ctx.open_stream() as stream:
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
# itersend +1 msg more then the buffer size
|
# itersend +1 msg more then the buffer size
|
||||||
|
|
@ -971,7 +901,7 @@ def test_one_end_stream_not_opened(
|
||||||
await trio.sleep_forever()
|
await trio.sleep_forever()
|
||||||
|
|
||||||
else:
|
else:
|
||||||
# child overruns parent case so we do nothing here
|
# callee overruns caller case so we do nothing here
|
||||||
await trio.sleep_forever()
|
await trio.sleep_forever()
|
||||||
|
|
||||||
await portal.cancel_actor()
|
await portal.cancel_actor()
|
||||||
|
|
@ -979,19 +909,19 @@ def test_one_end_stream_not_opened(
|
||||||
# 2 overrun cases and the no overrun case (which pushes right up to
|
# 2 overrun cases and the no overrun case (which pushes right up to
|
||||||
# the msg limit)
|
# the msg limit)
|
||||||
if (
|
if (
|
||||||
overrunner == 'parent'
|
overrunner == 'caller'
|
||||||
):
|
):
|
||||||
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
assert excinfo.value.boxed_type == StreamOverrun
|
assert excinfo.value.boxed_type == StreamOverrun
|
||||||
|
|
||||||
elif overrunner == 'child':
|
elif overrunner == 'callee':
|
||||||
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
# TODO: embedded remote errors so that we can verify the source
|
# TODO: embedded remote errors so that we can verify the source
|
||||||
# error? the child delivers an error which is an overrun
|
# error? the callee delivers an error which is an overrun
|
||||||
# wrapped in a remote actor error.
|
# wrapped in a remote actor error.
|
||||||
assert excinfo.value.boxed_type == tractor.RemoteActorError
|
assert excinfo.value.boxed_type == tractor.RemoteActorError
|
||||||
|
|
||||||
|
|
@ -1001,6 +931,7 @@ def test_one_end_stream_not_opened(
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
async def echo_back_sequence(
|
async def echo_back_sequence(
|
||||||
|
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
seq: list[int],
|
seq: list[int],
|
||||||
wait_for_cancel: bool,
|
wait_for_cancel: bool,
|
||||||
|
|
@ -1010,12 +941,12 @@ async def echo_back_sequence(
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
Send endlessly on the child stream using a small buffer size
|
Send endlessly on the calleee stream using a small buffer size
|
||||||
setting on the contex to simulate backlogging that would normally
|
setting on the contex to simulate backlogging that would normally
|
||||||
cause overruns.
|
cause overruns.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
# NOTE: ensure that if the parent is expecting to cancel this task
|
# NOTE: ensure that if the caller is expecting to cancel this task
|
||||||
# that we stay echoing much longer then they are so we don't
|
# that we stay echoing much longer then they are so we don't
|
||||||
# return early instead of receive the cancel msg.
|
# return early instead of receive the cancel msg.
|
||||||
total_batches: int = (
|
total_batches: int = (
|
||||||
|
|
@ -1065,18 +996,18 @@ async def echo_back_sequence(
|
||||||
if be_slow:
|
if be_slow:
|
||||||
await trio.sleep(0.05)
|
await trio.sleep(0.05)
|
||||||
|
|
||||||
print('child waiting on next')
|
print('callee waiting on next')
|
||||||
|
|
||||||
print(f'child echoing back latest batch\n{batch}')
|
print(f'callee echoing back latest batch\n{batch}')
|
||||||
for msg in batch:
|
for msg in batch:
|
||||||
print(f'child sending msg\n{msg}')
|
print(f'callee sending msg\n{msg}')
|
||||||
await stream.send(msg)
|
await stream.send(msg)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
return 'yo'
|
return 'yo'
|
||||||
finally:
|
finally:
|
||||||
print(
|
print(
|
||||||
'exiting child with context:\n'
|
'exiting callee with context:\n'
|
||||||
f'{pformat(ctx)}\n'
|
f'{pformat(ctx)}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -1130,7 +1061,7 @@ def test_maybe_allow_overruns_stream(
|
||||||
debug_mode=debug_mode,
|
debug_mode=debug_mode,
|
||||||
) as an:
|
) as an:
|
||||||
portal = await an.start_actor(
|
portal = await an.start_actor(
|
||||||
'child_sends_forever',
|
'callee_sends_forever',
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
loglevel=loglevel,
|
loglevel=loglevel,
|
||||||
debug_mode=debug_mode,
|
debug_mode=debug_mode,
|
||||||
|
|
|
||||||
|
|
@ -7,11 +7,8 @@ import platform
|
||||||
from functools import partial
|
from functools import partial
|
||||||
import itertools
|
import itertools
|
||||||
|
|
||||||
import psutil
|
|
||||||
import pytest
|
import pytest
|
||||||
import subprocess
|
|
||||||
import tractor
|
import tractor
|
||||||
from tractor.trionics import collapse_eg
|
|
||||||
from tractor._testing import tractor_test
|
from tractor._testing import tractor_test
|
||||||
import trio
|
import trio
|
||||||
|
|
||||||
|
|
@ -29,7 +26,7 @@ async def test_reg_then_unreg(reg_addr):
|
||||||
portal = await n.start_actor('actor', enable_modules=[__name__])
|
portal = await n.start_actor('actor', enable_modules=[__name__])
|
||||||
uid = portal.channel.uid
|
uid = portal.channel.uid
|
||||||
|
|
||||||
async with tractor.get_registry(reg_addr) as aportal:
|
async with tractor.get_registry(*reg_addr) as aportal:
|
||||||
# this local actor should be the arbiter
|
# this local actor should be the arbiter
|
||||||
assert actor is aportal.actor
|
assert actor is aportal.actor
|
||||||
|
|
||||||
|
|
@ -155,25 +152,15 @@ async def unpack_reg(actor_or_portal):
|
||||||
async def spawn_and_check_registry(
|
async def spawn_and_check_registry(
|
||||||
reg_addr: tuple,
|
reg_addr: tuple,
|
||||||
use_signal: bool,
|
use_signal: bool,
|
||||||
debug_mode: bool = False,
|
|
||||||
remote_arbiter: bool = False,
|
remote_arbiter: bool = False,
|
||||||
with_streaming: bool = False,
|
with_streaming: bool = False,
|
||||||
maybe_daemon: tuple[
|
|
||||||
subprocess.Popen,
|
|
||||||
psutil.Process,
|
|
||||||
]|None = None,
|
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
if maybe_daemon:
|
|
||||||
popen, proc = maybe_daemon
|
|
||||||
# breakpoint()
|
|
||||||
|
|
||||||
async with tractor.open_root_actor(
|
async with tractor.open_root_actor(
|
||||||
registry_addrs=[reg_addr],
|
registry_addrs=[reg_addr],
|
||||||
debug_mode=debug_mode,
|
|
||||||
):
|
):
|
||||||
async with tractor.get_registry(reg_addr) as portal:
|
async with tractor.get_registry(*reg_addr) as portal:
|
||||||
# runtime needs to be up to call this
|
# runtime needs to be up to call this
|
||||||
actor = tractor.current_actor()
|
actor = tractor.current_actor()
|
||||||
|
|
||||||
|
|
@ -189,30 +176,28 @@ async def spawn_and_check_registry(
|
||||||
extra = 2 # local root actor + remote arbiter
|
extra = 2 # local root actor + remote arbiter
|
||||||
|
|
||||||
# ensure current actor is registered
|
# ensure current actor is registered
|
||||||
registry: dict = await get_reg()
|
registry = await get_reg()
|
||||||
assert actor.uid in registry
|
assert actor.uid in registry
|
||||||
|
|
||||||
try:
|
try:
|
||||||
async with tractor.open_nursery() as an:
|
async with tractor.open_nursery() as n:
|
||||||
async with (
|
async with trio.open_nursery() as trion:
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as trion,
|
|
||||||
):
|
|
||||||
portals = {}
|
portals = {}
|
||||||
for i in range(3):
|
for i in range(3):
|
||||||
name = f'a{i}'
|
name = f'a{i}'
|
||||||
if with_streaming:
|
if with_streaming:
|
||||||
portals[name] = await an.start_actor(
|
portals[name] = await n.start_actor(
|
||||||
name=name, enable_modules=[__name__])
|
name=name, enable_modules=[__name__])
|
||||||
|
|
||||||
else: # no streaming
|
else: # no streaming
|
||||||
portals[name] = await an.run_in_actor(
|
portals[name] = await n.run_in_actor(
|
||||||
trio.sleep_forever, name=name)
|
trio.sleep_forever, name=name)
|
||||||
|
|
||||||
# wait on last actor to come up
|
# wait on last actor to come up
|
||||||
async with tractor.wait_for_actor(name):
|
async with tractor.wait_for_actor(name):
|
||||||
registry = await get_reg()
|
registry = await get_reg()
|
||||||
for uid in an._children:
|
for uid in n._children:
|
||||||
assert uid in registry
|
assert uid in registry
|
||||||
|
|
||||||
assert len(portals) + extra == len(registry)
|
assert len(portals) + extra == len(registry)
|
||||||
|
|
@ -245,7 +230,6 @@ async def spawn_and_check_registry(
|
||||||
@pytest.mark.parametrize('use_signal', [False, True])
|
@pytest.mark.parametrize('use_signal', [False, True])
|
||||||
@pytest.mark.parametrize('with_streaming', [False, True])
|
@pytest.mark.parametrize('with_streaming', [False, True])
|
||||||
def test_subactors_unregister_on_cancel(
|
def test_subactors_unregister_on_cancel(
|
||||||
debug_mode: bool,
|
|
||||||
start_method,
|
start_method,
|
||||||
use_signal,
|
use_signal,
|
||||||
reg_addr,
|
reg_addr,
|
||||||
|
|
@ -262,7 +246,6 @@ def test_subactors_unregister_on_cancel(
|
||||||
spawn_and_check_registry,
|
spawn_and_check_registry,
|
||||||
reg_addr,
|
reg_addr,
|
||||||
use_signal,
|
use_signal,
|
||||||
debug_mode=debug_mode,
|
|
||||||
remote_arbiter=False,
|
remote_arbiter=False,
|
||||||
with_streaming=with_streaming,
|
with_streaming=with_streaming,
|
||||||
),
|
),
|
||||||
|
|
@ -272,8 +255,7 @@ def test_subactors_unregister_on_cancel(
|
||||||
@pytest.mark.parametrize('use_signal', [False, True])
|
@pytest.mark.parametrize('use_signal', [False, True])
|
||||||
@pytest.mark.parametrize('with_streaming', [False, True])
|
@pytest.mark.parametrize('with_streaming', [False, True])
|
||||||
def test_subactors_unregister_on_cancel_remote_daemon(
|
def test_subactors_unregister_on_cancel_remote_daemon(
|
||||||
daemon: subprocess.Popen,
|
daemon,
|
||||||
debug_mode: bool,
|
|
||||||
start_method,
|
start_method,
|
||||||
use_signal,
|
use_signal,
|
||||||
reg_addr,
|
reg_addr,
|
||||||
|
|
@ -289,13 +271,8 @@ def test_subactors_unregister_on_cancel_remote_daemon(
|
||||||
spawn_and_check_registry,
|
spawn_and_check_registry,
|
||||||
reg_addr,
|
reg_addr,
|
||||||
use_signal,
|
use_signal,
|
||||||
debug_mode=debug_mode,
|
|
||||||
remote_arbiter=True,
|
remote_arbiter=True,
|
||||||
with_streaming=with_streaming,
|
with_streaming=with_streaming,
|
||||||
maybe_daemon=(
|
|
||||||
daemon,
|
|
||||||
psutil.Process(daemon.pid)
|
|
||||||
),
|
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -321,7 +298,7 @@ async def close_chans_before_nursery(
|
||||||
async with tractor.open_root_actor(
|
async with tractor.open_root_actor(
|
||||||
registry_addrs=[reg_addr],
|
registry_addrs=[reg_addr],
|
||||||
):
|
):
|
||||||
async with tractor.get_registry(reg_addr) as aportal:
|
async with tractor.get_registry(*reg_addr) as aportal:
|
||||||
try:
|
try:
|
||||||
get_reg = partial(unpack_reg, aportal)
|
get_reg = partial(unpack_reg, aportal)
|
||||||
|
|
||||||
|
|
@ -339,12 +316,9 @@ async def close_chans_before_nursery(
|
||||||
async with portal2.open_stream_from(
|
async with portal2.open_stream_from(
|
||||||
stream_forever
|
stream_forever
|
||||||
) as agen2:
|
) as agen2:
|
||||||
async with (
|
async with trio.open_nursery() as n:
|
||||||
collapse_eg(),
|
n.start_soon(streamer, agen1)
|
||||||
trio.open_nursery() as tn,
|
n.start_soon(cancel, use_signal, .5)
|
||||||
):
|
|
||||||
tn.start_soon(streamer, agen1)
|
|
||||||
tn.start_soon(cancel, use_signal, .5)
|
|
||||||
try:
|
try:
|
||||||
await streamer(agen2)
|
await streamer(agen2)
|
||||||
finally:
|
finally:
|
||||||
|
|
@ -395,7 +369,7 @@ def test_close_channel_explicit(
|
||||||
|
|
||||||
@pytest.mark.parametrize('use_signal', [False, True])
|
@pytest.mark.parametrize('use_signal', [False, True])
|
||||||
def test_close_channel_explicit_remote_arbiter(
|
def test_close_channel_explicit_remote_arbiter(
|
||||||
daemon: subprocess.Popen,
|
daemon,
|
||||||
start_method,
|
start_method,
|
||||||
use_signal,
|
use_signal,
|
||||||
reg_addr,
|
reg_addr,
|
||||||
|
|
|
||||||
|
|
@ -19,7 +19,7 @@ from tractor._testing import (
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def run_example_in_subproc(
|
def run_example_in_subproc(
|
||||||
loglevel: str,
|
loglevel: str,
|
||||||
testdir: pytest.Pytester,
|
testdir: pytest.Testdir,
|
||||||
reg_addr: tuple[str, int],
|
reg_addr: tuple[str, int],
|
||||||
):
|
):
|
||||||
|
|
||||||
|
|
@ -66,9 +66,6 @@ def run_example_in_subproc(
|
||||||
# due to backpressure!!!
|
# due to backpressure!!!
|
||||||
proc = testdir.popen(
|
proc = testdir.popen(
|
||||||
cmdargs,
|
cmdargs,
|
||||||
stdin=subprocess.PIPE,
|
|
||||||
stdout=subprocess.PIPE,
|
|
||||||
stderr=subprocess.PIPE,
|
|
||||||
**kwargs,
|
**kwargs,
|
||||||
)
|
)
|
||||||
assert not proc.returncode
|
assert not proc.returncode
|
||||||
|
|
@ -84,37 +81,27 @@ def run_example_in_subproc(
|
||||||
|
|
||||||
# walk yields: (dirpath, dirnames, filenames)
|
# walk yields: (dirpath, dirnames, filenames)
|
||||||
[
|
[
|
||||||
(p[0], f)
|
(p[0], f) for p in os.walk(examples_dir()) for f in p[2]
|
||||||
for p in os.walk(examples_dir())
|
|
||||||
for f in p[2]
|
|
||||||
|
|
||||||
if (
|
if '__' not in f
|
||||||
'__' not in f
|
|
||||||
and f[0] != '_'
|
and f[0] != '_'
|
||||||
and 'debugging' not in p[0]
|
and 'debugging' not in p[0]
|
||||||
and 'integration' not in p[0]
|
and 'integration' not in p[0]
|
||||||
and 'advanced_faults' not in p[0]
|
and 'advanced_faults' not in p[0]
|
||||||
and 'multihost' not in p[0]
|
|
||||||
and 'trio' not in p[0]
|
|
||||||
)
|
|
||||||
],
|
],
|
||||||
|
|
||||||
ids=lambda t: t[1],
|
ids=lambda t: t[1],
|
||||||
)
|
)
|
||||||
def test_example(
|
def test_example(run_example_in_subproc, example_script):
|
||||||
run_example_in_subproc,
|
"""Load and run scripts from this repo's ``examples/`` dir as a user
|
||||||
example_script,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Load and run scripts from this repo's ``examples/`` dir as a user
|
|
||||||
would copy and pasing them into their editor.
|
would copy and pasing them into their editor.
|
||||||
|
|
||||||
On windows a little more "finessing" is done to make
|
On windows a little more "finessing" is done to make
|
||||||
``multiprocessing`` play nice: we copy the ``__main__.py`` into the
|
``multiprocessing`` play nice: we copy the ``__main__.py`` into the
|
||||||
test directory and invoke the script as a module with ``python -m
|
test directory and invoke the script as a module with ``python -m
|
||||||
test_example``.
|
test_example``.
|
||||||
|
"""
|
||||||
'''
|
ex_file = os.path.join(*example_script)
|
||||||
ex_file: str = os.path.join(*example_script)
|
|
||||||
|
|
||||||
if 'rpc_bidir_streaming' in ex_file and sys.version_info < (3, 9):
|
if 'rpc_bidir_streaming' in ex_file and sys.version_info < (3, 9):
|
||||||
pytest.skip("2-way streaming example requires py3.9 async with syntax")
|
pytest.skip("2-way streaming example requires py3.9 async with syntax")
|
||||||
|
|
@ -123,14 +110,10 @@ def test_example(
|
||||||
code = ex.read()
|
code = ex.read()
|
||||||
|
|
||||||
with run_example_in_subproc(code) as proc:
|
with run_example_in_subproc(code) as proc:
|
||||||
err = None
|
proc.wait()
|
||||||
try:
|
err, _ = proc.stderr.read(), proc.stdout.read()
|
||||||
if not proc.poll():
|
# print(f'STDERR: {err}')
|
||||||
_, err = proc.communicate(timeout=15)
|
# print(f'STDOUT: {out}')
|
||||||
|
|
||||||
except subprocess.TimeoutExpired as e:
|
|
||||||
proc.kill()
|
|
||||||
err = e.stderr
|
|
||||||
|
|
||||||
# if we get some gnarly output let's aggregate and raise
|
# if we get some gnarly output let's aggregate and raise
|
||||||
if err:
|
if err:
|
||||||
|
|
@ -144,8 +127,7 @@ def test_example(
|
||||||
# shouldn't eventually once we figure out what's
|
# shouldn't eventually once we figure out what's
|
||||||
# a better way to be explicit about aio side
|
# a better way to be explicit about aio side
|
||||||
# cancels?
|
# cancels?
|
||||||
and
|
and 'asyncio.exceptions.CancelledError' not in last_error
|
||||||
'asyncio.exceptions.CancelledError' not in last_error
|
|
||||||
):
|
):
|
||||||
raise Exception(errmsg)
|
raise Exception(errmsg)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,946 +0,0 @@
|
||||||
'''
|
|
||||||
Low-level functional audits for our
|
|
||||||
"capability based messaging"-spec feats.
|
|
||||||
|
|
||||||
B~)
|
|
||||||
|
|
||||||
'''
|
|
||||||
from contextlib import (
|
|
||||||
contextmanager as cm,
|
|
||||||
# nullcontext,
|
|
||||||
)
|
|
||||||
import importlib
|
|
||||||
from typing import (
|
|
||||||
Any,
|
|
||||||
Type,
|
|
||||||
Union,
|
|
||||||
)
|
|
||||||
|
|
||||||
from msgspec import (
|
|
||||||
# structs,
|
|
||||||
# msgpack,
|
|
||||||
Raw,
|
|
||||||
# Struct,
|
|
||||||
ValidationError,
|
|
||||||
)
|
|
||||||
import pytest
|
|
||||||
import trio
|
|
||||||
|
|
||||||
import tractor
|
|
||||||
from tractor import (
|
|
||||||
Actor,
|
|
||||||
# _state,
|
|
||||||
MsgTypeError,
|
|
||||||
Context,
|
|
||||||
)
|
|
||||||
from tractor.msg import (
|
|
||||||
_codec,
|
|
||||||
_ctxvar_MsgCodec,
|
|
||||||
_exts,
|
|
||||||
|
|
||||||
NamespacePath,
|
|
||||||
MsgCodec,
|
|
||||||
MsgDec,
|
|
||||||
mk_codec,
|
|
||||||
mk_dec,
|
|
||||||
apply_codec,
|
|
||||||
current_codec,
|
|
||||||
)
|
|
||||||
from tractor.msg.types import (
|
|
||||||
log,
|
|
||||||
Started,
|
|
||||||
# _payload_msgs,
|
|
||||||
# PayloadMsg,
|
|
||||||
# mk_msg_spec,
|
|
||||||
)
|
|
||||||
from tractor.msg._ops import (
|
|
||||||
limit_plds,
|
|
||||||
)
|
|
||||||
|
|
||||||
def enc_nsp(obj: Any) -> Any:
|
|
||||||
actor: Actor = tractor.current_actor(
|
|
||||||
err_on_no_runtime=False,
|
|
||||||
)
|
|
||||||
uid: tuple[str, str]|None = None if not actor else actor.uid
|
|
||||||
print(f'{uid} ENC HOOK')
|
|
||||||
|
|
||||||
match obj:
|
|
||||||
# case NamespacePath()|str():
|
|
||||||
case NamespacePath():
|
|
||||||
encoded: str = str(obj)
|
|
||||||
print(
|
|
||||||
f'----- ENCODING `NamespacePath` as `str` ------\n'
|
|
||||||
f'|_obj:{type(obj)!r} = {obj!r}\n'
|
|
||||||
f'|_encoded: str = {encoded!r}\n'
|
|
||||||
)
|
|
||||||
# if type(obj) != NamespacePath:
|
|
||||||
# breakpoint()
|
|
||||||
return encoded
|
|
||||||
case _:
|
|
||||||
logmsg: str = (
|
|
||||||
f'{uid}\n'
|
|
||||||
'FAILED ENCODE\n'
|
|
||||||
f'obj-> `{obj}: {type(obj)}`\n'
|
|
||||||
)
|
|
||||||
raise NotImplementedError(logmsg)
|
|
||||||
|
|
||||||
|
|
||||||
def dec_nsp(
|
|
||||||
obj_type: Type,
|
|
||||||
obj: Any,
|
|
||||||
|
|
||||||
) -> Any:
|
|
||||||
# breakpoint()
|
|
||||||
actor: Actor = tractor.current_actor(
|
|
||||||
err_on_no_runtime=False,
|
|
||||||
)
|
|
||||||
uid: tuple[str, str]|None = None if not actor else actor.uid
|
|
||||||
print(
|
|
||||||
f'{uid}\n'
|
|
||||||
'CUSTOM DECODE\n'
|
|
||||||
f'type-arg-> {obj_type}\n'
|
|
||||||
f'obj-arg-> `{obj}`: {type(obj)}\n'
|
|
||||||
)
|
|
||||||
nsp = None
|
|
||||||
# XXX, never happens right?
|
|
||||||
if obj_type is Raw:
|
|
||||||
breakpoint()
|
|
||||||
|
|
||||||
if (
|
|
||||||
obj_type is NamespacePath
|
|
||||||
and isinstance(obj, str)
|
|
||||||
and ':' in obj
|
|
||||||
):
|
|
||||||
nsp = NamespacePath(obj)
|
|
||||||
# TODO: we could built a generic handler using
|
|
||||||
# JUST matching the obj_type part?
|
|
||||||
# nsp = obj_type(obj)
|
|
||||||
|
|
||||||
if nsp:
|
|
||||||
print(f'Returning NSP instance: {nsp}')
|
|
||||||
return nsp
|
|
||||||
|
|
||||||
logmsg: str = (
|
|
||||||
f'{uid}\n'
|
|
||||||
'FAILED DECODE\n'
|
|
||||||
f'type-> {obj_type}\n'
|
|
||||||
f'obj-arg-> `{obj}`: {type(obj)}\n\n'
|
|
||||||
f'current codec:\n'
|
|
||||||
f'{current_codec()}\n'
|
|
||||||
)
|
|
||||||
# TODO: figure out the ignore subsys for this!
|
|
||||||
# -[ ] option whether to defense-relay backc the msg
|
|
||||||
# inside an `Invalid`/`Ignore`
|
|
||||||
# -[ ] how to make this handling pluggable such that a
|
|
||||||
# `Channel`/`MsgTransport` can intercept and process
|
|
||||||
# back msgs either via exception handling or some other
|
|
||||||
# signal?
|
|
||||||
log.warning(logmsg)
|
|
||||||
# NOTE: this delivers the invalid
|
|
||||||
# value up to `msgspec`'s decoding
|
|
||||||
# machinery for error raising.
|
|
||||||
return obj
|
|
||||||
# raise NotImplementedError(logmsg)
|
|
||||||
|
|
||||||
|
|
||||||
def ex_func(*args):
|
|
||||||
'''
|
|
||||||
A mod level func we can ref and load via our `NamespacePath`
|
|
||||||
python-object pointer `str` subtype.
|
|
||||||
|
|
||||||
'''
|
|
||||||
print(f'ex_func({args})')
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'add_codec_hooks',
|
|
||||||
[
|
|
||||||
True,
|
|
||||||
False,
|
|
||||||
],
|
|
||||||
ids=['use_codec_hooks', 'no_codec_hooks'],
|
|
||||||
)
|
|
||||||
def test_custom_extension_types(
|
|
||||||
debug_mode: bool,
|
|
||||||
add_codec_hooks: bool
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Verify that a `MsgCodec` (used for encoding all outbound IPC msgs
|
|
||||||
and decoding all inbound `PayloadMsg`s) and a paired `MsgDec`
|
|
||||||
(used for decoding the `PayloadMsg.pld: Raw` received within a given
|
|
||||||
task's ipc `Context` scope) can both send and receive "extension types"
|
|
||||||
as supported via custom converter hooks passed to `msgspec`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
nsp_pld_dec: MsgDec = mk_dec(
|
|
||||||
spec=None, # ONLY support the ext type
|
|
||||||
dec_hook=dec_nsp if add_codec_hooks else None,
|
|
||||||
ext_types=[NamespacePath],
|
|
||||||
)
|
|
||||||
nsp_codec: MsgCodec = mk_codec(
|
|
||||||
# ipc_pld_spec=Raw, # default!
|
|
||||||
|
|
||||||
# NOTE XXX: the encode hook MUST be used no matter what since
|
|
||||||
# our `NamespacePath` is not any of a `Any` native type nor
|
|
||||||
# a `msgspec.Struct` subtype - so `msgspec` has no way to know
|
|
||||||
# how to encode it unless we provide the custom hook.
|
|
||||||
#
|
|
||||||
# AGAIN that is, regardless of whether we spec an
|
|
||||||
# `Any`-decoded-pld the enc has no knowledge (by default)
|
|
||||||
# how to enc `NamespacePath` (nsp), so we add a custom
|
|
||||||
# hook to do that ALWAYS.
|
|
||||||
enc_hook=enc_nsp if add_codec_hooks else None,
|
|
||||||
|
|
||||||
# XXX NOTE: pretty sure this is mutex with the `type=` to
|
|
||||||
# `Decoder`? so it won't work in tandem with the
|
|
||||||
# `ipc_pld_spec` passed above?
|
|
||||||
ext_types=[NamespacePath],
|
|
||||||
|
|
||||||
# TODO? is it useful to have the `.pld` decoded *prior* to
|
|
||||||
# the `PldRx`?? like perf or mem related?
|
|
||||||
# ext_dec=nsp_pld_dec,
|
|
||||||
)
|
|
||||||
if add_codec_hooks:
|
|
||||||
assert nsp_codec.dec.dec_hook is None
|
|
||||||
|
|
||||||
# TODO? if we pass `ext_dec` above?
|
|
||||||
# assert nsp_codec.dec.dec_hook is dec_nsp
|
|
||||||
|
|
||||||
assert nsp_codec.enc.enc_hook is enc_nsp
|
|
||||||
|
|
||||||
nsp = NamespacePath.from_ref(ex_func)
|
|
||||||
|
|
||||||
try:
|
|
||||||
nsp_bytes: bytes = nsp_codec.encode(nsp)
|
|
||||||
nsp_rt_sin_msg = nsp_pld_dec.decode(nsp_bytes)
|
|
||||||
nsp_rt_sin_msg.load_ref() is ex_func
|
|
||||||
except TypeError:
|
|
||||||
if not add_codec_hooks:
|
|
||||||
pass
|
|
||||||
|
|
||||||
try:
|
|
||||||
msg_bytes: bytes = nsp_codec.encode(
|
|
||||||
Started(
|
|
||||||
cid='cid',
|
|
||||||
pld=nsp,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
# since the ext-type obj should also be set as the msg.pld
|
|
||||||
assert nsp_bytes in msg_bytes
|
|
||||||
started_rt: Started = nsp_codec.decode(msg_bytes)
|
|
||||||
pld: Raw = started_rt.pld
|
|
||||||
assert isinstance(pld, Raw)
|
|
||||||
nsp_rt: NamespacePath = nsp_pld_dec.decode(pld)
|
|
||||||
assert isinstance(nsp_rt, NamespacePath)
|
|
||||||
# in obj comparison terms they should be the same
|
|
||||||
assert nsp_rt == nsp
|
|
||||||
# ensure we've decoded to ext type!
|
|
||||||
assert nsp_rt.load_ref() is ex_func
|
|
||||||
|
|
||||||
except TypeError:
|
|
||||||
if not add_codec_hooks:
|
|
||||||
pass
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def sleep_forever_in_sub(
|
|
||||||
ctx: Context,
|
|
||||||
) -> None:
|
|
||||||
await trio.sleep_forever()
|
|
||||||
|
|
||||||
|
|
||||||
def mk_custom_codec(
|
|
||||||
add_hooks: bool,
|
|
||||||
|
|
||||||
) -> tuple[
|
|
||||||
MsgCodec, # encode to send
|
|
||||||
MsgDec, # pld receive-n-decode
|
|
||||||
]:
|
|
||||||
'''
|
|
||||||
Create custom `msgpack` enc/dec-hooks and set a `Decoder`
|
|
||||||
which only loads `pld_spec` (like `NamespacePath`) types.
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
# XXX NOTE XXX: despite defining `NamespacePath` as a type
|
|
||||||
# field on our `PayloadMsg.pld`, we still need a enc/dec_hook() pair
|
|
||||||
# to cast to/from that type on the wire. See the docs:
|
|
||||||
# https://jcristharif.com/msgspec/extending.html#mapping-to-from-native-types
|
|
||||||
|
|
||||||
# if pld_spec is Any:
|
|
||||||
# pld_spec = Raw
|
|
||||||
|
|
||||||
nsp_codec: MsgCodec = mk_codec(
|
|
||||||
# ipc_pld_spec=Raw, # default!
|
|
||||||
|
|
||||||
# NOTE XXX: the encode hook MUST be used no matter what since
|
|
||||||
# our `NamespacePath` is not any of a `Any` native type nor
|
|
||||||
# a `msgspec.Struct` subtype - so `msgspec` has no way to know
|
|
||||||
# how to encode it unless we provide the custom hook.
|
|
||||||
#
|
|
||||||
# AGAIN that is, regardless of whether we spec an
|
|
||||||
# `Any`-decoded-pld the enc has no knowledge (by default)
|
|
||||||
# how to enc `NamespacePath` (nsp), so we add a custom
|
|
||||||
# hook to do that ALWAYS.
|
|
||||||
enc_hook=enc_nsp if add_hooks else None,
|
|
||||||
|
|
||||||
# XXX NOTE: pretty sure this is mutex with the `type=` to
|
|
||||||
# `Decoder`? so it won't work in tandem with the
|
|
||||||
# `ipc_pld_spec` passed above?
|
|
||||||
ext_types=[NamespacePath],
|
|
||||||
)
|
|
||||||
# dec_hook=dec_nsp if add_hooks else None,
|
|
||||||
return nsp_codec
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'limit_plds_args',
|
|
||||||
[
|
|
||||||
(
|
|
||||||
{'dec_hook': None, 'ext_types': None},
|
|
||||||
None,
|
|
||||||
),
|
|
||||||
(
|
|
||||||
{'dec_hook': dec_nsp, 'ext_types': None},
|
|
||||||
TypeError,
|
|
||||||
),
|
|
||||||
(
|
|
||||||
{'dec_hook': dec_nsp, 'ext_types': [NamespacePath]},
|
|
||||||
None,
|
|
||||||
),
|
|
||||||
(
|
|
||||||
{'dec_hook': dec_nsp, 'ext_types': [NamespacePath|None]},
|
|
||||||
None,
|
|
||||||
),
|
|
||||||
],
|
|
||||||
ids=[
|
|
||||||
'no_hook_no_ext_types',
|
|
||||||
'only_hook',
|
|
||||||
'hook_and_ext_types',
|
|
||||||
'hook_and_ext_types_w_null',
|
|
||||||
]
|
|
||||||
)
|
|
||||||
def test_pld_limiting_usage(
|
|
||||||
limit_plds_args: tuple[dict, Exception|None],
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Verify `dec_hook()` and `ext_types` need to either both be
|
|
||||||
provided or we raise a explanator type-error.
|
|
||||||
|
|
||||||
'''
|
|
||||||
kwargs, maybe_err = limit_plds_args
|
|
||||||
async def main():
|
|
||||||
async with tractor.open_nursery() as an: # just to open runtime
|
|
||||||
|
|
||||||
# XXX SHOULD NEVER WORK outside an ipc ctx scope!
|
|
||||||
try:
|
|
||||||
with limit_plds(**kwargs):
|
|
||||||
pass
|
|
||||||
except RuntimeError:
|
|
||||||
pass
|
|
||||||
|
|
||||||
p: tractor.Portal = await an.start_actor(
|
|
||||||
'sub',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
)
|
|
||||||
async with (
|
|
||||||
p.open_context(
|
|
||||||
sleep_forever_in_sub
|
|
||||||
) as (ctx, first),
|
|
||||||
):
|
|
||||||
try:
|
|
||||||
with limit_plds(**kwargs):
|
|
||||||
pass
|
|
||||||
except maybe_err as exc:
|
|
||||||
assert type(exc) is maybe_err
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
def chk_codec_applied(
|
|
||||||
expect_codec: MsgCodec|None,
|
|
||||||
enter_value: MsgCodec|None = None,
|
|
||||||
|
|
||||||
) -> MsgCodec:
|
|
||||||
'''
|
|
||||||
buncha sanity checks ensuring that the IPC channel's
|
|
||||||
context-vars are set to the expected codec and that are
|
|
||||||
ctx-var wrapper APIs match the same.
|
|
||||||
|
|
||||||
'''
|
|
||||||
# TODO: play with tricyle again, bc this is supposed to work
|
|
||||||
# the way we want?
|
|
||||||
#
|
|
||||||
# TreeVar
|
|
||||||
# task: trio.Task = trio.lowlevel.current_task()
|
|
||||||
# curr_codec = _ctxvar_MsgCodec.get_in(task)
|
|
||||||
|
|
||||||
# ContextVar
|
|
||||||
# task_ctx: Context = task.context
|
|
||||||
# assert _ctxvar_MsgCodec in task_ctx
|
|
||||||
# curr_codec: MsgCodec = task.context[_ctxvar_MsgCodec]
|
|
||||||
if expect_codec is None:
|
|
||||||
assert enter_value is None
|
|
||||||
return
|
|
||||||
|
|
||||||
# NOTE: currently we use this!
|
|
||||||
# RunVar
|
|
||||||
curr_codec: MsgCodec = current_codec()
|
|
||||||
last_read_codec = _ctxvar_MsgCodec.get()
|
|
||||||
# assert curr_codec is last_read_codec
|
|
||||||
|
|
||||||
assert (
|
|
||||||
(same_codec := expect_codec) is
|
|
||||||
# returned from `mk_codec()`
|
|
||||||
|
|
||||||
# yielded value from `apply_codec()`
|
|
||||||
|
|
||||||
# read from current task's `contextvars.Context`
|
|
||||||
curr_codec is
|
|
||||||
last_read_codec
|
|
||||||
|
|
||||||
# the default `msgspec` settings
|
|
||||||
is not _codec._def_msgspec_codec
|
|
||||||
is not _codec._def_tractor_codec
|
|
||||||
)
|
|
||||||
|
|
||||||
if enter_value:
|
|
||||||
assert enter_value is same_codec
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def send_back_values(
|
|
||||||
ctx: Context,
|
|
||||||
rent_pld_spec_type_strs: list[str],
|
|
||||||
add_hooks: bool,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
Setup up a custom codec to load instances of `NamespacePath`
|
|
||||||
and ensure we can round trip a func ref with our parent.
|
|
||||||
|
|
||||||
'''
|
|
||||||
uid: tuple = tractor.current_actor().uid
|
|
||||||
|
|
||||||
# init state in sub-actor should be default
|
|
||||||
chk_codec_applied(
|
|
||||||
expect_codec=_codec._def_tractor_codec,
|
|
||||||
)
|
|
||||||
|
|
||||||
# load pld spec from input str
|
|
||||||
rent_pld_spec = _exts.dec_type_union(
|
|
||||||
rent_pld_spec_type_strs,
|
|
||||||
mods=[
|
|
||||||
importlib.import_module(__name__),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
rent_pld_spec_types: set[Type] = _codec.unpack_spec_types(
|
|
||||||
rent_pld_spec,
|
|
||||||
)
|
|
||||||
|
|
||||||
# ONLY add ext-hooks if the rent specified a non-std type!
|
|
||||||
add_hooks: bool = (
|
|
||||||
NamespacePath in rent_pld_spec_types
|
|
||||||
and
|
|
||||||
add_hooks
|
|
||||||
)
|
|
||||||
|
|
||||||
# same as on parent side config.
|
|
||||||
nsp_codec: MsgCodec|None = None
|
|
||||||
if add_hooks:
|
|
||||||
nsp_codec = mk_codec(
|
|
||||||
enc_hook=enc_nsp,
|
|
||||||
ext_types=[NamespacePath],
|
|
||||||
)
|
|
||||||
|
|
||||||
with (
|
|
||||||
maybe_apply_codec(nsp_codec) as codec,
|
|
||||||
limit_plds(
|
|
||||||
rent_pld_spec,
|
|
||||||
dec_hook=dec_nsp if add_hooks else None,
|
|
||||||
ext_types=[NamespacePath] if add_hooks else None,
|
|
||||||
) as pld_dec,
|
|
||||||
):
|
|
||||||
# ?XXX? SHOULD WE NOT be swapping the global codec since it
|
|
||||||
# breaks `Context.started()` roundtripping checks??
|
|
||||||
chk_codec_applied(
|
|
||||||
expect_codec=nsp_codec,
|
|
||||||
enter_value=codec,
|
|
||||||
)
|
|
||||||
|
|
||||||
# ?TODO, mismatch case(s)?
|
|
||||||
#
|
|
||||||
# ensure pld spec matches on both sides
|
|
||||||
ctx_pld_dec: MsgDec = ctx._pld_rx._pld_dec
|
|
||||||
assert pld_dec is ctx_pld_dec
|
|
||||||
child_pld_spec: Type = pld_dec.spec
|
|
||||||
child_pld_spec_types: set[Type] = _codec.unpack_spec_types(
|
|
||||||
child_pld_spec,
|
|
||||||
)
|
|
||||||
assert (
|
|
||||||
child_pld_spec_types.issuperset(
|
|
||||||
rent_pld_spec_types
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
# ?TODO, try loop for each of the types in pld-superset?
|
|
||||||
#
|
|
||||||
# for send_value in [
|
|
||||||
# nsp,
|
|
||||||
# str(nsp),
|
|
||||||
# None,
|
|
||||||
# ]:
|
|
||||||
nsp = NamespacePath.from_ref(ex_func)
|
|
||||||
try:
|
|
||||||
print(
|
|
||||||
f'{uid}: attempting to `.started({nsp})`\n'
|
|
||||||
f'\n'
|
|
||||||
f'rent_pld_spec: {rent_pld_spec}\n'
|
|
||||||
f'child_pld_spec: {child_pld_spec}\n'
|
|
||||||
f'codec: {codec}\n'
|
|
||||||
)
|
|
||||||
# await tractor.pause()
|
|
||||||
await ctx.started(nsp)
|
|
||||||
|
|
||||||
except tractor.MsgTypeError as _mte:
|
|
||||||
mte = _mte
|
|
||||||
|
|
||||||
# false -ve case
|
|
||||||
if add_hooks:
|
|
||||||
raise RuntimeError(
|
|
||||||
f'EXPECTED to `.started()` value given spec ??\n\n'
|
|
||||||
f'child_pld_spec -> {child_pld_spec}\n'
|
|
||||||
f'value = {nsp}: {type(nsp)}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# true -ve case
|
|
||||||
raise mte
|
|
||||||
|
|
||||||
# TODO: maybe we should add our own wrapper error so as to
|
|
||||||
# be interchange-lib agnostic?
|
|
||||||
# -[ ] the error type is wtv is raised from the hook so we
|
|
||||||
# could also require a type-class of errors for
|
|
||||||
# indicating whether the hook-failure can be handled by
|
|
||||||
# a nasty-dialog-unprot sub-sys?
|
|
||||||
except TypeError as typerr:
|
|
||||||
# false -ve
|
|
||||||
if add_hooks:
|
|
||||||
raise RuntimeError('Should have been able to send `nsp`??')
|
|
||||||
|
|
||||||
# true -ve
|
|
||||||
print('Failed to send `nsp` due to no ext hooks set!')
|
|
||||||
raise typerr
|
|
||||||
|
|
||||||
# now try sending a set of valid and invalid plds to ensure
|
|
||||||
# the pld spec is respected.
|
|
||||||
sent: list[Any] = []
|
|
||||||
async with ctx.open_stream() as ipc:
|
|
||||||
print(
|
|
||||||
f'{uid}: streaming all pld types to rent..'
|
|
||||||
)
|
|
||||||
|
|
||||||
# for send_value, expect_send in iter_send_val_items:
|
|
||||||
for send_value in [
|
|
||||||
nsp,
|
|
||||||
str(nsp),
|
|
||||||
None,
|
|
||||||
]:
|
|
||||||
send_type: Type = type(send_value)
|
|
||||||
print(
|
|
||||||
f'{uid}: SENDING NEXT pld\n'
|
|
||||||
f'send_type: {send_type}\n'
|
|
||||||
f'send_value: {send_value}\n'
|
|
||||||
)
|
|
||||||
try:
|
|
||||||
await ipc.send(send_value)
|
|
||||||
sent.append(send_value)
|
|
||||||
|
|
||||||
except ValidationError as valerr:
|
|
||||||
print(f'{uid} FAILED TO SEND {send_value}!')
|
|
||||||
|
|
||||||
# false -ve
|
|
||||||
if add_hooks:
|
|
||||||
raise RuntimeError(
|
|
||||||
f'EXPECTED to roundtrip value given spec:\n'
|
|
||||||
f'rent_pld_spec -> {rent_pld_spec}\n'
|
|
||||||
f'child_pld_spec -> {child_pld_spec}\n'
|
|
||||||
f'value = {send_value}: {send_type}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# true -ve
|
|
||||||
raise valerr
|
|
||||||
# continue
|
|
||||||
|
|
||||||
else:
|
|
||||||
print(
|
|
||||||
f'{uid}: finished sending all values\n'
|
|
||||||
'Should be exiting stream block!\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
print(f'{uid}: exited streaming block!')
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@cm
|
|
||||||
def maybe_apply_codec(codec: MsgCodec|None) -> MsgCodec|None:
|
|
||||||
if codec is None:
|
|
||||||
yield None
|
|
||||||
return
|
|
||||||
|
|
||||||
with apply_codec(codec) as codec:
|
|
||||||
yield codec
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'pld_spec',
|
|
||||||
[
|
|
||||||
Any,
|
|
||||||
NamespacePath,
|
|
||||||
NamespacePath|None, # the "maybe" spec Bo
|
|
||||||
],
|
|
||||||
ids=[
|
|
||||||
'any_type',
|
|
||||||
'only_nsp_ext',
|
|
||||||
'maybe_nsp_ext',
|
|
||||||
]
|
|
||||||
)
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'add_hooks',
|
|
||||||
[
|
|
||||||
True,
|
|
||||||
False,
|
|
||||||
],
|
|
||||||
ids=[
|
|
||||||
'use_codec_hooks',
|
|
||||||
'no_codec_hooks',
|
|
||||||
],
|
|
||||||
)
|
|
||||||
def test_ext_types_over_ipc(
|
|
||||||
debug_mode: bool,
|
|
||||||
pld_spec: Union[Type],
|
|
||||||
add_hooks: bool,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Ensure we can support extension types coverted using
|
|
||||||
`enc/dec_hook()`s passed to the `.msg.limit_plds()` API
|
|
||||||
and that sane errors happen when we try do the same without
|
|
||||||
the codec hooks.
|
|
||||||
|
|
||||||
'''
|
|
||||||
pld_types: set[Type] = _codec.unpack_spec_types(pld_spec)
|
|
||||||
|
|
||||||
async def main():
|
|
||||||
|
|
||||||
# sanity check the default pld-spec beforehand
|
|
||||||
chk_codec_applied(
|
|
||||||
expect_codec=_codec._def_tractor_codec,
|
|
||||||
)
|
|
||||||
|
|
||||||
# extension type we want to send as msg payload
|
|
||||||
nsp = NamespacePath.from_ref(ex_func)
|
|
||||||
|
|
||||||
# ^NOTE, 2 cases:
|
|
||||||
# - codec hooks noto added -> decode nsp as `str`
|
|
||||||
# - codec with hooks -> decode nsp as `NamespacePath`
|
|
||||||
nsp_codec: MsgCodec|None = None
|
|
||||||
if (
|
|
||||||
NamespacePath in pld_types
|
|
||||||
and
|
|
||||||
add_hooks
|
|
||||||
):
|
|
||||||
nsp_codec = mk_codec(
|
|
||||||
enc_hook=enc_nsp,
|
|
||||||
ext_types=[NamespacePath],
|
|
||||||
)
|
|
||||||
|
|
||||||
async with tractor.open_nursery(
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
) as an:
|
|
||||||
p: tractor.Portal = await an.start_actor(
|
|
||||||
'sub',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
)
|
|
||||||
with (
|
|
||||||
maybe_apply_codec(nsp_codec) as codec,
|
|
||||||
):
|
|
||||||
chk_codec_applied(
|
|
||||||
expect_codec=nsp_codec,
|
|
||||||
enter_value=codec,
|
|
||||||
)
|
|
||||||
rent_pld_spec_type_strs: list[str] = _exts.enc_type_union(pld_spec)
|
|
||||||
|
|
||||||
# XXX should raise an mte (`MsgTypeError`)
|
|
||||||
# when `add_hooks == False` bc the input
|
|
||||||
# `expect_ipc_send` kwarg has a nsp which can't be
|
|
||||||
# serialized!
|
|
||||||
#
|
|
||||||
# TODO:can we ensure this happens from the
|
|
||||||
# `Return`-side (aka the sub) as well?
|
|
||||||
try:
|
|
||||||
ctx: tractor.Context
|
|
||||||
ipc: tractor.MsgStream
|
|
||||||
async with (
|
|
||||||
|
|
||||||
# XXX should raise an mte (`MsgTypeError`)
|
|
||||||
# when `add_hooks == False`..
|
|
||||||
p.open_context(
|
|
||||||
send_back_values,
|
|
||||||
# expect_debug=debug_mode,
|
|
||||||
rent_pld_spec_type_strs=rent_pld_spec_type_strs,
|
|
||||||
add_hooks=add_hooks,
|
|
||||||
# expect_ipc_send=expect_ipc_send,
|
|
||||||
) as (ctx, first),
|
|
||||||
|
|
||||||
ctx.open_stream() as ipc,
|
|
||||||
):
|
|
||||||
with (
|
|
||||||
limit_plds(
|
|
||||||
pld_spec,
|
|
||||||
dec_hook=dec_nsp if add_hooks else None,
|
|
||||||
ext_types=[NamespacePath] if add_hooks else None,
|
|
||||||
) as pld_dec,
|
|
||||||
):
|
|
||||||
ctx_pld_dec: MsgDec = ctx._pld_rx._pld_dec
|
|
||||||
assert pld_dec is ctx_pld_dec
|
|
||||||
|
|
||||||
# if (
|
|
||||||
# not add_hooks
|
|
||||||
# and
|
|
||||||
# NamespacePath in
|
|
||||||
# ):
|
|
||||||
# pytest.fail('ctx should fail to open without custom enc_hook!?')
|
|
||||||
|
|
||||||
await ipc.send(nsp)
|
|
||||||
nsp_rt = await ipc.receive()
|
|
||||||
|
|
||||||
assert nsp_rt == nsp
|
|
||||||
assert nsp_rt.load_ref() is ex_func
|
|
||||||
|
|
||||||
# this test passes bc we can go no further!
|
|
||||||
except MsgTypeError as mte:
|
|
||||||
# if not add_hooks:
|
|
||||||
# # teardown nursery
|
|
||||||
# await p.cancel_actor()
|
|
||||||
# return
|
|
||||||
|
|
||||||
raise mte
|
|
||||||
|
|
||||||
await p.cancel_actor()
|
|
||||||
|
|
||||||
if (
|
|
||||||
NamespacePath in pld_types
|
|
||||||
and
|
|
||||||
add_hooks
|
|
||||||
):
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
else:
|
|
||||||
with pytest.raises(
|
|
||||||
expected_exception=tractor.RemoteActorError,
|
|
||||||
) as excinfo:
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
exc = excinfo.value
|
|
||||||
# bc `.started(nsp: NamespacePath)` will raise
|
|
||||||
assert exc.boxed_type is TypeError
|
|
||||||
|
|
||||||
|
|
||||||
# def chk_pld_type(
|
|
||||||
# payload_spec: Type[Struct]|Any,
|
|
||||||
# pld: Any,
|
|
||||||
|
|
||||||
# expect_roundtrip: bool|None = None,
|
|
||||||
|
|
||||||
# ) -> bool:
|
|
||||||
|
|
||||||
# pld_val_type: Type = type(pld)
|
|
||||||
|
|
||||||
# # TODO: verify that the overridden subtypes
|
|
||||||
# # DO NOT have modified type-annots from original!
|
|
||||||
# # 'Start', .pld: FuncSpec
|
|
||||||
# # 'StartAck', .pld: IpcCtxSpec
|
|
||||||
# # 'Stop', .pld: UNSEt
|
|
||||||
# # 'Error', .pld: ErrorData
|
|
||||||
|
|
||||||
# codec: MsgCodec = mk_codec(
|
|
||||||
# # NOTE: this ONLY accepts `PayloadMsg.pld` fields of a specified
|
|
||||||
# # type union.
|
|
||||||
# ipc_pld_spec=payload_spec,
|
|
||||||
# )
|
|
||||||
|
|
||||||
# # make a one-off dec to compare with our `MsgCodec` instance
|
|
||||||
# # which does the below `mk_msg_spec()` call internally
|
|
||||||
# ipc_msg_spec: Union[Type[Struct]]
|
|
||||||
# msg_types: list[PayloadMsg[payload_spec]]
|
|
||||||
# (
|
|
||||||
# ipc_msg_spec,
|
|
||||||
# msg_types,
|
|
||||||
# ) = mk_msg_spec(
|
|
||||||
# payload_type_union=payload_spec,
|
|
||||||
# )
|
|
||||||
# _enc = msgpack.Encoder()
|
|
||||||
# _dec = msgpack.Decoder(
|
|
||||||
# type=ipc_msg_spec or Any, # like `PayloadMsg[Any]`
|
|
||||||
# )
|
|
||||||
|
|
||||||
# assert (
|
|
||||||
# payload_spec
|
|
||||||
# ==
|
|
||||||
# codec.pld_spec
|
|
||||||
# )
|
|
||||||
|
|
||||||
# # assert codec.dec == dec
|
|
||||||
# #
|
|
||||||
# # ^-XXX-^ not sure why these aren't "equal" but when cast
|
|
||||||
# # to `str` they seem to match ?? .. kk
|
|
||||||
|
|
||||||
# assert (
|
|
||||||
# str(ipc_msg_spec)
|
|
||||||
# ==
|
|
||||||
# str(codec.msg_spec)
|
|
||||||
# ==
|
|
||||||
# str(_dec.type)
|
|
||||||
# ==
|
|
||||||
# str(codec.dec.type)
|
|
||||||
# )
|
|
||||||
|
|
||||||
# # verify the boxed-type for all variable payload-type msgs.
|
|
||||||
# if not msg_types:
|
|
||||||
# breakpoint()
|
|
||||||
|
|
||||||
# roundtrip: bool|None = None
|
|
||||||
# pld_spec_msg_names: list[str] = [
|
|
||||||
# td.__name__ for td in _payload_msgs
|
|
||||||
# ]
|
|
||||||
# for typedef in msg_types:
|
|
||||||
|
|
||||||
# skip_runtime_msg: bool = typedef.__name__ not in pld_spec_msg_names
|
|
||||||
# if skip_runtime_msg:
|
|
||||||
# continue
|
|
||||||
|
|
||||||
# pld_field = structs.fields(typedef)[1]
|
|
||||||
# assert pld_field.type is payload_spec # TODO-^ does this need to work to get all subtypes to adhere?
|
|
||||||
|
|
||||||
# kwargs: dict[str, Any] = {
|
|
||||||
# 'cid': '666',
|
|
||||||
# 'pld': pld,
|
|
||||||
# }
|
|
||||||
# enc_msg: PayloadMsg = typedef(**kwargs)
|
|
||||||
|
|
||||||
# _wire_bytes: bytes = _enc.encode(enc_msg)
|
|
||||||
# wire_bytes: bytes = codec.enc.encode(enc_msg)
|
|
||||||
# assert _wire_bytes == wire_bytes
|
|
||||||
|
|
||||||
# ve: ValidationError|None = None
|
|
||||||
# try:
|
|
||||||
# dec_msg = codec.dec.decode(wire_bytes)
|
|
||||||
# _dec_msg = _dec.decode(wire_bytes)
|
|
||||||
|
|
||||||
# # decoded msg and thus payload should be exactly same!
|
|
||||||
# assert (roundtrip := (
|
|
||||||
# _dec_msg
|
|
||||||
# ==
|
|
||||||
# dec_msg
|
|
||||||
# ==
|
|
||||||
# enc_msg
|
|
||||||
# ))
|
|
||||||
|
|
||||||
# if (
|
|
||||||
# expect_roundtrip is not None
|
|
||||||
# and expect_roundtrip != roundtrip
|
|
||||||
# ):
|
|
||||||
# breakpoint()
|
|
||||||
|
|
||||||
# assert (
|
|
||||||
# pld
|
|
||||||
# ==
|
|
||||||
# dec_msg.pld
|
|
||||||
# ==
|
|
||||||
# enc_msg.pld
|
|
||||||
# )
|
|
||||||
# # assert (roundtrip := (_dec_msg == enc_msg))
|
|
||||||
|
|
||||||
# except ValidationError as _ve:
|
|
||||||
# ve = _ve
|
|
||||||
# roundtrip: bool = False
|
|
||||||
# if pld_val_type is payload_spec:
|
|
||||||
# raise ValueError(
|
|
||||||
# 'Got `ValidationError` despite type-var match!?\n'
|
|
||||||
# f'pld_val_type: {pld_val_type}\n'
|
|
||||||
# f'payload_type: {payload_spec}\n'
|
|
||||||
# ) from ve
|
|
||||||
|
|
||||||
# else:
|
|
||||||
# # ow we good cuz the pld spec mismatched.
|
|
||||||
# print(
|
|
||||||
# 'Got expected `ValidationError` since,\n'
|
|
||||||
# f'{pld_val_type} is not {payload_spec}\n'
|
|
||||||
# )
|
|
||||||
# else:
|
|
||||||
# if (
|
|
||||||
# payload_spec is not Any
|
|
||||||
# and
|
|
||||||
# pld_val_type is not payload_spec
|
|
||||||
# ):
|
|
||||||
# raise ValueError(
|
|
||||||
# 'DID NOT `ValidationError` despite expected type match!?\n'
|
|
||||||
# f'pld_val_type: {pld_val_type}\n'
|
|
||||||
# f'payload_type: {payload_spec}\n'
|
|
||||||
# )
|
|
||||||
|
|
||||||
# # full code decode should always be attempted!
|
|
||||||
# if roundtrip is None:
|
|
||||||
# breakpoint()
|
|
||||||
|
|
||||||
# return roundtrip
|
|
||||||
|
|
||||||
|
|
||||||
# ?TODO? maybe remove since covered in the newer `test_pldrx_limiting`
|
|
||||||
# via end-2-end testing of all this?
|
|
||||||
# -[ ] IOW do we really NEED this lowlevel unit testing?
|
|
||||||
#
|
|
||||||
# def test_limit_msgspec(
|
|
||||||
# debug_mode: bool,
|
|
||||||
# ):
|
|
||||||
# '''
|
|
||||||
# Internals unit testing to verify that type-limiting an IPC ctx's
|
|
||||||
# msg spec with `Pldrx.limit_plds()` results in various
|
|
||||||
# encapsulated `msgspec` object settings and state.
|
|
||||||
|
|
||||||
# '''
|
|
||||||
# async def main():
|
|
||||||
# async with tractor.open_root_actor(
|
|
||||||
# debug_mode=debug_mode,
|
|
||||||
# ):
|
|
||||||
# # ensure we can round-trip a boxing `PayloadMsg`
|
|
||||||
# assert chk_pld_type(
|
|
||||||
# payload_spec=Any,
|
|
||||||
# pld=None,
|
|
||||||
# expect_roundtrip=True,
|
|
||||||
# )
|
|
||||||
|
|
||||||
# # verify that a mis-typed payload value won't decode
|
|
||||||
# assert not chk_pld_type(
|
|
||||||
# payload_spec=int,
|
|
||||||
# pld='doggy',
|
|
||||||
# )
|
|
||||||
|
|
||||||
# # parametrize the boxed `.pld` type as a custom-struct
|
|
||||||
# # and ensure that parametrization propagates
|
|
||||||
# # to all payload-msg-spec-able subtypes!
|
|
||||||
# class CustomPayload(Struct):
|
|
||||||
# name: str
|
|
||||||
# value: Any
|
|
||||||
|
|
||||||
# assert not chk_pld_type(
|
|
||||||
# payload_spec=CustomPayload,
|
|
||||||
# pld='doggy',
|
|
||||||
# )
|
|
||||||
|
|
||||||
# assert chk_pld_type(
|
|
||||||
# payload_spec=CustomPayload,
|
|
||||||
# pld=CustomPayload(name='doggy', value='urmom')
|
|
||||||
# )
|
|
||||||
|
|
||||||
# # yah, we can `.pause_from_sync()` now!
|
|
||||||
# # breakpoint()
|
|
||||||
|
|
||||||
# trio.run(main)
|
|
||||||
|
|
@ -32,16 +32,6 @@ from tractor.trionics import BroadcastReceiver
|
||||||
from tractor._testing import expect_ctxc
|
from tractor._testing import expect_ctxc
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(
|
|
||||||
scope='module',
|
|
||||||
)
|
|
||||||
def delay(debug_mode: bool) -> int:
|
|
||||||
if debug_mode:
|
|
||||||
return 999
|
|
||||||
else:
|
|
||||||
return 1
|
|
||||||
|
|
||||||
|
|
||||||
async def sleep_and_err(
|
async def sleep_and_err(
|
||||||
sleep_for: float = 0.1,
|
sleep_for: float = 0.1,
|
||||||
|
|
||||||
|
|
@ -69,23 +59,17 @@ async def trio_cancels_single_aio_task():
|
||||||
await tractor.to_asyncio.run_task(aio_sleep_forever)
|
await tractor.to_asyncio.run_task(aio_sleep_forever)
|
||||||
|
|
||||||
|
|
||||||
def test_trio_cancels_aio_on_actor_side(
|
def test_trio_cancels_aio_on_actor_side(reg_addr):
|
||||||
reg_addr: tuple[str, int],
|
|
||||||
delay: int,
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
'''
|
'''
|
||||||
Spawn an infected actor that is cancelled by the ``trio`` side
|
Spawn an infected actor that is cancelled by the ``trio`` side
|
||||||
task using std cancel scope apis.
|
task using std cancel scope apis.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
with trio.fail_after(1 + delay):
|
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
registry_addrs=[reg_addr],
|
registry_addrs=[reg_addr]
|
||||||
debug_mode=debug_mode,
|
) as n:
|
||||||
) as an:
|
await n.run_in_actor(
|
||||||
await an.run_in_actor(
|
|
||||||
trio_cancels_single_aio_task,
|
trio_cancels_single_aio_task,
|
||||||
infect_asyncio=True,
|
infect_asyncio=True,
|
||||||
)
|
)
|
||||||
|
|
@ -132,10 +116,7 @@ async def asyncio_actor(
|
||||||
raise
|
raise
|
||||||
|
|
||||||
|
|
||||||
def test_aio_simple_error(
|
def test_aio_simple_error(reg_addr):
|
||||||
reg_addr: tuple[str, int],
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
'''
|
'''
|
||||||
Verify a simple remote asyncio error propagates back through trio
|
Verify a simple remote asyncio error propagates back through trio
|
||||||
to the parent actor.
|
to the parent actor.
|
||||||
|
|
@ -144,10 +125,9 @@ def test_aio_simple_error(
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
registry_addrs=[reg_addr],
|
registry_addrs=[reg_addr]
|
||||||
debug_mode=debug_mode,
|
) as n:
|
||||||
) as an:
|
await n.run_in_actor(
|
||||||
await an.run_in_actor(
|
|
||||||
asyncio_actor,
|
asyncio_actor,
|
||||||
target='sleep_and_err',
|
target='sleep_and_err',
|
||||||
expect_err='AssertionError',
|
expect_err='AssertionError',
|
||||||
|
|
@ -173,19 +153,14 @@ def test_aio_simple_error(
|
||||||
assert err.boxed_type is AssertionError
|
assert err.boxed_type is AssertionError
|
||||||
|
|
||||||
|
|
||||||
def test_tractor_cancels_aio(
|
def test_tractor_cancels_aio(reg_addr):
|
||||||
reg_addr: tuple[str, int],
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
'''
|
'''
|
||||||
Verify we can cancel a spawned asyncio task gracefully.
|
Verify we can cancel a spawned asyncio task gracefully.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery() as n:
|
||||||
debug_mode=debug_mode,
|
portal = await n.run_in_actor(
|
||||||
) as an:
|
|
||||||
portal = await an.run_in_actor(
|
|
||||||
asyncio_actor,
|
asyncio_actor,
|
||||||
target='aio_sleep_forever',
|
target='aio_sleep_forever',
|
||||||
expect_err='trio.Cancelled',
|
expect_err='trio.Cancelled',
|
||||||
|
|
@ -197,9 +172,7 @@ def test_tractor_cancels_aio(
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
def test_trio_cancels_aio(
|
def test_trio_cancels_aio(reg_addr):
|
||||||
reg_addr: tuple[str, int],
|
|
||||||
):
|
|
||||||
'''
|
'''
|
||||||
Much like the above test with ``tractor.Portal.cancel_actor()``
|
Much like the above test with ``tractor.Portal.cancel_actor()``
|
||||||
except we just use a standard ``trio`` cancellation api.
|
except we just use a standard ``trio`` cancellation api.
|
||||||
|
|
@ -230,12 +203,13 @@ async def trio_ctx(
|
||||||
|
|
||||||
# this will block until the ``asyncio`` task sends a "first"
|
# this will block until the ``asyncio`` task sends a "first"
|
||||||
# message.
|
# message.
|
||||||
delay: int = 999 if tractor.debug_mode() else 1
|
with trio.fail_after(2):
|
||||||
with trio.fail_after(1 + delay):
|
|
||||||
try:
|
try:
|
||||||
async with (
|
async with (
|
||||||
tractor.trionics.collapse_eg(),
|
trio.open_nursery(
|
||||||
trio.open_nursery() as tn,
|
# TODO, for new `trio` / py3.13
|
||||||
|
# strict_exception_groups=False,
|
||||||
|
) as tn,
|
||||||
tractor.to_asyncio.open_channel_from(
|
tractor.to_asyncio.open_channel_from(
|
||||||
sleep_and_err,
|
sleep_and_err,
|
||||||
) as (first, chan),
|
) as (first, chan),
|
||||||
|
|
@ -265,10 +239,8 @@ async def trio_ctx(
|
||||||
ids='parent_actor_cancels_child={}'.format
|
ids='parent_actor_cancels_child={}'.format
|
||||||
)
|
)
|
||||||
def test_context_spawns_aio_task_that_errors(
|
def test_context_spawns_aio_task_that_errors(
|
||||||
reg_addr: tuple[str, int],
|
reg_addr,
|
||||||
delay: int,
|
|
||||||
parent_cancels: bool,
|
parent_cancels: bool,
|
||||||
debug_mode: bool,
|
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Verify that spawning a task via an intertask channel ctx mngr that
|
Verify that spawning a task via an intertask channel ctx mngr that
|
||||||
|
|
@ -277,13 +249,13 @@ def test_context_spawns_aio_task_that_errors(
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
with trio.fail_after(1 + delay):
|
with trio.fail_after(2):
|
||||||
async with tractor.open_nursery() as an:
|
async with tractor.open_nursery() as n:
|
||||||
p = await an.start_actor(
|
p = await n.start_actor(
|
||||||
'aio_daemon',
|
'aio_daemon',
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
infect_asyncio=True,
|
infect_asyncio=True,
|
||||||
debug_mode=debug_mode,
|
# debug_mode=True,
|
||||||
loglevel='cancel',
|
loglevel='cancel',
|
||||||
)
|
)
|
||||||
async with (
|
async with (
|
||||||
|
|
@ -350,12 +322,11 @@ async def aio_cancel():
|
||||||
|
|
||||||
def test_aio_cancelled_from_aio_causes_trio_cancelled(
|
def test_aio_cancelled_from_aio_causes_trio_cancelled(
|
||||||
reg_addr: tuple,
|
reg_addr: tuple,
|
||||||
delay: int,
|
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
When the `asyncio.Task` cancels itself the `trio` side should
|
When the `asyncio.Task` cancels itself the `trio` side cshould
|
||||||
also cancel and teardown and relay the cancellation cross-process
|
also cancel and teardown and relay the cancellation cross-process
|
||||||
to the parent caller.
|
to the caller (parent).
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
|
|
@ -371,7 +342,7 @@ def test_aio_cancelled_from_aio_causes_trio_cancelled(
|
||||||
# NOTE: normally the `an.__aexit__()` waits on the
|
# NOTE: normally the `an.__aexit__()` waits on the
|
||||||
# portal's result but we do it explicitly here
|
# portal's result but we do it explicitly here
|
||||||
# to avoid indent levels.
|
# to avoid indent levels.
|
||||||
with trio.fail_after(1 + delay):
|
with trio.fail_after(1):
|
||||||
await p.wait_for_result()
|
await p.wait_for_result()
|
||||||
|
|
||||||
with pytest.raises(
|
with pytest.raises(
|
||||||
|
|
@ -382,10 +353,11 @@ def test_aio_cancelled_from_aio_causes_trio_cancelled(
|
||||||
# might get multiple `trio.Cancelled`s as well inside an inception
|
# might get multiple `trio.Cancelled`s as well inside an inception
|
||||||
err: RemoteActorError|ExceptionGroup = excinfo.value
|
err: RemoteActorError|ExceptionGroup = excinfo.value
|
||||||
if isinstance(err, ExceptionGroup):
|
if isinstance(err, ExceptionGroup):
|
||||||
excs = err.exceptions
|
err = next(itertools.dropwhile(
|
||||||
assert len(excs) == 1
|
lambda exc: not isinstance(exc, tractor.RemoteActorError),
|
||||||
final_exc = excs[0]
|
err.exceptions
|
||||||
assert isinstance(final_exc, tractor.RemoteActorError)
|
))
|
||||||
|
assert err
|
||||||
|
|
||||||
# relayed boxed error should be our `trio`-task's
|
# relayed boxed error should be our `trio`-task's
|
||||||
# cancel-signal-proxy-equivalent of `asyncio.CancelledError`.
|
# cancel-signal-proxy-equivalent of `asyncio.CancelledError`.
|
||||||
|
|
@ -398,18 +370,15 @@ async def no_to_trio_in_args():
|
||||||
|
|
||||||
|
|
||||||
async def push_from_aio_task(
|
async def push_from_aio_task(
|
||||||
|
|
||||||
sequence: Iterable,
|
sequence: Iterable,
|
||||||
to_trio: trio.abc.SendChannel,
|
to_trio: trio.abc.SendChannel,
|
||||||
expect_cancel: False,
|
expect_cancel: False,
|
||||||
fail_early: bool,
|
fail_early: bool,
|
||||||
exit_early: bool,
|
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# print('trying breakpoint')
|
|
||||||
# breakpoint()
|
|
||||||
|
|
||||||
# sync caller ctx manager
|
# sync caller ctx manager
|
||||||
to_trio.send_nowait(True)
|
to_trio.send_nowait(True)
|
||||||
|
|
||||||
|
|
@ -418,27 +387,10 @@ async def push_from_aio_task(
|
||||||
to_trio.send_nowait(i)
|
to_trio.send_nowait(i)
|
||||||
await asyncio.sleep(0.001)
|
await asyncio.sleep(0.001)
|
||||||
|
|
||||||
if (
|
if i == 50 and fail_early:
|
||||||
i == 50
|
|
||||||
):
|
|
||||||
if fail_early:
|
|
||||||
print('Raising exc from aio side!')
|
|
||||||
raise Exception
|
raise Exception
|
||||||
|
|
||||||
if exit_early:
|
print('asyncio streamer complete!')
|
||||||
# TODO? really you could enforce the same
|
|
||||||
# SC-proto we use for actors here with asyncio
|
|
||||||
# such that a Return[None] msg would be
|
|
||||||
# implicitly delivered to the trio side?
|
|
||||||
#
|
|
||||||
# XXX => this might be the end-all soln for
|
|
||||||
# converting any-inter-task system (regardless
|
|
||||||
# of maybe-remote runtime or language) to be
|
|
||||||
# SC-compat no?
|
|
||||||
print(f'asyncio breaking early @ {i!r}')
|
|
||||||
break
|
|
||||||
|
|
||||||
print('asyncio streaming complete!')
|
|
||||||
|
|
||||||
except asyncio.CancelledError:
|
except asyncio.CancelledError:
|
||||||
if not expect_cancel:
|
if not expect_cancel:
|
||||||
|
|
@ -450,10 +402,9 @@ async def push_from_aio_task(
|
||||||
|
|
||||||
|
|
||||||
async def stream_from_aio(
|
async def stream_from_aio(
|
||||||
trio_exit_early: bool = False,
|
exit_early: bool = False,
|
||||||
trio_raise_err: bool = False,
|
raise_err: bool = False,
|
||||||
aio_raise_err: bool = False,
|
aio_raise_err: bool = False,
|
||||||
aio_exit_early: bool = False,
|
|
||||||
fan_out: bool = False,
|
fan_out: bool = False,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
@ -466,17 +417,8 @@ async def stream_from_aio(
|
||||||
async with to_asyncio.open_channel_from(
|
async with to_asyncio.open_channel_from(
|
||||||
push_from_aio_task,
|
push_from_aio_task,
|
||||||
sequence=seq,
|
sequence=seq,
|
||||||
expect_cancel=trio_raise_err or trio_exit_early,
|
expect_cancel=raise_err or exit_early,
|
||||||
fail_early=aio_raise_err,
|
fail_early=aio_raise_err,
|
||||||
exit_early=aio_exit_early,
|
|
||||||
|
|
||||||
# such that we can test exit early cases
|
|
||||||
# for each side explicitly.
|
|
||||||
suppress_graceful_exits=(not(
|
|
||||||
aio_exit_early
|
|
||||||
or
|
|
||||||
trio_exit_early
|
|
||||||
))
|
|
||||||
|
|
||||||
) as (first, chan):
|
) as (first, chan):
|
||||||
|
|
||||||
|
|
@ -489,19 +431,13 @@ async def stream_from_aio(
|
||||||
],
|
],
|
||||||
):
|
):
|
||||||
async for value in chan:
|
async for value in chan:
|
||||||
print(f'trio received: {value!r}')
|
print(f'trio received {value}')
|
||||||
|
|
||||||
# XXX, debugging EoC not being handled correctly
|
|
||||||
# in `transate_aio_errors()`..
|
|
||||||
# if value is None:
|
|
||||||
# await tractor.pause(shield=True)
|
|
||||||
|
|
||||||
pulled.append(value)
|
pulled.append(value)
|
||||||
|
|
||||||
if value == 50:
|
if value == 50:
|
||||||
if trio_raise_err:
|
if raise_err:
|
||||||
raise Exception
|
raise Exception
|
||||||
elif trio_exit_early:
|
elif exit_early:
|
||||||
print('`consume()` breaking early!\n')
|
print('`consume()` breaking early!\n')
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|
@ -518,11 +454,11 @@ async def stream_from_aio(
|
||||||
# tasks are joined..
|
# tasks are joined..
|
||||||
chan.subscribe() as br,
|
chan.subscribe() as br,
|
||||||
|
|
||||||
trio.open_nursery() as tn,
|
trio.open_nursery() as n,
|
||||||
):
|
):
|
||||||
# start 2nd task that get's broadcast the same
|
# start 2nd task that get's broadcast the same
|
||||||
# value set.
|
# value set.
|
||||||
tn.start_soon(consume, br)
|
n.start_soon(consume, br)
|
||||||
await consume(chan)
|
await consume(chan)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
|
|
@ -535,14 +471,10 @@ async def stream_from_aio(
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
|
|
||||||
if not (
|
if (
|
||||||
trio_raise_err
|
not raise_err and
|
||||||
or
|
not exit_early and
|
||||||
trio_exit_early
|
not aio_raise_err
|
||||||
or
|
|
||||||
aio_raise_err
|
|
||||||
or
|
|
||||||
aio_exit_early
|
|
||||||
):
|
):
|
||||||
if fan_out:
|
if fan_out:
|
||||||
# we get double the pulled values in the
|
# we get double the pulled values in the
|
||||||
|
|
@ -552,7 +484,6 @@ async def stream_from_aio(
|
||||||
assert list(sorted(pulled)) == expect
|
assert list(sorted(pulled)) == expect
|
||||||
|
|
||||||
else:
|
else:
|
||||||
# await tractor.pause()
|
|
||||||
assert pulled == expect
|
assert pulled == expect
|
||||||
else:
|
else:
|
||||||
assert not fan_out
|
assert not fan_out
|
||||||
|
|
@ -566,15 +497,10 @@ async def stream_from_aio(
|
||||||
'fan_out', [False, True],
|
'fan_out', [False, True],
|
||||||
ids='fan_out_w_chan_subscribe={}'.format
|
ids='fan_out_w_chan_subscribe={}'.format
|
||||||
)
|
)
|
||||||
def test_basic_interloop_channel_stream(
|
def test_basic_interloop_channel_stream(reg_addr, fan_out):
|
||||||
reg_addr: tuple[str, int],
|
|
||||||
fan_out: bool,
|
|
||||||
):
|
|
||||||
async def main():
|
async def main():
|
||||||
# TODO, figure out min timeout here!
|
async with tractor.open_nursery() as n:
|
||||||
with trio.fail_after(6):
|
portal = await n.run_in_actor(
|
||||||
async with tractor.open_nursery() as an:
|
|
||||||
portal = await an.run_in_actor(
|
|
||||||
stream_from_aio,
|
stream_from_aio,
|
||||||
infect_asyncio=True,
|
infect_asyncio=True,
|
||||||
fan_out=fan_out,
|
fan_out=fan_out,
|
||||||
|
|
@ -588,10 +514,10 @@ def test_basic_interloop_channel_stream(
|
||||||
# TODO: parametrize the above test and avoid the duplication here?
|
# TODO: parametrize the above test and avoid the duplication here?
|
||||||
def test_trio_error_cancels_intertask_chan(reg_addr):
|
def test_trio_error_cancels_intertask_chan(reg_addr):
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery() as an:
|
async with tractor.open_nursery() as n:
|
||||||
portal = await an.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
stream_from_aio,
|
stream_from_aio,
|
||||||
trio_raise_err=True,
|
raise_err=True,
|
||||||
infect_asyncio=True,
|
infect_asyncio=True,
|
||||||
)
|
)
|
||||||
# should trigger remote actor error
|
# should trigger remote actor error
|
||||||
|
|
@ -604,116 +530,43 @@ def test_trio_error_cancels_intertask_chan(reg_addr):
|
||||||
excinfo.value.boxed_type is Exception
|
excinfo.value.boxed_type is Exception
|
||||||
|
|
||||||
|
|
||||||
def test_trio_closes_early_causes_aio_checkpoint_raise(
|
def test_trio_closes_early_and_channel_exits(
|
||||||
reg_addr: tuple[str, int],
|
reg_addr: tuple[str, int],
|
||||||
delay: int,
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Check that if the `trio`-task "exits early and silently" (in this
|
Check that if the `trio`-task "exits early" on `async for`ing the
|
||||||
case during `async for`-ing the inter-task-channel via
|
inter-task-channel (via a `break`) we exit silently from the
|
||||||
a `break`-from-loop), we raise `TrioTaskExited` on the
|
`open_channel_from()` block and get a final `Return[None]` msg.
|
||||||
`asyncio`-side which also then bubbles up through the
|
|
||||||
`open_channel_from()` block indicating that the `asyncio.Task`
|
|
||||||
hit a ran another checkpoint despite the `trio.Task` exit.
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
with trio.fail_after(1 + delay):
|
with trio.fail_after(2):
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=debug_mode,
|
# debug_mode=True,
|
||||||
# enable_stack_on_sig=True,
|
# enable_stack_on_sig=True,
|
||||||
) as an:
|
) as n:
|
||||||
portal = await an.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
stream_from_aio,
|
stream_from_aio,
|
||||||
trio_exit_early=True,
|
exit_early=True,
|
||||||
infect_asyncio=True,
|
infect_asyncio=True,
|
||||||
)
|
)
|
||||||
# should raise RAE diectly
|
# should raise RAE diectly
|
||||||
print('waiting on final infected subactor result..')
|
print('waiting on final infected subactor result..')
|
||||||
res: None = await portal.wait_for_result()
|
res: None = await portal.wait_for_result()
|
||||||
assert res is None
|
assert res is None
|
||||||
print(f'infected subactor returned result: {res!r}\n')
|
print('infected subactor returned result: {res!r}\n')
|
||||||
|
|
||||||
# should be a quiet exit on a simple channel exit
|
# should be a quiet exit on a simple channel exit
|
||||||
with pytest.raises(RemoteActorError) as excinfo:
|
trio.run(
|
||||||
trio.run(main)
|
main,
|
||||||
|
# strict_exception_groups=False,
|
||||||
# ensure remote error is an explicit `AsyncioCancelled` sub-type
|
|
||||||
# which indicates to the aio task that the trio side exited
|
|
||||||
# silently WITHOUT raising a `trio.Cancelled` (which would
|
|
||||||
# normally be raised instead as a `AsyncioCancelled`).
|
|
||||||
excinfo.value.boxed_type is to_asyncio.TrioTaskExited
|
|
||||||
|
|
||||||
|
|
||||||
def test_aio_exits_early_relays_AsyncioTaskExited(
|
|
||||||
# TODO, parametrize the 3 possible trio side conditions:
|
|
||||||
# - trio blocking on receive, aio exits early
|
|
||||||
# - trio cancelled AND aio exits early on its next tick
|
|
||||||
# - trio errors AND aio exits early on its next tick
|
|
||||||
reg_addr: tuple[str, int],
|
|
||||||
debug_mode: bool,
|
|
||||||
delay: int,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Check that if the `asyncio`-task "exits early and silently" (in this
|
|
||||||
case during `push_from_aio_task()` pushing to the `InterLoopTaskChannel`
|
|
||||||
it `break`s from the loop), we raise `AsyncioTaskExited` on the
|
|
||||||
`trio`-side which then DOES NOT BUBBLE up through the
|
|
||||||
`open_channel_from()` block UNLESS,
|
|
||||||
|
|
||||||
- the trio.Task also errored/cancelled, in which case we wrap
|
|
||||||
both errors in an eg
|
|
||||||
- the trio.Task was blocking on rxing a value from the
|
|
||||||
`InterLoopTaskChannel`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
async def main():
|
|
||||||
with trio.fail_after(1 + delay):
|
|
||||||
async with tractor.open_nursery(
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
# enable_stack_on_sig=True,
|
|
||||||
) as an:
|
|
||||||
portal = await an.run_in_actor(
|
|
||||||
stream_from_aio,
|
|
||||||
infect_asyncio=True,
|
|
||||||
trio_exit_early=False,
|
|
||||||
aio_exit_early=True,
|
|
||||||
)
|
)
|
||||||
# should raise RAE diectly
|
|
||||||
print('waiting on final infected subactor result..')
|
|
||||||
res: None = await portal.wait_for_result()
|
|
||||||
assert res is None
|
|
||||||
print(f'infected subactor returned result: {res!r}\n')
|
|
||||||
|
|
||||||
# should be a quiet exit on a simple channel exit
|
|
||||||
with pytest.raises(RemoteActorError) as excinfo:
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
exc = excinfo.value
|
|
||||||
|
|
||||||
# TODO, wow bug!
|
|
||||||
# -[ ] bp handler not replaced!?!?
|
|
||||||
# breakpoint()
|
|
||||||
|
|
||||||
# import pdbp; pdbp.set_trace()
|
|
||||||
|
|
||||||
# ensure remote error is an explicit `AsyncioCancelled` sub-type
|
|
||||||
# which indicates to the aio task that the trio side exited
|
|
||||||
# silently WITHOUT raising a `trio.Cancelled` (which would
|
|
||||||
# normally be raised instead as a `AsyncioCancelled`).
|
|
||||||
assert exc.boxed_type is to_asyncio.AsyncioTaskExited
|
|
||||||
|
|
||||||
|
|
||||||
def test_aio_errors_and_channel_propagates_and_closes(
|
def test_aio_errors_and_channel_propagates_and_closes(reg_addr):
|
||||||
reg_addr: tuple[str, int],
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery() as n:
|
||||||
debug_mode=debug_mode,
|
portal = await n.run_in_actor(
|
||||||
) as an:
|
|
||||||
portal = await an.run_in_actor(
|
|
||||||
stream_from_aio,
|
stream_from_aio,
|
||||||
aio_raise_err=True,
|
aio_raise_err=True,
|
||||||
infect_asyncio=True,
|
infect_asyncio=True,
|
||||||
|
|
@ -739,13 +592,7 @@ async def aio_echo_server(
|
||||||
to_trio.send_nowait('start')
|
to_trio.send_nowait('start')
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
try:
|
|
||||||
msg = await from_trio.get()
|
msg = await from_trio.get()
|
||||||
except to_asyncio.TrioTaskExited:
|
|
||||||
print(
|
|
||||||
'breaking aio echo loop due to `trio` exit!'
|
|
||||||
)
|
|
||||||
break
|
|
||||||
|
|
||||||
# echo the msg back
|
# echo the msg back
|
||||||
to_trio.send_nowait(msg)
|
to_trio.send_nowait(msg)
|
||||||
|
|
@ -794,15 +641,13 @@ async def trio_to_aio_echo_server(
|
||||||
ids='raise_error={}'.format,
|
ids='raise_error={}'.format,
|
||||||
)
|
)
|
||||||
def test_echoserver_detailed_mechanics(
|
def test_echoserver_detailed_mechanics(
|
||||||
reg_addr: tuple[str, int],
|
reg_addr,
|
||||||
debug_mode: bool,
|
|
||||||
raise_error_mid_stream,
|
raise_error_mid_stream,
|
||||||
):
|
):
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery() as n:
|
||||||
debug_mode=debug_mode,
|
p = await n.start_actor(
|
||||||
) as an:
|
|
||||||
p = await an.start_actor(
|
|
||||||
'aio_server',
|
'aio_server',
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
infect_asyncio=True,
|
infect_asyncio=True,
|
||||||
|
|
@ -889,7 +734,7 @@ async def manage_file(
|
||||||
|
|
||||||
# NOTE: turns out you don't even need to sched an aio task
|
# NOTE: turns out you don't even need to sched an aio task
|
||||||
# since the original issue, even though seemingly was due to
|
# since the original issue, even though seemingly was due to
|
||||||
# the guest-run being abandoned + a `.debug.pause()` inside
|
# the guest-run being abandoned + a `._debug.pause()` inside
|
||||||
# `._runtime._async_main()` (which was originally trying to
|
# `._runtime._async_main()` (which was originally trying to
|
||||||
# debug the `.lifetime_stack` not closing), IS NOT actually
|
# debug the `.lifetime_stack` not closing), IS NOT actually
|
||||||
# the core issue?
|
# the core issue?
|
||||||
|
|
@ -1007,8 +852,6 @@ def test_sigint_closes_lifetime_stack(
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
delay = 999 if tractor.debug_mode() else 1
|
|
||||||
try:
|
try:
|
||||||
an: tractor.ActorNursery
|
an: tractor.ActorNursery
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
|
|
@ -1059,7 +902,7 @@ def test_sigint_closes_lifetime_stack(
|
||||||
if wait_for_ctx:
|
if wait_for_ctx:
|
||||||
print('waiting for ctx outcome in parent..')
|
print('waiting for ctx outcome in parent..')
|
||||||
try:
|
try:
|
||||||
with trio.fail_after(1 + delay):
|
with trio.fail_after(1):
|
||||||
await ctx.wait_for_result()
|
await ctx.wait_for_result()
|
||||||
except tractor.ContextCancelled as ctxc:
|
except tractor.ContextCancelled as ctxc:
|
||||||
assert ctxc.canceller == ctx.chan.uid
|
assert ctxc.canceller == ctx.chan.uid
|
||||||
|
|
@ -1088,108 +931,6 @@ def test_sigint_closes_lifetime_stack(
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# ?TODO asyncio.Task fn-deco?
|
|
||||||
# -[ ] do sig checkingat import time like @context?
|
|
||||||
# -[ ] maybe name it @aio_task ??
|
|
||||||
# -[ ] chan: to_asyncio.InterloopChannel ??
|
|
||||||
async def raise_before_started(
|
|
||||||
# from_trio: asyncio.Queue,
|
|
||||||
# to_trio: trio.abc.SendChannel,
|
|
||||||
chan: to_asyncio.LinkedTaskChannel,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
`asyncio.Task` entry point which RTEs before calling
|
|
||||||
`to_trio.send_nowait()`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
await asyncio.sleep(0.2)
|
|
||||||
raise RuntimeError('Some shite went wrong before `.send_nowait()`!!')
|
|
||||||
|
|
||||||
# to_trio.send_nowait('Uhh we shouldve RTE-d ^^ ??')
|
|
||||||
chan.started_nowait('Uhh we shouldve RTE-d ^^ ??')
|
|
||||||
await asyncio.sleep(float('inf'))
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def caching_ep(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
):
|
|
||||||
|
|
||||||
log = tractor.log.get_logger('caching_ep')
|
|
||||||
log.info('syncing via `ctx.started()`')
|
|
||||||
await ctx.started()
|
|
||||||
|
|
||||||
# XXX, allocate the `open_channel_from()` inside
|
|
||||||
# a `.trionics.maybe_open_context()`.
|
|
||||||
chan: to_asyncio.LinkedTaskChannel
|
|
||||||
async with (
|
|
||||||
tractor.trionics.maybe_open_context(
|
|
||||||
acm_func=tractor.to_asyncio.open_channel_from,
|
|
||||||
kwargs={
|
|
||||||
'target': raise_before_started,
|
|
||||||
# ^XXX, kwarg to `open_channel_from()`
|
|
||||||
},
|
|
||||||
|
|
||||||
# lock around current actor task access
|
|
||||||
key=tractor.current_actor().uid,
|
|
||||||
|
|
||||||
) as (cache_hit, (clients, chan)),
|
|
||||||
):
|
|
||||||
if cache_hit:
|
|
||||||
log.error(
|
|
||||||
'Re-using cached `.open_from_channel()` call!\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
else:
|
|
||||||
log.info(
|
|
||||||
'Allocating SHOULD-FAIL `.open_from_channel()`\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
await trio.sleep_forever()
|
|
||||||
|
|
||||||
|
|
||||||
def test_aio_side_raises_before_started(
|
|
||||||
reg_addr: tuple[str, int],
|
|
||||||
debug_mode: bool,
|
|
||||||
loglevel: str,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Simulates connection-err from `piker.brokers.ib.api`..
|
|
||||||
|
|
||||||
Ensure any error raised by child-`asyncio.Task` BEFORE
|
|
||||||
`chan.started()`
|
|
||||||
|
|
||||||
'''
|
|
||||||
# delay = 999 if debug_mode else 1
|
|
||||||
async def main():
|
|
||||||
with trio.fail_after(3):
|
|
||||||
an: tractor.ActorNursery
|
|
||||||
async with tractor.open_nursery(
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
loglevel=loglevel,
|
|
||||||
) as an:
|
|
||||||
p: tractor.Portal = await an.start_actor(
|
|
||||||
'lchan_cacher_that_raises_fast',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
infect_asyncio=True,
|
|
||||||
)
|
|
||||||
async with p.open_context(
|
|
||||||
caching_ep,
|
|
||||||
) as (ctx, first):
|
|
||||||
assert not first
|
|
||||||
|
|
||||||
with pytest.raises(
|
|
||||||
expected_exception=(RemoteActorError),
|
|
||||||
) as excinfo:
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
# ensure `asyncio.Task` exception is bubbled
|
|
||||||
# allll the way erp!!
|
|
||||||
rae = excinfo.value
|
|
||||||
assert rae.boxed_type is RuntimeError
|
|
||||||
|
|
||||||
# TODO: debug_mode tests once we get support for `asyncio`!
|
# TODO: debug_mode tests once we get support for `asyncio`!
|
||||||
#
|
#
|
||||||
# -[ ] need tests to wrap both scripts:
|
# -[ ] need tests to wrap both scripts:
|
||||||
|
|
@ -1203,7 +944,7 @@ def test_aio_side_raises_before_started(
|
||||||
# => completed using `.bestow_portal(task)` inside
|
# => completed using `.bestow_portal(task)` inside
|
||||||
# `.to_asyncio._run_asyncio_task()` right?
|
# `.to_asyncio._run_asyncio_task()` right?
|
||||||
# -[ ] translation func to get from `asyncio` task calling to
|
# -[ ] translation func to get from `asyncio` task calling to
|
||||||
# `.debug.wait_for_parent_stdin_hijack()` which does root
|
# `._debug.wait_for_parent_stdin_hijack()` which does root
|
||||||
# call to do TTY locking.
|
# call to do TTY locking.
|
||||||
#
|
#
|
||||||
def test_sync_breakpoint():
|
def test_sync_breakpoint():
|
||||||
|
|
|
||||||
|
|
@ -24,10 +24,14 @@ from tractor._testing import (
|
||||||
)
|
)
|
||||||
|
|
||||||
# XXX TODO cases:
|
# XXX TODO cases:
|
||||||
|
# - [ ] peer cancelled itself - so other peers should
|
||||||
|
# get errors reflecting that the peer was itself the .canceller?
|
||||||
|
|
||||||
# - [x] WE cancelled the peer and thus should not see any raised
|
# - [x] WE cancelled the peer and thus should not see any raised
|
||||||
# `ContextCancelled` as it should be reaped silently?
|
# `ContextCancelled` as it should be reaped silently?
|
||||||
# => pretty sure `test_context_stream_semantics::test_caller_cancels()`
|
# => pretty sure `test_context_stream_semantics::test_caller_cancels()`
|
||||||
# already covers this case?
|
# already covers this case?
|
||||||
|
|
||||||
# - [x] INTER-PEER: some arbitrary remote peer cancels via
|
# - [x] INTER-PEER: some arbitrary remote peer cancels via
|
||||||
# Portal.cancel_actor().
|
# Portal.cancel_actor().
|
||||||
# => all other connected peers should get that cancel requesting peer's
|
# => all other connected peers should get that cancel requesting peer's
|
||||||
|
|
@ -40,6 +44,16 @@ from tractor._testing import (
|
||||||
# that also spawned a remote task task in that same peer-parent.
|
# that also spawned a remote task task in that same peer-parent.
|
||||||
|
|
||||||
|
|
||||||
|
# def test_self_cancel():
|
||||||
|
# '''
|
||||||
|
# 2 cases:
|
||||||
|
# - calls `Actor.cancel()` locally in some task
|
||||||
|
# - calls LocalPortal.cancel_actor()` ?
|
||||||
|
|
||||||
|
# '''
|
||||||
|
# ...
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
async def open_stream_then_sleep_forever(
|
async def open_stream_then_sleep_forever(
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
|
|
@ -156,7 +170,7 @@ def test_do_not_swallow_error_before_started_by_remote_contextcancelled(
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
rae = excinfo.value
|
rae = excinfo.value
|
||||||
assert rae.boxed_type is TypeError
|
assert rae.boxed_type == TypeError
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
@tractor.context
|
||||||
|
|
@ -396,6 +410,7 @@ def test_peer_canceller(
|
||||||
'''
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
|
# NOTE: to halt the peer tasks on ctxc, uncomment this.
|
||||||
debug_mode=debug_mode,
|
debug_mode=debug_mode,
|
||||||
) as an:
|
) as an:
|
||||||
canceller: Portal = await an.start_actor(
|
canceller: Portal = await an.start_actor(
|
||||||
|
|
@ -792,7 +807,7 @@ async def basic_echo_server(
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
peer_name: str = 'wittle_bruv',
|
peer_name: str = 'wittle_bruv',
|
||||||
|
|
||||||
err_after_imsg: int|None = None,
|
err_after: int|None = None,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
|
|
@ -821,9 +836,8 @@ async def basic_echo_server(
|
||||||
await ipc.send(resp)
|
await ipc.send(resp)
|
||||||
|
|
||||||
if (
|
if (
|
||||||
err_after_imsg
|
err_after
|
||||||
and
|
and i > err_after
|
||||||
i > err_after_imsg
|
|
||||||
):
|
):
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
f'Simulated error in `{peer_name}`'
|
f'Simulated error in `{peer_name}`'
|
||||||
|
|
@ -857,7 +871,7 @@ async def serve_subactors(
|
||||||
)
|
)
|
||||||
await ipc.send((
|
await ipc.send((
|
||||||
peer.chan.uid,
|
peer.chan.uid,
|
||||||
peer.chan.raddr.unwrap(),
|
peer.chan.raddr,
|
||||||
))
|
))
|
||||||
|
|
||||||
print('Spawner exiting spawn serve loop!')
|
print('Spawner exiting spawn serve loop!')
|
||||||
|
|
@ -965,8 +979,7 @@ async def tell_little_bro(
|
||||||
actor_name: str,
|
actor_name: str,
|
||||||
|
|
||||||
caller: str = '',
|
caller: str = '',
|
||||||
err_after: float|None = None,
|
err_after: int|None = None,
|
||||||
rng_seed: int = 50,
|
|
||||||
):
|
):
|
||||||
# contact target actor, do a stream dialog.
|
# contact target actor, do a stream dialog.
|
||||||
async with (
|
async with (
|
||||||
|
|
@ -977,18 +990,14 @@ async def tell_little_bro(
|
||||||
basic_echo_server,
|
basic_echo_server,
|
||||||
|
|
||||||
# XXX proxy any delayed err condition
|
# XXX proxy any delayed err condition
|
||||||
err_after_imsg=(
|
err_after=err_after,
|
||||||
err_after * rng_seed
|
|
||||||
if err_after is not None
|
|
||||||
else None
|
|
||||||
),
|
|
||||||
) as (sub_ctx, first),
|
) as (sub_ctx, first),
|
||||||
|
|
||||||
sub_ctx.open_stream() as echo_ipc,
|
sub_ctx.open_stream() as echo_ipc,
|
||||||
):
|
):
|
||||||
actor: Actor = current_actor()
|
actor: Actor = current_actor()
|
||||||
uid: tuple = actor.uid
|
uid: tuple = actor.uid
|
||||||
for i in range(rng_seed):
|
for i in range(100):
|
||||||
msg: tuple = (
|
msg: tuple = (
|
||||||
uid,
|
uid,
|
||||||
i,
|
i,
|
||||||
|
|
@ -1013,13 +1022,13 @@ async def tell_little_bro(
|
||||||
)
|
)
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'raise_sub_spawn_error_after',
|
'raise_sub_spawn_error_after',
|
||||||
[None, 0.5],
|
[None, 50],
|
||||||
)
|
)
|
||||||
def test_peer_spawns_and_cancels_service_subactor(
|
def test_peer_spawns_and_cancels_service_subactor(
|
||||||
debug_mode: bool,
|
debug_mode: bool,
|
||||||
raise_client_error: str,
|
raise_client_error: str,
|
||||||
reg_addr: tuple[str, int],
|
reg_addr: tuple[str, int],
|
||||||
raise_sub_spawn_error_after: float|None,
|
raise_sub_spawn_error_after: int|None,
|
||||||
):
|
):
|
||||||
# NOTE: this tests for the modden `mod wks open piker` bug
|
# NOTE: this tests for the modden `mod wks open piker` bug
|
||||||
# discovered as part of implementing workspace ctx
|
# discovered as part of implementing workspace ctx
|
||||||
|
|
@ -1033,7 +1042,6 @@ def test_peer_spawns_and_cancels_service_subactor(
|
||||||
# and the server's spawned child should cancel and terminate!
|
# and the server's spawned child should cancel and terminate!
|
||||||
peer_name: str = 'little_bro'
|
peer_name: str = 'little_bro'
|
||||||
|
|
||||||
|
|
||||||
def check_inner_rte(rae: RemoteActorError):
|
def check_inner_rte(rae: RemoteActorError):
|
||||||
'''
|
'''
|
||||||
Validate the little_bro's relayed inception!
|
Validate the little_bro's relayed inception!
|
||||||
|
|
@ -1127,7 +1135,8 @@ def test_peer_spawns_and_cancels_service_subactor(
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
res = await client_ctx.wait_for_result(hide_tb=False)
|
res = await client_ctx.result(hide_tb=False)
|
||||||
|
|
||||||
# in remote (relayed inception) error
|
# in remote (relayed inception) error
|
||||||
# case, we should error on the line above!
|
# case, we should error on the line above!
|
||||||
if raise_sub_spawn_error_after:
|
if raise_sub_spawn_error_after:
|
||||||
|
|
@ -1138,23 +1147,6 @@ def test_peer_spawns_and_cancels_service_subactor(
|
||||||
assert isinstance(res, ContextCancelled)
|
assert isinstance(res, ContextCancelled)
|
||||||
assert client_ctx.cancel_acked
|
assert client_ctx.cancel_acked
|
||||||
assert res.canceller == root.uid
|
assert res.canceller == root.uid
|
||||||
assert not raise_sub_spawn_error_after
|
|
||||||
|
|
||||||
# cancelling the spawner sub should
|
|
||||||
# transitively cancel it's sub, the little
|
|
||||||
# bruv.
|
|
||||||
print('root cancelling server/client sub-actors')
|
|
||||||
await spawn_ctx.cancel()
|
|
||||||
async with tractor.find_actor(
|
|
||||||
name=peer_name,
|
|
||||||
) as sub:
|
|
||||||
assert not sub
|
|
||||||
|
|
||||||
# XXX, only for tracing
|
|
||||||
# except BaseException as _berr:
|
|
||||||
# berr = _berr
|
|
||||||
# await tractor.pause(shield=True)
|
|
||||||
# raise berr
|
|
||||||
|
|
||||||
except RemoteActorError as rae:
|
except RemoteActorError as rae:
|
||||||
_err = rae
|
_err = rae
|
||||||
|
|
@ -1183,8 +1175,19 @@ def test_peer_spawns_and_cancels_service_subactor(
|
||||||
raise
|
raise
|
||||||
# await tractor.pause()
|
# await tractor.pause()
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert not raise_sub_spawn_error_after
|
||||||
|
|
||||||
|
# cancelling the spawner sub should
|
||||||
|
# transitively cancel it's sub, the little
|
||||||
|
# bruv.
|
||||||
|
print('root cancelling server/client sub-actors')
|
||||||
|
await spawn_ctx.cancel()
|
||||||
|
async with tractor.find_actor(
|
||||||
|
name=peer_name,
|
||||||
|
) as sub:
|
||||||
|
assert not sub
|
||||||
|
|
||||||
# await tractor.pause()
|
|
||||||
# await server.cancel_actor()
|
# await server.cancel_actor()
|
||||||
|
|
||||||
except RemoteActorError as rae:
|
except RemoteActorError as rae:
|
||||||
|
|
@ -1197,7 +1200,7 @@ def test_peer_spawns_and_cancels_service_subactor(
|
||||||
|
|
||||||
# since we called `.cancel_actor()`, `.cancel_ack`
|
# since we called `.cancel_actor()`, `.cancel_ack`
|
||||||
# will not be set on the ctx bc `ctx.cancel()` was not
|
# will not be set on the ctx bc `ctx.cancel()` was not
|
||||||
# called directly for this confext.
|
# called directly fot this confext.
|
||||||
except ContextCancelled as ctxc:
|
except ContextCancelled as ctxc:
|
||||||
_ctxc = ctxc
|
_ctxc = ctxc
|
||||||
print(
|
print(
|
||||||
|
|
@ -1237,19 +1240,12 @@ def test_peer_spawns_and_cancels_service_subactor(
|
||||||
|
|
||||||
# assert spawn_ctx.cancelled_caught
|
# assert spawn_ctx.cancelled_caught
|
||||||
|
|
||||||
async def _main():
|
|
||||||
with trio.fail_after(
|
|
||||||
3 if not debug_mode
|
|
||||||
else 999
|
|
||||||
):
|
|
||||||
await main()
|
|
||||||
|
|
||||||
if raise_sub_spawn_error_after:
|
if raise_sub_spawn_error_after:
|
||||||
with pytest.raises(RemoteActorError) as excinfo:
|
with pytest.raises(RemoteActorError) as excinfo:
|
||||||
trio.run(_main)
|
trio.run(main)
|
||||||
|
|
||||||
rae: RemoteActorError = excinfo.value
|
rae: RemoteActorError = excinfo.value
|
||||||
check_inner_rte(rae)
|
check_inner_rte(rae)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
trio.run(_main)
|
trio.run(main)
|
||||||
|
|
|
||||||
|
|
@ -235,16 +235,10 @@ async def cancel_after(wait, reg_addr):
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='module')
|
@pytest.fixture(scope='module')
|
||||||
def time_quad_ex(
|
def time_quad_ex(reg_addr, ci_env, spawn_backend):
|
||||||
reg_addr: tuple,
|
|
||||||
ci_env: bool,
|
|
||||||
spawn_backend: str,
|
|
||||||
):
|
|
||||||
if spawn_backend == 'mp':
|
if spawn_backend == 'mp':
|
||||||
'''
|
"""no idea but the mp *nix runs are flaking out here often...
|
||||||
no idea but the mp *nix runs are flaking out here often...
|
"""
|
||||||
|
|
||||||
'''
|
|
||||||
pytest.skip("Test is too flaky on mp in CI")
|
pytest.skip("Test is too flaky on mp in CI")
|
||||||
|
|
||||||
timeout = 7 if platform.system() in ('Windows', 'Darwin') else 4
|
timeout = 7 if platform.system() in ('Windows', 'Darwin') else 4
|
||||||
|
|
@ -255,24 +249,12 @@ def time_quad_ex(
|
||||||
return results, diff
|
return results, diff
|
||||||
|
|
||||||
|
|
||||||
def test_a_quadruple_example(
|
def test_a_quadruple_example(time_quad_ex, ci_env, spawn_backend):
|
||||||
time_quad_ex: tuple,
|
"""This also serves as a kind of "we'd like to be this fast test"."""
|
||||||
ci_env: bool,
|
|
||||||
spawn_backend: str,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
This also serves as a kind of "we'd like to be this fast test".
|
|
||||||
|
|
||||||
'''
|
|
||||||
results, diff = time_quad_ex
|
results, diff = time_quad_ex
|
||||||
assert results
|
assert results
|
||||||
this_fast = (
|
this_fast = 6 if platform.system() in ('Windows', 'Darwin') else 3
|
||||||
6 if platform.system() in (
|
|
||||||
'Windows',
|
|
||||||
'Darwin',
|
|
||||||
)
|
|
||||||
else 3
|
|
||||||
)
|
|
||||||
assert diff < this_fast
|
assert diff < this_fast
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -38,7 +38,7 @@ async def test_self_is_registered_localportal(reg_addr):
|
||||||
"Verify waiting on the arbiter to register itself using a local portal."
|
"Verify waiting on the arbiter to register itself using a local portal."
|
||||||
actor = tractor.current_actor()
|
actor = tractor.current_actor()
|
||||||
assert actor.is_arbiter
|
assert actor.is_arbiter
|
||||||
async with tractor.get_registry(reg_addr) as portal:
|
async with tractor.get_registry(*reg_addr) as portal:
|
||||||
assert isinstance(portal, tractor._portal.LocalPortal)
|
assert isinstance(portal, tractor._portal.LocalPortal)
|
||||||
|
|
||||||
with trio.fail_after(0.2):
|
with trio.fail_after(0.2):
|
||||||
|
|
|
||||||
|
|
@ -10,7 +10,7 @@ import tractor
|
||||||
from tractor._testing import (
|
from tractor._testing import (
|
||||||
tractor_test,
|
tractor_test,
|
||||||
)
|
)
|
||||||
from .conftest import (
|
from conftest import (
|
||||||
sig_prog,
|
sig_prog,
|
||||||
_INT_SIGNAL,
|
_INT_SIGNAL,
|
||||||
_INT_RETURN_CODE,
|
_INT_RETURN_CODE,
|
||||||
|
|
@ -32,7 +32,7 @@ def test_abort_on_sigint(daemon):
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_cancel_remote_arbiter(daemon, reg_addr):
|
async def test_cancel_remote_arbiter(daemon, reg_addr):
|
||||||
assert not tractor.current_actor().is_arbiter
|
assert not tractor.current_actor().is_arbiter
|
||||||
async with tractor.get_registry(reg_addr) as portal:
|
async with tractor.get_registry(*reg_addr) as portal:
|
||||||
await portal.cancel_actor()
|
await portal.cancel_actor()
|
||||||
|
|
||||||
time.sleep(0.1)
|
time.sleep(0.1)
|
||||||
|
|
@ -41,7 +41,7 @@ async def test_cancel_remote_arbiter(daemon, reg_addr):
|
||||||
|
|
||||||
# no arbiter socket should exist
|
# no arbiter socket should exist
|
||||||
with pytest.raises(OSError):
|
with pytest.raises(OSError):
|
||||||
async with tractor.get_registry(reg_addr) as portal:
|
async with tractor.get_registry(*reg_addr) as portal:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,239 +0,0 @@
|
||||||
'''
|
|
||||||
Define the details of inter-actor "out-of-band" (OoB) cancel
|
|
||||||
semantics, that is how cancellation works when a cancel request comes
|
|
||||||
from the different concurrency (primitive's) "layer" then where the
|
|
||||||
eventual `trio.Task` actually raises a signal.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from functools import partial
|
|
||||||
# from contextlib import asynccontextmanager as acm
|
|
||||||
# import itertools
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
import trio
|
|
||||||
import tractor
|
|
||||||
from tractor import ( # typing
|
|
||||||
ActorNursery,
|
|
||||||
Portal,
|
|
||||||
Context,
|
|
||||||
# ContextCancelled,
|
|
||||||
# RemoteActorError,
|
|
||||||
)
|
|
||||||
# from tractor._testing import (
|
|
||||||
# tractor_test,
|
|
||||||
# expect_ctxc,
|
|
||||||
# )
|
|
||||||
|
|
||||||
# XXX TODO cases:
|
|
||||||
# - [ ] peer cancelled itself - so other peers should
|
|
||||||
# get errors reflecting that the peer was itself the .canceller?
|
|
||||||
|
|
||||||
# def test_self_cancel():
|
|
||||||
# '''
|
|
||||||
# 2 cases:
|
|
||||||
# - calls `Actor.cancel()` locally in some task
|
|
||||||
# - calls LocalPortal.cancel_actor()` ?
|
|
||||||
#
|
|
||||||
# things to ensure!
|
|
||||||
# -[ ] the ctxc raised in a child should ideally show the tb of the
|
|
||||||
# underlying `Cancelled` checkpoint, i.e.
|
|
||||||
# `raise scope_error from ctxc`?
|
|
||||||
#
|
|
||||||
# -[ ] a self-cancelled context, if not allowed to block on
|
|
||||||
# `ctx.result()` at some point will hang since the `ctx._scope`
|
|
||||||
# is never `.cancel_called`; cases for this include,
|
|
||||||
# - an `open_ctx()` which never starteds before being OoB actor
|
|
||||||
# cancelled.
|
|
||||||
# |_ parent task will be blocked in `.open_context()` for the
|
|
||||||
# `Started` msg, and when the OoB ctxc arrives `ctx._scope`
|
|
||||||
# will never have been signalled..
|
|
||||||
|
|
||||||
# '''
|
|
||||||
# ...
|
|
||||||
|
|
||||||
# TODO, sanity test against the case in `/examples/trio/lockacquire_not_unmasked.py`
|
|
||||||
# but with the `Lock.acquire()` from a `@context` to ensure the
|
|
||||||
# implicit ignore-case-non-unmasking.
|
|
||||||
#
|
|
||||||
# @tractor.context
|
|
||||||
# async def acquire_actor_global_lock(
|
|
||||||
# ctx: tractor.Context,
|
|
||||||
# ignore_special_cases: bool,
|
|
||||||
# ):
|
|
||||||
|
|
||||||
# async with maybe_unmask_excs(
|
|
||||||
# ignore_special_cases=ignore_special_cases,
|
|
||||||
# ):
|
|
||||||
# await ctx.started('locked')
|
|
||||||
|
|
||||||
# # block til cancelled
|
|
||||||
# await trio.sleep_forever()
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def sleep_forever(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
# ignore_special_cases: bool,
|
|
||||||
do_started: bool,
|
|
||||||
):
|
|
||||||
|
|
||||||
# async with maybe_unmask_excs(
|
|
||||||
# ignore_special_cases=ignore_special_cases,
|
|
||||||
# ):
|
|
||||||
# await ctx.started('locked')
|
|
||||||
if do_started:
|
|
||||||
await ctx.started()
|
|
||||||
|
|
||||||
# block til cancelled
|
|
||||||
print('sleepin on child-side..')
|
|
||||||
await trio.sleep_forever()
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'cancel_ctx',
|
|
||||||
[True, False],
|
|
||||||
)
|
|
||||||
def test_cancel_ctx_with_parent_side_entered_in_bg_task(
|
|
||||||
debug_mode: bool,
|
|
||||||
loglevel: str,
|
|
||||||
cancel_ctx: bool,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
The most "basic" out-of-band-task self-cancellation case where
|
|
||||||
`Portal.open_context()` is entered in a bg task and the
|
|
||||||
parent-task (of the containing nursery) calls `Context.cancel()`
|
|
||||||
without the child knowing; the `Context._scope` should be
|
|
||||||
`.cancel_called` when the IPC ctx's child-side relays
|
|
||||||
a `ContextCancelled` with a `.canceller` set to the parent
|
|
||||||
actor('s task).
|
|
||||||
|
|
||||||
'''
|
|
||||||
async def main():
|
|
||||||
with trio.fail_after(
|
|
||||||
2 if not debug_mode else 999,
|
|
||||||
):
|
|
||||||
an: ActorNursery
|
|
||||||
async with (
|
|
||||||
tractor.open_nursery(
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
loglevel='devx',
|
|
||||||
enable_stack_on_sig=True,
|
|
||||||
) as an,
|
|
||||||
trio.open_nursery() as tn,
|
|
||||||
):
|
|
||||||
ptl: Portal = await an.start_actor(
|
|
||||||
'sub',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
)
|
|
||||||
|
|
||||||
async def _open_ctx_async(
|
|
||||||
do_started: bool = True,
|
|
||||||
task_status=trio.TASK_STATUS_IGNORED,
|
|
||||||
):
|
|
||||||
# do we expect to never enter the
|
|
||||||
# `.open_context()` below.
|
|
||||||
if not do_started:
|
|
||||||
task_status.started()
|
|
||||||
|
|
||||||
async with ptl.open_context(
|
|
||||||
sleep_forever,
|
|
||||||
do_started=do_started,
|
|
||||||
) as (ctx, first):
|
|
||||||
task_status.started(ctx)
|
|
||||||
await trio.sleep_forever()
|
|
||||||
|
|
||||||
# XXX, this is the key OoB part!
|
|
||||||
#
|
|
||||||
# - start the `.open_context()` in a bg task which
|
|
||||||
# blocks inside the embedded scope-body,
|
|
||||||
#
|
|
||||||
# - when we call `Context.cancel()` it **is
|
|
||||||
# not** from the same task which eventually runs
|
|
||||||
# `.__aexit__()`,
|
|
||||||
#
|
|
||||||
# - since the bg "opener" task will be in
|
|
||||||
# a `trio.sleep_forever()`, it must be interrupted
|
|
||||||
# by the `ContextCancelled` delivered from the
|
|
||||||
# child-side; `Context._scope: CancelScope` MUST
|
|
||||||
# be `.cancel_called`!
|
|
||||||
#
|
|
||||||
print('ASYNC opening IPC context in subtask..')
|
|
||||||
maybe_ctx: Context|None = await tn.start(partial(
|
|
||||||
_open_ctx_async,
|
|
||||||
))
|
|
||||||
|
|
||||||
if (
|
|
||||||
maybe_ctx
|
|
||||||
and
|
|
||||||
cancel_ctx
|
|
||||||
):
|
|
||||||
print('cancelling first IPC ctx!')
|
|
||||||
await maybe_ctx.cancel()
|
|
||||||
|
|
||||||
# XXX, note that despite `maybe_context.cancel()`
|
|
||||||
# being called above, it's the parent (bg) task
|
|
||||||
# which was originally never interrupted in
|
|
||||||
# the `ctx._scope` body due to missing case logic in
|
|
||||||
# `ctx._maybe_cancel_and_set_remote_error()`.
|
|
||||||
#
|
|
||||||
# It didn't matter that the subactor process was
|
|
||||||
# already terminated and reaped, nothing was
|
|
||||||
# cancelling the ctx-parent task's scope!
|
|
||||||
#
|
|
||||||
print('cancelling subactor!')
|
|
||||||
await ptl.cancel_actor()
|
|
||||||
|
|
||||||
if maybe_ctx:
|
|
||||||
try:
|
|
||||||
await maybe_ctx.wait_for_result()
|
|
||||||
except tractor.ContextCancelled as ctxc:
|
|
||||||
assert not cancel_ctx
|
|
||||||
assert (
|
|
||||||
ctxc.canceller
|
|
||||||
==
|
|
||||||
tractor.current_actor().aid.uid
|
|
||||||
)
|
|
||||||
# don't re-raise since it'll trigger
|
|
||||||
# an EG from the above tn.
|
|
||||||
|
|
||||||
if cancel_ctx:
|
|
||||||
# graceful self-cancel
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
else:
|
|
||||||
# ctx parent task should see OoB ctxc due to
|
|
||||||
# `ptl.cancel_actor()`.
|
|
||||||
with pytest.raises(tractor.ContextCancelled) as excinfo:
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
assert 'root' in excinfo.value.canceller[0]
|
|
||||||
|
|
||||||
|
|
||||||
# def test_parent_actor_cancels_subactor_with_gt1_ctxs_open_to_it(
|
|
||||||
# debug_mode: bool,
|
|
||||||
# loglevel: str,
|
|
||||||
# ):
|
|
||||||
# '''
|
|
||||||
# Demos OoB cancellation from the perspective of a ctx opened with
|
|
||||||
# a child subactor where the parent cancels the child at the "actor
|
|
||||||
# layer" using `Portal.cancel_actor()` and thus the
|
|
||||||
# `ContextCancelled.canceller` received by the ctx's parent-side
|
|
||||||
# task will appear to be a "self cancellation" even though that
|
|
||||||
# specific task itself was not cancelled and thus
|
|
||||||
# `Context.cancel_called ==False`.
|
|
||||||
# '''
|
|
||||||
# TODO, do we have an existing implied ctx
|
|
||||||
# cancel test like this?
|
|
||||||
# with trio.move_on_after(0.5):# as cs:
|
|
||||||
# await _open_ctx_async(
|
|
||||||
# do_started=False,
|
|
||||||
# )
|
|
||||||
|
|
||||||
|
|
||||||
# in-line ctx scope should definitely raise
|
|
||||||
# a ctxc with `.canceller = 'root'`
|
|
||||||
# async with ptl.open_context(
|
|
||||||
# sleep_forever,
|
|
||||||
# do_started=True,
|
|
||||||
# ) as pair:
|
|
||||||
|
|
||||||
|
|
@ -1,237 +0,0 @@
|
||||||
'''
|
|
||||||
Special case testing for issues not (dis)covered in the primary
|
|
||||||
`Context` related functional/scenario suites.
|
|
||||||
|
|
||||||
**NOTE: this mod is a WIP** space for handling
|
|
||||||
odd/rare/undiscovered/not-yet-revealed faults which either
|
|
||||||
loudly (ideal case) breakl our supervision protocol
|
|
||||||
or (worst case) result in distributed sys hangs.
|
|
||||||
|
|
||||||
Suites here further try to clarify (if [partially] ill-defined) and
|
|
||||||
verify our edge case semantics for inter-actor-relayed-exceptions
|
|
||||||
including,
|
|
||||||
|
|
||||||
- lowlevel: what remote obj-data is interchanged for IPC and what is
|
|
||||||
native-obj form is expected from unpacking in the the new
|
|
||||||
mem-domain.
|
|
||||||
|
|
||||||
- which kinds of `RemoteActorError` (and its derivs) are expected by which
|
|
||||||
(types of) peers (parent, child, sibling, etc) with what
|
|
||||||
particular meta-data set such as,
|
|
||||||
|
|
||||||
- `.src_uid`: the original (maybe) peer who raised.
|
|
||||||
- `.relay_uid`: the next-hop-peer who sent it.
|
|
||||||
- `.relay_path`: the sequence of peer actor hops.
|
|
||||||
- `.is_inception`: a predicate that denotes multi-hop remote errors.
|
|
||||||
|
|
||||||
- when should `ExceptionGroup`s be relayed from a particular
|
|
||||||
remote endpoint, they should never be caused by implicit `._rpc`
|
|
||||||
nursery machinery!
|
|
||||||
|
|
||||||
- various special `trio` edge cases around its cancellation semantics
|
|
||||||
and how we (currently) leverage `trio.Cancelled` as a signal for
|
|
||||||
whether a `Context` task should raise `ContextCancelled` (ctx).
|
|
||||||
|
|
||||||
'''
|
|
||||||
import pytest
|
|
||||||
import trio
|
|
||||||
import tractor
|
|
||||||
from tractor import ( # typing
|
|
||||||
ActorNursery,
|
|
||||||
Portal,
|
|
||||||
Context,
|
|
||||||
ContextCancelled,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def sleep_n_chkpt_in_finally(
|
|
||||||
ctx: Context,
|
|
||||||
sleep_n_raise: bool,
|
|
||||||
|
|
||||||
chld_raise_delay: float,
|
|
||||||
chld_finally_delay: float,
|
|
||||||
|
|
||||||
rent_cancels: bool,
|
|
||||||
rent_ctxc_delay: float,
|
|
||||||
|
|
||||||
expect_exc: str|None = None,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
Sync, open a tn, then wait for cancel, run a chkpt inside
|
|
||||||
the user's `finally:` teardown.
|
|
||||||
|
|
||||||
This covers a footgun case that `trio` core doesn't seem to care about
|
|
||||||
wherein an exc can be masked by a `trio.Cancelled` raised inside a tn emedded
|
|
||||||
`finally:`.
|
|
||||||
|
|
||||||
Also see `test_trioisms::test_acm_embedded_nursery_propagates_enter_err`
|
|
||||||
for the down and gritty details.
|
|
||||||
|
|
||||||
Since a `@context` endpoint fn can also contain code like this,
|
|
||||||
**and** bc we currently have no easy way other then
|
|
||||||
`trio.Cancelled` to signal cancellation on each side of an IPC `Context`,
|
|
||||||
the footgun issue can compound itself as demonstrated in this suite..
|
|
||||||
|
|
||||||
Here are some edge cases codified with our WIP "sclang" syntax
|
|
||||||
(note the parent(rent)/child(chld) naming here is just
|
|
||||||
pragmatism, generally these most of these cases can occurr
|
|
||||||
regardless of the distributed-task's supervision hiearchy),
|
|
||||||
|
|
||||||
- rent c)=> chld.raises-then-taskc-in-finally
|
|
||||||
|_ chld's body raises an `exc: BaseException`.
|
|
||||||
_ in its `finally:` block it runs a chkpoint
|
|
||||||
which raises a taskc (`trio.Cancelled`) which
|
|
||||||
masks `exc` instead raising taskc up to the first tn.
|
|
||||||
_ the embedded/chld tn captures the masking taskc and then
|
|
||||||
raises it up to the ._rpc-ep-tn instead of `exc`.
|
|
||||||
_ the rent thinks the child ctxc-ed instead of errored..
|
|
||||||
|
|
||||||
'''
|
|
||||||
await ctx.started()
|
|
||||||
|
|
||||||
if expect_exc:
|
|
||||||
expect_exc: BaseException = tractor._exceptions.get_err_type(
|
|
||||||
type_name=expect_exc,
|
|
||||||
)
|
|
||||||
|
|
||||||
berr: BaseException|None = None
|
|
||||||
try:
|
|
||||||
if not sleep_n_raise:
|
|
||||||
await trio.sleep_forever()
|
|
||||||
elif sleep_n_raise:
|
|
||||||
|
|
||||||
# XXX this sleep is less then the sleep the parent
|
|
||||||
# does before calling `ctx.cancel()`
|
|
||||||
await trio.sleep(chld_raise_delay)
|
|
||||||
|
|
||||||
# XXX this will be masked by a taskc raised in
|
|
||||||
# the `finally:` if this fn doesn't terminate
|
|
||||||
# before any ctxc-req arrives AND a checkpoint is hit
|
|
||||||
# in that `finally:`.
|
|
||||||
raise RuntimeError('my app krurshed..')
|
|
||||||
|
|
||||||
except BaseException as _berr:
|
|
||||||
berr = _berr
|
|
||||||
|
|
||||||
# TODO: it'd sure be nice to be able to inject our own
|
|
||||||
# `ContextCancelled` here instead of of `trio.Cancelled`
|
|
||||||
# so that our runtime can expect it and this "user code"
|
|
||||||
# would be able to tell the diff between a generic trio
|
|
||||||
# cancel and a tractor runtime-IPC cancel.
|
|
||||||
if expect_exc:
|
|
||||||
if not isinstance(
|
|
||||||
berr,
|
|
||||||
expect_exc,
|
|
||||||
):
|
|
||||||
raise ValueError(
|
|
||||||
f'Unexpected exc type ??\n'
|
|
||||||
f'{berr!r}\n'
|
|
||||||
f'\n'
|
|
||||||
f'Expected a {expect_exc!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
raise berr
|
|
||||||
|
|
||||||
# simulate what user code might try even though
|
|
||||||
# it's a known boo-boo..
|
|
||||||
finally:
|
|
||||||
# maybe wait for rent ctxc to arrive
|
|
||||||
with trio.CancelScope(shield=True):
|
|
||||||
await trio.sleep(chld_finally_delay)
|
|
||||||
|
|
||||||
# !!XXX this will raise `trio.Cancelled` which
|
|
||||||
# will mask the RTE from above!!!
|
|
||||||
#
|
|
||||||
# YES, it's the same case as our extant
|
|
||||||
# `test_trioisms::test_acm_embedded_nursery_propagates_enter_err`
|
|
||||||
try:
|
|
||||||
await trio.lowlevel.checkpoint()
|
|
||||||
except trio.Cancelled as taskc:
|
|
||||||
if (scope_err := taskc.__context__):
|
|
||||||
print(
|
|
||||||
f'XXX MASKED REMOTE ERROR XXX\n'
|
|
||||||
f'ENDPOINT exception -> {scope_err!r}\n'
|
|
||||||
f'will be masked by -> {taskc!r}\n'
|
|
||||||
)
|
|
||||||
# await tractor.pause(shield=True)
|
|
||||||
|
|
||||||
raise taskc
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'chld_callspec',
|
|
||||||
[
|
|
||||||
dict(
|
|
||||||
sleep_n_raise=None,
|
|
||||||
chld_raise_delay=0.1,
|
|
||||||
chld_finally_delay=0.1,
|
|
||||||
expect_exc='Cancelled',
|
|
||||||
rent_cancels=True,
|
|
||||||
rent_ctxc_delay=0.1,
|
|
||||||
),
|
|
||||||
dict(
|
|
||||||
sleep_n_raise='RuntimeError',
|
|
||||||
chld_raise_delay=0.1,
|
|
||||||
chld_finally_delay=1,
|
|
||||||
expect_exc='RuntimeError',
|
|
||||||
rent_cancels=False,
|
|
||||||
rent_ctxc_delay=0.1,
|
|
||||||
),
|
|
||||||
],
|
|
||||||
ids=lambda item: f'chld_callspec={item!r}'
|
|
||||||
)
|
|
||||||
def test_unmasked_remote_exc(
|
|
||||||
debug_mode: bool,
|
|
||||||
chld_callspec: dict,
|
|
||||||
tpt_proto: str,
|
|
||||||
):
|
|
||||||
expect_exc_str: str|None = chld_callspec['sleep_n_raise']
|
|
||||||
rent_ctxc_delay: float|None = chld_callspec['rent_ctxc_delay']
|
|
||||||
async def main():
|
|
||||||
an: ActorNursery
|
|
||||||
async with tractor.open_nursery(
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
enable_transports=[tpt_proto],
|
|
||||||
) as an:
|
|
||||||
ptl: Portal = await an.start_actor(
|
|
||||||
'cancellee',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
)
|
|
||||||
ctx: Context
|
|
||||||
async with (
|
|
||||||
ptl.open_context(
|
|
||||||
sleep_n_chkpt_in_finally,
|
|
||||||
**chld_callspec,
|
|
||||||
) as (ctx, sent),
|
|
||||||
):
|
|
||||||
assert not sent
|
|
||||||
await trio.sleep(rent_ctxc_delay)
|
|
||||||
await ctx.cancel()
|
|
||||||
|
|
||||||
# recv error or result from chld
|
|
||||||
ctxc: ContextCancelled = await ctx.wait_for_result()
|
|
||||||
assert (
|
|
||||||
ctxc is ctx.outcome
|
|
||||||
and
|
|
||||||
isinstance(ctxc, ContextCancelled)
|
|
||||||
)
|
|
||||||
|
|
||||||
# always graceful terminate the sub in non-error cases
|
|
||||||
await an.cancel()
|
|
||||||
|
|
||||||
if expect_exc_str:
|
|
||||||
expect_exc: BaseException = tractor._exceptions.get_err_type(
|
|
||||||
type_name=expect_exc_str,
|
|
||||||
)
|
|
||||||
with pytest.raises(
|
|
||||||
expected_exception=tractor.RemoteActorError,
|
|
||||||
) as excinfo:
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
rae = excinfo.value
|
|
||||||
assert expect_exc == rae.boxed_type
|
|
||||||
|
|
||||||
else:
|
|
||||||
trio.run(main)
|
|
||||||
|
|
@ -1,6 +1,5 @@
|
||||||
'''
|
'''
|
||||||
Suites for our `.trionics.maybe_open_context()` multi-task
|
Async context manager cache api testing: ``trionics.maybe_open_context():``
|
||||||
shared-cached `@acm` API.
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from contextlib import asynccontextmanager as acm
|
from contextlib import asynccontextmanager as acm
|
||||||
|
|
@ -10,15 +9,6 @@ from typing import Awaitable
|
||||||
import pytest
|
import pytest
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
from tractor.trionics import (
|
|
||||||
maybe_open_context,
|
|
||||||
)
|
|
||||||
from tractor.log import (
|
|
||||||
get_console_log,
|
|
||||||
get_logger,
|
|
||||||
)
|
|
||||||
log = get_logger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
_resource: int = 0
|
_resource: int = 0
|
||||||
|
|
@ -62,7 +52,7 @@ def test_resource_only_entered_once(key_on):
|
||||||
# different task names per task will be used
|
# different task names per task will be used
|
||||||
kwargs = {'task_name': name}
|
kwargs = {'task_name': name}
|
||||||
|
|
||||||
async with maybe_open_context(
|
async with tractor.trionics.maybe_open_context(
|
||||||
maybe_increment_counter,
|
maybe_increment_counter,
|
||||||
kwargs=kwargs,
|
kwargs=kwargs,
|
||||||
key=key,
|
key=key,
|
||||||
|
|
@ -82,13 +72,11 @@ def test_resource_only_entered_once(key_on):
|
||||||
with trio.move_on_after(0.5):
|
with trio.move_on_after(0.5):
|
||||||
async with (
|
async with (
|
||||||
tractor.open_root_actor(),
|
tractor.open_root_actor(),
|
||||||
trio.open_nursery() as tn,
|
trio.open_nursery() as n,
|
||||||
):
|
):
|
||||||
|
|
||||||
for i in range(10):
|
for i in range(10):
|
||||||
tn.start_soon(
|
n.start_soon(enter_cached_mngr, f'task_{i}')
|
||||||
enter_cached_mngr,
|
|
||||||
f'task_{i}',
|
|
||||||
)
|
|
||||||
await trio.sleep(0.001)
|
await trio.sleep(0.001)
|
||||||
|
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
@ -110,55 +98,27 @@ async def streamer(
|
||||||
|
|
||||||
|
|
||||||
@acm
|
@acm
|
||||||
async def open_stream() -> Awaitable[
|
async def open_stream() -> Awaitable[tractor.MsgStream]:
|
||||||
tuple[
|
|
||||||
tractor.ActorNursery,
|
async with tractor.open_nursery() as tn:
|
||||||
tractor.MsgStream,
|
portal = await tn.start_actor('streamer', enable_modules=[__name__])
|
||||||
]
|
|
||||||
]:
|
|
||||||
try:
|
|
||||||
async with tractor.open_nursery() as an:
|
|
||||||
portal = await an.start_actor(
|
|
||||||
'streamer',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
)
|
|
||||||
try:
|
|
||||||
async with (
|
async with (
|
||||||
portal.open_context(streamer) as (ctx, first),
|
portal.open_context(streamer) as (ctx, first),
|
||||||
ctx.open_stream() as stream,
|
ctx.open_stream() as stream,
|
||||||
):
|
):
|
||||||
print('Entered open_stream() caller')
|
yield stream
|
||||||
yield an, stream
|
|
||||||
print('Exited open_stream() caller')
|
|
||||||
|
|
||||||
finally:
|
|
||||||
print(
|
|
||||||
'Cancelling streamer with,\n'
|
|
||||||
'=> `Portal.cancel_actor()`'
|
|
||||||
)
|
|
||||||
await portal.cancel_actor()
|
await portal.cancel_actor()
|
||||||
print('Cancelled streamer')
|
print('CANCELLED STREAMER')
|
||||||
|
|
||||||
except Exception as err:
|
|
||||||
print(
|
|
||||||
f'`open_stream()` errored?\n'
|
|
||||||
f'{err!r}\n'
|
|
||||||
)
|
|
||||||
await tractor.pause(shield=True)
|
|
||||||
raise err
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
@acm
|
||||||
async def maybe_open_stream(taskname: str):
|
async def maybe_open_stream(taskname: str):
|
||||||
async with maybe_open_context(
|
async with tractor.trionics.maybe_open_context(
|
||||||
# NOTE: all secondary tasks should cache hit on the same key
|
# NOTE: all secondary tasks should cache hit on the same key
|
||||||
acm_func=open_stream,
|
acm_func=open_stream,
|
||||||
) as (
|
) as (cache_hit, stream):
|
||||||
cache_hit,
|
|
||||||
(an, stream)
|
|
||||||
):
|
|
||||||
# when the actor + portal + ctx + stream has already been
|
|
||||||
# allocated we want to just bcast to this task.
|
|
||||||
if cache_hit:
|
if cache_hit:
|
||||||
print(f'{taskname} loaded from cache')
|
print(f'{taskname} loaded from cache')
|
||||||
|
|
||||||
|
|
@ -166,77 +126,27 @@ async def maybe_open_stream(taskname: str):
|
||||||
# if this feed is already allocated by the first
|
# if this feed is already allocated by the first
|
||||||
# task that entereed
|
# task that entereed
|
||||||
async with stream.subscribe() as bstream:
|
async with stream.subscribe() as bstream:
|
||||||
yield an, bstream
|
yield bstream
|
||||||
print(
|
|
||||||
f'cached task exited\n'
|
|
||||||
f')>\n'
|
|
||||||
f' |_{taskname}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# we should always unreg the "cloned" bcrc for this
|
|
||||||
# consumer-task
|
|
||||||
assert id(bstream) not in bstream._state.subs
|
|
||||||
|
|
||||||
else:
|
else:
|
||||||
# yield the actual stream
|
# yield the actual stream
|
||||||
try:
|
yield stream
|
||||||
yield an, stream
|
|
||||||
finally:
|
|
||||||
print(
|
|
||||||
f'NON-cached task exited\n'
|
|
||||||
f')>\n'
|
|
||||||
f' |_{taskname}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
first_bstream = stream._broadcaster
|
|
||||||
bcrx_state = first_bstream._state
|
|
||||||
subs: dict[int, int] = bcrx_state.subs
|
|
||||||
if len(subs) == 1:
|
|
||||||
assert id(first_bstream) in subs
|
|
||||||
# ^^TODO! the bcrx should always de-allocate all subs,
|
|
||||||
# including the implicit first one allocated on entry
|
|
||||||
# by the first subscribing peer task, no?
|
|
||||||
#
|
|
||||||
# -[ ] adjust `MsgStream.subscribe()` to do this mgmt!
|
|
||||||
# |_ allows reverting `MsgStream.receive()` to the
|
|
||||||
# non-bcaster method.
|
|
||||||
# |_ we can decide whether to reset `._broadcaster`?
|
|
||||||
#
|
|
||||||
# await tractor.pause(shield=True)
|
|
||||||
|
|
||||||
|
|
||||||
def test_open_local_sub_to_stream(
|
def test_open_local_sub_to_stream():
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
'''
|
'''
|
||||||
Verify a single inter-actor stream can can be fanned-out shared to
|
Verify a single inter-actor stream can can be fanned-out shared to
|
||||||
N local tasks using `trionics.maybe_open_context()`.
|
N local tasks using ``trionics.maybe_open_context():``.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
timeout: float = 3.6
|
timeout: float = 3.6 if platform.system() != "Windows" else 10
|
||||||
if platform.system() == "Windows":
|
|
||||||
timeout: float = 10
|
|
||||||
|
|
||||||
if debug_mode:
|
|
||||||
timeout = 999
|
|
||||||
print(f'IN debug_mode, setting large timeout={timeout!r}..')
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
full = list(range(1000))
|
full = list(range(1000))
|
||||||
an: tractor.ActorNursery|None = None
|
|
||||||
num_tasks: int = 10
|
|
||||||
|
|
||||||
async def get_sub_and_pull(taskname: str):
|
async def get_sub_and_pull(taskname: str):
|
||||||
|
|
||||||
nonlocal an
|
|
||||||
|
|
||||||
stream: tractor.MsgStream
|
|
||||||
async with (
|
async with (
|
||||||
maybe_open_stream(taskname) as (
|
maybe_open_stream(taskname) as stream,
|
||||||
an,
|
|
||||||
stream,
|
|
||||||
),
|
|
||||||
):
|
):
|
||||||
if '0' in taskname:
|
if '0' in taskname:
|
||||||
assert isinstance(stream, tractor.MsgStream)
|
assert isinstance(stream, tractor.MsgStream)
|
||||||
|
|
@ -248,159 +158,24 @@ def test_open_local_sub_to_stream(
|
||||||
|
|
||||||
first = await stream.receive()
|
first = await stream.receive()
|
||||||
print(f'{taskname} started with value {first}')
|
print(f'{taskname} started with value {first}')
|
||||||
seq: list[int] = []
|
seq = []
|
||||||
async for msg in stream:
|
async for msg in stream:
|
||||||
seq.append(msg)
|
seq.append(msg)
|
||||||
|
|
||||||
assert set(seq).issubset(set(full))
|
assert set(seq).issubset(set(full))
|
||||||
|
|
||||||
# end of @acm block
|
|
||||||
print(f'{taskname} finished')
|
print(f'{taskname} finished')
|
||||||
|
|
||||||
root: tractor.Actor
|
with trio.fail_after(timeout):
|
||||||
with trio.fail_after(timeout) as cs:
|
|
||||||
# TODO: turns out this isn't multi-task entrant XD
|
# TODO: turns out this isn't multi-task entrant XD
|
||||||
# We probably need an indepotent entry semantic?
|
# We probably need an indepotent entry semantic?
|
||||||
async with tractor.open_root_actor(
|
async with tractor.open_root_actor():
|
||||||
debug_mode=debug_mode,
|
|
||||||
# maybe_enable_greenback=True,
|
|
||||||
#
|
|
||||||
# ^TODO? doesn't seem to mk breakpoint() usage work
|
|
||||||
# bc each bg task needs to open a portal??
|
|
||||||
# - [ ] we should consider making this part of
|
|
||||||
# our taskman defaults?
|
|
||||||
# |_see https://github.com/goodboy/tractor/pull/363
|
|
||||||
#
|
|
||||||
) as root:
|
|
||||||
assert root.is_registrar
|
|
||||||
|
|
||||||
async with (
|
async with (
|
||||||
trio.open_nursery() as tn,
|
trio.open_nursery() as nurse,
|
||||||
):
|
):
|
||||||
for i in range(num_tasks):
|
for i in range(10):
|
||||||
tn.start_soon(
|
nurse.start_soon(get_sub_and_pull, f'task_{i}')
|
||||||
get_sub_and_pull,
|
|
||||||
f'task_{i}',
|
|
||||||
)
|
|
||||||
await trio.sleep(0.001)
|
await trio.sleep(0.001)
|
||||||
|
|
||||||
print('all consumer tasks finished!')
|
print('all consumer tasks finished')
|
||||||
|
|
||||||
# ?XXX, ensure actor-nursery is shutdown or we might
|
|
||||||
# hang here due to a minor task deadlock/race-condition?
|
|
||||||
#
|
|
||||||
# - seems that all we need is a checkpoint to ensure
|
|
||||||
# the last suspended task, which is inside
|
|
||||||
# `.maybe_open_context()`, can do the
|
|
||||||
# `Portal.cancel_actor()` call?
|
|
||||||
#
|
|
||||||
# - if that bg task isn't resumed, then this blocks
|
|
||||||
# timeout might hit before that?
|
|
||||||
#
|
|
||||||
if root.ipc_server.has_peers():
|
|
||||||
await trio.lowlevel.checkpoint()
|
|
||||||
|
|
||||||
# alt approach, cancel the entire `an`
|
|
||||||
# await tractor.pause()
|
|
||||||
# await an.cancel()
|
|
||||||
|
|
||||||
# end of runtime scope
|
|
||||||
print('root actor terminated.')
|
|
||||||
|
|
||||||
if cs.cancelled_caught:
|
|
||||||
pytest.fail(
|
|
||||||
'Should NOT time out in `open_root_actor()` ?'
|
|
||||||
)
|
|
||||||
|
|
||||||
print('exiting main.')
|
|
||||||
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
|
||||||
async def cancel_outer_cs(
|
|
||||||
cs: trio.CancelScope|None = None,
|
|
||||||
delay: float = 0,
|
|
||||||
):
|
|
||||||
# on first task delay this enough to block
|
|
||||||
# the 2nd task but then cancel it mid sleep
|
|
||||||
# so that the tn.start() inside the key-err handler block
|
|
||||||
# is cancelled and would previously corrupt the
|
|
||||||
# mutext state.
|
|
||||||
log.info(f'task entering sleep({delay})')
|
|
||||||
await trio.sleep(delay)
|
|
||||||
if cs:
|
|
||||||
log.info('task calling cs.cancel()')
|
|
||||||
cs.cancel()
|
|
||||||
trio.lowlevel.checkpoint()
|
|
||||||
yield
|
|
||||||
await trio.sleep_forever()
|
|
||||||
|
|
||||||
|
|
||||||
def test_lock_not_corrupted_on_fast_cancel(
|
|
||||||
debug_mode: bool,
|
|
||||||
loglevel: str,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Verify that if the caching-task (the first to enter
|
|
||||||
`maybe_open_context()`) is cancelled mid-cache-miss, the embedded
|
|
||||||
mutex can never be left in a corrupted state.
|
|
||||||
|
|
||||||
That is, the lock is always eventually released ensuring a peer
|
|
||||||
(cache-hitting) task will never,
|
|
||||||
|
|
||||||
- be left to inf-block/hang on the `lock.acquire()`.
|
|
||||||
- try to release the lock when still owned by the caching-task
|
|
||||||
due to it having erronously exited without calling
|
|
||||||
`lock.release()`.
|
|
||||||
|
|
||||||
|
|
||||||
'''
|
|
||||||
delay: float = 1.
|
|
||||||
|
|
||||||
async def use_moc(
|
|
||||||
cs: trio.CancelScope|None,
|
|
||||||
delay: float,
|
|
||||||
):
|
|
||||||
log.info('task entering moc')
|
|
||||||
async with maybe_open_context(
|
|
||||||
cancel_outer_cs,
|
|
||||||
kwargs={
|
|
||||||
'cs': cs,
|
|
||||||
'delay': delay,
|
|
||||||
},
|
|
||||||
) as (cache_hit, _null):
|
|
||||||
if cache_hit:
|
|
||||||
log.info('2nd task entered')
|
|
||||||
else:
|
|
||||||
log.info('1st task entered')
|
|
||||||
|
|
||||||
await trio.sleep_forever()
|
|
||||||
|
|
||||||
async def main():
|
|
||||||
with trio.fail_after(delay + 2):
|
|
||||||
async with (
|
|
||||||
tractor.open_root_actor(
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
loglevel=loglevel,
|
|
||||||
),
|
|
||||||
trio.open_nursery() as tn,
|
|
||||||
):
|
|
||||||
get_console_log('info')
|
|
||||||
log.info('yo starting')
|
|
||||||
cs = tn.cancel_scope
|
|
||||||
tn.start_soon(
|
|
||||||
use_moc,
|
|
||||||
cs,
|
|
||||||
delay,
|
|
||||||
name='child',
|
|
||||||
)
|
|
||||||
with trio.CancelScope() as rent_cs:
|
|
||||||
await use_moc(
|
|
||||||
cs=rent_cs,
|
|
||||||
delay=delay,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
trio.run(main)
|
trio.run(main)
|
||||||
|
|
|
||||||
|
|
@ -1,211 +0,0 @@
|
||||||
import time
|
|
||||||
|
|
||||||
import trio
|
|
||||||
import pytest
|
|
||||||
|
|
||||||
import tractor
|
|
||||||
from tractor.ipc._ringbuf import (
|
|
||||||
open_ringbuf,
|
|
||||||
RBToken,
|
|
||||||
RingBuffSender,
|
|
||||||
RingBuffReceiver
|
|
||||||
)
|
|
||||||
from tractor._testing.samples import (
|
|
||||||
generate_sample_messages,
|
|
||||||
)
|
|
||||||
|
|
||||||
# in case you don't want to melt your cores, uncomment dis!
|
|
||||||
pytestmark = pytest.mark.skip
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def child_read_shm(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
msg_amount: int,
|
|
||||||
token: RBToken,
|
|
||||||
total_bytes: int,
|
|
||||||
) -> None:
|
|
||||||
recvd_bytes = 0
|
|
||||||
await ctx.started()
|
|
||||||
start_ts = time.time()
|
|
||||||
async with RingBuffReceiver(token) as receiver:
|
|
||||||
while recvd_bytes < total_bytes:
|
|
||||||
msg = await receiver.receive_some()
|
|
||||||
recvd_bytes += len(msg)
|
|
||||||
|
|
||||||
# make sure we dont hold any memoryviews
|
|
||||||
# before the ctx manager aclose()
|
|
||||||
msg = None
|
|
||||||
|
|
||||||
end_ts = time.time()
|
|
||||||
elapsed = end_ts - start_ts
|
|
||||||
elapsed_ms = int(elapsed * 1000)
|
|
||||||
|
|
||||||
print(f'\n\telapsed ms: {elapsed_ms}')
|
|
||||||
print(f'\tmsg/sec: {int(msg_amount / elapsed):,}')
|
|
||||||
print(f'\tbytes/sec: {int(recvd_bytes / elapsed):,}')
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def child_write_shm(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
msg_amount: int,
|
|
||||||
rand_min: int,
|
|
||||||
rand_max: int,
|
|
||||||
token: RBToken,
|
|
||||||
) -> None:
|
|
||||||
msgs, total_bytes = generate_sample_messages(
|
|
||||||
msg_amount,
|
|
||||||
rand_min=rand_min,
|
|
||||||
rand_max=rand_max,
|
|
||||||
)
|
|
||||||
await ctx.started(total_bytes)
|
|
||||||
async with RingBuffSender(token) as sender:
|
|
||||||
for msg in msgs:
|
|
||||||
await sender.send_all(msg)
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'msg_amount,rand_min,rand_max,buf_size',
|
|
||||||
[
|
|
||||||
# simple case, fixed payloads, large buffer
|
|
||||||
(100_000, 0, 0, 10 * 1024),
|
|
||||||
|
|
||||||
# guaranteed wrap around on every write
|
|
||||||
(100, 10 * 1024, 20 * 1024, 10 * 1024),
|
|
||||||
|
|
||||||
# large payload size, but large buffer
|
|
||||||
(10_000, 256 * 1024, 512 * 1024, 10 * 1024 * 1024)
|
|
||||||
],
|
|
||||||
ids=[
|
|
||||||
'fixed_payloads_large_buffer',
|
|
||||||
'wrap_around_every_write',
|
|
||||||
'large_payloads_large_buffer',
|
|
||||||
]
|
|
||||||
)
|
|
||||||
def test_ringbuf(
|
|
||||||
msg_amount: int,
|
|
||||||
rand_min: int,
|
|
||||||
rand_max: int,
|
|
||||||
buf_size: int
|
|
||||||
):
|
|
||||||
async def main():
|
|
||||||
with open_ringbuf(
|
|
||||||
'test_ringbuf',
|
|
||||||
buf_size=buf_size
|
|
||||||
) as token:
|
|
||||||
proc_kwargs = {
|
|
||||||
'pass_fds': (token.write_eventfd, token.wrap_eventfd)
|
|
||||||
}
|
|
||||||
|
|
||||||
common_kwargs = {
|
|
||||||
'msg_amount': msg_amount,
|
|
||||||
'token': token,
|
|
||||||
}
|
|
||||||
async with tractor.open_nursery() as an:
|
|
||||||
send_p = await an.start_actor(
|
|
||||||
'ring_sender',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
proc_kwargs=proc_kwargs
|
|
||||||
)
|
|
||||||
recv_p = await an.start_actor(
|
|
||||||
'ring_receiver',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
proc_kwargs=proc_kwargs
|
|
||||||
)
|
|
||||||
async with (
|
|
||||||
send_p.open_context(
|
|
||||||
child_write_shm,
|
|
||||||
rand_min=rand_min,
|
|
||||||
rand_max=rand_max,
|
|
||||||
**common_kwargs
|
|
||||||
) as (sctx, total_bytes),
|
|
||||||
recv_p.open_context(
|
|
||||||
child_read_shm,
|
|
||||||
**common_kwargs,
|
|
||||||
total_bytes=total_bytes,
|
|
||||||
) as (sctx, _sent),
|
|
||||||
):
|
|
||||||
await recv_p.result()
|
|
||||||
|
|
||||||
await send_p.cancel_actor()
|
|
||||||
await recv_p.cancel_actor()
|
|
||||||
|
|
||||||
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def child_blocked_receiver(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
token: RBToken
|
|
||||||
):
|
|
||||||
async with RingBuffReceiver(token) as receiver:
|
|
||||||
await ctx.started()
|
|
||||||
await receiver.receive_some()
|
|
||||||
|
|
||||||
|
|
||||||
def test_ring_reader_cancel():
|
|
||||||
async def main():
|
|
||||||
with open_ringbuf('test_ring_cancel_reader') as token:
|
|
||||||
async with (
|
|
||||||
tractor.open_nursery() as an,
|
|
||||||
RingBuffSender(token) as _sender,
|
|
||||||
):
|
|
||||||
recv_p = await an.start_actor(
|
|
||||||
'ring_blocked_receiver',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
proc_kwargs={
|
|
||||||
'pass_fds': (token.write_eventfd, token.wrap_eventfd)
|
|
||||||
}
|
|
||||||
)
|
|
||||||
async with (
|
|
||||||
recv_p.open_context(
|
|
||||||
child_blocked_receiver,
|
|
||||||
token=token
|
|
||||||
) as (sctx, _sent),
|
|
||||||
):
|
|
||||||
await trio.sleep(1)
|
|
||||||
await an.cancel()
|
|
||||||
|
|
||||||
|
|
||||||
with pytest.raises(tractor._exceptions.ContextCancelled):
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def child_blocked_sender(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
token: RBToken
|
|
||||||
):
|
|
||||||
async with RingBuffSender(token) as sender:
|
|
||||||
await ctx.started()
|
|
||||||
await sender.send_all(b'this will wrap')
|
|
||||||
|
|
||||||
|
|
||||||
def test_ring_sender_cancel():
|
|
||||||
async def main():
|
|
||||||
with open_ringbuf(
|
|
||||||
'test_ring_cancel_sender',
|
|
||||||
buf_size=1
|
|
||||||
) as token:
|
|
||||||
async with tractor.open_nursery() as an:
|
|
||||||
recv_p = await an.start_actor(
|
|
||||||
'ring_blocked_sender',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
proc_kwargs={
|
|
||||||
'pass_fds': (token.write_eventfd, token.wrap_eventfd)
|
|
||||||
}
|
|
||||||
)
|
|
||||||
async with (
|
|
||||||
recv_p.open_context(
|
|
||||||
child_blocked_sender,
|
|
||||||
token=token
|
|
||||||
) as (sctx, _sent),
|
|
||||||
):
|
|
||||||
await trio.sleep(1)
|
|
||||||
await an.cancel()
|
|
||||||
|
|
||||||
|
|
||||||
with pytest.raises(tractor._exceptions.ContextCancelled):
|
|
||||||
trio.run(main)
|
|
||||||
|
|
@ -39,7 +39,7 @@ def test_infected_root_actor(
|
||||||
|
|
||||||
'''
|
'''
|
||||||
async def _trio_main():
|
async def _trio_main():
|
||||||
with trio.fail_after(2 if not debug_mode else 999):
|
with trio.fail_after(2):
|
||||||
first: str
|
first: str
|
||||||
chan: to_asyncio.LinkedTaskChannel
|
chan: to_asyncio.LinkedTaskChannel
|
||||||
async with (
|
async with (
|
||||||
|
|
@ -59,11 +59,7 @@ def test_infected_root_actor(
|
||||||
assert out == i
|
assert out == i
|
||||||
print(f'asyncio echoing {i}')
|
print(f'asyncio echoing {i}')
|
||||||
|
|
||||||
if (
|
if raise_error_mid_stream and i == 500:
|
||||||
raise_error_mid_stream
|
|
||||||
and
|
|
||||||
i == 500
|
|
||||||
):
|
|
||||||
raise raise_error_mid_stream
|
raise raise_error_mid_stream
|
||||||
|
|
||||||
if out is None:
|
if out is None:
|
||||||
|
|
@ -147,7 +143,8 @@ def test_trio_prestarted_task_bubbles(
|
||||||
await trio.sleep_forever()
|
await trio.sleep_forever()
|
||||||
|
|
||||||
async def _trio_main():
|
async def _trio_main():
|
||||||
with trio.fail_after(2 if not debug_mode else 999):
|
# with trio.fail_after(2):
|
||||||
|
with trio.fail_after(999):
|
||||||
first: str
|
first: str
|
||||||
chan: to_asyncio.LinkedTaskChannel
|
chan: to_asyncio.LinkedTaskChannel
|
||||||
aio_ev = asyncio.Event()
|
aio_ev = asyncio.Event()
|
||||||
|
|
@ -216,25 +213,32 @@ def test_trio_prestarted_task_bubbles(
|
||||||
):
|
):
|
||||||
aio_ev.set()
|
aio_ev.set()
|
||||||
|
|
||||||
|
with pytest.raises(
|
||||||
|
expected_exception=ExceptionGroup,
|
||||||
|
) as excinfo:
|
||||||
|
tractor.to_asyncio.run_as_asyncio_guest(
|
||||||
|
trio_main=_trio_main,
|
||||||
|
)
|
||||||
|
|
||||||
|
eg = excinfo.value
|
||||||
|
rte_eg, rest_eg = eg.split(RuntimeError)
|
||||||
|
|
||||||
# ensure the trio-task's error bubbled despite the aio-side
|
# ensure the trio-task's error bubbled despite the aio-side
|
||||||
# having (maybe) errored first.
|
# having (maybe) errored first.
|
||||||
if aio_err_trigger in (
|
if aio_err_trigger in (
|
||||||
'after_trio_task_starts',
|
'after_trio_task_starts',
|
||||||
'after_start_point',
|
'after_start_point',
|
||||||
):
|
):
|
||||||
patt: str = 'trio-side'
|
assert len(errs := rest_eg.exceptions) == 1
|
||||||
expect_exc = TypeError
|
typerr = errs[0]
|
||||||
|
assert (
|
||||||
|
type(typerr) is TypeError
|
||||||
|
and
|
||||||
|
'trio-side' in typerr.args
|
||||||
|
)
|
||||||
|
|
||||||
# when aio errors BEFORE (last) trio task is scheduled, we should
|
# when aio errors BEFORE (last) trio task is scheduled, we should
|
||||||
# never see anythinb but the aio-side.
|
# never see anythinb but the aio-side.
|
||||||
else:
|
else:
|
||||||
patt: str = 'asyncio-side'
|
assert len(rtes := rte_eg.exceptions) == 1
|
||||||
expect_exc = RuntimeError
|
assert 'asyncio-side' in rtes[0].args[0]
|
||||||
|
|
||||||
with pytest.raises(expect_exc) as excinfo:
|
|
||||||
tractor.to_asyncio.run_as_asyncio_guest(
|
|
||||||
trio_main=_trio_main,
|
|
||||||
)
|
|
||||||
|
|
||||||
caught_exc = excinfo.value
|
|
||||||
assert patt in caught_exc.args
|
|
||||||
|
|
|
||||||
|
|
@ -1,108 +0,0 @@
|
||||||
'''
|
|
||||||
Runtime boot/init sanity.
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
import trio
|
|
||||||
|
|
||||||
import tractor
|
|
||||||
from tractor._exceptions import RuntimeFailure
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def open_new_root_in_sub(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
) -> None:
|
|
||||||
|
|
||||||
async with tractor.open_root_actor():
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'open_root_in',
|
|
||||||
['root', 'sub'],
|
|
||||||
ids='open_2nd_root_in={}'.format,
|
|
||||||
)
|
|
||||||
def test_only_one_root_actor(
|
|
||||||
open_root_in: str,
|
|
||||||
reg_addr: tuple,
|
|
||||||
debug_mode: bool
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Verify we specially fail whenever more then one root actor
|
|
||||||
is attempted to be opened within an already opened tree.
|
|
||||||
|
|
||||||
'''
|
|
||||||
async def main():
|
|
||||||
async with tractor.open_nursery() as an:
|
|
||||||
|
|
||||||
if open_root_in == 'root':
|
|
||||||
async with tractor.open_root_actor(
|
|
||||||
registry_addrs=[reg_addr],
|
|
||||||
):
|
|
||||||
pass
|
|
||||||
|
|
||||||
ptl: tractor.Portal = await an.start_actor(
|
|
||||||
name='bad_rooty_boi',
|
|
||||||
enable_modules=[__name__],
|
|
||||||
)
|
|
||||||
|
|
||||||
async with ptl.open_context(
|
|
||||||
open_new_root_in_sub,
|
|
||||||
) as (ctx, first):
|
|
||||||
pass
|
|
||||||
|
|
||||||
if open_root_in == 'root':
|
|
||||||
with pytest.raises(
|
|
||||||
RuntimeFailure
|
|
||||||
) as excinfo:
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
else:
|
|
||||||
with pytest.raises(
|
|
||||||
tractor.RemoteActorError,
|
|
||||||
) as excinfo:
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
assert excinfo.value.boxed_type is RuntimeFailure
|
|
||||||
|
|
||||||
|
|
||||||
def test_implicit_root_via_first_nursery(
|
|
||||||
reg_addr: tuple,
|
|
||||||
debug_mode: bool
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
The first `ActorNursery` open should implicitly call
|
|
||||||
`_root.open_root_actor()`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
async def main():
|
|
||||||
async with tractor.open_nursery() as an:
|
|
||||||
assert an._implicit_runtime_started
|
|
||||||
assert tractor.current_actor().aid.name == 'root'
|
|
||||||
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
|
|
||||||
def test_runtime_vars_unset(
|
|
||||||
reg_addr: tuple,
|
|
||||||
debug_mode: bool
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Ensure any `._state._runtime_vars` are restored to default values
|
|
||||||
after the root actor-runtime exits!
|
|
||||||
|
|
||||||
'''
|
|
||||||
assert not tractor._state._runtime_vars['_debug_mode']
|
|
||||||
async def main():
|
|
||||||
assert not tractor._state._runtime_vars['_debug_mode']
|
|
||||||
async with tractor.open_nursery(
|
|
||||||
debug_mode=True,
|
|
||||||
):
|
|
||||||
assert tractor._state._runtime_vars['_debug_mode']
|
|
||||||
|
|
||||||
# after runtime closure, should be reverted!
|
|
||||||
assert not tractor._state._runtime_vars['_debug_mode']
|
|
||||||
|
|
||||||
trio.run(main)
|
|
||||||
|
|
@ -8,7 +8,7 @@ import uuid
|
||||||
import pytest
|
import pytest
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
from tractor.ipc._shm import (
|
from tractor._shm import (
|
||||||
open_shm_list,
|
open_shm_list,
|
||||||
attach_shm_list,
|
attach_shm_list,
|
||||||
)
|
)
|
||||||
|
|
|
||||||
|
|
@ -2,7 +2,6 @@
|
||||||
Spawning basics
|
Spawning basics
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from functools import partial
|
|
||||||
from typing import (
|
from typing import (
|
||||||
Any,
|
Any,
|
||||||
)
|
)
|
||||||
|
|
@ -13,99 +12,74 @@ import tractor
|
||||||
|
|
||||||
from tractor._testing import tractor_test
|
from tractor._testing import tractor_test
|
||||||
|
|
||||||
data_to_pass_down = {
|
data_to_pass_down = {'doggy': 10, 'kitty': 4}
|
||||||
'doggy': 10,
|
|
||||||
'kitty': 4,
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
async def spawn(
|
async def spawn(
|
||||||
should_be_root: bool,
|
is_arbiter: bool,
|
||||||
data: dict,
|
data: dict,
|
||||||
reg_addr: tuple[str, int],
|
reg_addr: tuple[str, int],
|
||||||
|
|
||||||
debug_mode: bool = False,
|
|
||||||
):
|
):
|
||||||
|
namespaces = [__name__]
|
||||||
|
|
||||||
await trio.sleep(0.1)
|
await trio.sleep(0.1)
|
||||||
actor = tractor.current_actor(err_on_no_runtime=False)
|
|
||||||
|
|
||||||
if should_be_root:
|
async with tractor.open_root_actor(
|
||||||
assert actor is None # no runtime yet
|
|
||||||
async with (
|
|
||||||
tractor.open_root_actor(
|
|
||||||
arbiter_addr=reg_addr,
|
arbiter_addr=reg_addr,
|
||||||
),
|
|
||||||
tractor.open_nursery() as an,
|
|
||||||
):
|
):
|
||||||
# now runtime exists
|
actor = tractor.current_actor()
|
||||||
actor: tractor.Actor = tractor.current_actor()
|
assert actor.is_arbiter == is_arbiter
|
||||||
assert actor.is_arbiter == should_be_root
|
data = data_to_pass_down
|
||||||
|
|
||||||
# spawns subproc here
|
if actor.is_arbiter:
|
||||||
portal: tractor.Portal = await an.run_in_actor(
|
async with tractor.open_nursery() as nursery:
|
||||||
fn=spawn,
|
|
||||||
|
|
||||||
# spawning args
|
# forks here
|
||||||
|
portal = await nursery.run_in_actor(
|
||||||
|
spawn,
|
||||||
|
is_arbiter=False,
|
||||||
name='sub-actor',
|
name='sub-actor',
|
||||||
enable_modules=[__name__],
|
data=data,
|
||||||
|
|
||||||
# passed to a subactor-recursive RPC invoke
|
|
||||||
# of this same `spawn()` fn.
|
|
||||||
should_be_root=False,
|
|
||||||
data=data_to_pass_down,
|
|
||||||
reg_addr=reg_addr,
|
reg_addr=reg_addr,
|
||||||
|
enable_modules=namespaces,
|
||||||
)
|
)
|
||||||
|
|
||||||
assert len(an._children) == 1
|
assert len(nursery._children) == 1
|
||||||
assert (
|
assert portal.channel.uid in tractor.current_actor()._peers
|
||||||
portal.channel.uid
|
# be sure we can still get the result
|
||||||
in
|
|
||||||
tractor.current_actor().ipc_server._peers
|
|
||||||
)
|
|
||||||
|
|
||||||
# get result from child subactor
|
|
||||||
result = await portal.result()
|
result = await portal.result()
|
||||||
assert result == 10
|
assert result == 10
|
||||||
return result
|
return result
|
||||||
else:
|
else:
|
||||||
assert actor.is_arbiter == should_be_root
|
|
||||||
return 10
|
return 10
|
||||||
|
|
||||||
|
|
||||||
def test_run_in_actor_same_func_in_child(
|
def test_local_arbiter_subactor_global_state(
|
||||||
reg_addr: tuple,
|
reg_addr,
|
||||||
debug_mode: bool,
|
|
||||||
):
|
):
|
||||||
result = trio.run(
|
result = trio.run(
|
||||||
partial(
|
|
||||||
spawn,
|
spawn,
|
||||||
should_be_root=True,
|
True,
|
||||||
data=data_to_pass_down,
|
data_to_pass_down,
|
||||||
reg_addr=reg_addr,
|
reg_addr,
|
||||||
debug_mode=debug_mode,
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
assert result == 10
|
assert result == 10
|
||||||
|
|
||||||
|
|
||||||
async def movie_theatre_question():
|
async def movie_theatre_question():
|
||||||
'''
|
"""A question asked in a dark theatre, in a tangent
|
||||||
A question asked in a dark theatre, in a tangent
|
|
||||||
(errr, I mean different) process.
|
(errr, I mean different) process.
|
||||||
|
"""
|
||||||
'''
|
|
||||||
return 'have you ever seen a portal?'
|
return 'have you ever seen a portal?'
|
||||||
|
|
||||||
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_movie_theatre_convo(start_method):
|
async def test_movie_theatre_convo(start_method):
|
||||||
'''
|
"""The main ``tractor`` routine.
|
||||||
The main ``tractor`` routine.
|
"""
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
'''
|
portal = await n.start_actor(
|
||||||
async with tractor.open_nursery(debug_mode=True) as an:
|
|
||||||
|
|
||||||
portal = await an.start_actor(
|
|
||||||
'frank',
|
'frank',
|
||||||
# enable the actor to run funcs from this current module
|
# enable the actor to run funcs from this current module
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
|
|
@ -144,8 +118,8 @@ async def test_most_beautiful_word(
|
||||||
with trio.fail_after(1):
|
with trio.fail_after(1):
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
debug_mode=debug_mode,
|
debug_mode=debug_mode,
|
||||||
) as an:
|
) as n:
|
||||||
portal = await an.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
cellar_door,
|
cellar_door,
|
||||||
return_value=return_value,
|
return_value=return_value,
|
||||||
name='some_linguist',
|
name='some_linguist',
|
||||||
|
|
|
||||||
|
|
@ -2,9 +2,7 @@
|
||||||
Broadcast channels for fan-out to local tasks.
|
Broadcast channels for fan-out to local tasks.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from contextlib import (
|
from contextlib import asynccontextmanager
|
||||||
asynccontextmanager as acm,
|
|
||||||
)
|
|
||||||
from functools import partial
|
from functools import partial
|
||||||
from itertools import cycle
|
from itertools import cycle
|
||||||
import time
|
import time
|
||||||
|
|
@ -17,7 +15,6 @@ import tractor
|
||||||
from tractor.trionics import (
|
from tractor.trionics import (
|
||||||
broadcast_receiver,
|
broadcast_receiver,
|
||||||
Lagged,
|
Lagged,
|
||||||
collapse_eg,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -65,7 +62,7 @@ async def ensure_sequence(
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|
||||||
@acm
|
@asynccontextmanager
|
||||||
async def open_sequence_streamer(
|
async def open_sequence_streamer(
|
||||||
|
|
||||||
sequence: list[int],
|
sequence: list[int],
|
||||||
|
|
@ -77,9 +74,9 @@ async def open_sequence_streamer(
|
||||||
async with tractor.open_nursery(
|
async with tractor.open_nursery(
|
||||||
arbiter_addr=reg_addr,
|
arbiter_addr=reg_addr,
|
||||||
start_method=start_method,
|
start_method=start_method,
|
||||||
) as an:
|
) as tn:
|
||||||
|
|
||||||
portal = await an.start_actor(
|
portal = await tn.start_actor(
|
||||||
'sequence_echoer',
|
'sequence_echoer',
|
||||||
enable_modules=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
@ -158,12 +155,9 @@ def test_consumer_and_parent_maybe_lag(
|
||||||
) as stream:
|
) as stream:
|
||||||
|
|
||||||
try:
|
try:
|
||||||
async with (
|
async with trio.open_nursery() as n:
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as tn,
|
|
||||||
):
|
|
||||||
|
|
||||||
tn.start_soon(
|
n.start_soon(
|
||||||
ensure_sequence,
|
ensure_sequence,
|
||||||
stream,
|
stream,
|
||||||
sequence.copy(),
|
sequence.copy(),
|
||||||
|
|
@ -236,8 +230,8 @@ def test_faster_task_to_recv_is_cancelled_by_slower(
|
||||||
|
|
||||||
) as stream:
|
) as stream:
|
||||||
|
|
||||||
async with trio.open_nursery() as tn:
|
async with trio.open_nursery() as n:
|
||||||
tn.start_soon(
|
n.start_soon(
|
||||||
ensure_sequence,
|
ensure_sequence,
|
||||||
stream,
|
stream,
|
||||||
sequence.copy(),
|
sequence.copy(),
|
||||||
|
|
@ -259,7 +253,7 @@ def test_faster_task_to_recv_is_cancelled_by_slower(
|
||||||
continue
|
continue
|
||||||
|
|
||||||
print('cancelling faster subtask')
|
print('cancelling faster subtask')
|
||||||
tn.cancel_scope.cancel()
|
n.cancel_scope.cancel()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
value = await stream.receive()
|
value = await stream.receive()
|
||||||
|
|
@ -377,13 +371,13 @@ def test_ensure_slow_consumers_lag_out(
|
||||||
f'on {lags}:{value}')
|
f'on {lags}:{value}')
|
||||||
return
|
return
|
||||||
|
|
||||||
async with trio.open_nursery() as tn:
|
async with trio.open_nursery() as nursery:
|
||||||
|
|
||||||
for i in range(1, num_laggers):
|
for i in range(1, num_laggers):
|
||||||
|
|
||||||
task_name = f'sub_{i}'
|
task_name = f'sub_{i}'
|
||||||
laggers[task_name] = 0
|
laggers[task_name] = 0
|
||||||
tn.start_soon(
|
nursery.start_soon(
|
||||||
partial(
|
partial(
|
||||||
sub_and_print,
|
sub_and_print,
|
||||||
delay=i*0.001,
|
delay=i*0.001,
|
||||||
|
|
@ -503,7 +497,6 @@ def test_no_raise_on_lag():
|
||||||
# internals when the no raise flag is set.
|
# internals when the no raise flag is set.
|
||||||
loglevel='warning',
|
loglevel='warning',
|
||||||
),
|
),
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as n,
|
trio.open_nursery() as n,
|
||||||
):
|
):
|
||||||
n.start_soon(slow)
|
n.start_soon(slow)
|
||||||
|
|
|
||||||
|
|
@ -6,18 +6,10 @@ want to see changed.
|
||||||
from contextlib import (
|
from contextlib import (
|
||||||
asynccontextmanager as acm,
|
asynccontextmanager as acm,
|
||||||
)
|
)
|
||||||
from types import ModuleType
|
|
||||||
|
|
||||||
from functools import partial
|
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from _pytest import pathlib
|
|
||||||
from tractor.trionics import collapse_eg
|
|
||||||
import trio
|
import trio
|
||||||
from trio import TaskStatus
|
from trio import TaskStatus
|
||||||
from tractor._testing import (
|
|
||||||
examples_dir,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
|
|
@ -72,7 +64,6 @@ def test_stashed_child_nursery(use_start_soon):
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
async with (
|
async with (
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as pn,
|
trio.open_nursery() as pn,
|
||||||
):
|
):
|
||||||
cn = await pn.start(mk_child_nursery)
|
cn = await pn.start(mk_child_nursery)
|
||||||
|
|
@ -110,24 +101,68 @@ def test_stashed_child_nursery(use_start_soon):
|
||||||
def test_acm_embedded_nursery_propagates_enter_err(
|
def test_acm_embedded_nursery_propagates_enter_err(
|
||||||
canc_from_finally: bool,
|
canc_from_finally: bool,
|
||||||
unmask_from_canc: bool,
|
unmask_from_canc: bool,
|
||||||
debug_mode: bool,
|
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Demo how a masking `trio.Cancelled` could be handled by unmasking
|
Demo how a masking `trio.Cancelled` could be handled by unmasking from the
|
||||||
from the `.__context__` field when a user (by accident) re-raises
|
`.__context__` field when a user (by accident) re-raises from a `finally:`.
|
||||||
from a `finally:`.
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def maybe_raise_from_masking_exc(
|
||||||
|
tn: trio.Nursery,
|
||||||
|
unmask_from: BaseException|None = trio.Cancelled
|
||||||
|
|
||||||
|
# TODO, maybe offer a collection?
|
||||||
|
# unmask_from: set[BaseException] = {
|
||||||
|
# trio.Cancelled,
|
||||||
|
# },
|
||||||
|
):
|
||||||
|
if not unmask_from:
|
||||||
|
yield
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
yield
|
||||||
|
except* unmask_from as be_eg:
|
||||||
|
|
||||||
|
# TODO, if we offer `unmask_from: set`
|
||||||
|
# for masker_exc_type in unmask_from:
|
||||||
|
|
||||||
|
matches, rest = be_eg.split(unmask_from)
|
||||||
|
if not matches:
|
||||||
|
raise
|
||||||
|
|
||||||
|
for exc_match in be_eg.exceptions:
|
||||||
|
if (
|
||||||
|
(exc_ctx := exc_match.__context__)
|
||||||
|
and
|
||||||
|
type(exc_ctx) not in {
|
||||||
|
# trio.Cancelled, # always by default?
|
||||||
|
unmask_from,
|
||||||
|
}
|
||||||
|
):
|
||||||
|
exc_ctx.add_note(
|
||||||
|
f'\n'
|
||||||
|
f'WARNING: the above error was masked by a {unmask_from!r} !?!\n'
|
||||||
|
f'Are you always cancelling? Say from a `finally:` ?\n\n'
|
||||||
|
|
||||||
|
f'{tn!r}'
|
||||||
|
)
|
||||||
|
raise exc_ctx from exc_match
|
||||||
|
|
||||||
|
|
||||||
@acm
|
@acm
|
||||||
async def wraps_tn_that_always_cancels():
|
async def wraps_tn_that_always_cancels():
|
||||||
async with (
|
async with (
|
||||||
trio.open_nursery() as tn,
|
trio.open_nursery() as tn,
|
||||||
tractor.trionics.maybe_raise_from_masking_exc(
|
maybe_raise_from_masking_exc(
|
||||||
|
tn=tn,
|
||||||
unmask_from=(
|
unmask_from=(
|
||||||
(trio.Cancelled,) if unmask_from_canc
|
trio.Cancelled
|
||||||
else ()
|
if unmask_from_canc
|
||||||
|
else None
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
|
|
@ -139,9 +174,7 @@ def test_acm_embedded_nursery_propagates_enter_err(
|
||||||
await trio.lowlevel.checkpoint()
|
await trio.lowlevel.checkpoint()
|
||||||
|
|
||||||
async def _main():
|
async def _main():
|
||||||
with tractor.devx.maybe_open_crash_handler(
|
with tractor.devx.open_crash_handler() as bxerr:
|
||||||
pdb=debug_mode,
|
|
||||||
) as bxerr:
|
|
||||||
assert not bxerr.value
|
assert not bxerr.value
|
||||||
|
|
||||||
async with (
|
async with (
|
||||||
|
|
@ -150,7 +183,6 @@ def test_acm_embedded_nursery_propagates_enter_err(
|
||||||
assert not tn.cancel_scope.cancel_called
|
assert not tn.cancel_scope.cancel_called
|
||||||
assert 0
|
assert 0
|
||||||
|
|
||||||
if debug_mode:
|
|
||||||
assert (
|
assert (
|
||||||
(err := bxerr.value)
|
(err := bxerr.value)
|
||||||
and
|
and
|
||||||
|
|
@ -164,139 +196,3 @@ def test_acm_embedded_nursery_propagates_enter_err(
|
||||||
assert_eg, rest_eg = eg.split(AssertionError)
|
assert_eg, rest_eg = eg.split(AssertionError)
|
||||||
|
|
||||||
assert len(assert_eg.exceptions) == 1
|
assert len(assert_eg.exceptions) == 1
|
||||||
|
|
||||||
|
|
||||||
def test_gatherctxs_with_memchan_breaks_multicancelled(
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Demo how a using an `async with sndchan` inside
|
|
||||||
a `.trionics.gather_contexts()` task will break a strict-eg-tn's
|
|
||||||
multi-cancelled absorption..
|
|
||||||
|
|
||||||
'''
|
|
||||||
from tractor import (
|
|
||||||
trionics,
|
|
||||||
)
|
|
||||||
|
|
||||||
@acm
|
|
||||||
async def open_memchan() -> trio.abc.ReceiveChannel:
|
|
||||||
|
|
||||||
task: trio.Task = trio.lowlevel.current_task()
|
|
||||||
print(
|
|
||||||
f'Opening {task!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# 1 to force eager sending
|
|
||||||
send, recv = trio.open_memory_channel(16)
|
|
||||||
|
|
||||||
try:
|
|
||||||
async with send:
|
|
||||||
yield recv
|
|
||||||
finally:
|
|
||||||
print(
|
|
||||||
f'Closed {task!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
async def main():
|
|
||||||
async with (
|
|
||||||
# XXX should ensure ONLY the KBI
|
|
||||||
# is relayed upward
|
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery(), # as tn,
|
|
||||||
|
|
||||||
trionics.gather_contexts([
|
|
||||||
open_memchan(),
|
|
||||||
open_memchan(),
|
|
||||||
]) as recv_chans,
|
|
||||||
):
|
|
||||||
assert len(recv_chans) == 2
|
|
||||||
|
|
||||||
await trio.sleep(1)
|
|
||||||
raise KeyboardInterrupt
|
|
||||||
# tn.cancel_scope.cancel()
|
|
||||||
|
|
||||||
with pytest.raises(KeyboardInterrupt):
|
|
||||||
trio.run(main)
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'raise_unmasked', [
|
|
||||||
True,
|
|
||||||
pytest.param(
|
|
||||||
False,
|
|
||||||
marks=pytest.mark.xfail(
|
|
||||||
reason="see examples/trio/send_chan_aclose_masks.py"
|
|
||||||
)
|
|
||||||
),
|
|
||||||
]
|
|
||||||
)
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'child_errors_mid_stream',
|
|
||||||
[True, False],
|
|
||||||
)
|
|
||||||
def test_unmask_aclose_as_checkpoint_on_aexit(
|
|
||||||
raise_unmasked: bool,
|
|
||||||
child_errors_mid_stream: bool,
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Verify that our unmasker util works over the common case where
|
|
||||||
a mem-chan's `.aclose()` is included in an `@acm` stack
|
|
||||||
and it being currently a checkpoint, can `trio.Cancelled`-mask an embedded
|
|
||||||
exception from user code resulting in a silent failure which
|
|
||||||
appears like graceful cancellation.
|
|
||||||
|
|
||||||
This test suite is mostly implemented as an example script so it
|
|
||||||
could more easily be shared with `trio`-core peeps as `tractor`-less
|
|
||||||
minimum reproducing example.
|
|
||||||
|
|
||||||
'''
|
|
||||||
mod: ModuleType = pathlib.import_path(
|
|
||||||
examples_dir()
|
|
||||||
/ 'trio'
|
|
||||||
/ 'send_chan_aclose_masks_beg.py',
|
|
||||||
root=examples_dir(),
|
|
||||||
consider_namespace_packages=False,
|
|
||||||
)
|
|
||||||
with pytest.raises(RuntimeError):
|
|
||||||
trio.run(partial(
|
|
||||||
mod.main,
|
|
||||||
raise_unmasked=raise_unmasked,
|
|
||||||
child_errors_mid_stream=child_errors_mid_stream,
|
|
||||||
))
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
'ignore_special_cases', [
|
|
||||||
True,
|
|
||||||
pytest.param(
|
|
||||||
False,
|
|
||||||
marks=pytest.mark.xfail(
|
|
||||||
reason="see examples/trio/lockacquire_not_umasked.py"
|
|
||||||
)
|
|
||||||
),
|
|
||||||
]
|
|
||||||
)
|
|
||||||
def test_cancelled_lockacquire_in_ipctx_not_unmasked(
|
|
||||||
ignore_special_cases: bool,
|
|
||||||
loglevel: str,
|
|
||||||
debug_mode: bool,
|
|
||||||
):
|
|
||||||
mod: ModuleType = pathlib.import_path(
|
|
||||||
examples_dir()
|
|
||||||
/ 'trio'
|
|
||||||
/ 'lockacquire_not_unmasked.py',
|
|
||||||
root=examples_dir(),
|
|
||||||
consider_namespace_packages=False,
|
|
||||||
)
|
|
||||||
async def _main():
|
|
||||||
with trio.fail_after(2):
|
|
||||||
await mod.main(
|
|
||||||
ignore_special_cases=ignore_special_cases,
|
|
||||||
loglevel=loglevel,
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
)
|
|
||||||
|
|
||||||
trio.run(_main)
|
|
||||||
|
|
|
||||||
|
|
@ -18,7 +18,6 @@
|
||||||
tractor: structured concurrent ``trio``-"actors".
|
tractor: structured concurrent ``trio``-"actors".
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from ._clustering import (
|
from ._clustering import (
|
||||||
open_actor_cluster as open_actor_cluster,
|
open_actor_cluster as open_actor_cluster,
|
||||||
)
|
)
|
||||||
|
|
@ -44,7 +43,6 @@ from ._state import (
|
||||||
current_actor as current_actor,
|
current_actor as current_actor,
|
||||||
is_root_process as is_root_process,
|
is_root_process as is_root_process,
|
||||||
current_ipc_ctx as current_ipc_ctx,
|
current_ipc_ctx as current_ipc_ctx,
|
||||||
debug_mode as debug_mode
|
|
||||||
)
|
)
|
||||||
from ._exceptions import (
|
from ._exceptions import (
|
||||||
ContextCancelled as ContextCancelled,
|
ContextCancelled as ContextCancelled,
|
||||||
|
|
@ -64,7 +62,6 @@ from ._root import (
|
||||||
run_daemon as run_daemon,
|
run_daemon as run_daemon,
|
||||||
open_root_actor as open_root_actor,
|
open_root_actor as open_root_actor,
|
||||||
)
|
)
|
||||||
from .ipc import Channel as Channel
|
from ._ipc import Channel as Channel
|
||||||
from ._portal import Portal as Portal
|
from ._portal import Portal as Portal
|
||||||
from ._runtime import Actor as Actor
|
from ._runtime import Actor as Actor
|
||||||
# from . import hilevel as hilevel
|
|
||||||
|
|
|
||||||
282
tractor/_addr.py
282
tractor/_addr.py
|
|
@ -1,282 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
|
||||||
# it under the terms of the GNU Affero General Public License as published by
|
|
||||||
# the Free Software Foundation, either version 3 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful,
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
# GNU Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public License
|
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
from __future__ import annotations
|
|
||||||
from uuid import uuid4
|
|
||||||
from typing import (
|
|
||||||
Protocol,
|
|
||||||
ClassVar,
|
|
||||||
Type,
|
|
||||||
TYPE_CHECKING,
|
|
||||||
)
|
|
||||||
|
|
||||||
from bidict import bidict
|
|
||||||
from trio import (
|
|
||||||
SocketListener,
|
|
||||||
)
|
|
||||||
|
|
||||||
from .log import get_logger
|
|
||||||
from ._state import (
|
|
||||||
_def_tpt_proto,
|
|
||||||
)
|
|
||||||
from .ipc._tcp import TCPAddress
|
|
||||||
from .ipc._uds import UDSAddress
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from ._runtime import Actor
|
|
||||||
|
|
||||||
log = get_logger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
# TODO, maybe breakout the netns key to a struct?
|
|
||||||
# class NetNs(Struct)[str, int]:
|
|
||||||
# ...
|
|
||||||
|
|
||||||
# TODO, can't we just use a type alias
|
|
||||||
# for this? namely just some `tuple[str, int, str, str]`?
|
|
||||||
#
|
|
||||||
# -[ ] would also just be simpler to keep this as SockAddr[tuple]
|
|
||||||
# or something, implying it's just a simple pair of values which can
|
|
||||||
# presumably be mapped to all transports?
|
|
||||||
# -[ ] `pydoc socket.socket.getsockname()` delivers a 4-tuple for
|
|
||||||
# ipv6 `(hostaddr, port, flowinfo, scope_id)`.. so how should we
|
|
||||||
# handle that?
|
|
||||||
# -[ ] as a further alternative to this wrap()/unwrap() approach we
|
|
||||||
# could just implement `enc/dec_hook()`s for the `Address`-types
|
|
||||||
# and just deal with our internal objs directly and always and
|
|
||||||
# leave it to the codec layer to figure out marshalling?
|
|
||||||
# |_ would mean only one spot to do the `.unwrap()` (which we may
|
|
||||||
# end up needing to call from the hook()s anyway?)
|
|
||||||
# -[x] rename to `UnwrappedAddress[Descriptor]` ??
|
|
||||||
# seems like the right name as per,
|
|
||||||
# https://www.geeksforgeeks.org/introduction-to-address-descriptor/
|
|
||||||
#
|
|
||||||
UnwrappedAddress = (
|
|
||||||
# tcp/udp/uds
|
|
||||||
tuple[
|
|
||||||
str, # host/domain(tcp), filesys-dir(uds)
|
|
||||||
int|str, # port/path(uds)
|
|
||||||
]
|
|
||||||
# ?TODO? should we also include another 2 fields from
|
|
||||||
# our `Aid` msg such that we include the runtime `Actor.uid`
|
|
||||||
# of `.name` and `.uuid`?
|
|
||||||
# - would ensure uniqueness across entire net?
|
|
||||||
# - allows for easier runtime-level filtering of "actors by
|
|
||||||
# service name"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# TODO, maybe rename to `SocketAddress`?
|
|
||||||
class Address(Protocol):
|
|
||||||
proto_key: ClassVar[str]
|
|
||||||
unwrapped_type: ClassVar[UnwrappedAddress]
|
|
||||||
|
|
||||||
# TODO, i feel like an `.is_bound()` is a better thing to
|
|
||||||
# support?
|
|
||||||
# Lke, what use does this have besides a noop and if it's not
|
|
||||||
# valid why aren't we erroring on creation/use?
|
|
||||||
@property
|
|
||||||
def is_valid(self) -> bool:
|
|
||||||
...
|
|
||||||
|
|
||||||
# TODO, maybe `.netns` is a better name?
|
|
||||||
@property
|
|
||||||
def namespace(self) -> tuple[str, int]|None:
|
|
||||||
'''
|
|
||||||
The if-available, OS-specific "network namespace" key.
|
|
||||||
|
|
||||||
'''
|
|
||||||
...
|
|
||||||
|
|
||||||
@property
|
|
||||||
def bindspace(self) -> str:
|
|
||||||
'''
|
|
||||||
Deliver the socket address' "bindable space" from
|
|
||||||
a `socket.socket.bind()` and thus from the perspective of
|
|
||||||
specific transport protocol domain.
|
|
||||||
|
|
||||||
I.e. for most (layer-4) network-socket protocols this is
|
|
||||||
normally the ipv4/6 address, for UDS this is normally
|
|
||||||
a filesystem (sub-directory).
|
|
||||||
|
|
||||||
For (distributed) network protocols this is normally the routing
|
|
||||||
layer's domain/(ip-)address, though it might also include a "network namespace"
|
|
||||||
key different then the default.
|
|
||||||
|
|
||||||
For local-host-only transports this is either an explicit
|
|
||||||
namespace (with types defined by the OS: netns, Cgroup, IPC,
|
|
||||||
pid, etc. on linux) or failing that the sub-directory in the
|
|
||||||
filesys in which socket/shm files are located *under*.
|
|
||||||
|
|
||||||
'''
|
|
||||||
...
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_addr(cls, addr: UnwrappedAddress) -> Address:
|
|
||||||
...
|
|
||||||
|
|
||||||
def unwrap(self) -> UnwrappedAddress:
|
|
||||||
'''
|
|
||||||
Deliver the underying minimum field set in
|
|
||||||
a primitive python data type-structure.
|
|
||||||
'''
|
|
||||||
...
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def get_random(
|
|
||||||
cls,
|
|
||||||
current_actor: Actor,
|
|
||||||
bindspace: str|None = None,
|
|
||||||
) -> Address:
|
|
||||||
...
|
|
||||||
|
|
||||||
# TODO, this should be something like a `.get_def_registar_addr()`
|
|
||||||
# or similar since,
|
|
||||||
# - it should be a **host singleton** (not root/tree singleton)
|
|
||||||
# - we **only need this value** when one isn't provided to the
|
|
||||||
# runtime at boot and we want to implicitly provide a host-wide
|
|
||||||
# registrar.
|
|
||||||
# - each rooted-actor-tree should likely have its own
|
|
||||||
# micro-registry (likely the root being it), also see
|
|
||||||
@classmethod
|
|
||||||
def get_root(cls) -> Address:
|
|
||||||
...
|
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
|
||||||
...
|
|
||||||
|
|
||||||
def __eq__(self, other) -> bool:
|
|
||||||
...
|
|
||||||
|
|
||||||
async def open_listener(
|
|
||||||
self,
|
|
||||||
**kwargs,
|
|
||||||
) -> SocketListener:
|
|
||||||
...
|
|
||||||
|
|
||||||
async def close_listener(self):
|
|
||||||
...
|
|
||||||
|
|
||||||
|
|
||||||
_address_types: bidict[str, Type[Address]] = {
|
|
||||||
'tcp': TCPAddress,
|
|
||||||
'uds': UDSAddress
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# TODO! really these are discovery sys default addrs ONLY useful for
|
|
||||||
# when none is provided to a root actor on first boot.
|
|
||||||
_default_lo_addrs: dict[
|
|
||||||
str,
|
|
||||||
UnwrappedAddress
|
|
||||||
] = {
|
|
||||||
'tcp': TCPAddress.get_root().unwrap(),
|
|
||||||
'uds': UDSAddress.get_root().unwrap(),
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def get_address_cls(name: str) -> Type[Address]:
|
|
||||||
return _address_types[name]
|
|
||||||
|
|
||||||
|
|
||||||
def is_wrapped_addr(addr: any) -> bool:
|
|
||||||
return type(addr) in _address_types.values()
|
|
||||||
|
|
||||||
|
|
||||||
def mk_uuid() -> str:
|
|
||||||
'''
|
|
||||||
Encapsulate creation of a uuid4 as `str` as used
|
|
||||||
for creating `Actor.uid: tuple[str, str]` and/or
|
|
||||||
`.msg.types.Aid`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return str(uuid4())
|
|
||||||
|
|
||||||
|
|
||||||
def wrap_address(
|
|
||||||
addr: UnwrappedAddress
|
|
||||||
) -> Address:
|
|
||||||
'''
|
|
||||||
Wrap an `UnwrappedAddress` as an `Address`-type based
|
|
||||||
on matching builtin python data-structures which we adhoc
|
|
||||||
use for each.
|
|
||||||
|
|
||||||
XXX NOTE, careful care must be placed to ensure
|
|
||||||
`UnwrappedAddress` cases are **definitely unique** otherwise the
|
|
||||||
wrong transport backend may be loaded and will break many
|
|
||||||
low-level things in our runtime in a not-fun-to-debug way!
|
|
||||||
|
|
||||||
XD
|
|
||||||
|
|
||||||
'''
|
|
||||||
if is_wrapped_addr(addr):
|
|
||||||
return addr
|
|
||||||
|
|
||||||
cls: Type|None = None
|
|
||||||
# if 'sock' in addr[0]:
|
|
||||||
# import pdbp; pdbp.set_trace()
|
|
||||||
match addr:
|
|
||||||
|
|
||||||
# classic network socket-address as tuple/list
|
|
||||||
case (
|
|
||||||
(str(), int())
|
|
||||||
|
|
|
||||||
[str(), int()]
|
|
||||||
):
|
|
||||||
cls = TCPAddress
|
|
||||||
|
|
||||||
case (
|
|
||||||
# (str()|Path(), str()|Path()),
|
|
||||||
# ^TODO? uhh why doesn't this work!?
|
|
||||||
|
|
||||||
(_, filename)
|
|
||||||
) if type(filename) is str:
|
|
||||||
cls = UDSAddress
|
|
||||||
|
|
||||||
# likely an unset UDS or TCP reg address as defaulted in
|
|
||||||
# `_state._runtime_vars['_root_mailbox']`
|
|
||||||
#
|
|
||||||
# TODO? figure out when/if we even need this?
|
|
||||||
case (
|
|
||||||
None
|
|
||||||
|
|
|
||||||
[None, None]
|
|
||||||
):
|
|
||||||
cls: Type[Address] = get_address_cls(_def_tpt_proto)
|
|
||||||
addr: UnwrappedAddress = cls.get_root().unwrap()
|
|
||||||
|
|
||||||
case _:
|
|
||||||
# import pdbp; pdbp.set_trace()
|
|
||||||
raise TypeError(
|
|
||||||
f'Can not wrap unwrapped-address ??\n'
|
|
||||||
f'type(addr): {type(addr)!r}\n'
|
|
||||||
f'addr: {addr!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
return cls.from_addr(addr)
|
|
||||||
|
|
||||||
|
|
||||||
def default_lo_addrs(
|
|
||||||
transports: list[str],
|
|
||||||
) -> list[Type[Address]]:
|
|
||||||
'''
|
|
||||||
Return the default, host-singleton, registry address
|
|
||||||
for an input transport key set.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return [
|
|
||||||
_default_lo_addrs[transport]
|
|
||||||
for transport in transports
|
|
||||||
]
|
|
||||||
|
|
@ -31,12 +31,8 @@ def parse_uid(arg):
|
||||||
return str(name), str(uuid) # ensures str encoding
|
return str(name), str(uuid) # ensures str encoding
|
||||||
|
|
||||||
def parse_ipaddr(arg):
|
def parse_ipaddr(arg):
|
||||||
try:
|
host, port = literal_eval(arg)
|
||||||
return literal_eval(arg)
|
return (str(host), int(port))
|
||||||
|
|
||||||
except (ValueError, SyntaxError):
|
|
||||||
# UDS: try to interpret as a straight up str
|
|
||||||
return arg
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
@ -50,8 +46,8 @@ if __name__ == "__main__":
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
subactor = Actor(
|
subactor = Actor(
|
||||||
name=args.uid[0],
|
args.uid[0],
|
||||||
uuid=args.uid[1],
|
uid=args.uid[1],
|
||||||
loglevel=args.loglevel,
|
loglevel=args.loglevel,
|
||||||
spawn_method="trio"
|
spawn_method="trio"
|
||||||
)
|
)
|
||||||
|
|
|
||||||
|
|
@ -19,13 +19,10 @@ Actor cluster helpers.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
from contextlib import (
|
|
||||||
asynccontextmanager as acm,
|
from contextlib import asynccontextmanager as acm
|
||||||
)
|
|
||||||
from multiprocessing import cpu_count
|
from multiprocessing import cpu_count
|
||||||
from typing import (
|
from typing import AsyncGenerator, Optional
|
||||||
AsyncGenerator,
|
|
||||||
)
|
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
@ -55,17 +52,10 @@ async def open_actor_cluster(
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
'Number of names is {len(names)} but count it {count}')
|
'Number of names is {len(names)} but count it {count}')
|
||||||
|
|
||||||
async with (
|
async with tractor.open_nursery(
|
||||||
# tractor.trionics.collapse_eg(),
|
|
||||||
tractor.open_nursery(
|
|
||||||
**runtime_kwargs,
|
**runtime_kwargs,
|
||||||
) as an
|
) as an:
|
||||||
):
|
async with trio.open_nursery() as n:
|
||||||
async with (
|
|
||||||
# tractor.trionics.collapse_eg(),
|
|
||||||
trio.open_nursery() as tn,
|
|
||||||
tractor.trionics.maybe_raise_from_masking_exc()
|
|
||||||
):
|
|
||||||
uid = tractor.current_actor().uid
|
uid = tractor.current_actor().uid
|
||||||
|
|
||||||
async def _start(name: str) -> None:
|
async def _start(name: str) -> None:
|
||||||
|
|
@ -76,8 +66,9 @@ async def open_actor_cluster(
|
||||||
)
|
)
|
||||||
|
|
||||||
for name in names:
|
for name in names:
|
||||||
tn.start_soon(_start, name)
|
n.start_soon(_start, name)
|
||||||
|
|
||||||
assert len(portals) == count
|
assert len(portals) == count
|
||||||
yield portals
|
yield portals
|
||||||
|
|
||||||
await an.cancel(hard_kill=hard_kill)
|
await an.cancel(hard_kill=hard_kill)
|
||||||
|
|
|
||||||
|
|
@ -82,14 +82,13 @@ from .msg import (
|
||||||
MsgType,
|
MsgType,
|
||||||
NamespacePath,
|
NamespacePath,
|
||||||
PayloadT,
|
PayloadT,
|
||||||
Return,
|
|
||||||
Started,
|
Started,
|
||||||
Stop,
|
Stop,
|
||||||
Yield,
|
Yield,
|
||||||
pretty_struct,
|
pretty_struct,
|
||||||
_ops as msgops,
|
_ops as msgops,
|
||||||
)
|
)
|
||||||
from .ipc import (
|
from ._ipc import (
|
||||||
Channel,
|
Channel,
|
||||||
)
|
)
|
||||||
from ._streaming import (
|
from ._streaming import (
|
||||||
|
|
@ -101,14 +100,11 @@ from ._state import (
|
||||||
debug_mode,
|
debug_mode,
|
||||||
_ctxvar_Context,
|
_ctxvar_Context,
|
||||||
)
|
)
|
||||||
from .trionics import (
|
|
||||||
collapse_eg,
|
|
||||||
)
|
|
||||||
# ------ - ------
|
# ------ - ------
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from ._portal import Portal
|
from ._portal import Portal
|
||||||
from ._runtime import Actor
|
from ._runtime import Actor
|
||||||
from .ipc._transport import MsgTransport
|
from ._ipc import MsgTransport
|
||||||
from .devx._frame_stack import (
|
from .devx._frame_stack import (
|
||||||
CallerInfo,
|
CallerInfo,
|
||||||
)
|
)
|
||||||
|
|
@ -154,7 +150,7 @@ class Context:
|
||||||
2 cancel-scope-linked, communicating and parallel executing
|
2 cancel-scope-linked, communicating and parallel executing
|
||||||
`Task`s. Contexts are allocated on each side of any task
|
`Task`s. Contexts are allocated on each side of any task
|
||||||
RPC-linked msg dialog, i.e. for every request to a remote
|
RPC-linked msg dialog, i.e. for every request to a remote
|
||||||
actor from a `Portal`. On the "child" side a context is
|
actor from a `Portal`. On the "callee" side a context is
|
||||||
always allocated inside `._rpc._invoke()`.
|
always allocated inside `._rpc._invoke()`.
|
||||||
|
|
||||||
TODO: more detailed writeup on cancellation, error and
|
TODO: more detailed writeup on cancellation, error and
|
||||||
|
|
@ -222,8 +218,8 @@ class Context:
|
||||||
# `._runtime.invoke()`.
|
# `._runtime.invoke()`.
|
||||||
_remote_func_type: str | None = None
|
_remote_func_type: str | None = None
|
||||||
|
|
||||||
# NOTE: (for now) only set (a portal) on the parent side since
|
# NOTE: (for now) only set (a portal) on the caller side since
|
||||||
# the child doesn't generally need a ref to one and should
|
# the callee doesn't generally need a ref to one and should
|
||||||
# normally need to explicitly ask for handle to its peer if
|
# normally need to explicitly ask for handle to its peer if
|
||||||
# more the the `Context` is needed?
|
# more the the `Context` is needed?
|
||||||
_portal: Portal | None = None
|
_portal: Portal | None = None
|
||||||
|
|
@ -249,15 +245,13 @@ class Context:
|
||||||
# a drain loop?
|
# a drain loop?
|
||||||
# _res_scope: trio.CancelScope|None = None
|
# _res_scope: trio.CancelScope|None = None
|
||||||
|
|
||||||
_outcome_msg: Return|Error|ContextCancelled = Unresolved
|
|
||||||
|
|
||||||
# on a clean exit there should be a final value
|
# on a clean exit there should be a final value
|
||||||
# delivered from the far end "child" task, so
|
# delivered from the far end "callee" task, so
|
||||||
# this value is only set on one side.
|
# this value is only set on one side.
|
||||||
# _result: Any | int = None
|
# _result: Any | int = None
|
||||||
_result: PayloadT|Unresolved = Unresolved
|
_result: Any|Unresolved = Unresolved
|
||||||
|
|
||||||
# if the local "parent" task errors this value is always set
|
# if the local "caller" task errors this value is always set
|
||||||
# to the error that was captured in the
|
# to the error that was captured in the
|
||||||
# `Portal.open_context().__aexit__()` teardown block OR, in
|
# `Portal.open_context().__aexit__()` teardown block OR, in
|
||||||
# 2 special cases when an (maybe) expected remote error
|
# 2 special cases when an (maybe) expected remote error
|
||||||
|
|
@ -293,9 +287,9 @@ class Context:
|
||||||
# a `ContextCancelled` due to a call to `.cancel()` triggering
|
# a `ContextCancelled` due to a call to `.cancel()` triggering
|
||||||
# "graceful closure" on either side:
|
# "graceful closure" on either side:
|
||||||
# - `._runtime._invoke()` will check this flag before engaging
|
# - `._runtime._invoke()` will check this flag before engaging
|
||||||
# the crash handler REPL in such cases where the "child"
|
# the crash handler REPL in such cases where the "callee"
|
||||||
# raises the cancellation,
|
# raises the cancellation,
|
||||||
# - `.devx.debug.lock_stdio_for_peer()` will set it to `False` if
|
# - `.devx._debug.lock_stdio_for_peer()` will set it to `False` if
|
||||||
# the global tty-lock has been configured to filter out some
|
# the global tty-lock has been configured to filter out some
|
||||||
# actors from being able to acquire the debugger lock.
|
# actors from being able to acquire the debugger lock.
|
||||||
_enter_debugger_on_cancel: bool = True
|
_enter_debugger_on_cancel: bool = True
|
||||||
|
|
@ -307,8 +301,8 @@ class Context:
|
||||||
_stream_opened: bool = False
|
_stream_opened: bool = False
|
||||||
_stream: MsgStream|None = None
|
_stream: MsgStream|None = None
|
||||||
|
|
||||||
# the parent-task's calling-fn's frame-info, the frame above
|
# caller of `Portal.open_context()` for
|
||||||
# `Portal.open_context()`, for introspection/logging.
|
# logging purposes mostly
|
||||||
_caller_info: CallerInfo|None = None
|
_caller_info: CallerInfo|None = None
|
||||||
|
|
||||||
# overrun handling machinery
|
# overrun handling machinery
|
||||||
|
|
@ -369,7 +363,7 @@ class Context:
|
||||||
# f' ---\n'
|
# f' ---\n'
|
||||||
f' |_ipc: {self.dst_maddr}\n'
|
f' |_ipc: {self.dst_maddr}\n'
|
||||||
# f' dst_maddr{ds}{self.dst_maddr}\n'
|
# f' dst_maddr{ds}{self.dst_maddr}\n'
|
||||||
f" uid{ds}'{self.chan.aid}'\n"
|
f" uid{ds}'{self.chan.uid}'\n"
|
||||||
f" cid{ds}'{self.cid}'\n"
|
f" cid{ds}'{self.cid}'\n"
|
||||||
# f' ---\n'
|
# f' ---\n'
|
||||||
f'\n'
|
f'\n'
|
||||||
|
|
@ -442,25 +436,25 @@ class Context:
|
||||||
'''
|
'''
|
||||||
Records whether cancellation has been requested for this context
|
Records whether cancellation has been requested for this context
|
||||||
by a call to `.cancel()` either due to,
|
by a call to `.cancel()` either due to,
|
||||||
- an explicit call by some local task,
|
- either an explicit call by some local task,
|
||||||
- or an implicit call due to an error caught inside
|
- or an implicit call due to an error caught inside
|
||||||
the `Portal.open_context()` block.
|
the ``Portal.open_context()`` block.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
return self._cancel_called
|
return self._cancel_called
|
||||||
|
|
||||||
# XXX, to debug who frickin sets it..
|
@cancel_called.setter
|
||||||
# @cancel_called.setter
|
def cancel_called(self, val: bool) -> None:
|
||||||
# def cancel_called(self, val: bool) -> None:
|
'''
|
||||||
# '''
|
Set the self-cancelled request `bool` value.
|
||||||
# Set the self-cancelled request `bool` value.
|
|
||||||
|
|
||||||
# '''
|
'''
|
||||||
|
# to debug who frickin sets it..
|
||||||
# if val:
|
# if val:
|
||||||
# from .devx import pause_from_sync
|
# from .devx import pause_from_sync
|
||||||
# pause_from_sync()
|
# pause_from_sync()
|
||||||
|
|
||||||
# self._cancel_called = val
|
self._cancel_called = val
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def canceller(self) -> tuple[str, str]|None:
|
def canceller(self) -> tuple[str, str]|None:
|
||||||
|
|
@ -529,11 +523,11 @@ class Context:
|
||||||
'''
|
'''
|
||||||
Exactly the value of `self._scope.cancelled_caught`
|
Exactly the value of `self._scope.cancelled_caught`
|
||||||
(delegation) and should only be (able to be read as)
|
(delegation) and should only be (able to be read as)
|
||||||
`True` for a `.side == "parent"` ctx wherein the
|
`True` for a `.side == "caller"` ctx wherein the
|
||||||
`Portal.open_context()` block was exited due to a call to
|
`Portal.open_context()` block was exited due to a call to
|
||||||
`._scope.cancel()` - which should only ocurr in 2 cases:
|
`._scope.cancel()` - which should only ocurr in 2 cases:
|
||||||
|
|
||||||
- a parent side calls `.cancel()`, the far side cancels
|
- a caller side calls `.cancel()`, the far side cancels
|
||||||
and delivers back a `ContextCancelled` (making
|
and delivers back a `ContextCancelled` (making
|
||||||
`.cancel_acked == True`) and `._scope.cancel()` is
|
`.cancel_acked == True`) and `._scope.cancel()` is
|
||||||
called by `._maybe_cancel_and_set_remote_error()` which
|
called by `._maybe_cancel_and_set_remote_error()` which
|
||||||
|
|
@ -542,20 +536,20 @@ class Context:
|
||||||
=> `._scope.cancelled_caught == True` by normal `trio`
|
=> `._scope.cancelled_caught == True` by normal `trio`
|
||||||
cs semantics.
|
cs semantics.
|
||||||
|
|
||||||
- a parent side is delivered a `._remote_error:
|
- a caller side is delivered a `._remote_error:
|
||||||
RemoteActorError` via `._deliver_msg()` and a transitive
|
RemoteActorError` via `._deliver_msg()` and a transitive
|
||||||
call to `_maybe_cancel_and_set_remote_error()` calls
|
call to `_maybe_cancel_and_set_remote_error()` calls
|
||||||
`._scope.cancel()` and that cancellation eventually
|
`._scope.cancel()` and that cancellation eventually
|
||||||
results in `trio.Cancelled`(s) caught in the
|
results in `trio.Cancelled`(s) caught in the
|
||||||
`.open_context()` handling around the @acm's `yield`.
|
`.open_context()` handling around the @acm's `yield`.
|
||||||
|
|
||||||
Only as an FYI, in the "child" side case it can also be
|
Only as an FYI, in the "callee" side case it can also be
|
||||||
set but never is readable by any task outside the RPC
|
set but never is readable by any task outside the RPC
|
||||||
machinery in `._invoke()` since,:
|
machinery in `._invoke()` since,:
|
||||||
- when a child side calls `.cancel()`, `._scope.cancel()`
|
- when a callee side calls `.cancel()`, `._scope.cancel()`
|
||||||
is called immediately and handled specially inside
|
is called immediately and handled specially inside
|
||||||
`._invoke()` to raise a `ContextCancelled` which is then
|
`._invoke()` to raise a `ContextCancelled` which is then
|
||||||
sent to the parent side.
|
sent to the caller side.
|
||||||
|
|
||||||
However, `._scope.cancelled_caught` can NEVER be
|
However, `._scope.cancelled_caught` can NEVER be
|
||||||
accessed/read as `True` by any RPC invoked task since it
|
accessed/read as `True` by any RPC invoked task since it
|
||||||
|
|
@ -635,71 +629,6 @@ class Context:
|
||||||
'''
|
'''
|
||||||
await self.chan.send(Stop(cid=self.cid))
|
await self.chan.send(Stop(cid=self.cid))
|
||||||
|
|
||||||
@property
|
|
||||||
def parent_task(self) -> trio.Task:
|
|
||||||
'''
|
|
||||||
This IPC context's "owning task" which is a `trio.Task`
|
|
||||||
on one of the "sides" of the IPC.
|
|
||||||
|
|
||||||
Note that the "parent_" prefix here refers to the local
|
|
||||||
`trio` task tree using the same interface as
|
|
||||||
`trio.Nursery.parent_task` whereas for IPC contexts,
|
|
||||||
a different cross-actor task hierarchy exists:
|
|
||||||
|
|
||||||
- a "parent"-side which originally entered
|
|
||||||
`Portal.open_context()`,
|
|
||||||
|
|
||||||
- the "child"-side which was spawned and scheduled to invoke
|
|
||||||
a function decorated with `@tractor.context`.
|
|
||||||
|
|
||||||
This task is thus a handle to mem-domain-distinct/per-process
|
|
||||||
`Nursery.parent_task` depending on in which of the above
|
|
||||||
"sides" this context exists.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return self._task
|
|
||||||
|
|
||||||
def _is_blocked_on_rx_chan(self) -> bool:
|
|
||||||
'''
|
|
||||||
Predicate to indicate whether the owner `._task: trio.Task` is
|
|
||||||
currently blocked (by `.receive()`-ing) on its underlying RPC
|
|
||||||
feeder `._rx_chan`.
|
|
||||||
|
|
||||||
This knowledge is highly useful when handling so called
|
|
||||||
"out-of-band" (OoB) cancellation conditions where a peer
|
|
||||||
actor's task transmitted some remote error/cancel-msg and we
|
|
||||||
must know whether to signal-via-cancel currently executing
|
|
||||||
"user-code" (user defined code embedded in `ctx._scope`) or
|
|
||||||
simply to forward the IPC-msg-as-error **without calling**
|
|
||||||
`._scope.cancel()`.
|
|
||||||
|
|
||||||
In the latter case it is presumed that if the owner task is
|
|
||||||
blocking for the next IPC msg, it will eventually receive,
|
|
||||||
process and raise the equivalent local error **without**
|
|
||||||
requiring `._scope.cancel()` to be explicitly called by the
|
|
||||||
*delivering OoB RPC-task* (via `_deliver_msg()`).
|
|
||||||
|
|
||||||
'''
|
|
||||||
# NOTE, see the mem-chan meth-impls for *why* this
|
|
||||||
# logic works,
|
|
||||||
# `trio._channel.MemoryReceiveChannel.receive[_nowait]()`
|
|
||||||
#
|
|
||||||
# XXX realize that this is NOT an
|
|
||||||
# official/will-be-loudly-deprecated API:
|
|
||||||
# - https://trio.readthedocs.io/en/stable/reference-lowlevel.html#trio.lowlevel.Task.custom_sleep_data
|
|
||||||
# |_https://trio.readthedocs.io/en/stable/reference-lowlevel.html#trio.lowlevel.wait_task_rescheduled
|
|
||||||
#
|
|
||||||
# orig repo intro in the mem-chan change over patch:
|
|
||||||
# - https://github.com/python-trio/trio/pull/586#issuecomment-414039117
|
|
||||||
# |_https://github.com/python-trio/trio/pull/616
|
|
||||||
# |_https://github.com/njsmith/trio/commit/98c38cef6f62e731bf8c7190e8756976bface8f0
|
|
||||||
#
|
|
||||||
return (
|
|
||||||
self._task.custom_sleep_data
|
|
||||||
is
|
|
||||||
self._rx_chan
|
|
||||||
)
|
|
||||||
|
|
||||||
def _maybe_cancel_and_set_remote_error(
|
def _maybe_cancel_and_set_remote_error(
|
||||||
self,
|
self,
|
||||||
error: BaseException,
|
error: BaseException,
|
||||||
|
|
@ -731,7 +660,7 @@ class Context:
|
||||||
when called/closed by actor local task(s).
|
when called/closed by actor local task(s).
|
||||||
|
|
||||||
NOTEs:
|
NOTEs:
|
||||||
- It is expected that the parent has previously unwrapped
|
- It is expected that the caller has previously unwrapped
|
||||||
the remote error using a call to `unpack_error()` and
|
the remote error using a call to `unpack_error()` and
|
||||||
provides that output exception value as the input
|
provides that output exception value as the input
|
||||||
`error` argument *here*.
|
`error` argument *here*.
|
||||||
|
|
@ -741,7 +670,7 @@ class Context:
|
||||||
`Portal.open_context()` (ideally) we want to interrupt
|
`Portal.open_context()` (ideally) we want to interrupt
|
||||||
any ongoing local tasks operating within that
|
any ongoing local tasks operating within that
|
||||||
`Context`'s cancel-scope so as to be notified ASAP of
|
`Context`'s cancel-scope so as to be notified ASAP of
|
||||||
the remote error and engage any parent handling (eg.
|
the remote error and engage any caller handling (eg.
|
||||||
for cross-process task supervision).
|
for cross-process task supervision).
|
||||||
|
|
||||||
- In some cases we may want to raise the remote error
|
- In some cases we may want to raise the remote error
|
||||||
|
|
@ -808,8 +737,6 @@ class Context:
|
||||||
# cancelled, NOT their reported canceller. IOW in the
|
# cancelled, NOT their reported canceller. IOW in the
|
||||||
# latter case we're cancelled by someone else getting
|
# latter case we're cancelled by someone else getting
|
||||||
# cancelled.
|
# cancelled.
|
||||||
#
|
|
||||||
# !TODO, switching to `Actor.aid` here!
|
|
||||||
if (canc := error.canceller) == self._actor.uid:
|
if (canc := error.canceller) == self._actor.uid:
|
||||||
whom: str = 'us'
|
whom: str = 'us'
|
||||||
self._canceller = canc
|
self._canceller = canc
|
||||||
|
|
@ -852,27 +779,13 @@ class Context:
|
||||||
if self._canceller is None:
|
if self._canceller is None:
|
||||||
log.error('Ctx has no canceller set!?')
|
log.error('Ctx has no canceller set!?')
|
||||||
|
|
||||||
cs: trio.CancelScope = self._scope
|
|
||||||
|
|
||||||
# ?TODO? see comment @ .start_remote_task()`
|
|
||||||
#
|
|
||||||
# if not cs:
|
|
||||||
# from .devx import mk_pdb
|
|
||||||
# mk_pdb().set_trace()
|
|
||||||
# raise RuntimeError(
|
|
||||||
# f'IPC ctx was not be opened prior to remote error delivery !?\n'
|
|
||||||
# f'{self}\n'
|
|
||||||
# f'\n'
|
|
||||||
# f'`Portal.open_context()` must be entered (somewhere) beforehand!\n'
|
|
||||||
# )
|
|
||||||
|
|
||||||
# Cancel the local `._scope`, catch that
|
# Cancel the local `._scope`, catch that
|
||||||
# `._scope.cancelled_caught` and re-raise any remote error
|
# `._scope.cancelled_caught` and re-raise any remote error
|
||||||
# once exiting (or manually calling `.wait_for_result()`) the
|
# once exiting (or manually calling `.wait_for_result()`) the
|
||||||
# `.open_context()` block.
|
# `.open_context()` block.
|
||||||
|
cs: trio.CancelScope = self._scope
|
||||||
if (
|
if (
|
||||||
cs
|
cs
|
||||||
and not cs.cancel_called
|
|
||||||
|
|
||||||
# XXX this is an expected cancel request response
|
# XXX this is an expected cancel request response
|
||||||
# message and we **don't need to raise it** in the
|
# message and we **don't need to raise it** in the
|
||||||
|
|
@ -881,7 +794,8 @@ class Context:
|
||||||
# if `._cancel_called` then `.cancel_acked and .cancel_called`
|
# if `._cancel_called` then `.cancel_acked and .cancel_called`
|
||||||
# always should be set.
|
# always should be set.
|
||||||
and not self._is_self_cancelled()
|
and not self._is_self_cancelled()
|
||||||
# and not cs.cancelled_caught
|
and not cs.cancel_called
|
||||||
|
and not cs.cancelled_caught
|
||||||
):
|
):
|
||||||
if (
|
if (
|
||||||
msgerr
|
msgerr
|
||||||
|
|
@ -892,7 +806,7 @@ class Context:
|
||||||
not self._cancel_on_msgerr
|
not self._cancel_on_msgerr
|
||||||
):
|
):
|
||||||
message: str = (
|
message: str = (
|
||||||
f'NOT Cancelling `Context._scope` since,\n'
|
'NOT Cancelling `Context._scope` since,\n'
|
||||||
f'Context._cancel_on_msgerr = {self._cancel_on_msgerr}\n\n'
|
f'Context._cancel_on_msgerr = {self._cancel_on_msgerr}\n\n'
|
||||||
f'AND we got a msg-type-error!\n'
|
f'AND we got a msg-type-error!\n'
|
||||||
f'{error}\n'
|
f'{error}\n'
|
||||||
|
|
@ -902,43 +816,13 @@ class Context:
|
||||||
# `trio.Cancelled` subtype here ;)
|
# `trio.Cancelled` subtype here ;)
|
||||||
# https://github.com/goodboy/tractor/issues/368
|
# https://github.com/goodboy/tractor/issues/368
|
||||||
message: str = 'Cancelling `Context._scope` !\n\n'
|
message: str = 'Cancelling `Context._scope` !\n\n'
|
||||||
cs.cancel()
|
# from .devx import pause_from_sync
|
||||||
|
# pause_from_sync()
|
||||||
# TODO, explicit condition for OoB (self-)cancellation?
|
self._scope.cancel()
|
||||||
# - we called `Portal.cancel_actor()` from this actor
|
else:
|
||||||
# and the peer ctx task delivered ctxc due to it.
|
message: str = 'NOT cancelling `Context._scope` !\n\n'
|
||||||
# - currently `self._is_self_cancelled()` will be true
|
|
||||||
# since the ctxc.canceller check will match us even though it
|
|
||||||
# wasn't from this ctx specifically!
|
|
||||||
elif (
|
|
||||||
cs
|
|
||||||
and self._is_self_cancelled()
|
|
||||||
and not cs.cancel_called
|
|
||||||
):
|
|
||||||
message: str = (
|
|
||||||
'Cancelling `ctx._scope` due to OoB self-cancel ?!\n'
|
|
||||||
'\n'
|
|
||||||
)
|
|
||||||
# from .devx import mk_pdb
|
# from .devx import mk_pdb
|
||||||
# mk_pdb().set_trace()
|
# mk_pdb().set_trace()
|
||||||
# TODO XXX, required to fix timeout failure in
|
|
||||||
# `test_cancelled_lockacquire_in_ipctx_not_unmaskeed`
|
|
||||||
#
|
|
||||||
|
|
||||||
# XXX NOTE XXX, this is SUPER SUBTLE!
|
|
||||||
# we only want to cancel our embedded `._scope`
|
|
||||||
# if the ctx's current/using task is NOT blocked
|
|
||||||
# on `._rx_chan.receive()` and on some other
|
|
||||||
# `trio`-checkpoint since in the former case
|
|
||||||
# any `._remote_error` will be relayed through
|
|
||||||
# the rx-chan and appropriately raised by the owning
|
|
||||||
# `._task` directly. IF the owner task is however
|
|
||||||
# blocking elsewhere we need to interrupt it **now**.
|
|
||||||
if not self._is_blocked_on_rx_chan():
|
|
||||||
cs.cancel()
|
|
||||||
else:
|
|
||||||
# rx_stats = self._rx_chan.statistics()
|
|
||||||
message: str = 'NOT cancelling `Context._scope` !\n\n'
|
|
||||||
|
|
||||||
fmt_str: str = 'No `self._scope: CancelScope` was set/used ?\n'
|
fmt_str: str = 'No `self._scope: CancelScope` was set/used ?\n'
|
||||||
if (
|
if (
|
||||||
|
|
@ -962,7 +846,6 @@ class Context:
|
||||||
+
|
+
|
||||||
cs_fmt
|
cs_fmt
|
||||||
)
|
)
|
||||||
|
|
||||||
log.cancel(
|
log.cancel(
|
||||||
message
|
message
|
||||||
+
|
+
|
||||||
|
|
@ -973,10 +856,19 @@ class Context:
|
||||||
@property
|
@property
|
||||||
def dst_maddr(self) -> str:
|
def dst_maddr(self) -> str:
|
||||||
chan: Channel = self.chan
|
chan: Channel = self.chan
|
||||||
|
dst_addr, dst_port = chan.raddr
|
||||||
trans: MsgTransport = chan.transport
|
trans: MsgTransport = chan.transport
|
||||||
# cid: str = self.cid
|
# cid: str = self.cid
|
||||||
# cid_head, cid_tail = cid[:6], cid[-6:]
|
# cid_head, cid_tail = cid[:6], cid[-6:]
|
||||||
return trans.maddr
|
return (
|
||||||
|
f'/ipv4/{dst_addr}'
|
||||||
|
f'/{trans.name_key}/{dst_port}'
|
||||||
|
# f'/{self.chan.uid[0]}'
|
||||||
|
# f'/{self.cid}'
|
||||||
|
|
||||||
|
# f'/cid={cid_head}..{cid_tail}'
|
||||||
|
# TODO: ? not use this ^ right ?
|
||||||
|
)
|
||||||
|
|
||||||
dmaddr = dst_maddr
|
dmaddr = dst_maddr
|
||||||
|
|
||||||
|
|
@ -995,11 +887,6 @@ class Context:
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def repr_caller(self) -> str:
|
def repr_caller(self) -> str:
|
||||||
'''
|
|
||||||
Render a "namespace-path" style representation of the calling
|
|
||||||
task-fn.
|
|
||||||
|
|
||||||
'''
|
|
||||||
ci: CallerInfo|None = self._caller_info
|
ci: CallerInfo|None = self._caller_info
|
||||||
if ci:
|
if ci:
|
||||||
return (
|
return (
|
||||||
|
|
@ -1013,7 +900,7 @@ class Context:
|
||||||
def repr_api(self) -> str:
|
def repr_api(self) -> str:
|
||||||
return 'Portal.open_context()'
|
return 'Portal.open_context()'
|
||||||
|
|
||||||
# TODO: use `.dev._frame_stack` scanning to find caller fn!
|
# TODO: use `.dev._frame_stack` scanning to find caller!
|
||||||
# ci: CallerInfo|None = self._caller_info
|
# ci: CallerInfo|None = self._caller_info
|
||||||
# if ci:
|
# if ci:
|
||||||
# return (
|
# return (
|
||||||
|
|
@ -1048,26 +935,25 @@ class Context:
|
||||||
=> That is, an IPC `Context` (this) **does not**
|
=> That is, an IPC `Context` (this) **does not**
|
||||||
have the same semantics as a `trio.CancelScope`.
|
have the same semantics as a `trio.CancelScope`.
|
||||||
|
|
||||||
If the parent (who entered the `Portal.open_context()`)
|
If the caller (who entered the `Portal.open_context()`)
|
||||||
desires that the internal block's cancel-scope be
|
desires that the internal block's cancel-scope be
|
||||||
cancelled it should open its own `trio.CancelScope` and
|
cancelled it should open its own `trio.CancelScope` and
|
||||||
manage it as needed.
|
manage it as needed.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
side: str = self.side
|
side: str = self.side
|
||||||
self._cancel_called = True
|
# XXX for debug via the `@.setter`
|
||||||
# ^ XXX for debug via the `@.setter`
|
self.cancel_called = True
|
||||||
# self.cancel_called = True
|
|
||||||
|
|
||||||
header: str = (
|
header: str = (
|
||||||
f'Cancelling ctx from {side!r}-side\n'
|
f'Cancelling ctx from {side.upper()}-side\n'
|
||||||
)
|
)
|
||||||
reminfo: str = (
|
reminfo: str = (
|
||||||
# ' =>\n'
|
# ' =>\n'
|
||||||
# f'Context.cancel() => {self.chan.uid}\n'
|
# f'Context.cancel() => {self.chan.uid}\n'
|
||||||
f'\n'
|
|
||||||
f'c)=> {self.chan.uid}\n'
|
f'c)=> {self.chan.uid}\n'
|
||||||
f' |_[{self.dst_maddr}\n'
|
# f'{self.chan.uid}\n'
|
||||||
|
f' |_ @{self.dst_maddr}\n'
|
||||||
f' >> {self.repr_rpc}\n'
|
f' >> {self.repr_rpc}\n'
|
||||||
# f' >> {self._nsf}() -> {codec}[dict]:\n\n'
|
# f' >> {self._nsf}() -> {codec}[dict]:\n\n'
|
||||||
# TODO: pull msg-type from spec re #320
|
# TODO: pull msg-type from spec re #320
|
||||||
|
|
@ -1120,7 +1006,7 @@ class Context:
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
log.cancel(
|
log.cancel(
|
||||||
f'Timed out on cancel request of remote task?\n'
|
'Timed out on cancel request of remote task?\n'
|
||||||
f'{reminfo}'
|
f'{reminfo}'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -1131,7 +1017,7 @@ class Context:
|
||||||
# `_invoke()` RPC task.
|
# `_invoke()` RPC task.
|
||||||
#
|
#
|
||||||
# NOTE: on this side we ALWAYS cancel the local scope
|
# NOTE: on this side we ALWAYS cancel the local scope
|
||||||
# since the parent expects a `ContextCancelled` to be sent
|
# since the caller expects a `ContextCancelled` to be sent
|
||||||
# from `._runtime._invoke()` back to the other side. The
|
# from `._runtime._invoke()` back to the other side. The
|
||||||
# logic for catching the result of the below
|
# logic for catching the result of the below
|
||||||
# `._scope.cancel()` is inside the `._runtime._invoke()`
|
# `._scope.cancel()` is inside the `._runtime._invoke()`
|
||||||
|
|
@ -1188,25 +1074,9 @@ class Context:
|
||||||
|RemoteActorError # stream overrun caused and ignored by us
|
|RemoteActorError # stream overrun caused and ignored by us
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
Maybe raise a remote error depending on the type of error and
|
Maybe raise a remote error depending on the type of error
|
||||||
*who*, i.e. which side of the task pair across actors,
|
and *who* (i.e. which task from which actor) requested
|
||||||
requested a cancellation (if any).
|
a cancellation (if any).
|
||||||
|
|
||||||
Depending on the input config-params suppress raising
|
|
||||||
certain remote excs:
|
|
||||||
|
|
||||||
- if `remote_error: ContextCancelled` (ctxc) AND this side's
|
|
||||||
task is the "requester", it at somem point called
|
|
||||||
`Context.cancel()`, then the peer's ctxc is treated
|
|
||||||
as a "cancel ack".
|
|
||||||
|
|
||||||
|_ this behaves exactly like how `trio.Nursery.cancel_scope`
|
|
||||||
absorbs any `BaseExceptionGroup[trio.Cancelled]` wherein the
|
|
||||||
owning parent task never will raise a `trio.Cancelled`
|
|
||||||
if `CancelScope.cancel_called == True`.
|
|
||||||
|
|
||||||
- `remote_error: StreamOverrrun` (overrun) AND
|
|
||||||
`raise_overrun_from_self` is set.
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
__tracebackhide__: bool = hide_tb
|
__tracebackhide__: bool = hide_tb
|
||||||
|
|
@ -1248,19 +1118,18 @@ class Context:
|
||||||
# for this ^, NO right?
|
# for this ^, NO right?
|
||||||
|
|
||||||
) or (
|
) or (
|
||||||
# NOTE: whenever this side is the cause of an
|
# NOTE: whenever this context is the cause of an
|
||||||
# overrun on the peer side, i.e. we sent msgs too
|
# overrun on the remote side (aka we sent msgs too
|
||||||
# fast and the peer task was overrun according
|
# fast that the remote task was overrun according
|
||||||
# to `MsgStream` buffer settings, AND this was
|
# to `MsgStream` buffer settings) AND the caller
|
||||||
# called with `raise_overrun_from_self=True` (the
|
# has requested to not raise overruns this side
|
||||||
# default), silently absorb any `StreamOverrun`.
|
# caused, we also silently absorb any remotely
|
||||||
#
|
# boxed `StreamOverrun`. This is mostly useful for
|
||||||
# XXX, this is namely useful for supressing such faults
|
# supressing such faults during
|
||||||
# during cancellation/error/final-result handling inside
|
# cancellation/error/final-result handling inside
|
||||||
# `.msg._ops.drain_to_final_msg()` such that we do not
|
# `msg._ops.drain_to_final_msg()` such that we do not
|
||||||
# raise during a cancellation-request, i.e. when
|
# raise such errors particularly in the case where
|
||||||
# `._cancel_called == True`.
|
# `._cancel_called == True`.
|
||||||
#
|
|
||||||
not raise_overrun_from_self
|
not raise_overrun_from_self
|
||||||
and isinstance(remote_error, RemoteActorError)
|
and isinstance(remote_error, RemoteActorError)
|
||||||
and remote_error.boxed_type is StreamOverrun
|
and remote_error.boxed_type is StreamOverrun
|
||||||
|
|
@ -1304,8 +1173,8 @@ class Context:
|
||||||
|
|
||||||
) -> Any|Exception:
|
) -> Any|Exception:
|
||||||
'''
|
'''
|
||||||
From some (parent) side task, wait for and return the final
|
From some (caller) side task, wait for and return the final
|
||||||
result from the remote (child) side's task.
|
result from the remote (callee) side's task.
|
||||||
|
|
||||||
This provides a mechanism for one task running in some actor to wait
|
This provides a mechanism for one task running in some actor to wait
|
||||||
on another task at the other side, in some other actor, to terminate.
|
on another task at the other side, in some other actor, to terminate.
|
||||||
|
|
@ -1329,10 +1198,8 @@ class Context:
|
||||||
|
|
||||||
'''
|
'''
|
||||||
__tracebackhide__: bool = hide_tb
|
__tracebackhide__: bool = hide_tb
|
||||||
if not self._portal:
|
assert self._portal, (
|
||||||
raise RuntimeError(
|
'`Context.wait_for_result()` can not be called from callee side!'
|
||||||
'Invalid usage of `Context.wait_for_result()`!\n'
|
|
||||||
'Not valid on child-side IPC ctx!\n'
|
|
||||||
)
|
)
|
||||||
if self._final_result_is_set():
|
if self._final_result_is_set():
|
||||||
return self._result
|
return self._result
|
||||||
|
|
@ -1354,8 +1221,6 @@ class Context:
|
||||||
# since every message should be delivered via the normal
|
# since every message should be delivered via the normal
|
||||||
# `._deliver_msg()` route which will appropriately set
|
# `._deliver_msg()` route which will appropriately set
|
||||||
# any `.maybe_error`.
|
# any `.maybe_error`.
|
||||||
outcome_msg: Return|Error|ContextCancelled
|
|
||||||
drained_msgs: list[MsgType]
|
|
||||||
(
|
(
|
||||||
outcome_msg,
|
outcome_msg,
|
||||||
drained_msgs,
|
drained_msgs,
|
||||||
|
|
@ -1363,19 +1228,11 @@ class Context:
|
||||||
ctx=self,
|
ctx=self,
|
||||||
hide_tb=hide_tb,
|
hide_tb=hide_tb,
|
||||||
)
|
)
|
||||||
|
|
||||||
drained_status: str = (
|
drained_status: str = (
|
||||||
'Ctx drained to final outcome msg\n\n'
|
'Ctx drained to final outcome msg\n\n'
|
||||||
f'{outcome_msg}\n'
|
f'{outcome_msg}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
# ?XXX, should already be set in `._deliver_msg()` right?
|
|
||||||
if self._outcome_msg is not Unresolved:
|
|
||||||
# from .devx import debug
|
|
||||||
# await debug.pause()
|
|
||||||
assert self._outcome_msg is outcome_msg
|
|
||||||
else:
|
|
||||||
self._outcome_msg = outcome_msg
|
|
||||||
|
|
||||||
if drained_msgs:
|
if drained_msgs:
|
||||||
drained_status += (
|
drained_status += (
|
||||||
'\n'
|
'\n'
|
||||||
|
|
@ -1601,12 +1458,6 @@ class Context:
|
||||||
):
|
):
|
||||||
status = 'peer-cancelled'
|
status = 'peer-cancelled'
|
||||||
|
|
||||||
case (
|
|
||||||
Unresolved,
|
|
||||||
trio.Cancelled(), # any error-type
|
|
||||||
) if self.canceller:
|
|
||||||
status = 'actor-cancelled'
|
|
||||||
|
|
||||||
# (remote) error condition
|
# (remote) error condition
|
||||||
case (
|
case (
|
||||||
Unresolved,
|
Unresolved,
|
||||||
|
|
@ -1712,15 +1563,15 @@ class Context:
|
||||||
strict_pld_parity=strict_pld_parity,
|
strict_pld_parity=strict_pld_parity,
|
||||||
hide_tb=hide_tb,
|
hide_tb=hide_tb,
|
||||||
)
|
)
|
||||||
except BaseException as _bexc:
|
except BaseException as err:
|
||||||
err = _bexc
|
|
||||||
if not isinstance(err, MsgTypeError):
|
if not isinstance(err, MsgTypeError):
|
||||||
__tracebackhide__: bool = False
|
__tracebackhide__: bool = False
|
||||||
|
|
||||||
raise err
|
raise
|
||||||
|
|
||||||
|
|
||||||
# TODO: maybe a flag to by-pass encode op if already done
|
# TODO: maybe a flag to by-pass encode op if already done
|
||||||
# here in parent?
|
# here in caller?
|
||||||
await self.chan.send(started_msg)
|
await self.chan.send(started_msg)
|
||||||
|
|
||||||
# set msg-related internal runtime-state
|
# set msg-related internal runtime-state
|
||||||
|
|
@ -1796,7 +1647,7 @@ class Context:
|
||||||
|
|
||||||
XXX RULES XXX
|
XXX RULES XXX
|
||||||
------ - ------
|
------ - ------
|
||||||
- NEVER raise remote errors from this method; a calling runtime-task.
|
- NEVER raise remote errors from this method; a runtime task caller.
|
||||||
An error "delivered" to a ctx should always be raised by
|
An error "delivered" to a ctx should always be raised by
|
||||||
the corresponding local task operating on the
|
the corresponding local task operating on the
|
||||||
`Portal`/`Context` APIs.
|
`Portal`/`Context` APIs.
|
||||||
|
|
@ -1872,7 +1723,7 @@ class Context:
|
||||||
|
|
||||||
else:
|
else:
|
||||||
report = (
|
report = (
|
||||||
'Queueing OVERRUN msg on parent task:\n\n'
|
'Queueing OVERRUN msg on caller task:\n\n'
|
||||||
+ report
|
+ report
|
||||||
)
|
)
|
||||||
log.debug(report)
|
log.debug(report)
|
||||||
|
|
@ -1889,6 +1740,7 @@ class Context:
|
||||||
|
|
||||||
f'{structfmt(msg)}\n'
|
f'{structfmt(msg)}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
# NOTE: if an error is deteced we should always still
|
# NOTE: if an error is deteced we should always still
|
||||||
# send it through the feeder-mem-chan and expect
|
# send it through the feeder-mem-chan and expect
|
||||||
# it to be raised by any context (stream) consumer
|
# it to be raised by any context (stream) consumer
|
||||||
|
|
@ -1900,21 +1752,6 @@ class Context:
|
||||||
# normally the task that should get cancelled/error
|
# normally the task that should get cancelled/error
|
||||||
# from some remote fault!
|
# from some remote fault!
|
||||||
send_chan.send_nowait(msg)
|
send_chan.send_nowait(msg)
|
||||||
match msg:
|
|
||||||
case Stop():
|
|
||||||
if (stream := self._stream):
|
|
||||||
stream._stop_msg = msg
|
|
||||||
|
|
||||||
case Return():
|
|
||||||
if not self._outcome_msg:
|
|
||||||
log.warning(
|
|
||||||
f'Setting final outcome msg AFTER '
|
|
||||||
f'`._rx_chan.send()`??\n'
|
|
||||||
f'\n'
|
|
||||||
f'{msg}'
|
|
||||||
)
|
|
||||||
self._outcome_msg = msg
|
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
except trio.BrokenResourceError:
|
except trio.BrokenResourceError:
|
||||||
|
|
@ -2068,12 +1905,12 @@ async def open_context_from_portal(
|
||||||
IPC protocol.
|
IPC protocol.
|
||||||
|
|
||||||
The yielded `tuple` is a pair delivering a `tractor.Context`
|
The yielded `tuple` is a pair delivering a `tractor.Context`
|
||||||
and any first value "sent" by the "child" task via a call
|
and any first value "sent" by the "callee" task via a call
|
||||||
to `Context.started(<value: Any>)`; this side of the
|
to `Context.started(<value: Any>)`; this side of the
|
||||||
context does not unblock until the "child" task calls
|
context does not unblock until the "callee" task calls
|
||||||
`.started()` in similar style to `trio.Nursery.start()`.
|
`.started()` in similar style to `trio.Nursery.start()`.
|
||||||
When the "child" (side that is "called"/started by a call
|
When the "callee" (side that is "called"/started by a call
|
||||||
to *this* method) returns, the parent side (this) unblocks
|
to *this* method) returns, the caller side (this) unblocks
|
||||||
and any final value delivered from the other end can be
|
and any final value delivered from the other end can be
|
||||||
retrieved using the `Contex.wait_for_result()` api.
|
retrieved using the `Contex.wait_for_result()` api.
|
||||||
|
|
||||||
|
|
@ -2086,7 +1923,7 @@ async def open_context_from_portal(
|
||||||
__tracebackhide__: bool = hide_tb
|
__tracebackhide__: bool = hide_tb
|
||||||
|
|
||||||
# denote this frame as a "runtime frame" for stack
|
# denote this frame as a "runtime frame" for stack
|
||||||
# introspection where we report the parent code in logging
|
# introspection where we report the caller code in logging
|
||||||
# and error message content.
|
# and error message content.
|
||||||
# NOTE: 2 bc of the wrapping `@acm`
|
# NOTE: 2 bc of the wrapping `@acm`
|
||||||
__runtimeframe__: int = 2 # noqa
|
__runtimeframe__: int = 2 # noqa
|
||||||
|
|
@ -2121,9 +1958,6 @@ async def open_context_from_portal(
|
||||||
f'|_{portal.actor}\n'
|
f'|_{portal.actor}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
# ?TODO? could we move this to inside the `tn` block?
|
|
||||||
# -> would allow doing `ctx.parent_task = tn.parent_task` ?
|
|
||||||
# -> would allow a `if not ._scope: => raise RTE` ?
|
|
||||||
ctx: Context = await portal.actor.start_remote_task(
|
ctx: Context = await portal.actor.start_remote_task(
|
||||||
portal.channel,
|
portal.channel,
|
||||||
nsf=nsf,
|
nsf=nsf,
|
||||||
|
|
@ -2148,11 +1982,9 @@ async def open_context_from_portal(
|
||||||
# placeholder for any exception raised in the runtime
|
# placeholder for any exception raised in the runtime
|
||||||
# or by user tasks which cause this context's closure.
|
# or by user tasks which cause this context's closure.
|
||||||
scope_err: BaseException|None = None
|
scope_err: BaseException|None = None
|
||||||
ctxc_from_child: ContextCancelled|None = None
|
ctxc_from_callee: ContextCancelled|None = None
|
||||||
try:
|
try:
|
||||||
# from .devx import pause
|
|
||||||
async with (
|
async with (
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as tn,
|
trio.open_nursery() as tn,
|
||||||
msgops.maybe_limit_plds(
|
msgops.maybe_limit_plds(
|
||||||
ctx=ctx,
|
ctx=ctx,
|
||||||
|
|
@ -2173,11 +2005,7 @@ async def open_context_from_portal(
|
||||||
# the dialog, the `Error` msg should be raised from the `msg`
|
# the dialog, the `Error` msg should be raised from the `msg`
|
||||||
# handling block below.
|
# handling block below.
|
||||||
try:
|
try:
|
||||||
log.runtime(
|
started_msg, first = await ctx._pld_rx.recv_msg_w_pld(
|
||||||
f'IPC ctx parent waiting on Started msg..\n'
|
|
||||||
f'ctx.cid: {ctx.cid!r}\n'
|
|
||||||
)
|
|
||||||
started_msg, first = await ctx._pld_rx.recv_msg(
|
|
||||||
ipc=ctx,
|
ipc=ctx,
|
||||||
expect_msg=Started,
|
expect_msg=Started,
|
||||||
passthrough_non_pld_msgs=False,
|
passthrough_non_pld_msgs=False,
|
||||||
|
|
@ -2185,16 +2013,16 @@ async def open_context_from_portal(
|
||||||
)
|
)
|
||||||
except trio.Cancelled as taskc:
|
except trio.Cancelled as taskc:
|
||||||
ctx_cs: trio.CancelScope = ctx._scope
|
ctx_cs: trio.CancelScope = ctx._scope
|
||||||
log.cancel(
|
|
||||||
f'IPC ctx was cancelled during "child" task sync due to\n\n'
|
|
||||||
f'.cid: {ctx.cid!r}\n'
|
|
||||||
f'.maybe_error: {ctx.maybe_error!r}\n'
|
|
||||||
)
|
|
||||||
# await pause(shield=True)
|
|
||||||
|
|
||||||
if not ctx_cs.cancel_called:
|
if not ctx_cs.cancel_called:
|
||||||
raise
|
raise
|
||||||
|
|
||||||
|
# from .devx import pause
|
||||||
|
# await pause(shield=True)
|
||||||
|
|
||||||
|
log.cancel(
|
||||||
|
'IPC ctx was cancelled during "child" task sync due to\n\n'
|
||||||
|
f'{ctx.maybe_error}\n'
|
||||||
|
)
|
||||||
# OW if the ctx's scope was cancelled manually,
|
# OW if the ctx's scope was cancelled manually,
|
||||||
# likely the `Context` was cancelled via a call to
|
# likely the `Context` was cancelled via a call to
|
||||||
# `._maybe_cancel_and_set_remote_error()` so ensure
|
# `._maybe_cancel_and_set_remote_error()` so ensure
|
||||||
|
|
@ -2232,7 +2060,7 @@ async def open_context_from_portal(
|
||||||
# that we can re-use it around the `yield` ^ here
|
# that we can re-use it around the `yield` ^ here
|
||||||
# or vice versa?
|
# or vice versa?
|
||||||
#
|
#
|
||||||
# maybe TODO NOTE: between the parent exiting and
|
# maybe TODO NOTE: between the caller exiting and
|
||||||
# arriving here the far end may have sent a ctxc-msg or
|
# arriving here the far end may have sent a ctxc-msg or
|
||||||
# other error, so the quetion is whether we should check
|
# other error, so the quetion is whether we should check
|
||||||
# for it here immediately and maybe raise so as to engage
|
# for it here immediately and maybe raise so as to engage
|
||||||
|
|
@ -2298,16 +2126,16 @@ async def open_context_from_portal(
|
||||||
# request in which case we DO let the error bubble to the
|
# request in which case we DO let the error bubble to the
|
||||||
# opener.
|
# opener.
|
||||||
#
|
#
|
||||||
# 2-THIS "parent" task somewhere invoked `Context.cancel()`
|
# 2-THIS "caller" task somewhere invoked `Context.cancel()`
|
||||||
# and received a `ContextCanclled` from the "child"
|
# and received a `ContextCanclled` from the "callee"
|
||||||
# task, in which case we mask the `ContextCancelled` from
|
# task, in which case we mask the `ContextCancelled` from
|
||||||
# bubbling to this "parent" (much like how `trio.Nursery`
|
# bubbling to this "caller" (much like how `trio.Nursery`
|
||||||
# swallows any `trio.Cancelled` bubbled by a call to
|
# swallows any `trio.Cancelled` bubbled by a call to
|
||||||
# `Nursery.cancel_scope.cancel()`)
|
# `Nursery.cancel_scope.cancel()`)
|
||||||
except ContextCancelled as ctxc:
|
except ContextCancelled as ctxc:
|
||||||
scope_err = ctxc
|
scope_err = ctxc
|
||||||
ctx._local_error: BaseException = scope_err
|
ctx._local_error: BaseException = scope_err
|
||||||
ctxc_from_child = ctxc
|
ctxc_from_callee = ctxc
|
||||||
|
|
||||||
# XXX TODO XXX: FIX THIS debug_mode BUGGGG!!!
|
# XXX TODO XXX: FIX THIS debug_mode BUGGGG!!!
|
||||||
# using this code and then resuming the REPL will
|
# using this code and then resuming the REPL will
|
||||||
|
|
@ -2318,7 +2146,7 @@ async def open_context_from_portal(
|
||||||
# debugging the tractor-runtime itself using it's
|
# debugging the tractor-runtime itself using it's
|
||||||
# own `.devx.` tooling!
|
# own `.devx.` tooling!
|
||||||
#
|
#
|
||||||
# await debug.pause()
|
# await _debug.pause()
|
||||||
|
|
||||||
# CASE 2: context was cancelled by local task calling
|
# CASE 2: context was cancelled by local task calling
|
||||||
# `.cancel()`, we don't raise and the exit block should
|
# `.cancel()`, we don't raise and the exit block should
|
||||||
|
|
@ -2344,11 +2172,11 @@ async def open_context_from_portal(
|
||||||
# the above `._scope` can be cancelled due to:
|
# the above `._scope` can be cancelled due to:
|
||||||
# 1. an explicit self cancel via `Context.cancel()` or
|
# 1. an explicit self cancel via `Context.cancel()` or
|
||||||
# `Actor.cancel()`,
|
# `Actor.cancel()`,
|
||||||
# 2. any "child"-side remote error, possibly also a cancellation
|
# 2. any "callee"-side remote error, possibly also a cancellation
|
||||||
# request by some peer,
|
# request by some peer,
|
||||||
# 3. any "parent" (aka THIS scope's) local error raised in the above `yield`
|
# 3. any "caller" (aka THIS scope's) local error raised in the above `yield`
|
||||||
except (
|
except (
|
||||||
# CASE 3: standard local error in this parent/yieldee
|
# CASE 3: standard local error in this caller/yieldee
|
||||||
Exception,
|
Exception,
|
||||||
|
|
||||||
# CASES 1 & 2: can manifest as a `ctx._scope_nursery`
|
# CASES 1 & 2: can manifest as a `ctx._scope_nursery`
|
||||||
|
|
@ -2362,9 +2190,9 @@ async def open_context_from_portal(
|
||||||
# any `Context._maybe_raise_remote_err()` call.
|
# any `Context._maybe_raise_remote_err()` call.
|
||||||
#
|
#
|
||||||
# 2.-`BaseExceptionGroup[ContextCancelled | RemoteActorError]`
|
# 2.-`BaseExceptionGroup[ContextCancelled | RemoteActorError]`
|
||||||
# from any error delivered from the "child" side
|
# from any error delivered from the "callee" side
|
||||||
# AND a group-exc is only raised if there was > 1
|
# AND a group-exc is only raised if there was > 1
|
||||||
# tasks started *here* in the "parent" / opener
|
# tasks started *here* in the "caller" / opener
|
||||||
# block. If any one of those tasks calls
|
# block. If any one of those tasks calls
|
||||||
# `.wait_for_result()` or `MsgStream.receive()`
|
# `.wait_for_result()` or `MsgStream.receive()`
|
||||||
# `._maybe_raise_remote_err()` will be transitively
|
# `._maybe_raise_remote_err()` will be transitively
|
||||||
|
|
@ -2377,42 +2205,39 @@ async def open_context_from_portal(
|
||||||
trio.Cancelled, # NOTE: NOT from inside the ctx._scope
|
trio.Cancelled, # NOTE: NOT from inside the ctx._scope
|
||||||
KeyboardInterrupt,
|
KeyboardInterrupt,
|
||||||
|
|
||||||
) as rent_err:
|
) as caller_err:
|
||||||
scope_err = rent_err
|
scope_err = caller_err
|
||||||
ctx._local_error: BaseException = scope_err
|
ctx._local_error: BaseException = scope_err
|
||||||
|
|
||||||
# XXX: ALWAYS request the context to CANCEL ON any ERROR.
|
# XXX: ALWAYS request the context to CANCEL ON any ERROR.
|
||||||
# NOTE: `Context.cancel()` is conversely NEVER CALLED in
|
# NOTE: `Context.cancel()` is conversely NEVER CALLED in
|
||||||
# the `ContextCancelled` "self cancellation absorbed" case
|
# the `ContextCancelled` "self cancellation absorbed" case
|
||||||
# handled in the block above ^^^ !!
|
# handled in the block above ^^^ !!
|
||||||
# await debug.pause()
|
# await _debug.pause()
|
||||||
# log.cancel(
|
# log.cancel(
|
||||||
match scope_err:
|
match scope_err:
|
||||||
case trio.Cancelled():
|
case trio.Cancelled:
|
||||||
logmeth = log.cancel
|
logmeth = log.cancel
|
||||||
cause: str = 'cancelled'
|
|
||||||
|
|
||||||
# XXX explicitly report on any non-graceful-taskc cases
|
# XXX explicitly report on any non-graceful-taskc cases
|
||||||
case _:
|
case _:
|
||||||
cause: str = 'errored'
|
|
||||||
logmeth = log.exception
|
logmeth = log.exception
|
||||||
|
|
||||||
logmeth(
|
logmeth(
|
||||||
f'ctx {ctx.side!r}-side {cause!r} with,\n'
|
f'ctx {ctx.side!r}-side exited with {ctx.repr_outcome()}\n'
|
||||||
f'{ctx.repr_outcome()!r}\n'
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if debug_mode():
|
if debug_mode():
|
||||||
# async with debug.acquire_debug_lock(portal.actor.uid):
|
# async with _debug.acquire_debug_lock(portal.actor.uid):
|
||||||
# pass
|
# pass
|
||||||
# TODO: factor ^ into below for non-root cases?
|
# TODO: factor ^ into below for non-root cases?
|
||||||
#
|
#
|
||||||
from .devx.debug import maybe_wait_for_debugger
|
from .devx import maybe_wait_for_debugger
|
||||||
was_acquired: bool = await maybe_wait_for_debugger(
|
was_acquired: bool = await maybe_wait_for_debugger(
|
||||||
# header_msg=(
|
header_msg=(
|
||||||
# 'Delaying `ctx.cancel()` until debug lock '
|
'Delaying `ctx.cancel()` until debug lock '
|
||||||
# 'acquired..\n'
|
'acquired..\n'
|
||||||
# ),
|
),
|
||||||
)
|
)
|
||||||
if was_acquired:
|
if was_acquired:
|
||||||
log.pdb(
|
log.pdb(
|
||||||
|
|
@ -2420,11 +2245,10 @@ async def open_context_from_portal(
|
||||||
'Calling `ctx.cancel()`!\n'
|
'Calling `ctx.cancel()`!\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
# we don't need to cancel the child if it already
|
# we don't need to cancel the callee if it already
|
||||||
# told us it's cancelled ;p
|
# told us it's cancelled ;p
|
||||||
if ctxc_from_child is None:
|
if ctxc_from_callee is None:
|
||||||
try:
|
try:
|
||||||
# await pause(shield=True)
|
|
||||||
await ctx.cancel()
|
await ctx.cancel()
|
||||||
except (
|
except (
|
||||||
trio.BrokenResourceError,
|
trio.BrokenResourceError,
|
||||||
|
|
@ -2454,8 +2278,8 @@ async def open_context_from_portal(
|
||||||
# via a call to
|
# via a call to
|
||||||
# `Context._maybe_cancel_and_set_remote_error()`.
|
# `Context._maybe_cancel_and_set_remote_error()`.
|
||||||
# As per `Context._deliver_msg()`, that error IS
|
# As per `Context._deliver_msg()`, that error IS
|
||||||
# ALWAYS SET any time "child" side fails and causes
|
# ALWAYS SET any time "callee" side fails and causes "caller
|
||||||
# "parent side" cancellation via a `ContextCancelled` here.
|
# side" cancellation via a `ContextCancelled` here.
|
||||||
try:
|
try:
|
||||||
result_or_err: Exception|Any = await ctx.wait_for_result()
|
result_or_err: Exception|Any = await ctx.wait_for_result()
|
||||||
except BaseException as berr:
|
except BaseException as berr:
|
||||||
|
|
@ -2471,8 +2295,8 @@ async def open_context_from_portal(
|
||||||
raise
|
raise
|
||||||
|
|
||||||
# yes this worx!
|
# yes this worx!
|
||||||
# from .devx import debug
|
# from .devx import _debug
|
||||||
# await debug.pause()
|
# await _debug.pause()
|
||||||
|
|
||||||
# an exception type boxed in a `RemoteActorError`
|
# an exception type boxed in a `RemoteActorError`
|
||||||
# is returned (meaning it was obvi not raised)
|
# is returned (meaning it was obvi not raised)
|
||||||
|
|
@ -2491,7 +2315,7 @@ async def open_context_from_portal(
|
||||||
)
|
)
|
||||||
case (None, _):
|
case (None, _):
|
||||||
log.runtime(
|
log.runtime(
|
||||||
'Context returned final result from child task:\n'
|
'Context returned final result from callee task:\n'
|
||||||
f'<= peer: {uid}\n'
|
f'<= peer: {uid}\n'
|
||||||
f' |_ {nsf}()\n\n'
|
f' |_ {nsf}()\n\n'
|
||||||
|
|
||||||
|
|
@ -2507,7 +2331,7 @@ async def open_context_from_portal(
|
||||||
# where the root is waiting on the lock to clear but the
|
# where the root is waiting on the lock to clear but the
|
||||||
# child has already cleared it and clobbered IPC.
|
# child has already cleared it and clobbered IPC.
|
||||||
if debug_mode():
|
if debug_mode():
|
||||||
from .devx.debug import maybe_wait_for_debugger
|
from .devx import maybe_wait_for_debugger
|
||||||
await maybe_wait_for_debugger()
|
await maybe_wait_for_debugger()
|
||||||
|
|
||||||
# though it should be impossible for any tasks
|
# though it should be impossible for any tasks
|
||||||
|
|
@ -2546,8 +2370,7 @@ async def open_context_from_portal(
|
||||||
# displaying `ContextCancelled` traces where the
|
# displaying `ContextCancelled` traces where the
|
||||||
# cause of crash/exit IS due to something in
|
# cause of crash/exit IS due to something in
|
||||||
# user/app code on either end of the context.
|
# user/app code on either end of the context.
|
||||||
and
|
and not rxchan._closed
|
||||||
not rxchan._closed
|
|
||||||
):
|
):
|
||||||
# XXX NOTE XXX: and again as per above, we mask any
|
# XXX NOTE XXX: and again as per above, we mask any
|
||||||
# `trio.Cancelled` raised here so as to NOT mask
|
# `trio.Cancelled` raised here so as to NOT mask
|
||||||
|
|
@ -2581,14 +2404,12 @@ async def open_context_from_portal(
|
||||||
log.cancel(
|
log.cancel(
|
||||||
f'Context cancelled by local {ctx.side!r}-side task\n'
|
f'Context cancelled by local {ctx.side!r}-side task\n'
|
||||||
f'c)>\n'
|
f'c)>\n'
|
||||||
f' |_{ctx.parent_task}\n'
|
f' |_{ctx._task}\n\n'
|
||||||
f' .cid={ctx.cid!r}\n'
|
f'{repr(scope_err)}\n'
|
||||||
f'\n'
|
|
||||||
f'{scope_err!r}\n'
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# TODO: should we add a `._cancel_req_received`
|
# TODO: should we add a `._cancel_req_received`
|
||||||
# flag to determine if the child manually called
|
# flag to determine if the callee manually called
|
||||||
# `ctx.cancel()`?
|
# `ctx.cancel()`?
|
||||||
# -[ ] going to need a cid check no?
|
# -[ ] going to need a cid check no?
|
||||||
|
|
||||||
|
|
@ -2608,7 +2429,6 @@ async def open_context_from_portal(
|
||||||
# FINALLY, remove the context from runtime tracking and
|
# FINALLY, remove the context from runtime tracking and
|
||||||
# exit!
|
# exit!
|
||||||
log.runtime(
|
log.runtime(
|
||||||
# log.cancel(
|
|
||||||
f'De-allocating IPC ctx opened with {ctx.side!r} peer \n'
|
f'De-allocating IPC ctx opened with {ctx.side!r} peer \n'
|
||||||
f'uid: {uid}\n'
|
f'uid: {uid}\n'
|
||||||
f'cid: {ctx.cid}\n'
|
f'cid: {ctx.cid}\n'
|
||||||
|
|
@ -2644,7 +2464,7 @@ def mk_context(
|
||||||
recv_chan: trio.MemoryReceiveChannel
|
recv_chan: trio.MemoryReceiveChannel
|
||||||
send_chan, recv_chan = trio.open_memory_channel(msg_buffer_size)
|
send_chan, recv_chan = trio.open_memory_channel(msg_buffer_size)
|
||||||
|
|
||||||
# TODO: only scan parent-info if log level so high!
|
# TODO: only scan caller-info if log level so high!
|
||||||
from .devx._frame_stack import find_caller_info
|
from .devx._frame_stack import find_caller_info
|
||||||
caller_info: CallerInfo|None = find_caller_info()
|
caller_info: CallerInfo|None = find_caller_info()
|
||||||
|
|
||||||
|
|
@ -2664,6 +2484,7 @@ def mk_context(
|
||||||
_caller_info=caller_info,
|
_caller_info=caller_info,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
)
|
)
|
||||||
|
pld_rx._ctx = ctx
|
||||||
ctx._result = Unresolved
|
ctx._result = Unresolved
|
||||||
return ctx
|
return ctx
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -28,16 +28,8 @@ from typing import (
|
||||||
from contextlib import asynccontextmanager as acm
|
from contextlib import asynccontextmanager as acm
|
||||||
|
|
||||||
from tractor.log import get_logger
|
from tractor.log import get_logger
|
||||||
from .trionics import (
|
from .trionics import gather_contexts
|
||||||
gather_contexts,
|
from ._ipc import _connect_chan, Channel
|
||||||
collapse_eg,
|
|
||||||
)
|
|
||||||
from .ipc import _connect_chan, Channel
|
|
||||||
from ._addr import (
|
|
||||||
UnwrappedAddress,
|
|
||||||
Address,
|
|
||||||
wrap_address
|
|
||||||
)
|
|
||||||
from ._portal import (
|
from ._portal import (
|
||||||
Portal,
|
Portal,
|
||||||
open_portal,
|
open_portal,
|
||||||
|
|
@ -46,7 +38,6 @@ from ._portal import (
|
||||||
from ._state import (
|
from ._state import (
|
||||||
current_actor,
|
current_actor,
|
||||||
_runtime_vars,
|
_runtime_vars,
|
||||||
_def_tpt_proto,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
|
|
@ -58,7 +49,9 @@ log = get_logger(__name__)
|
||||||
|
|
||||||
@acm
|
@acm
|
||||||
async def get_registry(
|
async def get_registry(
|
||||||
addr: UnwrappedAddress|None = None,
|
host: str,
|
||||||
|
port: int,
|
||||||
|
|
||||||
) -> AsyncGenerator[
|
) -> AsyncGenerator[
|
||||||
Portal | LocalPortal | None,
|
Portal | LocalPortal | None,
|
||||||
None,
|
None,
|
||||||
|
|
@ -76,20 +69,19 @@ async def get_registry(
|
||||||
# (likely a re-entrant call from the arbiter actor)
|
# (likely a re-entrant call from the arbiter actor)
|
||||||
yield LocalPortal(
|
yield LocalPortal(
|
||||||
actor,
|
actor,
|
||||||
Channel(transport=None)
|
Channel((host, port))
|
||||||
# ^XXX, we DO NOT actually provide nor connect an
|
|
||||||
# underlying transport since this is merely an API shim.
|
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
# TODO: try to look pre-existing connection from
|
# TODO: try to look pre-existing connection from
|
||||||
# `Server._peers` and use it instead?
|
# `Actor._peers` and use it instead?
|
||||||
async with (
|
async with (
|
||||||
_connect_chan(addr) as chan,
|
_connect_chan(host, port) as chan,
|
||||||
open_portal(chan) as regstr_ptl,
|
open_portal(chan) as regstr_ptl,
|
||||||
):
|
):
|
||||||
yield regstr_ptl
|
yield regstr_ptl
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
@acm
|
||||||
async def get_root(
|
async def get_root(
|
||||||
**kwargs,
|
**kwargs,
|
||||||
|
|
@ -97,10 +89,11 @@ async def get_root(
|
||||||
|
|
||||||
# TODO: rename mailbox to `_root_maddr` when we finally
|
# TODO: rename mailbox to `_root_maddr` when we finally
|
||||||
# add and impl libp2p multi-addrs?
|
# add and impl libp2p multi-addrs?
|
||||||
addr = _runtime_vars['_root_mailbox']
|
host, port = _runtime_vars['_root_mailbox']
|
||||||
|
assert host is not None
|
||||||
|
|
||||||
async with (
|
async with (
|
||||||
_connect_chan(addr) as chan,
|
_connect_chan(host, port) as chan,
|
||||||
open_portal(chan, **kwargs) as portal,
|
open_portal(chan, **kwargs) as portal,
|
||||||
):
|
):
|
||||||
yield portal
|
yield portal
|
||||||
|
|
@ -113,23 +106,17 @@ def get_peer_by_name(
|
||||||
) -> list[Channel]|None: # at least 1
|
) -> list[Channel]|None: # at least 1
|
||||||
'''
|
'''
|
||||||
Scan for an existing connection (set) to a named actor
|
Scan for an existing connection (set) to a named actor
|
||||||
and return any channels from `Server._peers: dict`.
|
and return any channels from `Actor._peers`.
|
||||||
|
|
||||||
This is an optimization method over querying the registrar for
|
This is an optimization method over querying the registrar for
|
||||||
the same info.
|
the same info.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
actor: Actor = current_actor()
|
actor: Actor = current_actor()
|
||||||
to_scan: dict[tuple, list[Channel]] = actor.ipc_server._peers.copy()
|
to_scan: dict[tuple, list[Channel]] = actor._peers.copy()
|
||||||
|
pchan: Channel|None = actor._parent_chan
|
||||||
# TODO: is this ever needed? creates a duplicate channel on actor._peers
|
if pchan:
|
||||||
# when multiple find_actor calls are made to same actor from a single ctx
|
to_scan[pchan.uid].append(pchan)
|
||||||
# which causes actor exit to hang waiting forever on
|
|
||||||
# `actor._no_more_peers.wait()` in `_runtime.async_main`
|
|
||||||
|
|
||||||
# pchan: Channel|None = actor._parent_chan
|
|
||||||
# if pchan and pchan.uid not in to_scan:
|
|
||||||
# to_scan[pchan.uid].append(pchan)
|
|
||||||
|
|
||||||
for aid, chans in to_scan.items():
|
for aid, chans in to_scan.items():
|
||||||
_, peer_name = aid
|
_, peer_name = aid
|
||||||
|
|
@ -147,10 +134,10 @@ def get_peer_by_name(
|
||||||
@acm
|
@acm
|
||||||
async def query_actor(
|
async def query_actor(
|
||||||
name: str,
|
name: str,
|
||||||
regaddr: UnwrappedAddress|None = None,
|
regaddr: tuple[str, int]|None = None,
|
||||||
|
|
||||||
) -> AsyncGenerator[
|
) -> AsyncGenerator[
|
||||||
UnwrappedAddress|None,
|
tuple[str, int]|None,
|
||||||
None,
|
None,
|
||||||
]:
|
]:
|
||||||
'''
|
'''
|
||||||
|
|
@ -176,31 +163,31 @@ async def query_actor(
|
||||||
return
|
return
|
||||||
|
|
||||||
reg_portal: Portal
|
reg_portal: Portal
|
||||||
regaddr: Address = wrap_address(regaddr) or actor.reg_addrs[0]
|
regaddr: tuple[str, int] = regaddr or actor.reg_addrs[0]
|
||||||
async with get_registry(regaddr) as reg_portal:
|
async with get_registry(*regaddr) as reg_portal:
|
||||||
# TODO: return portals to all available actors - for now
|
# TODO: return portals to all available actors - for now
|
||||||
# just the last one that registered
|
# just the last one that registered
|
||||||
addr: UnwrappedAddress = await reg_portal.run_from_ns(
|
sockaddr: tuple[str, int] = await reg_portal.run_from_ns(
|
||||||
'self',
|
'self',
|
||||||
'find_actor',
|
'find_actor',
|
||||||
name=name,
|
name=name,
|
||||||
)
|
)
|
||||||
yield addr
|
yield sockaddr
|
||||||
|
|
||||||
|
|
||||||
@acm
|
@acm
|
||||||
async def maybe_open_portal(
|
async def maybe_open_portal(
|
||||||
addr: UnwrappedAddress,
|
addr: tuple[str, int],
|
||||||
name: str,
|
name: str,
|
||||||
):
|
):
|
||||||
async with query_actor(
|
async with query_actor(
|
||||||
name=name,
|
name=name,
|
||||||
regaddr=addr,
|
regaddr=addr,
|
||||||
) as addr:
|
) as sockaddr:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
if addr:
|
if sockaddr:
|
||||||
async with _connect_chan(addr) as chan:
|
async with _connect_chan(*sockaddr) as chan:
|
||||||
async with open_portal(chan) as portal:
|
async with open_portal(chan) as portal:
|
||||||
yield portal
|
yield portal
|
||||||
else:
|
else:
|
||||||
|
|
@ -210,8 +197,7 @@ async def maybe_open_portal(
|
||||||
@acm
|
@acm
|
||||||
async def find_actor(
|
async def find_actor(
|
||||||
name: str,
|
name: str,
|
||||||
registry_addrs: list[UnwrappedAddress]|None = None,
|
registry_addrs: list[tuple[str, int]]|None = None,
|
||||||
enable_transports: list[str] = [_def_tpt_proto],
|
|
||||||
|
|
||||||
only_first: bool = True,
|
only_first: bool = True,
|
||||||
raise_on_none: bool = False,
|
raise_on_none: bool = False,
|
||||||
|
|
@ -238,15 +224,15 @@ async def find_actor(
|
||||||
# XXX NOTE: make sure to dynamically read the value on
|
# XXX NOTE: make sure to dynamically read the value on
|
||||||
# every call since something may change it globally (eg.
|
# every call since something may change it globally (eg.
|
||||||
# like in our discovery test suite)!
|
# like in our discovery test suite)!
|
||||||
from ._addr import default_lo_addrs
|
from . import _root
|
||||||
registry_addrs = (
|
registry_addrs = (
|
||||||
_runtime_vars['_registry_addrs']
|
_runtime_vars['_registry_addrs']
|
||||||
or
|
or
|
||||||
default_lo_addrs(enable_transports)
|
_root._default_lo_addrs
|
||||||
)
|
)
|
||||||
|
|
||||||
maybe_portals: list[
|
maybe_portals: list[
|
||||||
AsyncContextManager[UnwrappedAddress]
|
AsyncContextManager[tuple[str, int]]
|
||||||
] = list(
|
] = list(
|
||||||
maybe_open_portal(
|
maybe_open_portal(
|
||||||
addr=addr,
|
addr=addr,
|
||||||
|
|
@ -255,12 +241,9 @@ async def find_actor(
|
||||||
for addr in registry_addrs
|
for addr in registry_addrs
|
||||||
)
|
)
|
||||||
portals: list[Portal]
|
portals: list[Portal]
|
||||||
async with (
|
async with gather_contexts(
|
||||||
collapse_eg(),
|
|
||||||
gather_contexts(
|
|
||||||
mngrs=maybe_portals,
|
mngrs=maybe_portals,
|
||||||
) as portals,
|
) as portals:
|
||||||
):
|
|
||||||
# log.runtime(
|
# log.runtime(
|
||||||
# 'Gathered portals:\n'
|
# 'Gathered portals:\n'
|
||||||
# f'{portals}'
|
# f'{portals}'
|
||||||
|
|
@ -291,7 +274,7 @@ async def find_actor(
|
||||||
@acm
|
@acm
|
||||||
async def wait_for_actor(
|
async def wait_for_actor(
|
||||||
name: str,
|
name: str,
|
||||||
registry_addr: UnwrappedAddress | None = None,
|
registry_addr: tuple[str, int] | None = None,
|
||||||
|
|
||||||
) -> AsyncGenerator[Portal, None]:
|
) -> AsyncGenerator[Portal, None]:
|
||||||
'''
|
'''
|
||||||
|
|
@ -308,7 +291,7 @@ async def wait_for_actor(
|
||||||
yield peer_portal
|
yield peer_portal
|
||||||
return
|
return
|
||||||
|
|
||||||
regaddr: UnwrappedAddress = (
|
regaddr: tuple[str, int] = (
|
||||||
registry_addr
|
registry_addr
|
||||||
or
|
or
|
||||||
actor.reg_addrs[0]
|
actor.reg_addrs[0]
|
||||||
|
|
@ -316,8 +299,8 @@ async def wait_for_actor(
|
||||||
# TODO: use `.trionics.gather_contexts()` like
|
# TODO: use `.trionics.gather_contexts()` like
|
||||||
# above in `find_actor()` as well?
|
# above in `find_actor()` as well?
|
||||||
reg_portal: Portal
|
reg_portal: Portal
|
||||||
async with get_registry(regaddr) as reg_portal:
|
async with get_registry(*regaddr) as reg_portal:
|
||||||
addrs = await reg_portal.run_from_ns(
|
sockaddrs = await reg_portal.run_from_ns(
|
||||||
'self',
|
'self',
|
||||||
'wait_for_actor',
|
'wait_for_actor',
|
||||||
name=name,
|
name=name,
|
||||||
|
|
@ -325,8 +308,8 @@ async def wait_for_actor(
|
||||||
|
|
||||||
# get latest registered addr by default?
|
# get latest registered addr by default?
|
||||||
# TODO: offer multi-portal yields in multi-homed case?
|
# TODO: offer multi-portal yields in multi-homed case?
|
||||||
addr: UnwrappedAddress = addrs[-1]
|
sockaddr: tuple[str, int] = sockaddrs[-1]
|
||||||
|
|
||||||
async with _connect_chan(addr) as chan:
|
async with _connect_chan(*sockaddr) as chan:
|
||||||
async with open_portal(chan) as portal:
|
async with open_portal(chan) as portal:
|
||||||
yield portal
|
yield portal
|
||||||
|
|
|
||||||
|
|
@ -21,7 +21,8 @@ Sub-process entry points.
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
from functools import partial
|
from functools import partial
|
||||||
import multiprocessing as mp
|
import multiprocessing as mp
|
||||||
# import os
|
import os
|
||||||
|
import textwrap
|
||||||
from typing import (
|
from typing import (
|
||||||
Any,
|
Any,
|
||||||
TYPE_CHECKING,
|
TYPE_CHECKING,
|
||||||
|
|
@ -34,13 +35,8 @@ from .log import (
|
||||||
get_logger,
|
get_logger,
|
||||||
)
|
)
|
||||||
from . import _state
|
from . import _state
|
||||||
from .devx import (
|
from .devx import _debug
|
||||||
_frame_stack,
|
|
||||||
pformat,
|
|
||||||
)
|
|
||||||
# from .msg import pretty_struct
|
|
||||||
from .to_asyncio import run_as_asyncio_guest
|
from .to_asyncio import run_as_asyncio_guest
|
||||||
from ._addr import UnwrappedAddress
|
|
||||||
from ._runtime import (
|
from ._runtime import (
|
||||||
async_main,
|
async_main,
|
||||||
Actor,
|
Actor,
|
||||||
|
|
@ -56,10 +52,10 @@ log = get_logger(__name__)
|
||||||
def _mp_main(
|
def _mp_main(
|
||||||
|
|
||||||
actor: Actor,
|
actor: Actor,
|
||||||
accept_addrs: list[UnwrappedAddress],
|
accept_addrs: list[tuple[str, int]],
|
||||||
forkserver_info: tuple[Any, Any, Any, Any, Any],
|
forkserver_info: tuple[Any, Any, Any, Any, Any],
|
||||||
start_method: SpawnMethodKey,
|
start_method: SpawnMethodKey,
|
||||||
parent_addr: UnwrappedAddress | None = None,
|
parent_addr: tuple[str, int] | None = None,
|
||||||
infect_asyncio: bool = False,
|
infect_asyncio: bool = False,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
@ -106,10 +102,111 @@ def _mp_main(
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: move this func to some kinda `.devx._conc_lang.py` eventually
|
||||||
|
# as we work out our multi-domain state-flow-syntax!
|
||||||
|
def nest_from_op(
|
||||||
|
input_op: str,
|
||||||
|
#
|
||||||
|
# ?TODO? an idea for a syntax to the state of concurrent systems
|
||||||
|
# as a "3-domain" (execution, scope, storage) model and using
|
||||||
|
# a minimal ascii/utf-8 operator-set.
|
||||||
|
#
|
||||||
|
# try not to take any of this seriously yet XD
|
||||||
|
#
|
||||||
|
# > is a "play operator" indicating (CPU bound)
|
||||||
|
# exec/work/ops required at the "lowest level computing"
|
||||||
|
#
|
||||||
|
# execution primititves (tasks, threads, actors..) denote their
|
||||||
|
# lifetime with '(' and ')' since parentheses normally are used
|
||||||
|
# in many langs to denote function calls.
|
||||||
|
#
|
||||||
|
# starting = (
|
||||||
|
# >( opening/starting; beginning of the thread-of-exec (toe?)
|
||||||
|
# (> opened/started, (finished spawning toe)
|
||||||
|
# |_<Task: blah blah..> repr of toe, in py these look like <objs>
|
||||||
|
#
|
||||||
|
# >) closing/exiting/stopping,
|
||||||
|
# )> closed/exited/stopped,
|
||||||
|
# |_<Task: blah blah..>
|
||||||
|
# [OR <), )< ?? ]
|
||||||
|
#
|
||||||
|
# ending = )
|
||||||
|
# >c) cancelling to close/exit
|
||||||
|
# c)> cancelled (caused close), OR?
|
||||||
|
# |_<Actor: ..>
|
||||||
|
# OR maybe "<c)" which better indicates the cancel being
|
||||||
|
# "delivered/returned" / returned" to LHS?
|
||||||
|
#
|
||||||
|
# >x) erroring to eventuall exit
|
||||||
|
# x)> errored and terminated
|
||||||
|
# |_<Actor: ...>
|
||||||
|
#
|
||||||
|
# scopes: supers/nurseries, IPC-ctxs, sessions, perms, etc.
|
||||||
|
# >{ opening
|
||||||
|
# {> opened
|
||||||
|
# }> closed
|
||||||
|
# >} closing
|
||||||
|
#
|
||||||
|
# storage: like queues, shm-buffers, files, etc..
|
||||||
|
# >[ opening
|
||||||
|
# [> opened
|
||||||
|
# |_<FileObj: ..>
|
||||||
|
#
|
||||||
|
# >] closing
|
||||||
|
# ]> closed
|
||||||
|
|
||||||
|
# IPC ops: channels, transports, msging
|
||||||
|
# => req msg
|
||||||
|
# <= resp msg
|
||||||
|
# <=> 2-way streaming (of msgs)
|
||||||
|
# <- recv 1 msg
|
||||||
|
# -> send 1 msg
|
||||||
|
#
|
||||||
|
# TODO: still not sure on R/L-HS approach..?
|
||||||
|
# =>( send-req to exec start (task, actor, thread..)
|
||||||
|
# (<= recv-req to ^
|
||||||
|
#
|
||||||
|
# (<= recv-req ^
|
||||||
|
# <=( recv-resp opened remote exec primitive
|
||||||
|
# <=) recv-resp closed
|
||||||
|
#
|
||||||
|
# )<=c req to stop due to cancel
|
||||||
|
# c=>) req to stop due to cancel
|
||||||
|
#
|
||||||
|
# =>{ recv-req to open
|
||||||
|
# <={ send-status that it closed
|
||||||
|
|
||||||
|
tree_str: str,
|
||||||
|
|
||||||
|
# NOTE: so move back-from-the-left of the `input_op` by
|
||||||
|
# this amount.
|
||||||
|
back_from_op: int = 0,
|
||||||
|
) -> str:
|
||||||
|
'''
|
||||||
|
Depth-increment the input (presumably hierarchy/supervision)
|
||||||
|
input "tree string" below the provided `input_op` execution
|
||||||
|
operator, so injecting a `"\n|_{input_op}\n"`and indenting the
|
||||||
|
`tree_str` to nest content aligned with the ops last char.
|
||||||
|
|
||||||
|
'''
|
||||||
|
return (
|
||||||
|
f'{input_op}\n'
|
||||||
|
+
|
||||||
|
textwrap.indent(
|
||||||
|
tree_str,
|
||||||
|
prefix=(
|
||||||
|
len(input_op)
|
||||||
|
-
|
||||||
|
(back_from_op + 1)
|
||||||
|
) * ' ',
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def _trio_main(
|
def _trio_main(
|
||||||
actor: Actor,
|
actor: Actor,
|
||||||
*,
|
*,
|
||||||
parent_addr: UnwrappedAddress|None = None,
|
parent_addr: tuple[str, int] | None = None,
|
||||||
infect_asyncio: bool = False,
|
infect_asyncio: bool = False,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
@ -117,7 +214,7 @@ def _trio_main(
|
||||||
Entry point for a `trio_run_in_process` subactor.
|
Entry point for a `trio_run_in_process` subactor.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
_frame_stack.hide_runtime_frames()
|
_debug.hide_runtime_frames()
|
||||||
|
|
||||||
_state._current_actor = actor
|
_state._current_actor = actor
|
||||||
trio_main = partial(
|
trio_main = partial(
|
||||||
|
|
@ -128,23 +225,30 @@ def _trio_main(
|
||||||
|
|
||||||
if actor.loglevel is not None:
|
if actor.loglevel is not None:
|
||||||
get_console_log(actor.loglevel)
|
get_console_log(actor.loglevel)
|
||||||
|
actor_info: str = (
|
||||||
|
f'|_{actor}\n'
|
||||||
|
f' uid: {actor.uid}\n'
|
||||||
|
f' pid: {os.getpid()}\n'
|
||||||
|
f' parent_addr: {parent_addr}\n'
|
||||||
|
f' loglevel: {actor.loglevel}\n'
|
||||||
|
)
|
||||||
log.info(
|
log.info(
|
||||||
f'Starting `trio` subactor from parent @ '
|
'Starting new `trio` subactor:\n'
|
||||||
f'{parent_addr}\n'
|
|
||||||
+
|
+
|
||||||
pformat.nest_from_op(
|
nest_from_op(
|
||||||
input_op='>(', # see syntax ideas above
|
input_op='>(', # see syntax ideas above
|
||||||
text=f'{actor}',
|
tree_str=actor_info,
|
||||||
|
back_from_op=1,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
logmeth = log.info
|
logmeth = log.info
|
||||||
exit_status: str = (
|
exit_status: str = (
|
||||||
'Subactor exited\n'
|
'Subactor exited\n'
|
||||||
+
|
+
|
||||||
pformat.nest_from_op(
|
nest_from_op(
|
||||||
input_op=')>', # like a "closed-to-play"-icon from super perspective
|
input_op=')>', # like a "closed-to-play"-icon from super perspective
|
||||||
text=f'{actor}',
|
tree_str=actor_info,
|
||||||
nest_indent=1,
|
back_from_op=1,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
|
|
@ -159,9 +263,9 @@ def _trio_main(
|
||||||
exit_status: str = (
|
exit_status: str = (
|
||||||
'Actor received KBI (aka an OS-cancel)\n'
|
'Actor received KBI (aka an OS-cancel)\n'
|
||||||
+
|
+
|
||||||
pformat.nest_from_op(
|
nest_from_op(
|
||||||
input_op='c)>', # closed due to cancel (see above)
|
input_op='c)>', # closed due to cancel (see above)
|
||||||
text=f'{actor}',
|
tree_str=actor_info,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
except BaseException as err:
|
except BaseException as err:
|
||||||
|
|
@ -169,9 +273,9 @@ def _trio_main(
|
||||||
exit_status: str = (
|
exit_status: str = (
|
||||||
'Main actor task exited due to crash?\n'
|
'Main actor task exited due to crash?\n'
|
||||||
+
|
+
|
||||||
pformat.nest_from_op(
|
nest_from_op(
|
||||||
input_op='x)>', # closed by error
|
input_op='x)>', # closed by error
|
||||||
text=f'{actor}',
|
tree_str=actor_info,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
# NOTE since we raise a tb will already be shown on the
|
# NOTE since we raise a tb will already be shown on the
|
||||||
|
|
|
||||||
|
|
@ -22,7 +22,7 @@ from __future__ import annotations
|
||||||
import builtins
|
import builtins
|
||||||
import importlib
|
import importlib
|
||||||
from pprint import pformat
|
from pprint import pformat
|
||||||
from pdb import bdb
|
import sys
|
||||||
from types import (
|
from types import (
|
||||||
TracebackType,
|
TracebackType,
|
||||||
)
|
)
|
||||||
|
|
@ -64,29 +64,15 @@ if TYPE_CHECKING:
|
||||||
from ._context import Context
|
from ._context import Context
|
||||||
from .log import StackLevelAdapter
|
from .log import StackLevelAdapter
|
||||||
from ._stream import MsgStream
|
from ._stream import MsgStream
|
||||||
from .ipc import Channel
|
from ._ipc import Channel
|
||||||
|
|
||||||
log = get_logger('tractor')
|
log = get_logger('tractor')
|
||||||
|
|
||||||
_this_mod = importlib.import_module(__name__)
|
_this_mod = importlib.import_module(__name__)
|
||||||
|
|
||||||
|
|
||||||
class RuntimeFailure(RuntimeError):
|
class ActorFailure(Exception):
|
||||||
'''
|
"General actor failure"
|
||||||
General `Actor`-runtime failure due to,
|
|
||||||
|
|
||||||
- a bad runtime-env,
|
|
||||||
- falied spawning (bad input to process),
|
|
||||||
- API usage.
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
|
||||||
class ActorFailure(RuntimeFailure):
|
|
||||||
'''
|
|
||||||
`Actor` failed to boot before/after spawn
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
|
||||||
class InternalError(RuntimeError):
|
class InternalError(RuntimeError):
|
||||||
|
|
@ -96,54 +82,6 @@ class InternalError(RuntimeError):
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
|
||||||
class AsyncioCancelled(Exception):
|
|
||||||
'''
|
|
||||||
Asyncio cancelled translation (non-base) error
|
|
||||||
for use with the ``to_asyncio`` module
|
|
||||||
to be raised in the ``trio`` side task
|
|
||||||
|
|
||||||
NOTE: this should NOT inherit from `asyncio.CancelledError` or
|
|
||||||
tests should break!
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
|
||||||
class AsyncioTaskExited(Exception):
|
|
||||||
'''
|
|
||||||
asyncio.Task "exited" translation error for use with the
|
|
||||||
`to_asyncio` APIs to be raised in the `trio` side task indicating
|
|
||||||
on `.run_task()`/`.open_channel_from()` exit that the aio side
|
|
||||||
exited early/silently.
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
class TrioCancelled(Exception):
|
|
||||||
'''
|
|
||||||
Trio cancelled translation (non-base) error
|
|
||||||
for use with the `to_asyncio` module
|
|
||||||
to be raised in the `asyncio.Task` to indicate
|
|
||||||
that the `trio` side raised `Cancelled` or an error.
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
class TrioTaskExited(Exception):
|
|
||||||
'''
|
|
||||||
The `trio`-side task exited without explicitly cancelling the
|
|
||||||
`asyncio.Task` peer.
|
|
||||||
|
|
||||||
This is very similar to how `trio.ClosedResource` acts as
|
|
||||||
a "clean shutdown" signal to the consumer side of a mem-chan,
|
|
||||||
|
|
||||||
https://trio.readthedocs.io/en/stable/reference-core.html#clean-shutdown-with-channels
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
|
|
||||||
class DebugRequestError(RuntimeError):
|
|
||||||
'''
|
|
||||||
Failed to request stdio lock from root actor!
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
# NOTE: more or less should be close to these:
|
# NOTE: more or less should be close to these:
|
||||||
# 'boxed_type',
|
# 'boxed_type',
|
||||||
|
|
@ -189,8 +127,8 @@ _body_fields: list[str] = list(
|
||||||
|
|
||||||
def get_err_type(type_name: str) -> BaseException|None:
|
def get_err_type(type_name: str) -> BaseException|None:
|
||||||
'''
|
'''
|
||||||
Look up an exception type by name from the set of locally known
|
Look up an exception type by name from the set of locally
|
||||||
namespaces:
|
known namespaces:
|
||||||
|
|
||||||
- `builtins`
|
- `builtins`
|
||||||
- `tractor._exceptions`
|
- `tractor._exceptions`
|
||||||
|
|
@ -201,7 +139,6 @@ def get_err_type(type_name: str) -> BaseException|None:
|
||||||
builtins,
|
builtins,
|
||||||
_this_mod,
|
_this_mod,
|
||||||
trio,
|
trio,
|
||||||
bdb,
|
|
||||||
]:
|
]:
|
||||||
if type_ref := getattr(
|
if type_ref := getattr(
|
||||||
ns,
|
ns,
|
||||||
|
|
@ -210,8 +147,6 @@ def get_err_type(type_name: str) -> BaseException|None:
|
||||||
):
|
):
|
||||||
return type_ref
|
return type_ref
|
||||||
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def pack_from_raise(
|
def pack_from_raise(
|
||||||
local_err: (
|
local_err: (
|
||||||
|
|
@ -423,13 +358,6 @@ class RemoteActorError(Exception):
|
||||||
self._ipc_msg.src_type_str
|
self._ipc_msg.src_type_str
|
||||||
)
|
)
|
||||||
|
|
||||||
if not self._src_type:
|
|
||||||
raise TypeError(
|
|
||||||
f'Failed to lookup src error type with '
|
|
||||||
f'`tractor._exceptions.get_err_type()` :\n'
|
|
||||||
f'{self.src_type_str}'
|
|
||||||
)
|
|
||||||
|
|
||||||
return self._src_type
|
return self._src_type
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|
@ -438,9 +366,6 @@ class RemoteActorError(Exception):
|
||||||
String-name of the (last hop's) boxed error type.
|
String-name of the (last hop's) boxed error type.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
# TODO, maybe support also serializing the
|
|
||||||
# `ExceptionGroup.exeptions: list[BaseException]` set under
|
|
||||||
# certain conditions?
|
|
||||||
bt: Type[BaseException] = self.boxed_type
|
bt: Type[BaseException] = self.boxed_type
|
||||||
if bt:
|
if bt:
|
||||||
return str(bt.__name__)
|
return str(bt.__name__)
|
||||||
|
|
@ -453,13 +378,9 @@ class RemoteActorError(Exception):
|
||||||
Error type boxed by last actor IPC hop.
|
Error type boxed by last actor IPC hop.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
if (
|
if self._boxed_type is None:
|
||||||
self._boxed_type is None
|
|
||||||
and
|
|
||||||
(ipc_msg := self._ipc_msg)
|
|
||||||
):
|
|
||||||
self._boxed_type = get_err_type(
|
self._boxed_type = get_err_type(
|
||||||
ipc_msg.boxed_type_str
|
self._ipc_msg.boxed_type_str
|
||||||
)
|
)
|
||||||
|
|
||||||
return self._boxed_type
|
return self._boxed_type
|
||||||
|
|
@ -542,6 +463,7 @@ class RemoteActorError(Exception):
|
||||||
if val:
|
if val:
|
||||||
_repr += f'{key}={val_str}{end_char}'
|
_repr += f'{key}={val_str}{end_char}'
|
||||||
|
|
||||||
|
|
||||||
return _repr
|
return _repr
|
||||||
|
|
||||||
def reprol(self) -> str:
|
def reprol(self) -> str:
|
||||||
|
|
@ -620,9 +542,56 @@ class RemoteActorError(Exception):
|
||||||
the type name is already implicitly shown by python).
|
the type name is already implicitly shown by python).
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
header: str = ''
|
||||||
|
body: str = ''
|
||||||
|
message: str = ''
|
||||||
|
|
||||||
|
# XXX when the currently raised exception is this instance,
|
||||||
|
# we do not ever use the "type header" style repr.
|
||||||
|
is_being_raised: bool = False
|
||||||
|
if (
|
||||||
|
(exc := sys.exception())
|
||||||
|
and
|
||||||
|
exc is self
|
||||||
|
):
|
||||||
|
is_being_raised: bool = True
|
||||||
|
|
||||||
|
with_type_header: bool = (
|
||||||
|
with_type_header
|
||||||
|
and
|
||||||
|
not is_being_raised
|
||||||
|
)
|
||||||
|
|
||||||
|
# <RemoteActorError( .. )> style
|
||||||
|
if with_type_header:
|
||||||
|
header: str = f'<{type(self).__name__}('
|
||||||
|
|
||||||
|
if message := self._message:
|
||||||
|
|
||||||
|
# split off the first line so, if needed, it isn't
|
||||||
|
# indented the same like the "boxed content" which
|
||||||
|
# since there is no `.tb_str` is just the `.message`.
|
||||||
|
lines: list[str] = message.splitlines()
|
||||||
|
first: str = lines[0]
|
||||||
|
message: str = message.removeprefix(first)
|
||||||
|
|
||||||
|
# with a type-style header we,
|
||||||
|
# - have no special message "first line" extraction/handling
|
||||||
|
# - place the message a space in from the header:
|
||||||
|
# `MsgTypeError( <message> ..`
|
||||||
|
# ^-here
|
||||||
|
# - indent the `.message` inside the type body.
|
||||||
|
if with_type_header:
|
||||||
|
first = f' {first} )>'
|
||||||
|
|
||||||
|
message: str = textwrap.indent(
|
||||||
|
message,
|
||||||
|
prefix=' '*2,
|
||||||
|
)
|
||||||
|
message: str = first + message
|
||||||
|
|
||||||
# IFF there is an embedded traceback-str we always
|
# IFF there is an embedded traceback-str we always
|
||||||
# draw the ascii-box around it.
|
# draw the ascii-box around it.
|
||||||
body: str = ''
|
|
||||||
if tb_str := self.tb_str:
|
if tb_str := self.tb_str:
|
||||||
fields: str = self._mk_fields_str(
|
fields: str = self._mk_fields_str(
|
||||||
_body_fields
|
_body_fields
|
||||||
|
|
@ -643,15 +612,21 @@ class RemoteActorError(Exception):
|
||||||
boxer_header=self.relay_uid,
|
boxer_header=self.relay_uid,
|
||||||
)
|
)
|
||||||
|
|
||||||
# !TODO, it'd be nice to import these top level without
|
tail = ''
|
||||||
# cycles!
|
if (
|
||||||
from tractor.devx.pformat import (
|
with_type_header
|
||||||
pformat_exc,
|
and not message
|
||||||
)
|
):
|
||||||
return pformat_exc(
|
tail: str = '>'
|
||||||
exc=self,
|
|
||||||
with_type_header=with_type_header,
|
return (
|
||||||
body=body,
|
header
|
||||||
|
+
|
||||||
|
message
|
||||||
|
+
|
||||||
|
f'{body}'
|
||||||
|
+
|
||||||
|
tail
|
||||||
)
|
)
|
||||||
|
|
||||||
__repr__ = pformat
|
__repr__ = pformat
|
||||||
|
|
@ -677,10 +652,16 @@ class RemoteActorError(Exception):
|
||||||
failing actor's remote env.
|
failing actor's remote env.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
src_type_ref: Type[BaseException] = self.src_type
|
||||||
|
if not src_type_ref:
|
||||||
|
raise TypeError(
|
||||||
|
'Failed to lookup src error type:\n'
|
||||||
|
f'{self.src_type_str}'
|
||||||
|
)
|
||||||
|
|
||||||
# TODO: better tb insertion and all the fancier dunder
|
# TODO: better tb insertion and all the fancier dunder
|
||||||
# metadata stuff as per `.__context__` etc. and friends:
|
# metadata stuff as per `.__context__` etc. and friends:
|
||||||
# https://github.com/python-trio/trio/issues/611
|
# https://github.com/python-trio/trio/issues/611
|
||||||
src_type_ref: Type[BaseException] = self.src_type
|
|
||||||
return src_type_ref(self.tb_str)
|
return src_type_ref(self.tb_str)
|
||||||
|
|
||||||
# TODO: local recontruction of nested inception for a given
|
# TODO: local recontruction of nested inception for a given
|
||||||
|
|
@ -806,11 +787,8 @@ class MsgTypeError(
|
||||||
'''
|
'''
|
||||||
if (
|
if (
|
||||||
(_bad_msg := self.msgdata.get('_bad_msg'))
|
(_bad_msg := self.msgdata.get('_bad_msg'))
|
||||||
and (
|
and
|
||||||
isinstance(_bad_msg, PayloadMsg)
|
isinstance(_bad_msg, PayloadMsg)
|
||||||
or
|
|
||||||
isinstance(_bad_msg, msgtypes.Start)
|
|
||||||
)
|
|
||||||
):
|
):
|
||||||
return _bad_msg
|
return _bad_msg
|
||||||
|
|
||||||
|
|
@ -929,7 +907,7 @@ class StreamOverrun(
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
|
||||||
class TransportClosed(Exception):
|
class TransportClosed(trio.BrokenResourceError):
|
||||||
'''
|
'''
|
||||||
IPC transport (protocol) connection was closed or broke and
|
IPC transport (protocol) connection was closed or broke and
|
||||||
indicates that the wrapping communication `Channel` can no longer
|
indicates that the wrapping communication `Channel` can no longer
|
||||||
|
|
@ -940,39 +918,24 @@ class TransportClosed(Exception):
|
||||||
self,
|
self,
|
||||||
message: str,
|
message: str,
|
||||||
loglevel: str = 'transport',
|
loglevel: str = 'transport',
|
||||||
src_exc: Exception|None = None,
|
cause: BaseException|None = None,
|
||||||
raise_on_report: bool = False,
|
raise_on_report: bool = False,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
self.message: str = message
|
self.message: str = message
|
||||||
self._loglevel: str = loglevel
|
self._loglevel = loglevel
|
||||||
super().__init__(message)
|
super().__init__(message)
|
||||||
|
|
||||||
self._src_exc = src_exc
|
if cause is not None:
|
||||||
# set the cause manually if not already set by python
|
self.__cause__ = cause
|
||||||
if (
|
|
||||||
src_exc is not None
|
|
||||||
and
|
|
||||||
not self.__cause__
|
|
||||||
):
|
|
||||||
self.__cause__ = src_exc
|
|
||||||
|
|
||||||
# flag to toggle whether the msg loop should raise
|
# flag to toggle whether the msg loop should raise
|
||||||
# the exc in its `TransportClosed` handler block.
|
# the exc in its `TransportClosed` handler block.
|
||||||
self._raise_on_report = raise_on_report
|
self._raise_on_report = raise_on_report
|
||||||
|
|
||||||
@property
|
|
||||||
def src_exc(self) -> Exception:
|
|
||||||
return (
|
|
||||||
self.__cause__
|
|
||||||
or
|
|
||||||
self._src_exc
|
|
||||||
)
|
|
||||||
|
|
||||||
def report_n_maybe_raise(
|
def report_n_maybe_raise(
|
||||||
self,
|
self,
|
||||||
message: str|None = None,
|
message: str|None = None,
|
||||||
hide_tb: bool = True,
|
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
|
|
@ -980,10 +943,9 @@ class TransportClosed(Exception):
|
||||||
for this error.
|
for this error.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
__tracebackhide__: bool = hide_tb
|
|
||||||
message: str = message or self.message
|
message: str = message or self.message
|
||||||
# when a cause is set, slap it onto the log emission.
|
# when a cause is set, slap it onto the log emission.
|
||||||
if cause := self.src_exc:
|
if cause := self.__cause__:
|
||||||
cause_tb_str: str = ''.join(
|
cause_tb_str: str = ''.join(
|
||||||
traceback.format_tb(cause.__traceback__)
|
traceback.format_tb(cause.__traceback__)
|
||||||
)
|
)
|
||||||
|
|
@ -992,86 +954,13 @@ class TransportClosed(Exception):
|
||||||
f' {cause}\n' # exc repr
|
f' {cause}\n' # exc repr
|
||||||
)
|
)
|
||||||
|
|
||||||
getattr(
|
getattr(log, self._loglevel)(message)
|
||||||
log,
|
|
||||||
self._loglevel
|
|
||||||
)(message)
|
|
||||||
|
|
||||||
# some errors we want to blow up from
|
# some errors we want to blow up from
|
||||||
# inside the RPC msg loop
|
# inside the RPC msg loop
|
||||||
if self._raise_on_report:
|
if self._raise_on_report:
|
||||||
raise self from cause
|
raise self from cause
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def repr_src_exc(
|
|
||||||
self,
|
|
||||||
src_exc: Exception|None = None,
|
|
||||||
) -> str:
|
|
||||||
|
|
||||||
if src_exc is None:
|
|
||||||
return '<unknown>'
|
|
||||||
|
|
||||||
src_msg: tuple[str] = src_exc.args
|
|
||||||
src_exc_repr: str = (
|
|
||||||
f'{type(src_exc).__name__}[ {src_msg} ]'
|
|
||||||
)
|
|
||||||
return src_exc_repr
|
|
||||||
|
|
||||||
def pformat(self) -> str:
|
|
||||||
from tractor.devx.pformat import (
|
|
||||||
pformat_exc,
|
|
||||||
)
|
|
||||||
return pformat_exc(
|
|
||||||
exc=self,
|
|
||||||
)
|
|
||||||
|
|
||||||
# delegate to `str`-ified pformat
|
|
||||||
__repr__ = pformat
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_src_exc(
|
|
||||||
cls,
|
|
||||||
src_exc: (
|
|
||||||
Exception|
|
|
||||||
trio.ClosedResource|
|
|
||||||
trio.BrokenResourceError
|
|
||||||
),
|
|
||||||
message: str,
|
|
||||||
body: str = '',
|
|
||||||
**init_kws,
|
|
||||||
) -> TransportClosed:
|
|
||||||
'''
|
|
||||||
Convenience constructor for creation from an underlying
|
|
||||||
`trio`-sourced async-resource/chan/stream error.
|
|
||||||
|
|
||||||
Embeds the original `src_exc`'s repr within the
|
|
||||||
`Exception.args` via a first-line-in-`.message`-put-in-header
|
|
||||||
pre-processing and allows inserting additional content beyond
|
|
||||||
the main message via a `body: str`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
repr_src_exc: str = cls.repr_src_exc(
|
|
||||||
src_exc,
|
|
||||||
)
|
|
||||||
next_line: str = f' src_exc: {repr_src_exc}\n'
|
|
||||||
if body:
|
|
||||||
body: str = textwrap.indent(
|
|
||||||
body,
|
|
||||||
prefix=' '*2,
|
|
||||||
)
|
|
||||||
|
|
||||||
return TransportClosed(
|
|
||||||
message=(
|
|
||||||
message
|
|
||||||
+
|
|
||||||
next_line
|
|
||||||
+
|
|
||||||
body
|
|
||||||
),
|
|
||||||
src_exc=src_exc,
|
|
||||||
**init_kws,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class NoResult(RuntimeError):
|
class NoResult(RuntimeError):
|
||||||
"No final result is expected for this actor"
|
"No final result is expected for this actor"
|
||||||
|
|
@ -1092,6 +981,18 @@ class MessagingError(Exception):
|
||||||
|
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
class AsyncioCancelled(Exception):
|
||||||
|
'''
|
||||||
|
Asyncio cancelled translation (non-base) error
|
||||||
|
for use with the ``to_asyncio`` module
|
||||||
|
to be raised in the ``trio`` side task
|
||||||
|
|
||||||
|
NOTE: this should NOT inherit from `asyncio.CancelledError` or
|
||||||
|
tests should break!
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
|
||||||
def pack_error(
|
def pack_error(
|
||||||
exc: BaseException|RemoteActorError,
|
exc: BaseException|RemoteActorError,
|
||||||
|
|
||||||
|
|
@ -1203,8 +1104,6 @@ def unpack_error(
|
||||||
which is the responsibilitiy of the caller.
|
which is the responsibilitiy of the caller.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
# XXX, apparently we pass all sorts of msgs here?
|
|
||||||
# kinda odd but seems like maybe they shouldn't be?
|
|
||||||
if not isinstance(msg, Error):
|
if not isinstance(msg, Error):
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
@ -1246,6 +1145,55 @@ def unpack_error(
|
||||||
return exc
|
return exc
|
||||||
|
|
||||||
|
|
||||||
|
def is_multi_cancelled(
|
||||||
|
exc: BaseException|BaseExceptionGroup,
|
||||||
|
|
||||||
|
ignore_nested: set[BaseException] = set(),
|
||||||
|
|
||||||
|
) -> bool|BaseExceptionGroup:
|
||||||
|
'''
|
||||||
|
Predicate to determine if an `BaseExceptionGroup` only contains
|
||||||
|
some (maybe nested) set of sub-grouped exceptions (like only
|
||||||
|
`trio.Cancelled`s which get swallowed silently by default) and is
|
||||||
|
thus the result of "gracefully cancelling" a collection of
|
||||||
|
sub-tasks (or other conc primitives) and receiving a "cancelled
|
||||||
|
ACK" from each after termination.
|
||||||
|
|
||||||
|
Docs:
|
||||||
|
----
|
||||||
|
- https://docs.python.org/3/library/exceptions.html#exception-groups
|
||||||
|
- https://docs.python.org/3/library/exceptions.html#BaseExceptionGroup.subgroup
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
if (
|
||||||
|
not ignore_nested
|
||||||
|
or
|
||||||
|
trio.Cancelled in ignore_nested
|
||||||
|
# XXX always count-in `trio`'s native signal
|
||||||
|
):
|
||||||
|
ignore_nested |= {trio.Cancelled}
|
||||||
|
|
||||||
|
if isinstance(exc, BaseExceptionGroup):
|
||||||
|
matched_exc: BaseExceptionGroup|None = exc.subgroup(
|
||||||
|
tuple(ignore_nested),
|
||||||
|
|
||||||
|
# TODO, complain about why not allowed XD
|
||||||
|
# condition=tuple(ignore_nested),
|
||||||
|
)
|
||||||
|
if matched_exc is not None:
|
||||||
|
return matched_exc
|
||||||
|
|
||||||
|
# NOTE, IFF no excs types match (throughout the error-tree)
|
||||||
|
# -> return `False`, OW return the matched sub-eg.
|
||||||
|
#
|
||||||
|
# IOW, for the inverse of ^ for the purpose of
|
||||||
|
# maybe-enter-REPL--logic: "only debug when the err-tree contains
|
||||||
|
# at least one exc-type NOT in `ignore_nested`" ; i.e. the case where
|
||||||
|
# we fallthrough and return `False` here.
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
def _raise_from_unexpected_msg(
|
def _raise_from_unexpected_msg(
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
msg: MsgType,
|
msg: MsgType,
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,820 @@
|
||||||
|
# tractor: structured concurrent "actors".
|
||||||
|
# Copyright 2018-eternity Tyler Goodlet.
|
||||||
|
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU Affero General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU Affero General Public License for more details.
|
||||||
|
|
||||||
|
# You should have received a copy of the GNU Affero General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
"""
|
||||||
|
Inter-process comms abstractions
|
||||||
|
|
||||||
|
"""
|
||||||
|
from __future__ import annotations
|
||||||
|
from collections.abc import (
|
||||||
|
AsyncGenerator,
|
||||||
|
AsyncIterator,
|
||||||
|
)
|
||||||
|
from contextlib import (
|
||||||
|
asynccontextmanager as acm,
|
||||||
|
contextmanager as cm,
|
||||||
|
)
|
||||||
|
import platform
|
||||||
|
from pprint import pformat
|
||||||
|
import struct
|
||||||
|
import typing
|
||||||
|
from typing import (
|
||||||
|
Any,
|
||||||
|
Callable,
|
||||||
|
runtime_checkable,
|
||||||
|
Protocol,
|
||||||
|
Type,
|
||||||
|
TypeVar,
|
||||||
|
)
|
||||||
|
|
||||||
|
import msgspec
|
||||||
|
from tricycle import BufferedReceiveStream
|
||||||
|
import trio
|
||||||
|
|
||||||
|
from tractor.log import get_logger
|
||||||
|
from tractor._exceptions import (
|
||||||
|
MsgTypeError,
|
||||||
|
pack_from_raise,
|
||||||
|
TransportClosed,
|
||||||
|
_mk_send_mte,
|
||||||
|
_mk_recv_mte,
|
||||||
|
)
|
||||||
|
from tractor.msg import (
|
||||||
|
_ctxvar_MsgCodec,
|
||||||
|
# _codec, XXX see `self._codec` sanity/debug checks
|
||||||
|
MsgCodec,
|
||||||
|
types as msgtypes,
|
||||||
|
pretty_struct,
|
||||||
|
)
|
||||||
|
|
||||||
|
log = get_logger(__name__)
|
||||||
|
|
||||||
|
_is_windows = platform.system() == 'Windows'
|
||||||
|
|
||||||
|
|
||||||
|
def get_stream_addrs(
|
||||||
|
stream: trio.SocketStream
|
||||||
|
) -> tuple[
|
||||||
|
tuple[str, int], # local
|
||||||
|
tuple[str, int], # remote
|
||||||
|
]:
|
||||||
|
'''
|
||||||
|
Return the `trio` streaming transport prot's socket-addrs for
|
||||||
|
both the local and remote sides as a pair.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# rn, should both be IP sockets
|
||||||
|
lsockname = stream.socket.getsockname()
|
||||||
|
rsockname = stream.socket.getpeername()
|
||||||
|
return (
|
||||||
|
tuple(lsockname[:2]),
|
||||||
|
tuple(rsockname[:2]),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# from tractor.msg.types import MsgType
|
||||||
|
# ?TODO? this should be our `Union[*msgtypes.__spec__]` alias now right..?
|
||||||
|
# => BLEH, except can't bc prots must inherit typevar or param-spec
|
||||||
|
# vars..
|
||||||
|
MsgType = TypeVar('MsgType')
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: break up this mod into a subpkg so we can start adding new
|
||||||
|
# backends and move this type stuff into a dedicated file.. Bo
|
||||||
|
#
|
||||||
|
@runtime_checkable
|
||||||
|
class MsgTransport(Protocol[MsgType]):
|
||||||
|
#
|
||||||
|
# ^-TODO-^ consider using a generic def and indexing with our
|
||||||
|
# eventual msg definition/types?
|
||||||
|
# - https://docs.python.org/3/library/typing.html#typing.Protocol
|
||||||
|
|
||||||
|
stream: trio.SocketStream
|
||||||
|
drained: list[MsgType]
|
||||||
|
|
||||||
|
def __init__(self, stream: trio.SocketStream) -> None:
|
||||||
|
...
|
||||||
|
|
||||||
|
# XXX: should this instead be called `.sendall()`?
|
||||||
|
async def send(self, msg: MsgType) -> None:
|
||||||
|
...
|
||||||
|
|
||||||
|
async def recv(self) -> MsgType:
|
||||||
|
...
|
||||||
|
|
||||||
|
def __aiter__(self) -> MsgType:
|
||||||
|
...
|
||||||
|
|
||||||
|
def connected(self) -> bool:
|
||||||
|
...
|
||||||
|
|
||||||
|
# defining this sync otherwise it causes a mypy error because it
|
||||||
|
# can't figure out it's a generator i guess?..?
|
||||||
|
def drain(self) -> AsyncIterator[dict]:
|
||||||
|
...
|
||||||
|
|
||||||
|
@property
|
||||||
|
def laddr(self) -> tuple[str, int]:
|
||||||
|
...
|
||||||
|
|
||||||
|
@property
|
||||||
|
def raddr(self) -> tuple[str, int]:
|
||||||
|
...
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: typing oddity.. not sure why we have to inherit here, but it
|
||||||
|
# seems to be an issue with `get_msg_transport()` returning
|
||||||
|
# a `Type[Protocol]`; probably should make a `mypy` issue?
|
||||||
|
class MsgpackTCPStream(MsgTransport):
|
||||||
|
'''
|
||||||
|
A ``trio.SocketStream`` delivering ``msgpack`` formatted data
|
||||||
|
using the ``msgspec`` codec lib.
|
||||||
|
|
||||||
|
'''
|
||||||
|
layer_key: int = 4
|
||||||
|
name_key: str = 'tcp'
|
||||||
|
|
||||||
|
# TODO: better naming for this?
|
||||||
|
# -[ ] check how libp2p does naming for such things?
|
||||||
|
codec_key: str = 'msgpack'
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
stream: trio.SocketStream,
|
||||||
|
prefix_size: int = 4,
|
||||||
|
|
||||||
|
# XXX optionally provided codec pair for `msgspec`:
|
||||||
|
# https://jcristharif.com/msgspec/extending.html#mapping-to-from-native-types
|
||||||
|
#
|
||||||
|
# TODO: define this as a `Codec` struct which can be
|
||||||
|
# overriden dynamically by the application/runtime?
|
||||||
|
codec: tuple[
|
||||||
|
Callable[[Any], Any]|None, # coder
|
||||||
|
Callable[[type, Any], Any]|None, # decoder
|
||||||
|
]|None = None,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
self.stream = stream
|
||||||
|
assert self.stream.socket
|
||||||
|
|
||||||
|
# should both be IP sockets
|
||||||
|
self._laddr, self._raddr = get_stream_addrs(stream)
|
||||||
|
|
||||||
|
# create read loop instance
|
||||||
|
self._aiter_pkts = self._iter_packets()
|
||||||
|
self._send_lock = trio.StrictFIFOLock()
|
||||||
|
|
||||||
|
# public i guess?
|
||||||
|
self.drained: list[dict] = []
|
||||||
|
|
||||||
|
self.recv_stream = BufferedReceiveStream(
|
||||||
|
transport_stream=stream
|
||||||
|
)
|
||||||
|
self.prefix_size = prefix_size
|
||||||
|
|
||||||
|
# allow for custom IPC msg interchange format
|
||||||
|
# dynamic override Bo
|
||||||
|
self._task = trio.lowlevel.current_task()
|
||||||
|
|
||||||
|
# XXX for ctxvar debug only!
|
||||||
|
# self._codec: MsgCodec = (
|
||||||
|
# codec
|
||||||
|
# or
|
||||||
|
# _codec._ctxvar_MsgCodec.get()
|
||||||
|
# )
|
||||||
|
|
||||||
|
async def _iter_packets(self) -> AsyncGenerator[dict, None]:
|
||||||
|
'''
|
||||||
|
Yield `bytes`-blob decoded packets from the underlying TCP
|
||||||
|
stream using the current task's `MsgCodec`.
|
||||||
|
|
||||||
|
This is a streaming routine implemented as an async generator
|
||||||
|
func (which was the original design, but could be changed?)
|
||||||
|
and is allocated by a `.__call__()` inside `.__init__()` where
|
||||||
|
it is assigned to the `._aiter_pkts` attr.
|
||||||
|
|
||||||
|
'''
|
||||||
|
decodes_failed: int = 0
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
header: bytes = await self.recv_stream.receive_exactly(4)
|
||||||
|
except (
|
||||||
|
ValueError,
|
||||||
|
ConnectionResetError,
|
||||||
|
|
||||||
|
# not sure entirely why we need this but without it we
|
||||||
|
# seem to be getting racy failures here on
|
||||||
|
# arbiter/registry name subs..
|
||||||
|
trio.BrokenResourceError,
|
||||||
|
|
||||||
|
) as trans_err:
|
||||||
|
|
||||||
|
loglevel = 'transport'
|
||||||
|
match trans_err:
|
||||||
|
# case (
|
||||||
|
# ConnectionResetError()
|
||||||
|
# ):
|
||||||
|
# loglevel = 'transport'
|
||||||
|
|
||||||
|
# peer actor (graceful??) TCP EOF but `tricycle`
|
||||||
|
# seems to raise a 0-bytes-read?
|
||||||
|
case ValueError() if (
|
||||||
|
'unclean EOF' in trans_err.args[0]
|
||||||
|
):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# peer actor (task) prolly shutdown quickly due
|
||||||
|
# to cancellation
|
||||||
|
case trio.BrokenResourceError() if (
|
||||||
|
'Connection reset by peer' in trans_err.args[0]
|
||||||
|
):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# unless the disconnect condition falls under "a
|
||||||
|
# normal operation breakage" we usualy console warn
|
||||||
|
# about it.
|
||||||
|
case _:
|
||||||
|
loglevel: str = 'warning'
|
||||||
|
|
||||||
|
|
||||||
|
raise TransportClosed(
|
||||||
|
message=(
|
||||||
|
f'IPC transport already closed by peer\n'
|
||||||
|
f'x)> {type(trans_err)}\n'
|
||||||
|
f' |_{self}\n'
|
||||||
|
),
|
||||||
|
loglevel=loglevel,
|
||||||
|
) from trans_err
|
||||||
|
|
||||||
|
# XXX definitely can happen if transport is closed
|
||||||
|
# manually by another `trio.lowlevel.Task` in the
|
||||||
|
# same actor; we use this in some simulated fault
|
||||||
|
# testing for ex, but generally should never happen
|
||||||
|
# under normal operation!
|
||||||
|
#
|
||||||
|
# NOTE: as such we always re-raise this error from the
|
||||||
|
# RPC msg loop!
|
||||||
|
except trio.ClosedResourceError as closure_err:
|
||||||
|
raise TransportClosed(
|
||||||
|
message=(
|
||||||
|
f'IPC transport already manually closed locally?\n'
|
||||||
|
f'x)> {type(closure_err)} \n'
|
||||||
|
f' |_{self}\n'
|
||||||
|
),
|
||||||
|
loglevel='error',
|
||||||
|
raise_on_report=(
|
||||||
|
closure_err.args[0] == 'another task closed this fd'
|
||||||
|
or
|
||||||
|
closure_err.args[0] in ['another task closed this fd']
|
||||||
|
),
|
||||||
|
) from closure_err
|
||||||
|
|
||||||
|
# graceful TCP EOF disconnect
|
||||||
|
if header == b'':
|
||||||
|
raise TransportClosed(
|
||||||
|
message=(
|
||||||
|
f'IPC transport already gracefully closed\n'
|
||||||
|
f')>\n'
|
||||||
|
f'|_{self}\n'
|
||||||
|
),
|
||||||
|
loglevel='transport',
|
||||||
|
# cause=??? # handy or no?
|
||||||
|
)
|
||||||
|
|
||||||
|
size: int
|
||||||
|
size, = struct.unpack("<I", header)
|
||||||
|
|
||||||
|
log.transport(f'received header {size}') # type: ignore
|
||||||
|
msg_bytes: bytes = await self.recv_stream.receive_exactly(size)
|
||||||
|
|
||||||
|
log.transport(f"received {msg_bytes}") # type: ignore
|
||||||
|
try:
|
||||||
|
# NOTE: lookup the `trio.Task.context`'s var for
|
||||||
|
# the current `MsgCodec`.
|
||||||
|
codec: MsgCodec = _ctxvar_MsgCodec.get()
|
||||||
|
|
||||||
|
# XXX for ctxvar debug only!
|
||||||
|
# if self._codec.pld_spec != codec.pld_spec:
|
||||||
|
# assert (
|
||||||
|
# task := trio.lowlevel.current_task()
|
||||||
|
# ) is not self._task
|
||||||
|
# self._task = task
|
||||||
|
# self._codec = codec
|
||||||
|
# log.runtime(
|
||||||
|
# f'Using new codec in {self}.recv()\n'
|
||||||
|
# f'codec: {self._codec}\n\n'
|
||||||
|
# f'msg_bytes: {msg_bytes}\n'
|
||||||
|
# )
|
||||||
|
yield codec.decode(msg_bytes)
|
||||||
|
|
||||||
|
# XXX NOTE: since the below error derives from
|
||||||
|
# `DecodeError` we need to catch is specially
|
||||||
|
# and always raise such that spec violations
|
||||||
|
# are never allowed to be caught silently!
|
||||||
|
except msgspec.ValidationError as verr:
|
||||||
|
msgtyperr: MsgTypeError = _mk_recv_mte(
|
||||||
|
msg=msg_bytes,
|
||||||
|
codec=codec,
|
||||||
|
src_validation_error=verr,
|
||||||
|
)
|
||||||
|
# XXX deliver up to `Channel.recv()` where
|
||||||
|
# a re-raise and `Error`-pack can inject the far
|
||||||
|
# end actor `.uid`.
|
||||||
|
yield msgtyperr
|
||||||
|
|
||||||
|
except (
|
||||||
|
msgspec.DecodeError,
|
||||||
|
UnicodeDecodeError,
|
||||||
|
):
|
||||||
|
if decodes_failed < 4:
|
||||||
|
# ignore decoding errors for now and assume they have to
|
||||||
|
# do with a channel drop - hope that receiving from the
|
||||||
|
# channel will raise an expected error and bubble up.
|
||||||
|
try:
|
||||||
|
msg_str: str|bytes = msg_bytes.decode()
|
||||||
|
except UnicodeDecodeError:
|
||||||
|
msg_str = msg_bytes
|
||||||
|
|
||||||
|
log.exception(
|
||||||
|
'Failed to decode msg?\n'
|
||||||
|
f'{codec}\n\n'
|
||||||
|
'Rxed bytes from wire:\n\n'
|
||||||
|
f'{msg_str!r}\n'
|
||||||
|
)
|
||||||
|
decodes_failed += 1
|
||||||
|
else:
|
||||||
|
raise
|
||||||
|
|
||||||
|
async def send(
|
||||||
|
self,
|
||||||
|
msg: msgtypes.MsgType,
|
||||||
|
|
||||||
|
strict_types: bool = True,
|
||||||
|
hide_tb: bool = False,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Send a msgpack encoded py-object-blob-as-msg over TCP.
|
||||||
|
|
||||||
|
If `strict_types == True` then a `MsgTypeError` will be raised on any
|
||||||
|
invalid msg type
|
||||||
|
|
||||||
|
'''
|
||||||
|
__tracebackhide__: bool = hide_tb
|
||||||
|
|
||||||
|
# XXX see `trio._sync.AsyncContextManagerMixin` for details
|
||||||
|
# on the `.acquire()`/`.release()` sequencing..
|
||||||
|
async with self._send_lock:
|
||||||
|
|
||||||
|
# NOTE: lookup the `trio.Task.context`'s var for
|
||||||
|
# the current `MsgCodec`.
|
||||||
|
codec: MsgCodec = _ctxvar_MsgCodec.get()
|
||||||
|
|
||||||
|
# XXX for ctxvar debug only!
|
||||||
|
# if self._codec.pld_spec != codec.pld_spec:
|
||||||
|
# self._codec = codec
|
||||||
|
# log.runtime(
|
||||||
|
# f'Using new codec in {self}.send()\n'
|
||||||
|
# f'codec: {self._codec}\n\n'
|
||||||
|
# f'msg: {msg}\n'
|
||||||
|
# )
|
||||||
|
|
||||||
|
if type(msg) not in msgtypes.__msg_types__:
|
||||||
|
if strict_types:
|
||||||
|
raise _mk_send_mte(
|
||||||
|
msg,
|
||||||
|
codec=codec,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
log.warning(
|
||||||
|
'Sending non-`Msg`-spec msg?\n\n'
|
||||||
|
f'{msg}\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
bytes_data: bytes = codec.encode(msg)
|
||||||
|
except TypeError as _err:
|
||||||
|
typerr = _err
|
||||||
|
msgtyperr: MsgTypeError = _mk_send_mte(
|
||||||
|
msg,
|
||||||
|
codec=codec,
|
||||||
|
message=(
|
||||||
|
f'IPC-msg-spec violation in\n\n'
|
||||||
|
f'{pretty_struct.Struct.pformat(msg)}'
|
||||||
|
),
|
||||||
|
src_type_error=typerr,
|
||||||
|
)
|
||||||
|
raise msgtyperr from typerr
|
||||||
|
|
||||||
|
# supposedly the fastest says,
|
||||||
|
# https://stackoverflow.com/a/54027962
|
||||||
|
size: bytes = struct.pack("<I", len(bytes_data))
|
||||||
|
return await self.stream.send_all(size + bytes_data)
|
||||||
|
|
||||||
|
# ?TODO? does it help ever to dynamically show this
|
||||||
|
# frame?
|
||||||
|
# try:
|
||||||
|
# <the-above_code>
|
||||||
|
# except BaseException as _err:
|
||||||
|
# err = _err
|
||||||
|
# if not isinstance(err, MsgTypeError):
|
||||||
|
# __tracebackhide__: bool = False
|
||||||
|
# raise
|
||||||
|
|
||||||
|
@property
|
||||||
|
def laddr(self) -> tuple[str, int]:
|
||||||
|
return self._laddr
|
||||||
|
|
||||||
|
@property
|
||||||
|
def raddr(self) -> tuple[str, int]:
|
||||||
|
return self._raddr
|
||||||
|
|
||||||
|
async def recv(self) -> Any:
|
||||||
|
return await self._aiter_pkts.asend(None)
|
||||||
|
|
||||||
|
async def drain(self) -> AsyncIterator[dict]:
|
||||||
|
'''
|
||||||
|
Drain the stream's remaining messages sent from
|
||||||
|
the far end until the connection is closed by
|
||||||
|
the peer.
|
||||||
|
|
||||||
|
'''
|
||||||
|
try:
|
||||||
|
async for msg in self._iter_packets():
|
||||||
|
self.drained.append(msg)
|
||||||
|
except TransportClosed:
|
||||||
|
for msg in self.drained:
|
||||||
|
yield msg
|
||||||
|
|
||||||
|
def __aiter__(self):
|
||||||
|
return self._aiter_pkts
|
||||||
|
|
||||||
|
def connected(self) -> bool:
|
||||||
|
return self.stream.socket.fileno() != -1
|
||||||
|
|
||||||
|
|
||||||
|
def get_msg_transport(
|
||||||
|
|
||||||
|
key: tuple[str, str],
|
||||||
|
|
||||||
|
) -> Type[MsgTransport]:
|
||||||
|
|
||||||
|
return {
|
||||||
|
('msgpack', 'tcp'): MsgpackTCPStream,
|
||||||
|
}[key]
|
||||||
|
|
||||||
|
|
||||||
|
class Channel:
|
||||||
|
'''
|
||||||
|
An inter-process channel for communication between (remote) actors.
|
||||||
|
|
||||||
|
Wraps a ``MsgStream``: transport + encoding IPC connection.
|
||||||
|
|
||||||
|
Currently we only support ``trio.SocketStream`` for transport
|
||||||
|
(aka TCP) and the ``msgpack`` interchange format via the ``msgspec``
|
||||||
|
codec libary.
|
||||||
|
|
||||||
|
'''
|
||||||
|
def __init__(
|
||||||
|
|
||||||
|
self,
|
||||||
|
destaddr: tuple[str, int]|None,
|
||||||
|
|
||||||
|
msg_transport_type_key: tuple[str, str] = ('msgpack', 'tcp'),
|
||||||
|
|
||||||
|
# TODO: optional reconnection support?
|
||||||
|
# auto_reconnect: bool = False,
|
||||||
|
# on_reconnect: typing.Callable[..., typing.Awaitable] = None,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
# self._recon_seq = on_reconnect
|
||||||
|
# self._autorecon = auto_reconnect
|
||||||
|
|
||||||
|
self._destaddr = destaddr
|
||||||
|
self._transport_key = msg_transport_type_key
|
||||||
|
|
||||||
|
# Either created in ``.connect()`` or passed in by
|
||||||
|
# user in ``.from_stream()``.
|
||||||
|
self._stream: trio.SocketStream|None = None
|
||||||
|
self._transport: MsgTransport|None = None
|
||||||
|
|
||||||
|
# set after handshake - always uid of far end
|
||||||
|
self.uid: tuple[str, str]|None = None
|
||||||
|
|
||||||
|
self._aiter_msgs = self._iter_msgs()
|
||||||
|
self._exc: Exception|None = None # set if far end actor errors
|
||||||
|
self._closed: bool = False
|
||||||
|
|
||||||
|
# flag set by ``Portal.cancel_actor()`` indicating remote
|
||||||
|
# (possibly peer) cancellation of the far end actor
|
||||||
|
# runtime.
|
||||||
|
self._cancel_called: bool = False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def msgstream(self) -> MsgTransport:
|
||||||
|
log.info(
|
||||||
|
'`Channel.msgstream` is an old name, use `._transport`'
|
||||||
|
)
|
||||||
|
return self._transport
|
||||||
|
|
||||||
|
@property
|
||||||
|
def transport(self) -> MsgTransport:
|
||||||
|
return self._transport
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_stream(
|
||||||
|
cls,
|
||||||
|
stream: trio.SocketStream,
|
||||||
|
**kwargs,
|
||||||
|
|
||||||
|
) -> Channel:
|
||||||
|
|
||||||
|
src, dst = get_stream_addrs(stream)
|
||||||
|
chan = Channel(
|
||||||
|
destaddr=dst,
|
||||||
|
**kwargs,
|
||||||
|
)
|
||||||
|
|
||||||
|
# set immediately here from provided instance
|
||||||
|
chan._stream: trio.SocketStream = stream
|
||||||
|
chan.set_msg_transport(stream)
|
||||||
|
return chan
|
||||||
|
|
||||||
|
def set_msg_transport(
|
||||||
|
self,
|
||||||
|
stream: trio.SocketStream,
|
||||||
|
type_key: tuple[str, str]|None = None,
|
||||||
|
|
||||||
|
# XXX optionally provided codec pair for `msgspec`:
|
||||||
|
# https://jcristharif.com/msgspec/extending.html#mapping-to-from-native-types
|
||||||
|
codec: MsgCodec|None = None,
|
||||||
|
|
||||||
|
) -> MsgTransport:
|
||||||
|
type_key = (
|
||||||
|
type_key
|
||||||
|
or
|
||||||
|
self._transport_key
|
||||||
|
)
|
||||||
|
# get transport type, then
|
||||||
|
self._transport = get_msg_transport(
|
||||||
|
type_key
|
||||||
|
# instantiate an instance of the msg-transport
|
||||||
|
)(
|
||||||
|
stream,
|
||||||
|
codec=codec,
|
||||||
|
)
|
||||||
|
return self._transport
|
||||||
|
|
||||||
|
@cm
|
||||||
|
def apply_codec(
|
||||||
|
self,
|
||||||
|
codec: MsgCodec,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Temporarily override the underlying IPC msg codec for
|
||||||
|
dynamic enforcement of messaging schema.
|
||||||
|
|
||||||
|
'''
|
||||||
|
orig: MsgCodec = self._transport.codec
|
||||||
|
try:
|
||||||
|
self._transport.codec = codec
|
||||||
|
yield
|
||||||
|
finally:
|
||||||
|
self._transport.codec = orig
|
||||||
|
|
||||||
|
# TODO: do a .src/.dst: str for maddrs?
|
||||||
|
def __repr__(self) -> str:
|
||||||
|
if not self._transport:
|
||||||
|
return '<Channel with inactive transport?>'
|
||||||
|
|
||||||
|
return repr(
|
||||||
|
self._transport.stream.socket._sock
|
||||||
|
).replace( # type: ignore
|
||||||
|
"socket.socket",
|
||||||
|
"Channel",
|
||||||
|
)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def laddr(self) -> tuple[str, int]|None:
|
||||||
|
return self._transport.laddr if self._transport else None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def raddr(self) -> tuple[str, int]|None:
|
||||||
|
return self._transport.raddr if self._transport else None
|
||||||
|
|
||||||
|
async def connect(
|
||||||
|
self,
|
||||||
|
destaddr: tuple[Any, ...] | None = None,
|
||||||
|
**kwargs
|
||||||
|
|
||||||
|
) -> MsgTransport:
|
||||||
|
|
||||||
|
if self.connected():
|
||||||
|
raise RuntimeError("channel is already connected?")
|
||||||
|
|
||||||
|
destaddr = destaddr or self._destaddr
|
||||||
|
assert isinstance(destaddr, tuple)
|
||||||
|
|
||||||
|
stream = await trio.open_tcp_stream(
|
||||||
|
*destaddr,
|
||||||
|
**kwargs
|
||||||
|
)
|
||||||
|
transport = self.set_msg_transport(stream)
|
||||||
|
|
||||||
|
log.transport(
|
||||||
|
f'Opened channel[{type(transport)}]: {self.laddr} -> {self.raddr}'
|
||||||
|
)
|
||||||
|
return transport
|
||||||
|
|
||||||
|
# TODO: something like,
|
||||||
|
# `pdbp.hideframe_on(errors=[MsgTypeError])`
|
||||||
|
# instead of the `try/except` hack we have rn..
|
||||||
|
# seems like a pretty useful thing to have in general
|
||||||
|
# along with being able to filter certain stack frame(s / sets)
|
||||||
|
# possibly based on the current log-level?
|
||||||
|
async def send(
|
||||||
|
self,
|
||||||
|
payload: Any,
|
||||||
|
|
||||||
|
hide_tb: bool = False,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Send a coded msg-blob over the transport.
|
||||||
|
|
||||||
|
'''
|
||||||
|
__tracebackhide__: bool = hide_tb
|
||||||
|
try:
|
||||||
|
log.transport(
|
||||||
|
'=> send IPC msg:\n\n'
|
||||||
|
f'{pformat(payload)}\n'
|
||||||
|
)
|
||||||
|
# assert self._transport # but why typing?
|
||||||
|
await self._transport.send(
|
||||||
|
payload,
|
||||||
|
hide_tb=hide_tb,
|
||||||
|
)
|
||||||
|
except BaseException as _err:
|
||||||
|
err = _err # bind for introspection
|
||||||
|
if not isinstance(_err, MsgTypeError):
|
||||||
|
# assert err
|
||||||
|
__tracebackhide__: bool = False
|
||||||
|
else:
|
||||||
|
assert err.cid
|
||||||
|
|
||||||
|
raise
|
||||||
|
|
||||||
|
async def recv(self) -> Any:
|
||||||
|
assert self._transport
|
||||||
|
return await self._transport.recv()
|
||||||
|
|
||||||
|
# TODO: auto-reconnect features like 0mq/nanomsg?
|
||||||
|
# -[ ] implement it manually with nods to SC prot
|
||||||
|
# possibly on multiple transport backends?
|
||||||
|
# -> seems like that might be re-inventing scalability
|
||||||
|
# prots tho no?
|
||||||
|
# try:
|
||||||
|
# return await self._transport.recv()
|
||||||
|
# except trio.BrokenResourceError:
|
||||||
|
# if self._autorecon:
|
||||||
|
# await self._reconnect()
|
||||||
|
# return await self.recv()
|
||||||
|
# raise
|
||||||
|
|
||||||
|
async def aclose(self) -> None:
|
||||||
|
|
||||||
|
log.transport(
|
||||||
|
f'Closing channel to {self.uid} '
|
||||||
|
f'{self.laddr} -> {self.raddr}'
|
||||||
|
)
|
||||||
|
assert self._transport
|
||||||
|
await self._transport.stream.aclose()
|
||||||
|
self._closed = True
|
||||||
|
|
||||||
|
async def __aenter__(self):
|
||||||
|
await self.connect()
|
||||||
|
return self
|
||||||
|
|
||||||
|
async def __aexit__(self, *args):
|
||||||
|
await self.aclose(*args)
|
||||||
|
|
||||||
|
def __aiter__(self):
|
||||||
|
return self._aiter_msgs
|
||||||
|
|
||||||
|
# ?TODO? run any reconnection sequence?
|
||||||
|
# -[ ] prolly should be impl-ed as deco-API?
|
||||||
|
#
|
||||||
|
# async def _reconnect(self) -> None:
|
||||||
|
# """Handle connection failures by polling until a reconnect can be
|
||||||
|
# established.
|
||||||
|
# """
|
||||||
|
# down = False
|
||||||
|
# while True:
|
||||||
|
# try:
|
||||||
|
# with trio.move_on_after(3) as cancel_scope:
|
||||||
|
# await self.connect()
|
||||||
|
# cancelled = cancel_scope.cancelled_caught
|
||||||
|
# if cancelled:
|
||||||
|
# log.transport(
|
||||||
|
# "Reconnect timed out after 3 seconds, retrying...")
|
||||||
|
# continue
|
||||||
|
# else:
|
||||||
|
# log.transport("Stream connection re-established!")
|
||||||
|
|
||||||
|
# # on_recon = self._recon_seq
|
||||||
|
# # if on_recon:
|
||||||
|
# # await on_recon(self)
|
||||||
|
|
||||||
|
# break
|
||||||
|
# except (OSError, ConnectionRefusedError):
|
||||||
|
# if not down:
|
||||||
|
# down = True
|
||||||
|
# log.transport(
|
||||||
|
# f"Connection to {self.raddr} went down, waiting"
|
||||||
|
# " for re-establishment")
|
||||||
|
# await trio.sleep(1)
|
||||||
|
|
||||||
|
async def _iter_msgs(
|
||||||
|
self
|
||||||
|
) -> AsyncGenerator[Any, None]:
|
||||||
|
'''
|
||||||
|
Yield `MsgType` IPC msgs decoded and deliverd from
|
||||||
|
an underlying `MsgTransport` protocol.
|
||||||
|
|
||||||
|
This is a streaming routine alo implemented as an async-gen
|
||||||
|
func (same a `MsgTransport._iter_pkts()`) gets allocated by
|
||||||
|
a `.__call__()` inside `.__init__()` where it is assigned to
|
||||||
|
the `._aiter_msgs` attr.
|
||||||
|
|
||||||
|
'''
|
||||||
|
assert self._transport
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
async for msg in self._transport:
|
||||||
|
match msg:
|
||||||
|
# NOTE: if transport/interchange delivers
|
||||||
|
# a type error, we pack it with the far
|
||||||
|
# end peer `Actor.uid` and relay the
|
||||||
|
# `Error`-msg upward to the `._rpc` stack
|
||||||
|
# for normal RAE handling.
|
||||||
|
case MsgTypeError():
|
||||||
|
yield pack_from_raise(
|
||||||
|
local_err=msg,
|
||||||
|
cid=msg.cid,
|
||||||
|
|
||||||
|
# XXX we pack it here bc lower
|
||||||
|
# layers have no notion of an
|
||||||
|
# actor-id ;)
|
||||||
|
src_uid=self.uid,
|
||||||
|
)
|
||||||
|
case _:
|
||||||
|
yield msg
|
||||||
|
|
||||||
|
except trio.BrokenResourceError:
|
||||||
|
|
||||||
|
# if not self._autorecon:
|
||||||
|
raise
|
||||||
|
|
||||||
|
await self.aclose()
|
||||||
|
|
||||||
|
# if self._autorecon: # attempt reconnect
|
||||||
|
# await self._reconnect()
|
||||||
|
# continue
|
||||||
|
|
||||||
|
def connected(self) -> bool:
|
||||||
|
return self._transport.connected() if self._transport else False
|
||||||
|
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def _connect_chan(
|
||||||
|
host: str,
|
||||||
|
port: int
|
||||||
|
|
||||||
|
) -> typing.AsyncGenerator[Channel, None]:
|
||||||
|
'''
|
||||||
|
Create and connect a channel with disconnect on context manager
|
||||||
|
teardown.
|
||||||
|
|
||||||
|
'''
|
||||||
|
chan = Channel((host, port))
|
||||||
|
await chan.connect()
|
||||||
|
yield chan
|
||||||
|
with trio.CancelScope(shield=True):
|
||||||
|
await chan.aclose()
|
||||||
|
|
@ -39,14 +39,11 @@ import warnings
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
|
|
||||||
from .trionics import (
|
from .trionics import maybe_open_nursery
|
||||||
maybe_open_nursery,
|
|
||||||
collapse_eg,
|
|
||||||
)
|
|
||||||
from ._state import (
|
from ._state import (
|
||||||
current_actor,
|
current_actor,
|
||||||
)
|
)
|
||||||
from .ipc import Channel
|
from ._ipc import Channel
|
||||||
from .log import get_logger
|
from .log import get_logger
|
||||||
from .msg import (
|
from .msg import (
|
||||||
# Error,
|
# Error,
|
||||||
|
|
@ -55,8 +52,8 @@ from .msg import (
|
||||||
Return,
|
Return,
|
||||||
)
|
)
|
||||||
from ._exceptions import (
|
from ._exceptions import (
|
||||||
|
# unpack_error,
|
||||||
NoResult,
|
NoResult,
|
||||||
TransportClosed,
|
|
||||||
)
|
)
|
||||||
from ._context import (
|
from ._context import (
|
||||||
Context,
|
Context,
|
||||||
|
|
@ -110,18 +107,10 @@ class Portal:
|
||||||
# point.
|
# point.
|
||||||
self._expect_result_ctx: Context|None = None
|
self._expect_result_ctx: Context|None = None
|
||||||
self._streams: set[MsgStream] = set()
|
self._streams: set[MsgStream] = set()
|
||||||
|
|
||||||
# TODO, this should be PRIVATE (and never used publicly)! since it's just
|
|
||||||
# a cached ref to the local runtime instead of calling
|
|
||||||
# `current_actor()` everywhere.. XD
|
|
||||||
self.actor: Actor = current_actor()
|
self.actor: Actor = current_actor()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def chan(self) -> Channel:
|
def chan(self) -> Channel:
|
||||||
'''
|
|
||||||
Ref to this ctx's underlying `tractor.ipc.Channel`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return self._chan
|
return self._chan
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|
@ -181,17 +170,10 @@ class Portal:
|
||||||
|
|
||||||
# not expecting a "main" result
|
# not expecting a "main" result
|
||||||
if self._expect_result_ctx is None:
|
if self._expect_result_ctx is None:
|
||||||
peer_id: str = f'{self.channel.aid.reprol()!r}'
|
|
||||||
log.warning(
|
log.warning(
|
||||||
f'Portal to peer {peer_id} will not deliver a final result?\n'
|
f"Portal for {self.channel.uid} not expecting a final"
|
||||||
f'\n'
|
" result?\nresult() should only be called if subactor"
|
||||||
f'Context.result() can only be called by the parent of '
|
" was spawned with `ActorNursery.run_in_actor()`")
|
||||||
f'a sub-actor when it was spawned with '
|
|
||||||
f'`ActorNursery.run_in_actor()`'
|
|
||||||
f'\n'
|
|
||||||
f'Further this `ActorNursery`-method-API will deprecated in the'
|
|
||||||
f'near fututre!\n'
|
|
||||||
)
|
|
||||||
return NoResult
|
return NoResult
|
||||||
|
|
||||||
# expecting a "main" result
|
# expecting a "main" result
|
||||||
|
|
@ -202,7 +184,7 @@ class Portal:
|
||||||
(
|
(
|
||||||
self._final_result_msg,
|
self._final_result_msg,
|
||||||
self._final_result_pld,
|
self._final_result_pld,
|
||||||
) = await self._expect_result_ctx._pld_rx.recv_msg(
|
) = await self._expect_result_ctx._pld_rx.recv_msg_w_pld(
|
||||||
ipc=self._expect_result_ctx,
|
ipc=self._expect_result_ctx,
|
||||||
expect_msg=Return,
|
expect_msg=Return,
|
||||||
)
|
)
|
||||||
|
|
@ -224,7 +206,6 @@ class Portal:
|
||||||
typname: str = type(self).__name__
|
typname: str = type(self).__name__
|
||||||
log.warning(
|
log.warning(
|
||||||
f'`{typname}.result()` is DEPRECATED!\n'
|
f'`{typname}.result()` is DEPRECATED!\n'
|
||||||
f'\n'
|
|
||||||
f'Use `{typname}.wait_for_result()` instead!\n'
|
f'Use `{typname}.wait_for_result()` instead!\n'
|
||||||
)
|
)
|
||||||
return await self.wait_for_result(
|
return await self.wait_for_result(
|
||||||
|
|
@ -236,10 +217,8 @@ class Portal:
|
||||||
# terminate all locally running async generator
|
# terminate all locally running async generator
|
||||||
# IPC calls
|
# IPC calls
|
||||||
if self._streams:
|
if self._streams:
|
||||||
peer_id: str = f'{self.channel.aid.reprol()!r}'
|
log.cancel(
|
||||||
report: str = (
|
f"Cancelling all streams with {self.channel.uid}")
|
||||||
f'Cancelling all msg-streams with {peer_id}\n'
|
|
||||||
)
|
|
||||||
for stream in self._streams.copy():
|
for stream in self._streams.copy():
|
||||||
try:
|
try:
|
||||||
await stream.aclose()
|
await stream.aclose()
|
||||||
|
|
@ -248,18 +227,10 @@ class Portal:
|
||||||
# (unless of course at some point down the road we
|
# (unless of course at some point down the road we
|
||||||
# won't expect this to always be the case or need to
|
# won't expect this to always be the case or need to
|
||||||
# detect it for respawning purposes?)
|
# detect it for respawning purposes?)
|
||||||
report += (
|
log.debug(f"{stream} was already closed.")
|
||||||
f'->) {stream!r} already closed\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
log.cancel(report)
|
|
||||||
|
|
||||||
async def aclose(self):
|
async def aclose(self):
|
||||||
log.debug(
|
log.debug(f"Closing {self}")
|
||||||
f'Closing portal\n'
|
|
||||||
f'>}}\n'
|
|
||||||
f'|_{self}\n'
|
|
||||||
)
|
|
||||||
# TODO: once we move to implementing our own `ReceiveChannel`
|
# TODO: once we move to implementing our own `ReceiveChannel`
|
||||||
# (including remote task cancellation inside its `.aclose()`)
|
# (including remote task cancellation inside its `.aclose()`)
|
||||||
# we'll need to .aclose all those channels here
|
# we'll need to .aclose all those channels here
|
||||||
|
|
@ -285,22 +256,23 @@ class Portal:
|
||||||
__runtimeframe__: int = 1 # noqa
|
__runtimeframe__: int = 1 # noqa
|
||||||
|
|
||||||
chan: Channel = self.channel
|
chan: Channel = self.channel
|
||||||
peer_id: str = f'{self.channel.aid.reprol()!r}'
|
|
||||||
if not chan.connected():
|
if not chan.connected():
|
||||||
log.runtime(
|
log.runtime(
|
||||||
'Peer {peer_id} is already disconnected\n'
|
'This channel is already closed, skipping cancel request..'
|
||||||
'-> skipping cancel request..\n'
|
|
||||||
)
|
)
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
reminfo: str = (
|
||||||
|
f'c)=> {self.channel.uid}\n'
|
||||||
|
f' |_{chan}\n'
|
||||||
|
)
|
||||||
log.cancel(
|
log.cancel(
|
||||||
f'Sending actor-runtime-cancel-req to peer\n'
|
f'Requesting actor-runtime cancel for peer\n\n'
|
||||||
f'\n'
|
f'{reminfo}'
|
||||||
f'c)=> {peer_id}\n'
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# XXX the one spot we set it?
|
# XXX the one spot we set it?
|
||||||
chan._cancel_called: bool = True
|
self.channel._cancel_called: bool = True
|
||||||
try:
|
try:
|
||||||
# send cancel cmd - might not get response
|
# send cancel cmd - might not get response
|
||||||
# XXX: sure would be nice to make this work with
|
# XXX: sure would be nice to make this work with
|
||||||
|
|
@ -321,43 +293,22 @@ class Portal:
|
||||||
# may timeout and we never get an ack (obvi racy)
|
# may timeout and we never get an ack (obvi racy)
|
||||||
# but that doesn't mean it wasn't cancelled.
|
# but that doesn't mean it wasn't cancelled.
|
||||||
log.debug(
|
log.debug(
|
||||||
f'May have failed to cancel peer?\n'
|
'May have failed to cancel peer?\n'
|
||||||
f'\n'
|
f'{reminfo}'
|
||||||
f'c)=?> {peer_id}\n'
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# if we get here some weird cancellation case happened
|
# if we get here some weird cancellation case happened
|
||||||
return False
|
return False
|
||||||
|
|
||||||
except (
|
except (
|
||||||
# XXX, should never really get raised unless we aren't
|
|
||||||
# wrapping them in the below type by mistake?
|
|
||||||
#
|
|
||||||
# Leaving the catch here for now until we're very sure
|
|
||||||
# all the cases (for various tpt protos) have indeed been
|
|
||||||
# re-wrapped ;p
|
|
||||||
trio.ClosedResourceError,
|
trio.ClosedResourceError,
|
||||||
trio.BrokenResourceError,
|
trio.BrokenResourceError,
|
||||||
|
):
|
||||||
TransportClosed,
|
log.debug(
|
||||||
) as tpt_err:
|
'IPC chan for actor already closed or broken?\n\n'
|
||||||
ipc_borked_report: str = (
|
f'{self.channel.uid}\n'
|
||||||
f'IPC for actor already closed/broken?\n\n'
|
f' |_{self.channel}\n'
|
||||||
f'\n'
|
|
||||||
f'c)=x> {peer_id}\n'
|
|
||||||
)
|
)
|
||||||
match tpt_err:
|
|
||||||
case TransportClosed():
|
|
||||||
log.debug(ipc_borked_report)
|
|
||||||
case _:
|
|
||||||
ipc_borked_report += (
|
|
||||||
f'\n'
|
|
||||||
f'Unhandled low-level transport-closed/error during\n'
|
|
||||||
f'Portal.cancel_actor()` request?\n'
|
|
||||||
f'<{type(tpt_err).__name__}( {tpt_err} )>\n'
|
|
||||||
)
|
|
||||||
log.warning(ipc_borked_report)
|
|
||||||
|
|
||||||
return False
|
return False
|
||||||
|
|
||||||
# TODO: do we still need this for low level `Actor`-runtime
|
# TODO: do we still need this for low level `Actor`-runtime
|
||||||
|
|
@ -513,13 +464,10 @@ class Portal:
|
||||||
with trio.CancelScope(shield=True):
|
with trio.CancelScope(shield=True):
|
||||||
await ctx.cancel()
|
await ctx.cancel()
|
||||||
|
|
||||||
except trio.ClosedResourceError as cre:
|
except trio.ClosedResourceError:
|
||||||
# if the far end terminates before we send a cancel the
|
# if the far end terminates before we send a cancel the
|
||||||
# underlying transport-channel may already be closed.
|
# underlying transport-channel may already be closed.
|
||||||
log.cancel(
|
log.cancel(f'Context {ctx} was already closed?')
|
||||||
f'Context.cancel() -> {cre!r}\n'
|
|
||||||
f'cid: {ctx.cid!r} already closed?\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# XXX: should this always be done?
|
# XXX: should this always be done?
|
||||||
# await recv_chan.aclose()
|
# await recv_chan.aclose()
|
||||||
|
|
@ -556,12 +504,8 @@ class LocalPortal:
|
||||||
return it's result.
|
return it's result.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
obj = (
|
obj = self.actor if ns == 'self' else importlib.import_module(ns)
|
||||||
self.actor
|
func = getattr(obj, func_name)
|
||||||
if ns == 'self'
|
|
||||||
else importlib.import_module(ns)
|
|
||||||
)
|
|
||||||
func: Callable = getattr(obj, func_name)
|
|
||||||
return await func(**kwargs)
|
return await func(**kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -586,30 +530,26 @@ async def open_portal(
|
||||||
assert actor
|
assert actor
|
||||||
was_connected: bool = False
|
was_connected: bool = False
|
||||||
|
|
||||||
async with (
|
async with maybe_open_nursery(
|
||||||
collapse_eg(),
|
|
||||||
maybe_open_nursery(
|
|
||||||
tn,
|
tn,
|
||||||
shield=shield,
|
shield=shield,
|
||||||
) as tn,
|
) as tn:
|
||||||
):
|
|
||||||
|
|
||||||
if not channel.connected():
|
if not channel.connected():
|
||||||
await channel.connect()
|
await channel.connect()
|
||||||
was_connected = True
|
was_connected = True
|
||||||
|
|
||||||
if channel.aid is None:
|
if channel.uid is None:
|
||||||
await channel._do_handshake(
|
await actor._do_handshake(channel)
|
||||||
aid=actor.aid,
|
|
||||||
)
|
|
||||||
|
|
||||||
msg_loop_cs: trio.CancelScope|None = None
|
msg_loop_cs: trio.CancelScope|None = None
|
||||||
if start_msg_loop:
|
if start_msg_loop:
|
||||||
from . import _rpc
|
from ._runtime import process_messages
|
||||||
msg_loop_cs = await tn.start(
|
msg_loop_cs = await tn.start(
|
||||||
partial(
|
partial(
|
||||||
_rpc.process_messages,
|
process_messages,
|
||||||
chan=channel,
|
actor,
|
||||||
|
channel,
|
||||||
# if the local task is cancelled we want to keep
|
# if the local task is cancelled we want to keep
|
||||||
# the msg loop running until our block ends
|
# the msg loop running until our block ends
|
||||||
shield=True,
|
shield=True,
|
||||||
|
|
|
||||||
439
tractor/_root.py
439
tractor/_root.py
|
|
@ -18,9 +18,7 @@
|
||||||
Root actor runtime ignition(s).
|
Root actor runtime ignition(s).
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from contextlib import (
|
from contextlib import asynccontextmanager as acm
|
||||||
asynccontextmanager as acm,
|
|
||||||
)
|
|
||||||
from functools import partial
|
from functools import partial
|
||||||
import importlib
|
import importlib
|
||||||
import inspect
|
import inspect
|
||||||
|
|
@ -28,55 +26,96 @@ import logging
|
||||||
import os
|
import os
|
||||||
import signal
|
import signal
|
||||||
import sys
|
import sys
|
||||||
from typing import (
|
from typing import Callable
|
||||||
Any,
|
|
||||||
Callable,
|
|
||||||
)
|
|
||||||
import warnings
|
import warnings
|
||||||
|
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
|
|
||||||
from . import _runtime
|
from ._runtime import (
|
||||||
from .devx import (
|
Actor,
|
||||||
debug,
|
Arbiter,
|
||||||
_frame_stack,
|
# TODO: rename and make a non-actor subtype?
|
||||||
pformat as _pformat,
|
# Arbiter as Registry,
|
||||||
|
async_main,
|
||||||
)
|
)
|
||||||
|
from .devx import _debug
|
||||||
from . import _spawn
|
from . import _spawn
|
||||||
from . import _state
|
from . import _state
|
||||||
from . import log
|
from . import log
|
||||||
from .ipc import (
|
from ._ipc import _connect_chan
|
||||||
_connect_chan,
|
from ._exceptions import is_multi_cancelled
|
||||||
)
|
|
||||||
from ._addr import (
|
|
||||||
Address,
|
# set at startup and after forks
|
||||||
UnwrappedAddress,
|
_default_host: str = '127.0.0.1'
|
||||||
default_lo_addrs,
|
_default_port: int = 1616
|
||||||
mk_uuid,
|
|
||||||
wrap_address,
|
# default registry always on localhost
|
||||||
)
|
_default_lo_addrs: list[tuple[str, int]] = [(
|
||||||
from .trionics import (
|
_default_host,
|
||||||
is_multi_cancelled,
|
_default_port,
|
||||||
collapse_eg,
|
)]
|
||||||
)
|
|
||||||
from ._exceptions import (
|
|
||||||
RuntimeFailure,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
logger = log.get_logger('tractor')
|
logger = log.get_logger('tractor')
|
||||||
|
|
||||||
|
|
||||||
# TODO: stick this in a `@acm` defined in `devx.debug`?
|
|
||||||
# -[ ] also maybe consider making this a `wrapt`-deco to
|
|
||||||
# save an indent level?
|
|
||||||
#
|
|
||||||
@acm
|
@acm
|
||||||
async def maybe_block_bp(
|
async def open_root_actor(
|
||||||
debug_mode: bool,
|
|
||||||
maybe_enable_greenback: bool,
|
*,
|
||||||
) -> bool:
|
# defaults are above
|
||||||
|
registry_addrs: list[tuple[str, int]]|None = None,
|
||||||
|
|
||||||
|
# defaults are above
|
||||||
|
arbiter_addr: tuple[str, int]|None = None,
|
||||||
|
|
||||||
|
name: str|None = 'root',
|
||||||
|
|
||||||
|
# either the `multiprocessing` start method:
|
||||||
|
# https://docs.python.org/3/library/multiprocessing.html#contexts-and-start-methods
|
||||||
|
# OR `trio` (the new default).
|
||||||
|
start_method: _spawn.SpawnMethodKey|None = None,
|
||||||
|
|
||||||
|
# enables the multi-process debugger support
|
||||||
|
debug_mode: bool = False,
|
||||||
|
maybe_enable_greenback: bool = True, # `.pause_from_sync()/breakpoint()` support
|
||||||
|
enable_stack_on_sig: bool = False,
|
||||||
|
|
||||||
|
# internal logging
|
||||||
|
loglevel: str|None = None,
|
||||||
|
|
||||||
|
enable_modules: list|None = None,
|
||||||
|
rpc_module_paths: list|None = None,
|
||||||
|
|
||||||
|
# NOTE: allow caller to ensure that only one registry exists
|
||||||
|
# and that this call creates it.
|
||||||
|
ensure_registry: bool = False,
|
||||||
|
|
||||||
|
hide_tb: bool = True,
|
||||||
|
|
||||||
|
# XXX, proxied directly to `.devx._debug._maybe_enter_pm()`
|
||||||
|
# for REPL-entry logic.
|
||||||
|
debug_filter: Callable[
|
||||||
|
[BaseException|BaseExceptionGroup],
|
||||||
|
bool,
|
||||||
|
] = lambda err: not is_multi_cancelled(err),
|
||||||
|
|
||||||
|
# TODO, a way for actors to augment passing derived
|
||||||
|
# read-only state to sublayers?
|
||||||
|
# extra_rt_vars: dict|None = None,
|
||||||
|
|
||||||
|
) -> Actor:
|
||||||
|
'''
|
||||||
|
Runtime init entry point for ``tractor``.
|
||||||
|
|
||||||
|
'''
|
||||||
|
__tracebackhide__: bool = hide_tb
|
||||||
|
_debug.hide_runtime_frames()
|
||||||
|
|
||||||
|
# TODO: stick this in a `@cm` defined in `devx._debug`?
|
||||||
|
#
|
||||||
# Override the global debugger hook to make it play nice with
|
# Override the global debugger hook to make it play nice with
|
||||||
# ``trio``, see much discussion in:
|
# ``trio``, see much discussion in:
|
||||||
# https://github.com/python-trio/trio/issues/1155#issuecomment-742964018
|
# https://github.com/python-trio/trio/issues/1155#issuecomment-742964018
|
||||||
|
|
@ -85,25 +124,23 @@ async def maybe_block_bp(
|
||||||
'PYTHONBREAKPOINT',
|
'PYTHONBREAKPOINT',
|
||||||
None,
|
None,
|
||||||
)
|
)
|
||||||
bp_blocked: bool
|
|
||||||
if (
|
if (
|
||||||
debug_mode
|
debug_mode
|
||||||
and maybe_enable_greenback
|
and maybe_enable_greenback
|
||||||
and (
|
and (
|
||||||
maybe_mod := await debug.maybe_init_greenback(
|
maybe_mod := await _debug.maybe_init_greenback(
|
||||||
raise_not_found=False,
|
raise_not_found=False,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
logger.info(
|
logger.info(
|
||||||
f'Found `greenback` installed @ {maybe_mod}\n'
|
f'Found `greenback` installed @ {maybe_mod}\n'
|
||||||
f'Enabling `tractor.pause_from_sync()` support!\n'
|
'Enabling `tractor.pause_from_sync()` support!\n'
|
||||||
)
|
)
|
||||||
os.environ['PYTHONBREAKPOINT'] = (
|
os.environ['PYTHONBREAKPOINT'] = (
|
||||||
'tractor.devx.debug._sync_pause_from_builtin'
|
'tractor.devx._debug._sync_pause_from_builtin'
|
||||||
)
|
)
|
||||||
_state._runtime_vars['use_greenback'] = True
|
_state._runtime_vars['use_greenback'] = True
|
||||||
bp_blocked = False
|
|
||||||
|
|
||||||
else:
|
else:
|
||||||
# TODO: disable `breakpoint()` by default (without
|
# TODO: disable `breakpoint()` by default (without
|
||||||
|
|
@ -122,127 +159,10 @@ async def maybe_block_bp(
|
||||||
# lol ok,
|
# lol ok,
|
||||||
# https://docs.python.org/3/library/sys.html#sys.breakpointhook
|
# https://docs.python.org/3/library/sys.html#sys.breakpointhook
|
||||||
os.environ['PYTHONBREAKPOINT'] = "0"
|
os.environ['PYTHONBREAKPOINT'] = "0"
|
||||||
bp_blocked = True
|
|
||||||
|
|
||||||
try:
|
|
||||||
yield bp_blocked
|
|
||||||
finally:
|
|
||||||
# restore any prior built-in `breakpoint()` hook state
|
|
||||||
if builtin_bp_handler is not None:
|
|
||||||
sys.breakpointhook = builtin_bp_handler
|
|
||||||
|
|
||||||
if orig_bp_path is not None:
|
|
||||||
os.environ['PYTHONBREAKPOINT'] = orig_bp_path
|
|
||||||
|
|
||||||
else:
|
|
||||||
# clear env back to having no entry
|
|
||||||
os.environ.pop('PYTHONBREAKPOINT', None)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
|
||||||
async def open_root_actor(
|
|
||||||
*,
|
|
||||||
# defaults are above
|
|
||||||
registry_addrs: list[UnwrappedAddress]|None = None,
|
|
||||||
|
|
||||||
# defaults are above
|
|
||||||
arbiter_addr: tuple[UnwrappedAddress]|None = None,
|
|
||||||
|
|
||||||
enable_transports: list[
|
|
||||||
# TODO, this should eventually be the pairs as
|
|
||||||
# defined by (codec, proto) as on `MsgTransport.
|
|
||||||
_state.TransportProtocolKey,
|
|
||||||
]|None = None,
|
|
||||||
|
|
||||||
name: str|None = 'root',
|
|
||||||
|
|
||||||
# either the `multiprocessing` start method:
|
|
||||||
# https://docs.python.org/3/library/multiprocessing.html#contexts-and-start-methods
|
|
||||||
# OR `trio` (the new default).
|
|
||||||
start_method: _spawn.SpawnMethodKey|None = None,
|
|
||||||
|
|
||||||
# enables the multi-process debugger support
|
|
||||||
debug_mode: bool = False,
|
|
||||||
maybe_enable_greenback: bool = False, # `.pause_from_sync()/breakpoint()` support
|
|
||||||
# ^XXX NOTE^ the perf implications of use,
|
|
||||||
# https://greenback.readthedocs.io/en/latest/principle.html#performance
|
|
||||||
enable_stack_on_sig: bool = False,
|
|
||||||
|
|
||||||
# internal logging
|
|
||||||
loglevel: str|None = None,
|
|
||||||
|
|
||||||
enable_modules: list|None = None,
|
|
||||||
rpc_module_paths: list|None = None,
|
|
||||||
|
|
||||||
# NOTE: allow caller to ensure that only one registry exists
|
|
||||||
# and that this call creates it.
|
|
||||||
ensure_registry: bool = False,
|
|
||||||
|
|
||||||
hide_tb: bool = True,
|
|
||||||
|
|
||||||
# XXX, proxied directly to `.devx.debug._maybe_enter_pm()`
|
|
||||||
# for REPL-entry logic.
|
|
||||||
debug_filter: Callable[
|
|
||||||
[BaseException|BaseExceptionGroup],
|
|
||||||
bool,
|
|
||||||
] = lambda err: not is_multi_cancelled(err),
|
|
||||||
|
|
||||||
# TODO, a way for actors to augment passing derived
|
|
||||||
# read-only state to sublayers?
|
|
||||||
# extra_rt_vars: dict|None = None,
|
|
||||||
|
|
||||||
) -> _runtime.Actor:
|
|
||||||
'''
|
|
||||||
Initialize the `tractor` runtime by starting a "root actor" in
|
|
||||||
a parent-most Python process.
|
|
||||||
|
|
||||||
All (disjoint) actor-process-trees-as-programs are created via
|
|
||||||
this entrypoint.
|
|
||||||
|
|
||||||
'''
|
|
||||||
# XXX NEVER allow nested actor-trees!
|
|
||||||
if already_actor := _state.current_actor(
|
|
||||||
err_on_no_runtime=False,
|
|
||||||
):
|
|
||||||
rtvs: dict[str, Any] = _state._runtime_vars
|
|
||||||
root_mailbox: list[str, int] = rtvs['_root_mailbox']
|
|
||||||
registry_addrs: list[list[str, int]] = rtvs['_registry_addrs']
|
|
||||||
raise RuntimeFailure(
|
|
||||||
f'A current actor already exists !?\n'
|
|
||||||
f'({already_actor}\n'
|
|
||||||
f'\n'
|
|
||||||
f'You can NOT open a second root actor from within '
|
|
||||||
f'an existing tree and the current root of this '
|
|
||||||
f'already exists !!\n'
|
|
||||||
f'\n'
|
|
||||||
f'_root_mailbox: {root_mailbox!r}\n'
|
|
||||||
f'_registry_addrs: {registry_addrs!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
async with maybe_block_bp(
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
maybe_enable_greenback=maybe_enable_greenback,
|
|
||||||
):
|
|
||||||
if enable_transports is None:
|
|
||||||
enable_transports: list[str] = _state.current_ipc_protos()
|
|
||||||
else:
|
|
||||||
_state._runtime_vars['_enable_tpts'] = enable_transports
|
|
||||||
|
|
||||||
# TODO! support multi-tpts per actor!
|
|
||||||
# Bo
|
|
||||||
if not len(enable_transports) == 1:
|
|
||||||
raise RuntimeError(
|
|
||||||
f'No multi-tpt support yet!\n'
|
|
||||||
f'enable_transports={enable_transports!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
_frame_stack.hide_runtime_frames()
|
|
||||||
__tracebackhide__: bool = hide_tb
|
|
||||||
|
|
||||||
# attempt to retreive ``trio``'s sigint handler and stash it
|
# attempt to retreive ``trio``'s sigint handler and stash it
|
||||||
# on our debugger lock state.
|
# on our debugger lock state.
|
||||||
debug.DebugStatus._trio_handler = signal.getsignal(signal.SIGINT)
|
_debug.DebugStatus._trio_handler = signal.getsignal(signal.SIGINT)
|
||||||
|
|
||||||
# mark top most level process as root actor
|
# mark top most level process as root actor
|
||||||
_state._runtime_vars['_is_root'] = True
|
_state._runtime_vars['_is_root'] = True
|
||||||
|
|
@ -266,7 +186,6 @@ async def open_root_actor(
|
||||||
if start_method is not None:
|
if start_method is not None:
|
||||||
_spawn.try_set_start_method(start_method)
|
_spawn.try_set_start_method(start_method)
|
||||||
|
|
||||||
# TODO! remove this ASAP!
|
|
||||||
if arbiter_addr is not None:
|
if arbiter_addr is not None:
|
||||||
warnings.warn(
|
warnings.warn(
|
||||||
'`arbiter_addr` is now deprecated\n'
|
'`arbiter_addr` is now deprecated\n'
|
||||||
|
|
@ -274,20 +193,14 @@ async def open_root_actor(
|
||||||
DeprecationWarning,
|
DeprecationWarning,
|
||||||
stacklevel=2,
|
stacklevel=2,
|
||||||
)
|
)
|
||||||
uw_reg_addrs = [arbiter_addr]
|
registry_addrs = [arbiter_addr]
|
||||||
|
|
||||||
uw_reg_addrs = registry_addrs
|
registry_addrs: list[tuple[str, int]] = (
|
||||||
if not uw_reg_addrs:
|
registry_addrs
|
||||||
uw_reg_addrs: list[UnwrappedAddress] = default_lo_addrs(
|
or
|
||||||
enable_transports
|
_default_lo_addrs
|
||||||
)
|
)
|
||||||
|
assert registry_addrs
|
||||||
# must exist by now since all below code is dependent
|
|
||||||
assert uw_reg_addrs
|
|
||||||
registry_addrs: list[Address] = [
|
|
||||||
wrap_address(uw_addr)
|
|
||||||
for uw_addr in uw_reg_addrs
|
|
||||||
]
|
|
||||||
|
|
||||||
loglevel = (
|
loglevel = (
|
||||||
loglevel
|
loglevel
|
||||||
|
|
@ -296,14 +209,13 @@ async def open_root_actor(
|
||||||
|
|
||||||
if (
|
if (
|
||||||
debug_mode
|
debug_mode
|
||||||
and
|
and _spawn._spawn_method == 'trio'
|
||||||
_spawn._spawn_method == 'trio'
|
|
||||||
):
|
):
|
||||||
_state._runtime_vars['_debug_mode'] = True
|
_state._runtime_vars['_debug_mode'] = True
|
||||||
|
|
||||||
# expose internal debug module to every actor allowing for
|
# expose internal debug module to every actor allowing for
|
||||||
# use of ``await tractor.pause()``
|
# use of ``await tractor.pause()``
|
||||||
enable_modules.append('tractor.devx.debug._tty_lock')
|
enable_modules.append('tractor.devx._debug')
|
||||||
|
|
||||||
# if debug mode get's enabled *at least* use that level of
|
# if debug mode get's enabled *at least* use that level of
|
||||||
# logging for some informative console prompts.
|
# logging for some informative console prompts.
|
||||||
|
|
@ -336,10 +248,10 @@ async def open_root_actor(
|
||||||
enable_stack_on_sig()
|
enable_stack_on_sig()
|
||||||
|
|
||||||
# closed into below ping task-func
|
# closed into below ping task-func
|
||||||
ponged_addrs: list[Address] = []
|
ponged_addrs: list[tuple[str, int]] = []
|
||||||
|
|
||||||
async def ping_tpt_socket(
|
async def ping_tpt_socket(
|
||||||
addr: Address,
|
addr: tuple[str, int],
|
||||||
timeout: float = 1,
|
timeout: float = 1,
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
|
|
@ -359,28 +271,23 @@ async def open_root_actor(
|
||||||
# be better to eventually have a "discovery" protocol
|
# be better to eventually have a "discovery" protocol
|
||||||
# with basic handshake instead?
|
# with basic handshake instead?
|
||||||
with trio.move_on_after(timeout):
|
with trio.move_on_after(timeout):
|
||||||
async with _connect_chan(addr.unwrap()):
|
async with _connect_chan(*addr):
|
||||||
ponged_addrs.append(addr)
|
ponged_addrs.append(addr)
|
||||||
|
|
||||||
except OSError:
|
except OSError:
|
||||||
# ?TODO, make this a "discovery" log level?
|
# TODO: make this a "discovery" log level?
|
||||||
logger.info(
|
logger.info(
|
||||||
f'No root-actor registry found @ {addr!r}\n'
|
f'No actor registry found @ {addr}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
# !TODO, this is basically just another (abstract)
|
|
||||||
# happy-eyeballs, so we should try for formalize it somewhere
|
|
||||||
# in a `.[_]discovery` ya?
|
|
||||||
#
|
|
||||||
async with trio.open_nursery() as tn:
|
async with trio.open_nursery() as tn:
|
||||||
for uw_addr in uw_reg_addrs:
|
for addr in registry_addrs:
|
||||||
addr: Address = wrap_address(uw_addr)
|
|
||||||
tn.start_soon(
|
tn.start_soon(
|
||||||
ping_tpt_socket,
|
ping_tpt_socket,
|
||||||
addr,
|
tuple(addr), # TODO: just drop this requirement?
|
||||||
)
|
)
|
||||||
|
|
||||||
trans_bind_addrs: list[UnwrappedAddress] = []
|
trans_bind_addrs: list[tuple[str, int]] = []
|
||||||
|
|
||||||
# Create a new local root-actor instance which IS NOT THE
|
# Create a new local root-actor instance which IS NOT THE
|
||||||
# REGISTRAR
|
# REGISTRAR
|
||||||
|
|
@ -396,35 +303,28 @@ async def open_root_actor(
|
||||||
f'Registry(s) seem(s) to exist @ {ponged_addrs}'
|
f'Registry(s) seem(s) to exist @ {ponged_addrs}'
|
||||||
)
|
)
|
||||||
|
|
||||||
actor = _runtime.Actor(
|
actor = Actor(
|
||||||
name=name or 'anonymous',
|
name=name or 'anonymous',
|
||||||
uuid=mk_uuid(),
|
|
||||||
registry_addrs=ponged_addrs,
|
registry_addrs=ponged_addrs,
|
||||||
loglevel=loglevel,
|
loglevel=loglevel,
|
||||||
enable_modules=enable_modules,
|
enable_modules=enable_modules,
|
||||||
)
|
)
|
||||||
# **DO NOT** use the registry_addrs as the
|
# DO NOT use the registry_addrs as the transport server
|
||||||
# ipc-transport-server's bind-addrs as this is
|
# addrs for this new non-registar, root-actor.
|
||||||
# a new NON-registrar, ROOT-actor.
|
for host, port in ponged_addrs:
|
||||||
#
|
# NOTE: zero triggers dynamic OS port allocation
|
||||||
# XXX INSTEAD, bind random addrs using the same tpt
|
trans_bind_addrs.append((host, 0))
|
||||||
# proto.
|
|
||||||
for addr in ponged_addrs:
|
|
||||||
trans_bind_addrs.append(
|
|
||||||
addr.get_random(
|
|
||||||
bindspace=addr.bindspace,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
# Start this local actor as the "registrar", aka a regular
|
# Start this local actor as the "registrar", aka a regular
|
||||||
# actor who manages the local registry of "mailboxes" of
|
# actor who manages the local registry of "mailboxes" of
|
||||||
# other process-tree-local sub-actors.
|
# other process-tree-local sub-actors.
|
||||||
else:
|
else:
|
||||||
|
|
||||||
# NOTE that if the current actor IS THE REGISTAR, the
|
# NOTE that if the current actor IS THE REGISTAR, the
|
||||||
# following init steps are taken:
|
# following init steps are taken:
|
||||||
# - the tranport layer server is bound to each addr
|
# - the tranport layer server is bound to each (host, port)
|
||||||
# pair defined in provided registry_addrs, or the default.
|
# pair defined in provided registry_addrs, or the default.
|
||||||
trans_bind_addrs = uw_reg_addrs
|
trans_bind_addrs = registry_addrs
|
||||||
|
|
||||||
# - it is normally desirable for any registrar to stay up
|
# - it is normally desirable for any registrar to stay up
|
||||||
# indefinitely until either all registered (child/sub)
|
# indefinitely until either all registered (child/sub)
|
||||||
|
|
@ -435,10 +335,8 @@ async def open_root_actor(
|
||||||
# https://github.com/goodboy/tractor/pull/348
|
# https://github.com/goodboy/tractor/pull/348
|
||||||
# https://github.com/goodboy/tractor/issues/296
|
# https://github.com/goodboy/tractor/issues/296
|
||||||
|
|
||||||
# TODO: rename as `RootActor` or is that even necessary?
|
actor = Arbiter(
|
||||||
actor = _runtime.Arbiter(
|
name or 'registrar',
|
||||||
name=name or 'registrar',
|
|
||||||
uuid=mk_uuid(),
|
|
||||||
registry_addrs=registry_addrs,
|
registry_addrs=registry_addrs,
|
||||||
loglevel=loglevel,
|
loglevel=loglevel,
|
||||||
enable_modules=enable_modules,
|
enable_modules=enable_modules,
|
||||||
|
|
@ -448,16 +346,6 @@ async def open_root_actor(
|
||||||
# `.trio.run()`.
|
# `.trio.run()`.
|
||||||
actor._infected_aio = _state._runtime_vars['_is_infected_aio']
|
actor._infected_aio = _state._runtime_vars['_is_infected_aio']
|
||||||
|
|
||||||
# NOTE, only set the loopback addr for the
|
|
||||||
# process-tree-global "root" mailbox since all sub-actors
|
|
||||||
# should be able to speak to their root actor over that
|
|
||||||
# channel.
|
|
||||||
raddrs: list[Address] = _state._runtime_vars['_root_addrs']
|
|
||||||
raddrs.extend(trans_bind_addrs)
|
|
||||||
# TODO, remove once we have also removed all usage;
|
|
||||||
# eventually all (root-)registry apis should expect > 1 addr.
|
|
||||||
_state._runtime_vars['_root_mailbox'] = raddrs[0]
|
|
||||||
|
|
||||||
# Start up main task set via core actor-runtime nurseries.
|
# Start up main task set via core actor-runtime nurseries.
|
||||||
try:
|
try:
|
||||||
# assign process-local actor
|
# assign process-local actor
|
||||||
|
|
@ -465,28 +353,18 @@ async def open_root_actor(
|
||||||
|
|
||||||
# start local channel-server and fake the portal API
|
# start local channel-server and fake the portal API
|
||||||
# NOTE: this won't block since we provide the nursery
|
# NOTE: this won't block since we provide the nursery
|
||||||
report: str = f'Starting actor-runtime for {actor.aid.reprol()!r}\n'
|
ml_addrs_str: str = '\n'.join(
|
||||||
if reg_addrs := actor.registry_addrs:
|
f'@{addr}' for addr in trans_bind_addrs
|
||||||
report += (
|
|
||||||
'-> Opening new registry @ '
|
|
||||||
+
|
|
||||||
'\n'.join(
|
|
||||||
f'{addr}' for addr in reg_addrs
|
|
||||||
)
|
)
|
||||||
|
logger.info(
|
||||||
|
f'Starting local {actor.uid} on the following transport addrs:\n'
|
||||||
|
f'{ml_addrs_str}'
|
||||||
)
|
)
|
||||||
logger.info(f'{report}\n')
|
|
||||||
|
|
||||||
# start runtime in a bg sub-task, yield to caller.
|
# start the actor runtime in a new task
|
||||||
async with (
|
async with trio.open_nursery() as nursery:
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as root_tn,
|
|
||||||
|
|
||||||
# ?TODO? finally-footgun below?
|
# ``_runtime.async_main()`` creates an internal nursery
|
||||||
# -> see note on why shielding.
|
|
||||||
# maybe_raise_from_masking_exc(),
|
|
||||||
):
|
|
||||||
actor._root_tn = root_tn
|
|
||||||
# `_runtime.async_main()` creates an internal nursery
|
|
||||||
# and blocks here until any underlying actor(-process)
|
# and blocks here until any underlying actor(-process)
|
||||||
# tree has terminated thereby conducting so called
|
# tree has terminated thereby conducting so called
|
||||||
# "end-to-end" structured concurrency throughout an
|
# "end-to-end" structured concurrency throughout an
|
||||||
|
|
@ -494,9 +372,9 @@ async def open_root_actor(
|
||||||
# "actor runtime" primitives are SC-compat and thus all
|
# "actor runtime" primitives are SC-compat and thus all
|
||||||
# transitively spawned actors/processes must be as
|
# transitively spawned actors/processes must be as
|
||||||
# well.
|
# well.
|
||||||
await root_tn.start(
|
await nursery.start(
|
||||||
partial(
|
partial(
|
||||||
_runtime.async_main,
|
async_main,
|
||||||
actor,
|
actor,
|
||||||
accept_addrs=trans_bind_addrs,
|
accept_addrs=trans_bind_addrs,
|
||||||
parent_addr=None
|
parent_addr=None
|
||||||
|
|
@ -509,26 +387,15 @@ async def open_root_actor(
|
||||||
BaseExceptionGroup,
|
BaseExceptionGroup,
|
||||||
) as err:
|
) as err:
|
||||||
|
|
||||||
# TODO, in beginning to handle the subsubactor with
|
|
||||||
# crashed grandparent cases..
|
|
||||||
#
|
|
||||||
# was_locked: bool = await debug.maybe_wait_for_debugger(
|
|
||||||
# child_in_debug=True,
|
|
||||||
# )
|
|
||||||
# XXX NOTE XXX see equiv note inside
|
# XXX NOTE XXX see equiv note inside
|
||||||
# `._runtime.Actor._stream_handler()` where in the
|
# `._runtime.Actor._stream_handler()` where in the
|
||||||
# non-root or root-that-opened-this-mahually case we
|
# non-root or root-that-opened-this-mahually case we
|
||||||
# wait for the local actor-nursery to exit before
|
# wait for the local actor-nursery to exit before
|
||||||
# exiting the transport channel handler.
|
# exiting the transport channel handler.
|
||||||
entered: bool = await debug._maybe_enter_pm(
|
entered: bool = await _debug._maybe_enter_pm(
|
||||||
err,
|
err,
|
||||||
api_frame=inspect.currentframe(),
|
api_frame=inspect.currentframe(),
|
||||||
debug_filter=debug_filter,
|
debug_filter=debug_filter,
|
||||||
|
|
||||||
# XXX NOTE, required to debug root-actor
|
|
||||||
# crashes under cancellation conditions; so
|
|
||||||
# most of them!
|
|
||||||
shield=root_tn.cancel_scope.cancel_called,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if (
|
if (
|
||||||
|
|
@ -538,18 +405,14 @@ async def open_root_actor(
|
||||||
err,
|
err,
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
logger.exception(
|
logger.exception('Root actor crashed\n')
|
||||||
'Root actor crashed\n'
|
|
||||||
f'>x)\n'
|
|
||||||
f' |_{actor}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# ALWAYS re-raise any error bubbled up from the
|
# ALWAYS re-raise any error bubbled up from the
|
||||||
# runtime!
|
# runtime!
|
||||||
raise
|
raise
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
# NOTE/TODO?, not sure if we'll ever need this but it's
|
# NOTE: not sure if we'll ever need this but it's
|
||||||
# possibly better for even more determinism?
|
# possibly better for even more determinism?
|
||||||
# logger.cancel(
|
# logger.cancel(
|
||||||
# f'Waiting on {len(nurseries)} nurseries in root..')
|
# f'Waiting on {len(nurseries)} nurseries in root..')
|
||||||
|
|
@ -558,45 +421,31 @@ async def open_root_actor(
|
||||||
# for an in nurseries:
|
# for an in nurseries:
|
||||||
# tempn.start_soon(an.exited.wait)
|
# tempn.start_soon(an.exited.wait)
|
||||||
|
|
||||||
op_nested_actor_repr: str = _pformat.nest_from_op(
|
|
||||||
input_op='>) ',
|
|
||||||
text=actor.pformat(),
|
|
||||||
nest_prefix='|_',
|
|
||||||
)
|
|
||||||
logger.info(
|
logger.info(
|
||||||
f'Closing down root actor\n'
|
'Closing down root actor'
|
||||||
f'{op_nested_actor_repr}'
|
|
||||||
)
|
)
|
||||||
# XXX, THIS IS A *finally-footgun*!
|
|
||||||
# (also mentioned in with-block above)
|
|
||||||
# -> though already shields iternally it can
|
|
||||||
# taskc here and mask underlying errors raised in
|
|
||||||
# the try-block above?
|
|
||||||
with trio.CancelScope(shield=True):
|
|
||||||
await actor.cancel(None) # self cancel
|
await actor.cancel(None) # self cancel
|
||||||
finally:
|
finally:
|
||||||
# revert all process-global runtime state
|
|
||||||
if (
|
|
||||||
debug_mode
|
|
||||||
and
|
|
||||||
_spawn._spawn_method == 'trio'
|
|
||||||
):
|
|
||||||
_state._runtime_vars['_debug_mode'] = False
|
|
||||||
|
|
||||||
_state._current_actor = None
|
_state._current_actor = None
|
||||||
_state._last_actor_terminated = actor
|
_state._last_actor_terminated = actor
|
||||||
|
|
||||||
sclang_repr: str = _pformat.nest_from_op(
|
# restore built-in `breakpoint()` hook state
|
||||||
input_op=')>',
|
if (
|
||||||
text=actor.pformat(),
|
debug_mode
|
||||||
nest_prefix='|_',
|
and
|
||||||
nest_indent=1,
|
maybe_enable_greenback
|
||||||
)
|
):
|
||||||
|
if builtin_bp_handler is not None:
|
||||||
|
sys.breakpointhook = builtin_bp_handler
|
||||||
|
|
||||||
logger.info(
|
if orig_bp_path is not None:
|
||||||
f'Root actor terminated\n'
|
os.environ['PYTHONBREAKPOINT'] = orig_bp_path
|
||||||
f'{sclang_repr}'
|
|
||||||
)
|
else:
|
||||||
|
# clear env back to having no entry
|
||||||
|
os.environ.pop('PYTHONBREAKPOINT', None)
|
||||||
|
|
||||||
|
logger.runtime("Root actor terminated")
|
||||||
|
|
||||||
|
|
||||||
def run_daemon(
|
def run_daemon(
|
||||||
|
|
@ -604,23 +453,16 @@ def run_daemon(
|
||||||
|
|
||||||
# runtime kwargs
|
# runtime kwargs
|
||||||
name: str | None = 'root',
|
name: str | None = 'root',
|
||||||
registry_addrs: list[UnwrappedAddress]|None = None,
|
registry_addrs: list[tuple[str, int]] = _default_lo_addrs,
|
||||||
|
|
||||||
start_method: str | None = None,
|
start_method: str | None = None,
|
||||||
debug_mode: bool = False,
|
debug_mode: bool = False,
|
||||||
|
|
||||||
# TODO, support `infected_aio=True` mode by,
|
|
||||||
# - calling the appropriate entrypoint-func from `.to_asyncio`
|
|
||||||
# - maybe init-ing `greenback` as done above in
|
|
||||||
# `open_root_actor()`.
|
|
||||||
|
|
||||||
**kwargs
|
**kwargs
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
Spawn a root (daemon) actor which will respond to RPC; the main
|
Spawn daemon actor which will respond to RPC; the main task simply
|
||||||
task simply starts the runtime and then blocks via embedded
|
starts the runtime and then sleeps forever.
|
||||||
`trio.sleep_forever()`.
|
|
||||||
|
|
||||||
This is a very minimal convenience wrapper around starting
|
This is a very minimal convenience wrapper around starting
|
||||||
a "run-until-cancelled" root actor which can be started with a set
|
a "run-until-cancelled" root actor which can be started with a set
|
||||||
|
|
@ -633,6 +475,7 @@ def run_daemon(
|
||||||
importlib.import_module(path)
|
importlib.import_module(path)
|
||||||
|
|
||||||
async def _main():
|
async def _main():
|
||||||
|
|
||||||
async with open_root_actor(
|
async with open_root_actor(
|
||||||
registry_addrs=registry_addrs,
|
registry_addrs=registry_addrs,
|
||||||
name=name,
|
name=name,
|
||||||
|
|
|
||||||
291
tractor/_rpc.py
291
tractor/_rpc.py
|
|
@ -37,13 +37,12 @@ import warnings
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
from trio import (
|
from trio import (
|
||||||
Cancelled,
|
|
||||||
CancelScope,
|
CancelScope,
|
||||||
Nursery,
|
Nursery,
|
||||||
TaskStatus,
|
TaskStatus,
|
||||||
)
|
)
|
||||||
|
|
||||||
from .ipc import Channel
|
from ._ipc import Channel
|
||||||
from ._context import (
|
from ._context import (
|
||||||
Context,
|
Context,
|
||||||
)
|
)
|
||||||
|
|
@ -53,18 +52,14 @@ from ._exceptions import (
|
||||||
ModuleNotExposed,
|
ModuleNotExposed,
|
||||||
MsgTypeError,
|
MsgTypeError,
|
||||||
TransportClosed,
|
TransportClosed,
|
||||||
|
is_multi_cancelled,
|
||||||
pack_error,
|
pack_error,
|
||||||
unpack_error,
|
unpack_error,
|
||||||
)
|
)
|
||||||
from .trionics import (
|
|
||||||
collapse_eg,
|
|
||||||
is_multi_cancelled,
|
|
||||||
maybe_raise_from_masking_exc,
|
|
||||||
)
|
|
||||||
from .devx import (
|
from .devx import (
|
||||||
debug,
|
maybe_wait_for_debugger,
|
||||||
|
_debug,
|
||||||
add_div,
|
add_div,
|
||||||
pformat as _pformat,
|
|
||||||
)
|
)
|
||||||
from . import _state
|
from . import _state
|
||||||
from .log import get_logger
|
from .log import get_logger
|
||||||
|
|
@ -73,7 +68,7 @@ from .msg import (
|
||||||
MsgCodec,
|
MsgCodec,
|
||||||
PayloadT,
|
PayloadT,
|
||||||
NamespacePath,
|
NamespacePath,
|
||||||
pretty_struct,
|
# pretty_struct,
|
||||||
_ops as msgops,
|
_ops as msgops,
|
||||||
)
|
)
|
||||||
from tractor.msg.types import (
|
from tractor.msg.types import (
|
||||||
|
|
@ -221,18 +216,11 @@ async def _invoke_non_context(
|
||||||
task_status.started(ctx)
|
task_status.started(ctx)
|
||||||
result = await coro
|
result = await coro
|
||||||
fname: str = func.__name__
|
fname: str = func.__name__
|
||||||
|
|
||||||
op_nested_task: str = _pformat.nest_from_op(
|
|
||||||
input_op=f')> cid: {ctx.cid!r}',
|
|
||||||
text=f'{ctx._task}',
|
|
||||||
nest_indent=1, # under >
|
|
||||||
)
|
|
||||||
log.runtime(
|
log.runtime(
|
||||||
f'RPC task complete\n'
|
'RPC complete:\n'
|
||||||
f'\n'
|
f'task: {ctx._task}\n'
|
||||||
f'{op_nested_task}\n'
|
f'|_cid={ctx.cid}\n'
|
||||||
f'\n'
|
f'|_{fname}() -> {pformat(result)}\n'
|
||||||
f')> {fname}() -> {pformat(result)}\n'
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# NOTE: only send result if we know IPC isn't down
|
# NOTE: only send result if we know IPC isn't down
|
||||||
|
|
@ -263,7 +251,7 @@ async def _errors_relayed_via_ipc(
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
is_rpc: bool,
|
is_rpc: bool,
|
||||||
|
|
||||||
hide_tb: bool = True,
|
hide_tb: bool = False,
|
||||||
debug_kbis: bool = False,
|
debug_kbis: bool = False,
|
||||||
task_status: TaskStatus[
|
task_status: TaskStatus[
|
||||||
Context | BaseException
|
Context | BaseException
|
||||||
|
|
@ -279,7 +267,7 @@ async def _errors_relayed_via_ipc(
|
||||||
|
|
||||||
# TODO: a debug nursery when in debug mode!
|
# TODO: a debug nursery when in debug mode!
|
||||||
# async with maybe_open_debugger_nursery() as debug_tn:
|
# async with maybe_open_debugger_nursery() as debug_tn:
|
||||||
# => see matching comment in side `.debug._pause()`
|
# => see matching comment in side `._debug._pause()`
|
||||||
rpc_err: BaseException|None = None
|
rpc_err: BaseException|None = None
|
||||||
try:
|
try:
|
||||||
yield # run RPC invoke body
|
yield # run RPC invoke body
|
||||||
|
|
@ -331,7 +319,7 @@ async def _errors_relayed_via_ipc(
|
||||||
'RPC task crashed, attempting to enter debugger\n'
|
'RPC task crashed, attempting to enter debugger\n'
|
||||||
f'|_{ctx}'
|
f'|_{ctx}'
|
||||||
)
|
)
|
||||||
entered_debug = await debug._maybe_enter_pm(
|
entered_debug = await _debug._maybe_enter_pm(
|
||||||
err,
|
err,
|
||||||
api_frame=inspect.currentframe(),
|
api_frame=inspect.currentframe(),
|
||||||
)
|
)
|
||||||
|
|
@ -384,13 +372,13 @@ async def _errors_relayed_via_ipc(
|
||||||
|
|
||||||
# RPC task bookeeping.
|
# RPC task bookeeping.
|
||||||
# since RPC tasks are scheduled inside a flat
|
# since RPC tasks are scheduled inside a flat
|
||||||
# `Actor._service_tn`, we add "handles" to each such that
|
# `Actor._service_n`, we add "handles" to each such that
|
||||||
# they can be individually ccancelled.
|
# they can be individually ccancelled.
|
||||||
finally:
|
finally:
|
||||||
|
|
||||||
# if the error is not from user code and instead a failure of
|
# if the error is not from user code and instead a failure
|
||||||
# an internal-runtime-RPC or IPC-connection, we do (prolly) want
|
# of a runtime RPC or transport failure we do prolly want to
|
||||||
# to show this frame!
|
# show this frame
|
||||||
if (
|
if (
|
||||||
rpc_err
|
rpc_err
|
||||||
and (
|
and (
|
||||||
|
|
@ -462,7 +450,7 @@ async def _invoke(
|
||||||
connected IPC channel.
|
connected IPC channel.
|
||||||
|
|
||||||
This is the core "RPC" `trio.Task` scheduling machinery used to start every
|
This is the core "RPC" `trio.Task` scheduling machinery used to start every
|
||||||
remotely invoked function, normally in `Actor._service_tn: Nursery`.
|
remotely invoked function, normally in `Actor._service_n: Nursery`.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
__tracebackhide__: bool = hide_tb
|
__tracebackhide__: bool = hide_tb
|
||||||
|
|
@ -475,7 +463,7 @@ async def _invoke(
|
||||||
):
|
):
|
||||||
# XXX for .pause_from_sync()` usage we need to make sure
|
# XXX for .pause_from_sync()` usage we need to make sure
|
||||||
# `greenback` is boostrapped in the subactor!
|
# `greenback` is boostrapped in the subactor!
|
||||||
await debug.maybe_init_greenback()
|
await _debug.maybe_init_greenback()
|
||||||
|
|
||||||
# TODO: possibly a specially formatted traceback
|
# TODO: possibly a specially formatted traceback
|
||||||
# (not sure what typing is for this..)?
|
# (not sure what typing is for this..)?
|
||||||
|
|
@ -629,39 +617,28 @@ async def _invoke(
|
||||||
# -> the below scope is never exposed to the
|
# -> the below scope is never exposed to the
|
||||||
# `@context` marked RPC function.
|
# `@context` marked RPC function.
|
||||||
# - `._portal` is never set.
|
# - `._portal` is never set.
|
||||||
scope_err: BaseException|None = None
|
|
||||||
try:
|
try:
|
||||||
# TODO: better `trionics` primitive/tooling usage here!
|
tn: trio.Nursery
|
||||||
# -[ ] should would be nice to have our `TaskMngr`
|
|
||||||
# nursery here!
|
|
||||||
# -[ ] payload value checking like we do with
|
|
||||||
# `.started()` such that the debbuger can engage
|
|
||||||
# here in the child task instead of waiting for the
|
|
||||||
# parent to crash with it's own MTE..
|
|
||||||
#
|
|
||||||
tn: Nursery
|
|
||||||
rpc_ctx_cs: CancelScope
|
rpc_ctx_cs: CancelScope
|
||||||
async with (
|
async with (
|
||||||
collapse_eg(hide_tb=False),
|
|
||||||
trio.open_nursery() as tn,
|
trio.open_nursery() as tn,
|
||||||
msgops.maybe_limit_plds(
|
msgops.maybe_limit_plds(
|
||||||
ctx=ctx,
|
ctx=ctx,
|
||||||
spec=ctx_meta.get('pld_spec'),
|
spec=ctx_meta.get('pld_spec'),
|
||||||
dec_hook=ctx_meta.get('dec_hook'),
|
dec_hook=ctx_meta.get('dec_hook'),
|
||||||
),
|
),
|
||||||
|
|
||||||
# XXX NOTE, this being the "most embedded"
|
|
||||||
# scope ensures unasking of the `await coro` below
|
|
||||||
# *should* never be interfered with!!
|
|
||||||
maybe_raise_from_masking_exc(
|
|
||||||
unmask_from=(Cancelled,),
|
|
||||||
) as _mbme, # maybe boxed masked exc
|
|
||||||
):
|
):
|
||||||
ctx._scope_nursery = tn
|
ctx._scope_nursery = tn
|
||||||
rpc_ctx_cs = ctx._scope = tn.cancel_scope
|
rpc_ctx_cs = ctx._scope = tn.cancel_scope
|
||||||
task_status.started(ctx)
|
task_status.started(ctx)
|
||||||
|
|
||||||
# invoke user endpoint fn.
|
# TODO: better `trionics` tooling:
|
||||||
|
# -[ ] should would be nice to have our `TaskMngr`
|
||||||
|
# nursery here!
|
||||||
|
# -[ ] payload value checking like we do with
|
||||||
|
# `.started()` such that the debbuger can engage
|
||||||
|
# here in the child task instead of waiting for the
|
||||||
|
# parent to crash with it's own MTE..
|
||||||
res: Any|PayloadT = await coro
|
res: Any|PayloadT = await coro
|
||||||
return_msg: Return|CancelAck = return_msg_type(
|
return_msg: Return|CancelAck = return_msg_type(
|
||||||
cid=cid,
|
cid=cid,
|
||||||
|
|
@ -669,11 +646,6 @@ async def _invoke(
|
||||||
)
|
)
|
||||||
# set and shuttle final result to "parent"-side task.
|
# set and shuttle final result to "parent"-side task.
|
||||||
ctx._result = res
|
ctx._result = res
|
||||||
log.runtime(
|
|
||||||
f'Sending result msg and exiting {ctx.side!r}\n'
|
|
||||||
f'\n'
|
|
||||||
f'{pretty_struct.pformat(return_msg)}\n'
|
|
||||||
)
|
|
||||||
await chan.send(return_msg)
|
await chan.send(return_msg)
|
||||||
|
|
||||||
# NOTE: this happens IFF `ctx._scope.cancel()` is
|
# NOTE: this happens IFF `ctx._scope.cancel()` is
|
||||||
|
|
@ -762,54 +734,45 @@ async def _invoke(
|
||||||
# XXX: do we ever trigger this block any more?
|
# XXX: do we ever trigger this block any more?
|
||||||
except (
|
except (
|
||||||
BaseExceptionGroup,
|
BaseExceptionGroup,
|
||||||
BaseException,
|
|
||||||
trio.Cancelled,
|
trio.Cancelled,
|
||||||
) as _scope_err:
|
BaseException,
|
||||||
scope_err = _scope_err
|
|
||||||
|
) as scope_error:
|
||||||
if (
|
if (
|
||||||
isinstance(scope_err, RuntimeError)
|
isinstance(scope_error, RuntimeError)
|
||||||
and
|
and scope_error.args
|
||||||
scope_err.args
|
and 'Cancel scope stack corrupted' in scope_error.args[0]
|
||||||
and
|
|
||||||
'Cancel scope stack corrupted' in scope_err.args[0]
|
|
||||||
):
|
):
|
||||||
log.exception('Cancel scope stack corrupted!?\n')
|
log.exception('Cancel scope stack corrupted!?\n')
|
||||||
# debug.mk_pdb().set_trace()
|
# _debug.mk_pdb().set_trace()
|
||||||
|
|
||||||
# always set this (child) side's exception as the
|
# always set this (child) side's exception as the
|
||||||
# local error on the context
|
# local error on the context
|
||||||
ctx._local_error: BaseException = scope_err
|
ctx._local_error: BaseException = scope_error
|
||||||
# ^-TODO-^ question,
|
# ^-TODO-^ question,
|
||||||
# does this matter other then for
|
# does this matter other then for
|
||||||
# consistentcy/testing?
|
# consistentcy/testing?
|
||||||
# |_ no user code should be in this scope at this point
|
# |_ no user code should be in this scope at this point
|
||||||
# AND we already set this in the block below?
|
# AND we already set this in the block below?
|
||||||
|
|
||||||
# XXX if a remote error was set then likely the
|
# if a remote error was set then likely the
|
||||||
# exc group was raised due to that, so
|
# exception group was raised due to that, so
|
||||||
# and we instead raise that error immediately!
|
# and we instead raise that error immediately!
|
||||||
maybe_re: (
|
ctx.maybe_raise()
|
||||||
ContextCancelled|RemoteActorError
|
|
||||||
) = ctx.maybe_raise()
|
|
||||||
if maybe_re:
|
|
||||||
log.cancel(
|
|
||||||
f'Suppressing remote-exc from peer,\n'
|
|
||||||
f'{maybe_re!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# maybe TODO: pack in come kinda
|
# maybe TODO: pack in come kinda
|
||||||
# `trio.Cancelled.__traceback__` here so they can be
|
# `trio.Cancelled.__traceback__` here so they can be
|
||||||
# unwrapped and displayed on the caller side? no se..
|
# unwrapped and displayed on the caller side? no se..
|
||||||
raise scope_err
|
raise
|
||||||
|
|
||||||
# `@context` entrypoint task bookeeping.
|
# `@context` entrypoint task bookeeping.
|
||||||
# i.e. only pop the context tracking if used ;)
|
# i.e. only pop the context tracking if used ;)
|
||||||
finally:
|
finally:
|
||||||
assert chan.aid
|
assert chan.uid
|
||||||
|
|
||||||
# don't pop the local context until we know the
|
# don't pop the local context until we know the
|
||||||
# associated child isn't in debug any more
|
# associated child isn't in debug any more
|
||||||
await debug.maybe_wait_for_debugger()
|
await maybe_wait_for_debugger()
|
||||||
ctx: Context = actor._contexts.pop((
|
ctx: Context = actor._contexts.pop((
|
||||||
chan.uid,
|
chan.uid,
|
||||||
cid,
|
cid,
|
||||||
|
|
@ -822,49 +785,26 @@ async def _invoke(
|
||||||
f'after having {ctx.repr_state!r}\n'
|
f'after having {ctx.repr_state!r}\n'
|
||||||
)
|
)
|
||||||
if merr:
|
if merr:
|
||||||
|
|
||||||
logmeth: Callable = log.error
|
logmeth: Callable = log.error
|
||||||
if (
|
if isinstance(merr, ContextCancelled):
|
||||||
# ctxc: by `Context.cancel()`
|
logmeth: Callable = log.runtime
|
||||||
isinstance(merr, ContextCancelled)
|
|
||||||
|
|
||||||
# out-of-layer cancellation, one of:
|
if not isinstance(merr, RemoteActorError):
|
||||||
# - actorc: by `Portal.cancel_actor()`
|
tb_str: str = ''.join(traceback.format_exception(merr))
|
||||||
# - OSc: by SIGINT or `Process.signal()`
|
|
||||||
or (
|
|
||||||
isinstance(merr, trio.Cancelled)
|
|
||||||
and
|
|
||||||
ctx.canceller
|
|
||||||
)
|
|
||||||
):
|
|
||||||
logmeth: Callable = log.cancel
|
|
||||||
descr_str += (
|
|
||||||
f' with {merr!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
elif (
|
|
||||||
not isinstance(merr, RemoteActorError)
|
|
||||||
):
|
|
||||||
tb_str: str = ''.join(
|
|
||||||
traceback.format_exception(merr)
|
|
||||||
)
|
|
||||||
descr_str += (
|
descr_str += (
|
||||||
f'\n{merr!r}\n' # needed?
|
f'\n{merr!r}\n' # needed?
|
||||||
f'{tb_str}\n'
|
f'{tb_str}\n'
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
descr_str += (
|
descr_str += f'\n{merr!r}\n'
|
||||||
f'{merr!r}\n'
|
|
||||||
)
|
|
||||||
else:
|
else:
|
||||||
descr_str += (
|
descr_str += f'\nand final result {ctx.outcome!r}\n'
|
||||||
f'\n'
|
|
||||||
f'with final result {ctx.outcome!r}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
logmeth(
|
logmeth(
|
||||||
f'{message}\n'
|
message
|
||||||
f'\n'
|
+
|
||||||
f'{descr_str}\n'
|
descr_str
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -908,8 +848,8 @@ async def try_ship_error_to_remote(
|
||||||
log.critical(
|
log.critical(
|
||||||
'IPC transport failure -> '
|
'IPC transport failure -> '
|
||||||
f'failed to ship error to {remote_descr}!\n\n'
|
f'failed to ship error to {remote_descr}!\n\n'
|
||||||
f'{type(msg)!r}[{msg.boxed_type_str}] X=> {channel.uid}\n'
|
f'X=> {channel.uid}\n\n'
|
||||||
f'\n'
|
|
||||||
# TODO: use `.msg.preetty_struct` for this!
|
# TODO: use `.msg.preetty_struct` for this!
|
||||||
f'{msg}\n'
|
f'{msg}\n'
|
||||||
)
|
)
|
||||||
|
|
@ -922,6 +862,7 @@ async def try_ship_error_to_remote(
|
||||||
|
|
||||||
|
|
||||||
async def process_messages(
|
async def process_messages(
|
||||||
|
actor: Actor,
|
||||||
chan: Channel,
|
chan: Channel,
|
||||||
shield: bool = False,
|
shield: bool = False,
|
||||||
task_status: TaskStatus[CancelScope] = trio.TASK_STATUS_IGNORED,
|
task_status: TaskStatus[CancelScope] = trio.TASK_STATUS_IGNORED,
|
||||||
|
|
@ -935,7 +876,7 @@ async def process_messages(
|
||||||
|
|
||||||
Receive (multiplexed) per-`Channel` RPC requests as msgs from
|
Receive (multiplexed) per-`Channel` RPC requests as msgs from
|
||||||
remote processes; schedule target async funcs as local
|
remote processes; schedule target async funcs as local
|
||||||
`trio.Task`s inside the `Actor._service_tn: Nursery`.
|
`trio.Task`s inside the `Actor._service_n: Nursery`.
|
||||||
|
|
||||||
Depending on msg type, non-`cmd` (task spawning/starting)
|
Depending on msg type, non-`cmd` (task spawning/starting)
|
||||||
request payloads (eg. `started`, `yield`, `return`, `error`)
|
request payloads (eg. `started`, `yield`, `return`, `error`)
|
||||||
|
|
@ -959,8 +900,7 @@ async def process_messages(
|
||||||
(as utilized inside `Portal.cancel_actor()` ).
|
(as utilized inside `Portal.cancel_actor()` ).
|
||||||
|
|
||||||
'''
|
'''
|
||||||
actor: Actor = _state.current_actor()
|
assert actor._service_n # runtime state sanity
|
||||||
assert actor._service_tn # runtime state sanity
|
|
||||||
|
|
||||||
# TODO: once `trio` get's an "obvious way" for req/resp we
|
# TODO: once `trio` get's an "obvious way" for req/resp we
|
||||||
# should use it?
|
# should use it?
|
||||||
|
|
@ -1031,10 +971,12 @@ async def process_messages(
|
||||||
cid=cid,
|
cid=cid,
|
||||||
kwargs=kwargs,
|
kwargs=kwargs,
|
||||||
):
|
):
|
||||||
|
kwargs |= {'req_chan': chan}
|
||||||
|
|
||||||
# XXX NOTE XXX don't start entire actor
|
# XXX NOTE XXX don't start entire actor
|
||||||
# runtime cancellation if this actor is
|
# runtime cancellation if this actor is
|
||||||
# currently in debug mode!
|
# currently in debug mode!
|
||||||
pdb_complete: trio.Event|None = debug.DebugStatus.repl_release
|
pdb_complete: trio.Event|None = _debug.DebugStatus.repl_release
|
||||||
if pdb_complete:
|
if pdb_complete:
|
||||||
await pdb_complete.wait()
|
await pdb_complete.wait()
|
||||||
|
|
||||||
|
|
@ -1049,14 +991,14 @@ async def process_messages(
|
||||||
cid,
|
cid,
|
||||||
chan,
|
chan,
|
||||||
actor.cancel,
|
actor.cancel,
|
||||||
kwargs | {'req_chan': chan},
|
kwargs,
|
||||||
is_rpc=False,
|
is_rpc=False,
|
||||||
return_msg_type=CancelAck,
|
return_msg_type=CancelAck,
|
||||||
)
|
)
|
||||||
|
|
||||||
log.runtime(
|
log.runtime(
|
||||||
'Cancelling RPC-msg-loop with peer\n'
|
'Cancelling IPC transport msg-loop with peer:\n'
|
||||||
f'->c}} {chan.aid.reprol()}@[{chan.maddr}]\n'
|
f'|_{chan}\n'
|
||||||
)
|
)
|
||||||
loop_cs.cancel()
|
loop_cs.cancel()
|
||||||
break
|
break
|
||||||
|
|
@ -1069,7 +1011,7 @@ async def process_messages(
|
||||||
):
|
):
|
||||||
target_cid: str = kwargs['cid']
|
target_cid: str = kwargs['cid']
|
||||||
kwargs |= {
|
kwargs |= {
|
||||||
'requesting_aid': chan.aid,
|
'requesting_uid': chan.uid,
|
||||||
'ipc_msg': msg,
|
'ipc_msg': msg,
|
||||||
|
|
||||||
# XXX NOTE! ONLY the rpc-task-owning
|
# XXX NOTE! ONLY the rpc-task-owning
|
||||||
|
|
@ -1105,34 +1047,21 @@ async def process_messages(
|
||||||
ns=ns,
|
ns=ns,
|
||||||
func=funcname,
|
func=funcname,
|
||||||
kwargs=kwargs, # type-spec this? see `msg.types`
|
kwargs=kwargs, # type-spec this? see `msg.types`
|
||||||
uid=actor_uuid,
|
uid=actorid,
|
||||||
):
|
):
|
||||||
if actor_uuid != chan.aid.uid:
|
|
||||||
raise RuntimeError(
|
|
||||||
f'IPC <Start> msg <-> chan.aid mismatch!?\n'
|
|
||||||
f'Channel.aid = {chan.aid!r}\n'
|
|
||||||
f'Start.uid = {actor_uuid!r}\n'
|
|
||||||
)
|
|
||||||
# await debug.pause()
|
|
||||||
op_repr: str = 'Start <=) '
|
|
||||||
req_repr: str = _pformat.nest_from_op(
|
|
||||||
input_op=op_repr,
|
|
||||||
op_suffix='',
|
|
||||||
nest_prefix='',
|
|
||||||
text=f'{chan}',
|
|
||||||
|
|
||||||
nest_indent=len(op_repr)-1,
|
|
||||||
rm_from_first_ln='<',
|
|
||||||
# ^XXX, subtract -1 to account for
|
|
||||||
# <Channel
|
|
||||||
# ^_chevron to be stripped
|
|
||||||
)
|
|
||||||
start_status: str = (
|
start_status: str = (
|
||||||
'Handling RPC request\n'
|
'Handling RPC `Start` request\n'
|
||||||
f'{req_repr}\n'
|
f'<= peer: {actorid}\n\n'
|
||||||
f'\n'
|
f' |_{chan}\n'
|
||||||
f'->{{ ipc-context-id: {cid!r}\n'
|
f' |_cid: {cid}\n\n'
|
||||||
f'->{{ nsp for fn: `{ns}.{funcname}({kwargs})`\n'
|
# f' |_{ns}.{funcname}({kwargs})\n'
|
||||||
|
f'>> {actor.uid}\n'
|
||||||
|
f' |_{actor}\n'
|
||||||
|
f' -> nsp: `{ns}.{funcname}({kwargs})`\n'
|
||||||
|
|
||||||
|
# f' |_{ns}.{funcname}({kwargs})\n\n'
|
||||||
|
|
||||||
|
# f'{pretty_struct.pformat(msg)}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
# runtime-internal endpoint: `Actor.<funcname>`
|
# runtime-internal endpoint: `Actor.<funcname>`
|
||||||
|
|
@ -1161,6 +1090,10 @@ async def process_messages(
|
||||||
await chan.send(err_msg)
|
await chan.send(err_msg)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
start_status += (
|
||||||
|
f' -> func: {func}\n'
|
||||||
|
)
|
||||||
|
|
||||||
# schedule a task for the requested RPC function
|
# schedule a task for the requested RPC function
|
||||||
# in the actor's main "service nursery".
|
# in the actor's main "service nursery".
|
||||||
#
|
#
|
||||||
|
|
@ -1168,10 +1101,10 @@ async def process_messages(
|
||||||
# supervision isolation? would avoid having to
|
# supervision isolation? would avoid having to
|
||||||
# manage RPC tasks individually in `._rpc_tasks`
|
# manage RPC tasks individually in `._rpc_tasks`
|
||||||
# table?
|
# table?
|
||||||
start_status += '->( scheduling new task..\n'
|
start_status += ' -> scheduling new task..\n'
|
||||||
log.runtime(start_status)
|
log.runtime(start_status)
|
||||||
try:
|
try:
|
||||||
ctx: Context = await actor._service_tn.start(
|
ctx: Context = await actor._service_n.start(
|
||||||
partial(
|
partial(
|
||||||
_invoke,
|
_invoke,
|
||||||
actor,
|
actor,
|
||||||
|
|
@ -1216,7 +1149,7 @@ async def process_messages(
|
||||||
trio.Event(),
|
trio.Event(),
|
||||||
)
|
)
|
||||||
|
|
||||||
# XXX RUNTIME-SCOPED! remote (likely internal) error
|
# runtime-scoped remote (internal) error
|
||||||
# (^- bc no `Error.cid` -^)
|
# (^- bc no `Error.cid` -^)
|
||||||
#
|
#
|
||||||
# NOTE: this is the non-rpc error case, that
|
# NOTE: this is the non-rpc error case, that
|
||||||
|
|
@ -1252,24 +1185,12 @@ async def process_messages(
|
||||||
# END-OF `async for`:
|
# END-OF `async for`:
|
||||||
# IPC disconnected via `trio.EndOfChannel`, likely
|
# IPC disconnected via `trio.EndOfChannel`, likely
|
||||||
# due to a (graceful) `Channel.aclose()`.
|
# due to a (graceful) `Channel.aclose()`.
|
||||||
|
|
||||||
chan_op_repr: str = '<=x] '
|
|
||||||
chan_repr: str = _pformat.nest_from_op(
|
|
||||||
input_op=chan_op_repr,
|
|
||||||
op_suffix='',
|
|
||||||
nest_prefix='',
|
|
||||||
text=chan.pformat(),
|
|
||||||
nest_indent=len(chan_op_repr)-1,
|
|
||||||
rm_from_first_ln='<',
|
|
||||||
)
|
|
||||||
log.runtime(
|
log.runtime(
|
||||||
f'IPC channel disconnected\n'
|
f'channel for {chan.uid} disconnected, cancelling RPC tasks\n'
|
||||||
f'{chan_repr}\n'
|
f'|_{chan}\n'
|
||||||
f'\n'
|
|
||||||
f'->c) cancelling RPC tasks.\n'
|
|
||||||
)
|
)
|
||||||
await actor.cancel_rpc_tasks(
|
await actor.cancel_rpc_tasks(
|
||||||
req_aid=actor.aid,
|
req_uid=actor.uid,
|
||||||
# a "self cancel" in terms of the lifetime of the
|
# a "self cancel" in terms of the lifetime of the
|
||||||
# IPC connection which is presumed to be the
|
# IPC connection which is presumed to be the
|
||||||
# source of any requests for spawned tasks.
|
# source of any requests for spawned tasks.
|
||||||
|
|
@ -1291,10 +1212,8 @@ async def process_messages(
|
||||||
# -[ ] figure out how this will break with other transports?
|
# -[ ] figure out how this will break with other transports?
|
||||||
tc.report_n_maybe_raise(
|
tc.report_n_maybe_raise(
|
||||||
message=(
|
message=(
|
||||||
f'peer IPC channel closed abruptly?\n'
|
f'peer IPC channel closed abruptly?\n\n'
|
||||||
f'\n'
|
f'<=x {chan}\n'
|
||||||
f'<=x[\n'
|
|
||||||
f' {chan}\n'
|
|
||||||
f' |_{chan.raddr}\n\n'
|
f' |_{chan.raddr}\n\n'
|
||||||
)
|
)
|
||||||
+
|
+
|
||||||
|
|
@ -1311,7 +1230,7 @@ async def process_messages(
|
||||||
) as err:
|
) as err:
|
||||||
|
|
||||||
if nursery_cancelled_before_task:
|
if nursery_cancelled_before_task:
|
||||||
sn: Nursery = actor._service_tn
|
sn: Nursery = actor._service_n
|
||||||
assert sn and sn.cancel_scope.cancel_called # sanity
|
assert sn and sn.cancel_scope.cancel_called # sanity
|
||||||
log.cancel(
|
log.cancel(
|
||||||
f'Service nursery cancelled before it handled {funcname}'
|
f'Service nursery cancelled before it handled {funcname}'
|
||||||
|
|
@ -1341,37 +1260,13 @@ async def process_messages(
|
||||||
finally:
|
finally:
|
||||||
# msg debugging for when he machinery is brokey
|
# msg debugging for when he machinery is brokey
|
||||||
if msg is None:
|
if msg is None:
|
||||||
message: str = 'Exiting RPC-loop without receiving a msg?'
|
message: str = 'Exiting IPC msg loop without receiving a msg?'
|
||||||
else:
|
else:
|
||||||
task_op_repr: str = ')>'
|
|
||||||
task: trio.Task = trio.lowlevel.current_task()
|
|
||||||
|
|
||||||
# maybe add cancelled opt prefix
|
|
||||||
if task._cancel_status.effectively_cancelled:
|
|
||||||
task_op_repr = 'c' + task_op_repr
|
|
||||||
|
|
||||||
task_repr: str = _pformat.nest_from_op(
|
|
||||||
input_op=task_op_repr,
|
|
||||||
text=f'{task!r}',
|
|
||||||
nest_indent=1,
|
|
||||||
)
|
|
||||||
# chan_op_repr: str = '<=} '
|
|
||||||
# chan_repr: str = _pformat.nest_from_op(
|
|
||||||
# input_op=chan_op_repr,
|
|
||||||
# op_suffix='',
|
|
||||||
# nest_prefix='',
|
|
||||||
# text=chan.pformat(),
|
|
||||||
# nest_indent=len(chan_op_repr)-1,
|
|
||||||
# rm_from_first_ln='<',
|
|
||||||
# )
|
|
||||||
message: str = (
|
message: str = (
|
||||||
f'Exiting RPC-loop with final msg\n'
|
'Exiting IPC msg loop with final msg\n\n'
|
||||||
f'\n'
|
f'<= peer: {chan.uid}\n'
|
||||||
# f'{chan_repr}\n'
|
f' |_{chan}\n\n'
|
||||||
f'{task_repr}\n'
|
# f'{pretty_struct.pformat(msg)}'
|
||||||
f'\n'
|
|
||||||
f'{pretty_struct.pformat(msg)}'
|
|
||||||
f'\n'
|
|
||||||
)
|
)
|
||||||
|
|
||||||
log.runtime(message)
|
log.runtime(message)
|
||||||
|
|
|
||||||
1455
tractor/_runtime.py
1455
tractor/_runtime.py
File diff suppressed because it is too large
Load Diff
|
|
@ -23,24 +23,19 @@ considered optional within the context of this runtime-library.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
from multiprocessing import shared_memory as shm
|
|
||||||
from multiprocessing.shared_memory import (
|
|
||||||
# SharedMemory,
|
|
||||||
ShareableList,
|
|
||||||
)
|
|
||||||
import platform
|
|
||||||
from sys import byteorder
|
from sys import byteorder
|
||||||
import time
|
import time
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
from multiprocessing import shared_memory as shm
|
||||||
from msgspec import (
|
from multiprocessing.shared_memory import (
|
||||||
Struct,
|
SharedMemory,
|
||||||
to_builtins
|
ShareableList,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
from msgspec import Struct
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
from tractor.ipc._mp_bs import disable_mantracker
|
from .log import get_logger
|
||||||
from tractor.log import get_logger
|
|
||||||
|
|
||||||
|
|
||||||
_USE_POSIX = getattr(shm, '_USE_POSIX', False)
|
_USE_POSIX = getattr(shm, '_USE_POSIX', False)
|
||||||
|
|
@ -51,10 +46,7 @@ if _USE_POSIX:
|
||||||
try:
|
try:
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from numpy.lib import recfunctions as rfn
|
from numpy.lib import recfunctions as rfn
|
||||||
# TODO ruff complains with,
|
# import nptyping
|
||||||
# warning| F401: `nptyping` imported but unused; consider using
|
|
||||||
# `importlib.util.find_spec` to test for availability
|
|
||||||
import nptyping # noqa
|
|
||||||
except ImportError:
|
except ImportError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
@ -62,7 +54,35 @@ except ImportError:
|
||||||
log = get_logger(__name__)
|
log = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
SharedMemory = disable_mantracker()
|
def disable_mantracker():
|
||||||
|
'''
|
||||||
|
Disable all ``multiprocessing``` "resource tracking" machinery since
|
||||||
|
it's an absolute multi-threaded mess of non-SC madness.
|
||||||
|
|
||||||
|
'''
|
||||||
|
from multiprocessing import resource_tracker as mantracker
|
||||||
|
|
||||||
|
# Tell the "resource tracker" thing to fuck off.
|
||||||
|
class ManTracker(mantracker.ResourceTracker):
|
||||||
|
def register(self, name, rtype):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def unregister(self, name, rtype):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def ensure_running(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# "know your land and know your prey"
|
||||||
|
# https://www.dailymotion.com/video/x6ozzco
|
||||||
|
mantracker._resource_tracker = ManTracker()
|
||||||
|
mantracker.register = mantracker._resource_tracker.register
|
||||||
|
mantracker.ensure_running = mantracker._resource_tracker.ensure_running
|
||||||
|
mantracker.unregister = mantracker._resource_tracker.unregister
|
||||||
|
mantracker.getfd = mantracker._resource_tracker.getfd
|
||||||
|
|
||||||
|
|
||||||
|
disable_mantracker()
|
||||||
|
|
||||||
|
|
||||||
class SharedInt:
|
class SharedInt:
|
||||||
|
|
@ -122,7 +142,7 @@ class NDToken(Struct, frozen=True):
|
||||||
).descr
|
).descr
|
||||||
|
|
||||||
def as_msg(self):
|
def as_msg(self):
|
||||||
return to_builtins(self)
|
return self.to_dict()
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_msg(cls, msg: dict) -> NDToken:
|
def from_msg(cls, msg: dict) -> NDToken:
|
||||||
|
|
@ -790,22 +810,10 @@ def open_shm_list(
|
||||||
readonly=readonly,
|
readonly=readonly,
|
||||||
)
|
)
|
||||||
|
|
||||||
# TODO, factor into a @actor_fixture acm-API?
|
|
||||||
# -[ ] also `@maybe_actor_fixture()` which inludes
|
|
||||||
# the .current_actor() convenience check?
|
|
||||||
# |_ orr can that just be in the sin-maybe-version?
|
|
||||||
#
|
|
||||||
# "close" attached shm on actor teardown
|
# "close" attached shm on actor teardown
|
||||||
try:
|
try:
|
||||||
actor = tractor.current_actor()
|
actor = tractor.current_actor()
|
||||||
|
|
||||||
actor.lifetime_stack.callback(shml.shm.close)
|
actor.lifetime_stack.callback(shml.shm.close)
|
||||||
|
|
||||||
# XXX on 3.13+ we don't need to call this?
|
|
||||||
# -> bc we pass `track=False` for `SharedMemeory` orr?
|
|
||||||
if (
|
|
||||||
platform.python_version_tuple()[:-1] < ('3', '13')
|
|
||||||
):
|
|
||||||
actor.lifetime_stack.callback(shml.shm.unlink)
|
actor.lifetime_stack.callback(shml.shm.unlink)
|
||||||
except RuntimeError:
|
except RuntimeError:
|
||||||
log.warning('tractor runtime not active, skipping teardown steps')
|
log.warning('tractor runtime not active, skipping teardown steps')
|
||||||
|
|
@ -34,9 +34,9 @@ from typing import (
|
||||||
import trio
|
import trio
|
||||||
from trio import TaskStatus
|
from trio import TaskStatus
|
||||||
|
|
||||||
from .devx import (
|
from tractor.devx import (
|
||||||
debug,
|
maybe_wait_for_debugger,
|
||||||
pformat as _pformat
|
acquire_debug_lock,
|
||||||
)
|
)
|
||||||
from tractor._state import (
|
from tractor._state import (
|
||||||
current_actor,
|
current_actor,
|
||||||
|
|
@ -46,26 +46,19 @@ from tractor._state import (
|
||||||
_runtime_vars,
|
_runtime_vars,
|
||||||
)
|
)
|
||||||
from tractor.log import get_logger
|
from tractor.log import get_logger
|
||||||
from tractor._addr import UnwrappedAddress
|
|
||||||
from tractor._portal import Portal
|
from tractor._portal import Portal
|
||||||
from tractor._runtime import Actor
|
from tractor._runtime import Actor
|
||||||
from tractor._entry import _mp_main
|
from tractor._entry import _mp_main
|
||||||
from tractor._exceptions import ActorFailure
|
from tractor._exceptions import ActorFailure
|
||||||
from tractor.msg import (
|
from tractor.msg.types import (
|
||||||
types as msgtypes,
|
SpawnSpec,
|
||||||
pretty_struct,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from ipc import (
|
|
||||||
_server,
|
|
||||||
Channel,
|
|
||||||
)
|
|
||||||
from ._supervise import ActorNursery
|
from ._supervise import ActorNursery
|
||||||
ProcessType = TypeVar('ProcessType', mp.Process, trio.Process)
|
ProcessType = TypeVar('ProcessType', mp.Process, trio.Process)
|
||||||
|
|
||||||
|
|
||||||
log = get_logger('tractor')
|
log = get_logger('tractor')
|
||||||
|
|
||||||
# placeholder for an mp start context if so using that backend
|
# placeholder for an mp start context if so using that backend
|
||||||
|
|
@ -170,7 +163,7 @@ async def exhaust_portal(
|
||||||
# TODO: merge with above?
|
# TODO: merge with above?
|
||||||
log.warning(
|
log.warning(
|
||||||
'Cancelled portal result waiter task:\n'
|
'Cancelled portal result waiter task:\n'
|
||||||
f'uid: {portal.channel.aid}\n'
|
f'uid: {portal.channel.uid}\n'
|
||||||
f'error: {err}\n'
|
f'error: {err}\n'
|
||||||
)
|
)
|
||||||
return err
|
return err
|
||||||
|
|
@ -178,7 +171,7 @@ async def exhaust_portal(
|
||||||
else:
|
else:
|
||||||
log.debug(
|
log.debug(
|
||||||
f'Returning final result from portal:\n'
|
f'Returning final result from portal:\n'
|
||||||
f'uid: {portal.channel.aid}\n'
|
f'uid: {portal.channel.uid}\n'
|
||||||
f'result: {final}\n'
|
f'result: {final}\n'
|
||||||
)
|
)
|
||||||
return final
|
return final
|
||||||
|
|
@ -297,23 +290,6 @@ async def hard_kill(
|
||||||
# zombies (as a feature) we ask the OS to do send in the
|
# zombies (as a feature) we ask the OS to do send in the
|
||||||
# removal swad as the last resort.
|
# removal swad as the last resort.
|
||||||
if cs.cancelled_caught:
|
if cs.cancelled_caught:
|
||||||
|
|
||||||
# TODO? attempt at intermediary-rent-sub
|
|
||||||
# with child in debug lock?
|
|
||||||
# |_https://github.com/goodboy/tractor/issues/320
|
|
||||||
#
|
|
||||||
# if not is_root_process():
|
|
||||||
# log.warning(
|
|
||||||
# 'Attempting to acquire debug-REPL-lock before zombie reap!'
|
|
||||||
# )
|
|
||||||
# with trio.CancelScope(shield=True):
|
|
||||||
# async with debug.acquire_debug_lock(
|
|
||||||
# subactor_uid=current_actor().uid,
|
|
||||||
# ) as _ctx:
|
|
||||||
# log.warning(
|
|
||||||
# 'Acquired debug lock, child ready to be killed ??\n'
|
|
||||||
# )
|
|
||||||
|
|
||||||
# TODO: toss in the skynet-logo face as ascii art?
|
# TODO: toss in the skynet-logo face as ascii art?
|
||||||
log.critical(
|
log.critical(
|
||||||
# 'Well, the #ZOMBIE_LORD_IS_HERE# to collect\n'
|
# 'Well, the #ZOMBIE_LORD_IS_HERE# to collect\n'
|
||||||
|
|
@ -344,13 +320,11 @@ async def soft_kill(
|
||||||
see `.hard_kill()`).
|
see `.hard_kill()`).
|
||||||
|
|
||||||
'''
|
'''
|
||||||
chan: Channel = portal.channel
|
uid: tuple[str, str] = portal.channel.uid
|
||||||
peer_aid: msgtypes.Aid = chan.aid
|
|
||||||
try:
|
try:
|
||||||
log.cancel(
|
log.cancel(
|
||||||
f'Soft killing sub-actor via portal request\n'
|
'Soft killing sub-actor via portal request\n'
|
||||||
f'\n'
|
f'c)> {portal.chan.uid}\n'
|
||||||
f'c)=> {peer_aid.reprol()}@[{chan.maddr}]\n'
|
|
||||||
f' |_{proc}\n'
|
f' |_{proc}\n'
|
||||||
)
|
)
|
||||||
# wait on sub-proc to signal termination
|
# wait on sub-proc to signal termination
|
||||||
|
|
@ -358,7 +332,7 @@ async def soft_kill(
|
||||||
|
|
||||||
except trio.Cancelled:
|
except trio.Cancelled:
|
||||||
with trio.CancelScope(shield=True):
|
with trio.CancelScope(shield=True):
|
||||||
await debug.maybe_wait_for_debugger(
|
await maybe_wait_for_debugger(
|
||||||
child_in_debug=_runtime_vars.get(
|
child_in_debug=_runtime_vars.get(
|
||||||
'_debug_mode', False
|
'_debug_mode', False
|
||||||
),
|
),
|
||||||
|
|
@ -399,7 +373,7 @@ async def soft_kill(
|
||||||
if proc.poll() is None: # type: ignore
|
if proc.poll() is None: # type: ignore
|
||||||
log.warning(
|
log.warning(
|
||||||
'Subactor still alive after cancel request?\n\n'
|
'Subactor still alive after cancel request?\n\n'
|
||||||
f'uid: {peer_aid}\n'
|
f'uid: {uid}\n'
|
||||||
f'|_{proc}\n'
|
f'|_{proc}\n'
|
||||||
)
|
)
|
||||||
n.cancel_scope.cancel()
|
n.cancel_scope.cancel()
|
||||||
|
|
@ -413,15 +387,14 @@ async def new_proc(
|
||||||
errors: dict[tuple[str, str], Exception],
|
errors: dict[tuple[str, str], Exception],
|
||||||
|
|
||||||
# passed through to actor main
|
# passed through to actor main
|
||||||
bind_addrs: list[UnwrappedAddress],
|
bind_addrs: list[tuple[str, int]],
|
||||||
parent_addr: UnwrappedAddress,
|
parent_addr: tuple[str, int],
|
||||||
_runtime_vars: dict[str, Any], # serialized and sent to _child
|
_runtime_vars: dict[str, Any], # serialized and sent to _child
|
||||||
|
|
||||||
*,
|
*,
|
||||||
|
|
||||||
infect_asyncio: bool = False,
|
infect_asyncio: bool = False,
|
||||||
task_status: TaskStatus[Portal] = trio.TASK_STATUS_IGNORED,
|
task_status: TaskStatus[Portal] = trio.TASK_STATUS_IGNORED
|
||||||
proc_kwargs: dict[str, any] = {}
|
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
|
|
@ -441,7 +414,6 @@ async def new_proc(
|
||||||
_runtime_vars, # run time vars
|
_runtime_vars, # run time vars
|
||||||
infect_asyncio=infect_asyncio,
|
infect_asyncio=infect_asyncio,
|
||||||
task_status=task_status,
|
task_status=task_status,
|
||||||
proc_kwargs=proc_kwargs
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -452,13 +424,12 @@ async def trio_proc(
|
||||||
errors: dict[tuple[str, str], Exception],
|
errors: dict[tuple[str, str], Exception],
|
||||||
|
|
||||||
# passed through to actor main
|
# passed through to actor main
|
||||||
bind_addrs: list[UnwrappedAddress],
|
bind_addrs: list[tuple[str, int]],
|
||||||
parent_addr: UnwrappedAddress,
|
parent_addr: tuple[str, int],
|
||||||
_runtime_vars: dict[str, Any], # serialized and sent to _child
|
_runtime_vars: dict[str, Any], # serialized and sent to _child
|
||||||
*,
|
*,
|
||||||
infect_asyncio: bool = False,
|
infect_asyncio: bool = False,
|
||||||
task_status: TaskStatus[Portal] = trio.TASK_STATUS_IGNORED,
|
task_status: TaskStatus[Portal] = trio.TASK_STATUS_IGNORED
|
||||||
proc_kwargs: dict[str, any] = {}
|
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
'''
|
'''
|
||||||
|
|
@ -480,9 +451,6 @@ async def trio_proc(
|
||||||
# the OS; it otherwise can be passed via the parent channel if
|
# the OS; it otherwise can be passed via the parent channel if
|
||||||
# we prefer in the future (for privacy).
|
# we prefer in the future (for privacy).
|
||||||
"--uid",
|
"--uid",
|
||||||
# TODO, how to pass this over "wire" encodings like
|
|
||||||
# cmdline args?
|
|
||||||
# -[ ] maybe we can add an `msgtypes.Aid.min_tuple()` ?
|
|
||||||
str(subactor.uid),
|
str(subactor.uid),
|
||||||
# Address the child must connect to on startup
|
# Address the child must connect to on startup
|
||||||
"--parent_addr",
|
"--parent_addr",
|
||||||
|
|
@ -500,20 +468,18 @@ async def trio_proc(
|
||||||
|
|
||||||
cancelled_during_spawn: bool = False
|
cancelled_during_spawn: bool = False
|
||||||
proc: trio.Process|None = None
|
proc: trio.Process|None = None
|
||||||
ipc_server: _server.Server = actor_nursery._actor.ipc_server
|
|
||||||
try:
|
try:
|
||||||
try:
|
try:
|
||||||
proc: trio.Process = await trio.lowlevel.open_process(spawn_cmd, **proc_kwargs)
|
proc: trio.Process = await trio.lowlevel.open_process(spawn_cmd)
|
||||||
log.runtime(
|
log.runtime(
|
||||||
f'Started new child subproc\n'
|
'Started new child\n'
|
||||||
f'(>\n'
|
|
||||||
f'|_{proc}\n'
|
f'|_{proc}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
# wait for actor to spawn and connect back to us
|
# wait for actor to spawn and connect back to us
|
||||||
# channel should have handshake completed by the
|
# channel should have handshake completed by the
|
||||||
# local actor by the time we get a ref to it
|
# local actor by the time we get a ref to it
|
||||||
event, chan = await ipc_server.wait_for_peer(
|
event, chan = await actor_nursery._actor.wait_for_peer(
|
||||||
subactor.uid
|
subactor.uid
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -525,10 +491,10 @@ async def trio_proc(
|
||||||
with trio.CancelScope(shield=True):
|
with trio.CancelScope(shield=True):
|
||||||
# don't clobber an ongoing pdb
|
# don't clobber an ongoing pdb
|
||||||
if is_root_process():
|
if is_root_process():
|
||||||
await debug.maybe_wait_for_debugger()
|
await maybe_wait_for_debugger()
|
||||||
|
|
||||||
elif proc is not None:
|
elif proc is not None:
|
||||||
async with debug.acquire_debug_lock(subactor.uid):
|
async with acquire_debug_lock(subactor.uid):
|
||||||
# soft wait on the proc to terminate
|
# soft wait on the proc to terminate
|
||||||
with trio.move_on_after(0.5):
|
with trio.move_on_after(0.5):
|
||||||
await proc.wait()
|
await proc.wait()
|
||||||
|
|
@ -546,20 +512,15 @@ async def trio_proc(
|
||||||
|
|
||||||
# send a "spawning specification" which configures the
|
# send a "spawning specification" which configures the
|
||||||
# initial runtime state of the child.
|
# initial runtime state of the child.
|
||||||
sspec = msgtypes.SpawnSpec(
|
await chan.send(
|
||||||
|
SpawnSpec(
|
||||||
_parent_main_data=subactor._parent_main_data,
|
_parent_main_data=subactor._parent_main_data,
|
||||||
enable_modules=subactor.enable_modules,
|
enable_modules=subactor.enable_modules,
|
||||||
reg_addrs=subactor.reg_addrs,
|
reg_addrs=subactor.reg_addrs,
|
||||||
bind_addrs=bind_addrs,
|
bind_addrs=bind_addrs,
|
||||||
_runtime_vars=_runtime_vars,
|
_runtime_vars=_runtime_vars,
|
||||||
)
|
)
|
||||||
log.runtime(
|
|
||||||
f'Sending spawn spec to child\n'
|
|
||||||
f'{{}}=> {chan.aid.reprol()!r}\n'
|
|
||||||
f'\n'
|
|
||||||
f'{pretty_struct.pformat(sspec)}\n'
|
|
||||||
)
|
)
|
||||||
await chan.send(sspec)
|
|
||||||
|
|
||||||
# track subactor in current nursery
|
# track subactor in current nursery
|
||||||
curr_actor: Actor = current_actor()
|
curr_actor: Actor = current_actor()
|
||||||
|
|
@ -586,7 +547,7 @@ async def trio_proc(
|
||||||
# condition.
|
# condition.
|
||||||
await soft_kill(
|
await soft_kill(
|
||||||
proc,
|
proc,
|
||||||
trio.Process.wait, # XXX, uses `pidfd_open()` below.
|
trio.Process.wait,
|
||||||
portal
|
portal
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -594,7 +555,8 @@ async def trio_proc(
|
||||||
# tandem if not done already
|
# tandem if not done already
|
||||||
log.cancel(
|
log.cancel(
|
||||||
'Cancelling portal result reaper task\n'
|
'Cancelling portal result reaper task\n'
|
||||||
f'c)> {subactor.aid.reprol()!r}\n'
|
f'>c)\n'
|
||||||
|
f' |_{subactor.uid}\n'
|
||||||
)
|
)
|
||||||
nursery.cancel_scope.cancel()
|
nursery.cancel_scope.cancel()
|
||||||
|
|
||||||
|
|
@ -603,24 +565,21 @@ async def trio_proc(
|
||||||
# allowed! Do this **after** cancellation/teardown to avoid
|
# allowed! Do this **after** cancellation/teardown to avoid
|
||||||
# killing the process too early.
|
# killing the process too early.
|
||||||
if proc:
|
if proc:
|
||||||
reap_repr: str = _pformat.nest_from_op(
|
|
||||||
input_op='>x)',
|
|
||||||
text=subactor.pformat(),
|
|
||||||
)
|
|
||||||
log.cancel(
|
log.cancel(
|
||||||
f'Hard reap sequence starting for subactor\n'
|
f'Hard reap sequence starting for subactor\n'
|
||||||
f'{reap_repr}'
|
f'>x)\n'
|
||||||
|
f' |_{subactor}@{subactor.uid}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
with trio.CancelScope(shield=True):
|
with trio.CancelScope(shield=True):
|
||||||
# don't clobber an ongoing pdb
|
# don't clobber an ongoing pdb
|
||||||
if cancelled_during_spawn:
|
if cancelled_during_spawn:
|
||||||
# Try again to avoid TTY clobbering.
|
# Try again to avoid TTY clobbering.
|
||||||
async with debug.acquire_debug_lock(subactor.uid):
|
async with acquire_debug_lock(subactor.uid):
|
||||||
with trio.move_on_after(0.5):
|
with trio.move_on_after(0.5):
|
||||||
await proc.wait()
|
await proc.wait()
|
||||||
|
|
||||||
await debug.maybe_wait_for_debugger(
|
await maybe_wait_for_debugger(
|
||||||
child_in_debug=_runtime_vars.get(
|
child_in_debug=_runtime_vars.get(
|
||||||
'_debug_mode', False
|
'_debug_mode', False
|
||||||
),
|
),
|
||||||
|
|
@ -649,7 +608,7 @@ async def trio_proc(
|
||||||
# acquire the lock and get notified of who has it,
|
# acquire the lock and get notified of who has it,
|
||||||
# check that uid against our known children?
|
# check that uid against our known children?
|
||||||
# this_uid: tuple[str, str] = current_actor().uid
|
# this_uid: tuple[str, str] = current_actor().uid
|
||||||
# await debug.acquire_debug_lock(this_uid)
|
# await acquire_debug_lock(this_uid)
|
||||||
|
|
||||||
if proc.poll() is None:
|
if proc.poll() is None:
|
||||||
log.cancel(f"Attempting to hard kill {proc}")
|
log.cancel(f"Attempting to hard kill {proc}")
|
||||||
|
|
@ -671,13 +630,12 @@ async def mp_proc(
|
||||||
subactor: Actor,
|
subactor: Actor,
|
||||||
errors: dict[tuple[str, str], Exception],
|
errors: dict[tuple[str, str], Exception],
|
||||||
# passed through to actor main
|
# passed through to actor main
|
||||||
bind_addrs: list[UnwrappedAddress],
|
bind_addrs: list[tuple[str, int]],
|
||||||
parent_addr: UnwrappedAddress,
|
parent_addr: tuple[str, int],
|
||||||
_runtime_vars: dict[str, Any], # serialized and sent to _child
|
_runtime_vars: dict[str, Any], # serialized and sent to _child
|
||||||
*,
|
*,
|
||||||
infect_asyncio: bool = False,
|
infect_asyncio: bool = False,
|
||||||
task_status: TaskStatus[Portal] = trio.TASK_STATUS_IGNORED,
|
task_status: TaskStatus[Portal] = trio.TASK_STATUS_IGNORED
|
||||||
proc_kwargs: dict[str, any] = {}
|
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
|
|
@ -752,14 +710,12 @@ async def mp_proc(
|
||||||
|
|
||||||
log.runtime(f"Started {proc}")
|
log.runtime(f"Started {proc}")
|
||||||
|
|
||||||
ipc_server: _server.Server = actor_nursery._actor.ipc_server
|
|
||||||
try:
|
try:
|
||||||
# wait for actor to spawn and connect back to us
|
# wait for actor to spawn and connect back to us
|
||||||
# channel should have handshake completed by the
|
# channel should have handshake completed by the
|
||||||
# local actor by the time we get a ref to it
|
# local actor by the time we get a ref to it
|
||||||
event, chan = await ipc_server.wait_for_peer(
|
event, chan = await actor_nursery._actor.wait_for_peer(
|
||||||
subactor.uid,
|
subactor.uid)
|
||||||
)
|
|
||||||
|
|
||||||
# XXX: monkey patch poll API to match the ``subprocess`` API..
|
# XXX: monkey patch poll API to match the ``subprocess`` API..
|
||||||
# not sure why they don't expose this but kk.
|
# not sure why they don't expose this but kk.
|
||||||
|
|
|
||||||
|
|
@ -14,19 +14,16 @@
|
||||||
# You should have received a copy of the GNU Affero General Public License
|
# You should have received a copy of the GNU Affero General Public License
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
'''
|
"""
|
||||||
Per actor-process runtime state mgmt APIs.
|
Per process state
|
||||||
|
|
||||||
'''
|
"""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
from contextvars import (
|
from contextvars import (
|
||||||
ContextVar,
|
ContextVar,
|
||||||
)
|
)
|
||||||
import os
|
|
||||||
from pathlib import Path
|
|
||||||
from typing import (
|
from typing import (
|
||||||
Any,
|
Any,
|
||||||
Literal,
|
|
||||||
TYPE_CHECKING,
|
TYPE_CHECKING,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -37,39 +34,20 @@ if TYPE_CHECKING:
|
||||||
from ._context import Context
|
from ._context import Context
|
||||||
|
|
||||||
|
|
||||||
# default IPC transport protocol settings
|
|
||||||
TransportProtocolKey = Literal[
|
|
||||||
'tcp',
|
|
||||||
'uds',
|
|
||||||
]
|
|
||||||
_def_tpt_proto: TransportProtocolKey = 'tcp'
|
|
||||||
|
|
||||||
_current_actor: Actor|None = None # type: ignore # noqa
|
_current_actor: Actor|None = None # type: ignore # noqa
|
||||||
_last_actor_terminated: Actor|None = None
|
_last_actor_terminated: Actor|None = None
|
||||||
|
|
||||||
# TODO: mk this a `msgspec.Struct`!
|
# TODO: mk this a `msgspec.Struct`!
|
||||||
# -[ ] type out all fields obvi!
|
|
||||||
# -[ ] (eventually) mk wire-ready for monitoring?
|
|
||||||
_runtime_vars: dict[str, Any] = {
|
_runtime_vars: dict[str, Any] = {
|
||||||
# root of actor-process tree info
|
'_debug_mode': False,
|
||||||
'_is_root': False, # bool
|
'_is_root': False,
|
||||||
'_root_mailbox': (None, None), # tuple[str|None, str|None]
|
'_root_mailbox': (None, None),
|
||||||
'_root_addrs': [], # tuple[str|None, str|None]
|
|
||||||
|
|
||||||
# parent->chld ipc protocol caps
|
|
||||||
'_enable_tpts': [_def_tpt_proto],
|
|
||||||
|
|
||||||
# registrar info
|
|
||||||
'_registry_addrs': [],
|
'_registry_addrs': [],
|
||||||
|
|
||||||
# `debug_mode: bool` settings
|
'_is_infected_aio': False,
|
||||||
'_debug_mode': False, # bool
|
|
||||||
'repl_fixture': False, # |AbstractContextManager[bool]
|
|
||||||
# for `tractor.pause_from_sync()` & `breakpoint()` support
|
# for `tractor.pause_from_sync()` & `breakpoint()` support
|
||||||
'use_greenback': False,
|
'use_greenback': False,
|
||||||
|
|
||||||
# infected-`asyncio`-mode: `trio` running as guest.
|
|
||||||
'_is_infected_aio': False,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -121,7 +99,7 @@ def current_actor(
|
||||||
return _current_actor
|
return _current_actor
|
||||||
|
|
||||||
|
|
||||||
def is_root_process() -> bool:
|
def is_main_process() -> bool:
|
||||||
'''
|
'''
|
||||||
Bool determining if this actor is running in the top-most process.
|
Bool determining if this actor is running in the top-most process.
|
||||||
|
|
||||||
|
|
@ -130,10 +108,7 @@ def is_root_process() -> bool:
|
||||||
return mp.current_process().name == 'MainProcess'
|
return mp.current_process().name == 'MainProcess'
|
||||||
|
|
||||||
|
|
||||||
is_main_process = is_root_process
|
def debug_mode() -> bool:
|
||||||
|
|
||||||
|
|
||||||
def is_debug_mode() -> bool:
|
|
||||||
'''
|
'''
|
||||||
Bool determining if "debug mode" is on which enables
|
Bool determining if "debug mode" is on which enables
|
||||||
remote subactor pdb entry on crashes.
|
remote subactor pdb entry on crashes.
|
||||||
|
|
@ -142,9 +117,6 @@ def is_debug_mode() -> bool:
|
||||||
return bool(_runtime_vars['_debug_mode'])
|
return bool(_runtime_vars['_debug_mode'])
|
||||||
|
|
||||||
|
|
||||||
debug_mode = is_debug_mode
|
|
||||||
|
|
||||||
|
|
||||||
def is_root_process() -> bool:
|
def is_root_process() -> bool:
|
||||||
return _runtime_vars['_is_root']
|
return _runtime_vars['_is_root']
|
||||||
|
|
||||||
|
|
@ -170,34 +142,3 @@ def current_ipc_ctx(
|
||||||
f'|_{current_task()}\n'
|
f'|_{current_task()}\n'
|
||||||
)
|
)
|
||||||
return ctx
|
return ctx
|
||||||
|
|
||||||
|
|
||||||
# std ODE (mutable) app state location
|
|
||||||
_rtdir: Path = Path(os.environ['XDG_RUNTIME_DIR'])
|
|
||||||
|
|
||||||
|
|
||||||
def get_rt_dir(
|
|
||||||
subdir: str = 'tractor'
|
|
||||||
) -> Path:
|
|
||||||
'''
|
|
||||||
Return the user "runtime dir" where most userspace apps stick
|
|
||||||
their IPC and cache related system util-files; we take hold
|
|
||||||
of a `'XDG_RUNTIME_DIR'/tractor/` subdir by default.
|
|
||||||
|
|
||||||
'''
|
|
||||||
rtdir: Path = _rtdir / subdir
|
|
||||||
if not rtdir.is_dir():
|
|
||||||
rtdir.mkdir()
|
|
||||||
return rtdir
|
|
||||||
|
|
||||||
|
|
||||||
def current_ipc_protos() -> list[str]:
|
|
||||||
'''
|
|
||||||
Return the list of IPC transport protocol keys currently
|
|
||||||
in use by this actor.
|
|
||||||
|
|
||||||
The keys are as declared by `MsgTransport` and `Address`
|
|
||||||
concrete-backend sub-types defined throughout `tractor.ipc`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return _runtime_vars['_enable_tpts']
|
|
||||||
|
|
|
||||||
|
|
@ -45,18 +45,16 @@ from .trionics import (
|
||||||
BroadcastReceiver,
|
BroadcastReceiver,
|
||||||
)
|
)
|
||||||
from tractor.msg import (
|
from tractor.msg import (
|
||||||
Error,
|
# Return,
|
||||||
Return,
|
# Stop,
|
||||||
Stop,
|
|
||||||
MsgType,
|
MsgType,
|
||||||
PayloadT,
|
|
||||||
Yield,
|
Yield,
|
||||||
)
|
)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from ._runtime import Actor
|
from ._runtime import Actor
|
||||||
from ._context import Context
|
from ._context import Context
|
||||||
from .ipc import Channel
|
from ._ipc import Channel
|
||||||
|
|
||||||
|
|
||||||
log = get_logger(__name__)
|
log = get_logger(__name__)
|
||||||
|
|
@ -72,7 +70,8 @@ class MsgStream(trio.abc.Channel):
|
||||||
A bidirectional message stream for receiving logically sequenced
|
A bidirectional message stream for receiving logically sequenced
|
||||||
values over an inter-actor IPC `Channel`.
|
values over an inter-actor IPC `Channel`.
|
||||||
|
|
||||||
|
This is the type returned to a local task which entered either
|
||||||
|
`Portal.open_stream_from()` or `Context.open_stream()`.
|
||||||
|
|
||||||
Termination rules:
|
Termination rules:
|
||||||
|
|
||||||
|
|
@ -95,9 +94,6 @@ class MsgStream(trio.abc.Channel):
|
||||||
self._rx_chan = rx_chan
|
self._rx_chan = rx_chan
|
||||||
self._broadcaster = _broadcaster
|
self._broadcaster = _broadcaster
|
||||||
|
|
||||||
# any actual IPC msg which is effectively an `EndOfStream`
|
|
||||||
self._stop_msg: bool|Stop = False
|
|
||||||
|
|
||||||
# flag to denote end of stream
|
# flag to denote end of stream
|
||||||
self._eoc: bool|trio.EndOfChannel = False
|
self._eoc: bool|trio.EndOfChannel = False
|
||||||
self._closed: bool|trio.ClosedResourceError = False
|
self._closed: bool|trio.ClosedResourceError = False
|
||||||
|
|
@ -129,67 +125,16 @@ class MsgStream(trio.abc.Channel):
|
||||||
def receive_nowait(
|
def receive_nowait(
|
||||||
self,
|
self,
|
||||||
expect_msg: MsgType = Yield,
|
expect_msg: MsgType = Yield,
|
||||||
) -> PayloadT:
|
):
|
||||||
ctx: Context = self._ctx
|
ctx: Context = self._ctx
|
||||||
(
|
return ctx._pld_rx.recv_pld_nowait(
|
||||||
msg,
|
|
||||||
pld,
|
|
||||||
) = ctx._pld_rx.recv_msg_nowait(
|
|
||||||
ipc=self,
|
ipc=self,
|
||||||
expect_msg=expect_msg,
|
expect_msg=expect_msg,
|
||||||
)
|
)
|
||||||
|
|
||||||
# ?TODO, maybe factor this into a hyper-common `unwrap_pld()`
|
|
||||||
#
|
|
||||||
match msg:
|
|
||||||
|
|
||||||
# XXX, these never seems to ever hit? cool?
|
|
||||||
case Stop():
|
|
||||||
log.cancel(
|
|
||||||
f'Msg-stream was ended via stop msg\n'
|
|
||||||
f'{msg}'
|
|
||||||
)
|
|
||||||
case Error():
|
|
||||||
log.error(
|
|
||||||
f'Msg-stream was ended via error msg\n'
|
|
||||||
f'{msg}'
|
|
||||||
)
|
|
||||||
|
|
||||||
# XXX NOTE, always set any final result on the ctx to
|
|
||||||
# avoid teardown race conditions where previously this msg
|
|
||||||
# would be consumed silently (by `.aclose()` doing its
|
|
||||||
# own "msg drain loop" but WITHOUT those `drained: lists[MsgType]`
|
|
||||||
# being post-close-processed!
|
|
||||||
#
|
|
||||||
# !!TODO, see the equiv todo-comment in `.receive()`
|
|
||||||
# around the `if drained:` where we should prolly
|
|
||||||
# ACTUALLY be doing this post-close processing??
|
|
||||||
#
|
|
||||||
case Return(pld=pld):
|
|
||||||
log.warning(
|
|
||||||
f'Msg-stream final result msg for IPC ctx?\n'
|
|
||||||
f'{msg}'
|
|
||||||
)
|
|
||||||
# XXX TODO, this **should be covered** by higher
|
|
||||||
# scoped runtime-side method calls such as
|
|
||||||
# `Context._deliver_msg()`, so you should never
|
|
||||||
# really see the warning above or else something
|
|
||||||
# racy/out-of-order is likely going on between
|
|
||||||
# actor-runtime-side push tasks and the user-app-side
|
|
||||||
# consume tasks!
|
|
||||||
# -[ ] figure out that set of race cases and fix!
|
|
||||||
# -[ ] possibly return the `msg` given an input
|
|
||||||
# arg-flag is set so we can process the `Return`
|
|
||||||
# from the `.aclose()` caller?
|
|
||||||
#
|
|
||||||
# breakpoint() # to debug this RACE CASE!
|
|
||||||
ctx._result = pld
|
|
||||||
ctx._outcome_msg = msg
|
|
||||||
|
|
||||||
return pld
|
|
||||||
|
|
||||||
async def receive(
|
async def receive(
|
||||||
self,
|
self,
|
||||||
|
|
||||||
hide_tb: bool = False,
|
hide_tb: bool = False,
|
||||||
):
|
):
|
||||||
'''
|
'''
|
||||||
|
|
@ -209,7 +154,7 @@ class MsgStream(trio.abc.Channel):
|
||||||
# except trio.EndOfChannel:
|
# except trio.EndOfChannel:
|
||||||
# raise StopAsyncIteration
|
# raise StopAsyncIteration
|
||||||
#
|
#
|
||||||
# see `.aclose()` for notes on the old behaviour prior to
|
# see ``.aclose()`` for notes on the old behaviour prior to
|
||||||
# introducing this
|
# introducing this
|
||||||
if self._eoc:
|
if self._eoc:
|
||||||
raise self._eoc
|
raise self._eoc
|
||||||
|
|
@ -220,11 +165,7 @@ class MsgStream(trio.abc.Channel):
|
||||||
src_err: Exception|None = None # orig tb
|
src_err: Exception|None = None # orig tb
|
||||||
try:
|
try:
|
||||||
ctx: Context = self._ctx
|
ctx: Context = self._ctx
|
||||||
pld = await ctx._pld_rx.recv_pld(
|
return await ctx._pld_rx.recv_pld(ipc=self)
|
||||||
ipc=self,
|
|
||||||
expect_msg=Yield,
|
|
||||||
)
|
|
||||||
return pld
|
|
||||||
|
|
||||||
# XXX: the stream terminates on either of:
|
# XXX: the stream terminates on either of:
|
||||||
# - `self._rx_chan.receive()` raising after manual closure
|
# - `self._rx_chan.receive()` raising after manual closure
|
||||||
|
|
@ -233,7 +174,7 @@ class MsgStream(trio.abc.Channel):
|
||||||
# - via a `Stop`-msg received from remote peer task.
|
# - via a `Stop`-msg received from remote peer task.
|
||||||
# NOTE
|
# NOTE
|
||||||
# |_ previously this was triggered by calling
|
# |_ previously this was triggered by calling
|
||||||
# `._rx_chan.aclose()` on the send side of the channel
|
# ``._rx_chan.aclose()`` on the send side of the channel
|
||||||
# inside `Actor._deliver_ctx_payload()`, but now the 'stop'
|
# inside `Actor._deliver_ctx_payload()`, but now the 'stop'
|
||||||
# message handling gets delegated to `PldRFx.recv_pld()`
|
# message handling gets delegated to `PldRFx.recv_pld()`
|
||||||
# internals.
|
# internals.
|
||||||
|
|
@ -257,14 +198,11 @@ class MsgStream(trio.abc.Channel):
|
||||||
# terminated and signal this local iterator to stop
|
# terminated and signal this local iterator to stop
|
||||||
drained: list[Exception|dict] = await self.aclose()
|
drained: list[Exception|dict] = await self.aclose()
|
||||||
if drained:
|
if drained:
|
||||||
# ^^^^^^^^TODO? pass these to the `._ctx._drained_msgs:
|
# ?TODO? pass these to the `._ctx._drained_msgs: deque`
|
||||||
# deque` and then iterate them as part of any
|
# and then iterate them as part of any `.wait_for_result()` call?
|
||||||
# `.wait_for_result()` call?
|
|
||||||
#
|
|
||||||
# -[ ] move the match-case processing from
|
|
||||||
# `.receive_nowait()` instead to right here, use it from
|
|
||||||
# a for msg in drained:` post-proc loop?
|
|
||||||
#
|
#
|
||||||
|
# from .devx import pause
|
||||||
|
# await pause()
|
||||||
log.warning(
|
log.warning(
|
||||||
'Drained context msgs during closure\n\n'
|
'Drained context msgs during closure\n\n'
|
||||||
f'{drained}'
|
f'{drained}'
|
||||||
|
|
@ -327,6 +265,9 @@ class MsgStream(trio.abc.Channel):
|
||||||
- more or less we try to maintain adherance to trio's `.aclose()` semantics:
|
- more or less we try to maintain adherance to trio's `.aclose()` semantics:
|
||||||
https://trio.readthedocs.io/en/stable/reference-io.html#trio.abc.AsyncResource.aclose
|
https://trio.readthedocs.io/en/stable/reference-io.html#trio.abc.AsyncResource.aclose
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
# rx_chan = self._rx_chan
|
||||||
|
|
||||||
# XXX NOTE XXX
|
# XXX NOTE XXX
|
||||||
# it's SUPER IMPORTANT that we ensure we don't DOUBLE
|
# it's SUPER IMPORTANT that we ensure we don't DOUBLE
|
||||||
# DRAIN msgs on closure so avoid getting stuck handing on
|
# DRAIN msgs on closure so avoid getting stuck handing on
|
||||||
|
|
@ -338,16 +279,15 @@ class MsgStream(trio.abc.Channel):
|
||||||
# this stream has already been closed so silently succeed as
|
# this stream has already been closed so silently succeed as
|
||||||
# per ``trio.AsyncResource`` semantics.
|
# per ``trio.AsyncResource`` semantics.
|
||||||
# https://trio.readthedocs.io/en/stable/reference-io.html#trio.abc.AsyncResource.aclose
|
# https://trio.readthedocs.io/en/stable/reference-io.html#trio.abc.AsyncResource.aclose
|
||||||
# import tractor
|
|
||||||
# await tractor.pause()
|
|
||||||
return []
|
return []
|
||||||
|
|
||||||
ctx: Context = self._ctx
|
ctx: Context = self._ctx
|
||||||
drained: list[Exception|dict] = []
|
drained: list[Exception|dict] = []
|
||||||
while not drained:
|
while not drained:
|
||||||
try:
|
try:
|
||||||
maybe_final_msg: Yield|Return = self.receive_nowait(
|
maybe_final_msg = self.receive_nowait(
|
||||||
expect_msg=Yield|Return,
|
# allow_msgs=[Yield, Return],
|
||||||
|
expect_msg=Yield,
|
||||||
)
|
)
|
||||||
if maybe_final_msg:
|
if maybe_final_msg:
|
||||||
log.debug(
|
log.debug(
|
||||||
|
|
@ -426,37 +366,24 @@ class MsgStream(trio.abc.Channel):
|
||||||
self._closed = re
|
self._closed = re
|
||||||
|
|
||||||
# if caught_eoc:
|
# if caught_eoc:
|
||||||
# # from .devx import debug
|
# # from .devx import _debug
|
||||||
# # await debug.pause()
|
# # await _debug.pause()
|
||||||
# with trio.CancelScope(shield=True):
|
# with trio.CancelScope(shield=True):
|
||||||
# await rx_chan.aclose()
|
# await rx_chan.aclose()
|
||||||
|
|
||||||
if not self._eoc:
|
if not self._eoc:
|
||||||
this_side: str = self._ctx.side
|
|
||||||
peer_side: str = self._ctx.peer_side
|
|
||||||
message: str = (
|
message: str = (
|
||||||
f'Stream self-closed by {this_side!r}-side before EoC from {peer_side!r}\n'
|
f'Stream self-closed by {self._ctx.side!r}-side before EoC\n'
|
||||||
# } bc a stream is a "scope"/msging-phase inside an IPC
|
# } bc a stream is a "scope"/msging-phase inside an IPC
|
||||||
f'c}}>\n'
|
f'x}}>\n'
|
||||||
f'|_{self}\n'
|
f'|_{self}\n'
|
||||||
)
|
)
|
||||||
if (
|
|
||||||
(rx_chan := self._rx_chan)
|
|
||||||
and
|
|
||||||
(stats := rx_chan.statistics()).tasks_waiting_receive
|
|
||||||
):
|
|
||||||
message += (
|
|
||||||
f'AND there is still reader tasks,\n'
|
|
||||||
f'\n'
|
|
||||||
f'{stats}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
log.cancel(message)
|
log.cancel(message)
|
||||||
self._eoc = trio.EndOfChannel(message)
|
self._eoc = trio.EndOfChannel(message)
|
||||||
|
|
||||||
# ?XXX WAIT, why do we not close the local mem chan `._rx_chan` XXX?
|
# ?XXX WAIT, why do we not close the local mem chan `._rx_chan` XXX?
|
||||||
# => NO, DEFINITELY NOT! <=
|
# => NO, DEFINITELY NOT! <=
|
||||||
# if we're a bi-dir `MsgStream` BECAUSE this same
|
# if we're a bi-dir ``MsgStream`` BECAUSE this same
|
||||||
# core-msg-loop mem recv-chan is used to deliver the
|
# core-msg-loop mem recv-chan is used to deliver the
|
||||||
# potential final result from the surrounding inter-actor
|
# potential final result from the surrounding inter-actor
|
||||||
# `Context` so we don't want to close it until that
|
# `Context` so we don't want to close it until that
|
||||||
|
|
@ -596,17 +523,8 @@ class MsgStream(trio.abc.Channel):
|
||||||
trio.ClosedResourceError,
|
trio.ClosedResourceError,
|
||||||
trio.BrokenResourceError,
|
trio.BrokenResourceError,
|
||||||
BrokenPipeError,
|
BrokenPipeError,
|
||||||
) as _trans_err:
|
) as trans_err:
|
||||||
trans_err = _trans_err
|
if hide_tb:
|
||||||
if (
|
|
||||||
hide_tb
|
|
||||||
and
|
|
||||||
self._ctx.chan._exc is trans_err
|
|
||||||
# ^XXX, IOW, only if the channel is marked errored
|
|
||||||
# for the same reason as whatever its underlying
|
|
||||||
# transport raised, do we keep the full low-level tb
|
|
||||||
# suppressed from the user.
|
|
||||||
):
|
|
||||||
raise type(trans_err)(
|
raise type(trans_err)(
|
||||||
*trans_err.args
|
*trans_err.args
|
||||||
) from trans_err
|
) from trans_err
|
||||||
|
|
@ -812,12 +730,13 @@ async def open_stream_from_ctx(
|
||||||
# sanity, can remove?
|
# sanity, can remove?
|
||||||
assert eoc is stream._eoc
|
assert eoc is stream._eoc
|
||||||
|
|
||||||
log.runtime(
|
log.warning(
|
||||||
'Stream was terminated by EoC\n\n'
|
'Stream was terminated by EoC\n\n'
|
||||||
# NOTE: won't show the error <Type> but
|
# NOTE: won't show the error <Type> but
|
||||||
# does show txt followed by IPC msg.
|
# does show txt followed by IPC msg.
|
||||||
f'{str(eoc)}\n'
|
f'{str(eoc)}\n'
|
||||||
)
|
)
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
if ctx._portal:
|
if ctx._portal:
|
||||||
try:
|
try:
|
||||||
|
|
|
||||||
|
|
@ -21,49 +21,34 @@
|
||||||
from contextlib import asynccontextmanager as acm
|
from contextlib import asynccontextmanager as acm
|
||||||
from functools import partial
|
from functools import partial
|
||||||
import inspect
|
import inspect
|
||||||
from typing import (
|
from pprint import pformat
|
||||||
TYPE_CHECKING,
|
from typing import TYPE_CHECKING
|
||||||
)
|
|
||||||
import typing
|
import typing
|
||||||
import warnings
|
import warnings
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
|
|
||||||
|
from .devx._debug import maybe_wait_for_debugger
|
||||||
from .devx import (
|
|
||||||
debug,
|
|
||||||
pformat as _pformat,
|
|
||||||
)
|
|
||||||
from ._addr import (
|
|
||||||
UnwrappedAddress,
|
|
||||||
mk_uuid,
|
|
||||||
)
|
|
||||||
from ._state import current_actor, is_main_process
|
from ._state import current_actor, is_main_process
|
||||||
from .log import get_logger, get_loglevel
|
from .log import get_logger, get_loglevel
|
||||||
from ._runtime import Actor
|
from ._runtime import Actor
|
||||||
from ._portal import Portal
|
from ._portal import Portal
|
||||||
from .trionics import (
|
|
||||||
is_multi_cancelled,
|
|
||||||
collapse_eg,
|
|
||||||
)
|
|
||||||
from ._exceptions import (
|
from ._exceptions import (
|
||||||
|
is_multi_cancelled,
|
||||||
ContextCancelled,
|
ContextCancelled,
|
||||||
)
|
)
|
||||||
from ._root import (
|
from ._root import open_root_actor
|
||||||
open_root_actor,
|
|
||||||
)
|
|
||||||
from . import _state
|
from . import _state
|
||||||
from . import _spawn
|
from . import _spawn
|
||||||
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
import multiprocessing as mp
|
import multiprocessing as mp
|
||||||
# from .ipc._server import IPCServer
|
|
||||||
from .ipc import IPCServer
|
|
||||||
|
|
||||||
|
|
||||||
log = get_logger(__name__)
|
log = get_logger(__name__)
|
||||||
|
|
||||||
|
_default_bind_addr: tuple[str, int] = ('127.0.0.1', 0)
|
||||||
|
|
||||||
|
|
||||||
class ActorNursery:
|
class ActorNursery:
|
||||||
'''
|
'''
|
||||||
|
|
@ -117,6 +102,7 @@ class ActorNursery:
|
||||||
]
|
]
|
||||||
] = {}
|
] = {}
|
||||||
|
|
||||||
|
self.cancelled: bool = False
|
||||||
self._join_procs = trio.Event()
|
self._join_procs = trio.Event()
|
||||||
self._at_least_one_child_in_debug: bool = False
|
self._at_least_one_child_in_debug: bool = False
|
||||||
self.errors = errors
|
self.errors = errors
|
||||||
|
|
@ -134,62 +120,18 @@ class ActorNursery:
|
||||||
# TODO: remove the `.run_in_actor()` API and thus this 2ndary
|
# TODO: remove the `.run_in_actor()` API and thus this 2ndary
|
||||||
# nursery when that API get's moved outside this primitive!
|
# nursery when that API get's moved outside this primitive!
|
||||||
self._ria_nursery = ria_nursery
|
self._ria_nursery = ria_nursery
|
||||||
|
|
||||||
# TODO, factor this into a .hilevel api!
|
|
||||||
#
|
|
||||||
# portals spawned with ``run_in_actor()`` are
|
# portals spawned with ``run_in_actor()`` are
|
||||||
# cancelled when their "main" result arrives
|
# cancelled when their "main" result arrives
|
||||||
self._cancel_after_result_on_exit: set = set()
|
self._cancel_after_result_on_exit: set = set()
|
||||||
|
|
||||||
# trio.Nursery-like cancel (request) statuses
|
|
||||||
self._cancelled_caught: bool = False
|
|
||||||
self._cancel_called: bool = False
|
|
||||||
|
|
||||||
@property
|
|
||||||
def cancel_called(self) -> bool:
|
|
||||||
'''
|
|
||||||
Records whether cancellation has been requested for this
|
|
||||||
actor-nursery by a call to `.cancel()` either due to,
|
|
||||||
- an explicit call by some actor-local-task,
|
|
||||||
- an implicit call due to an error/cancel emited inside
|
|
||||||
the `tractor.open_nursery()` block.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return self._cancel_called
|
|
||||||
|
|
||||||
@property
|
|
||||||
def cancelled_caught(self) -> bool:
|
|
||||||
'''
|
|
||||||
Set when this nursery was able to cance all spawned subactors
|
|
||||||
gracefully via an (implicit) call to `.cancel()`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return self._cancelled_caught
|
|
||||||
|
|
||||||
# TODO! remove internal/test-suite usage!
|
|
||||||
@property
|
|
||||||
def cancelled(self) -> bool:
|
|
||||||
warnings.warn(
|
|
||||||
"`ActorNursery.cancelled` is now deprecated, use "
|
|
||||||
" `.cancel_called` instead.",
|
|
||||||
DeprecationWarning,
|
|
||||||
stacklevel=2,
|
|
||||||
)
|
|
||||||
return (
|
|
||||||
self._cancel_called
|
|
||||||
# and
|
|
||||||
# self._cancelled_caught
|
|
||||||
)
|
|
||||||
|
|
||||||
async def start_actor(
|
async def start_actor(
|
||||||
self,
|
self,
|
||||||
name: str,
|
name: str,
|
||||||
|
|
||||||
*,
|
*,
|
||||||
|
|
||||||
bind_addrs: list[UnwrappedAddress]|None = None,
|
bind_addrs: list[tuple[str, int]] = [_default_bind_addr],
|
||||||
rpc_module_paths: list[str]|None = None,
|
rpc_module_paths: list[str]|None = None,
|
||||||
enable_transports: list[str] = [_state._def_tpt_proto],
|
|
||||||
enable_modules: list[str]|None = None,
|
enable_modules: list[str]|None = None,
|
||||||
loglevel: str|None = None, # set log level per subactor
|
loglevel: str|None = None, # set log level per subactor
|
||||||
debug_mode: bool|None = None,
|
debug_mode: bool|None = None,
|
||||||
|
|
@ -199,7 +141,6 @@ class ActorNursery:
|
||||||
# a `._ria_nursery` since the dependent APIs have been
|
# a `._ria_nursery` since the dependent APIs have been
|
||||||
# removed!
|
# removed!
|
||||||
nursery: trio.Nursery|None = None,
|
nursery: trio.Nursery|None = None,
|
||||||
proc_kwargs: dict[str, any] = {}
|
|
||||||
|
|
||||||
) -> Portal:
|
) -> Portal:
|
||||||
'''
|
'''
|
||||||
|
|
@ -236,17 +177,15 @@ class ActorNursery:
|
||||||
enable_modules.extend(rpc_module_paths)
|
enable_modules.extend(rpc_module_paths)
|
||||||
|
|
||||||
subactor = Actor(
|
subactor = Actor(
|
||||||
name=name,
|
name,
|
||||||
uuid=mk_uuid(),
|
|
||||||
|
|
||||||
# modules allowed to invoked funcs from
|
# modules allowed to invoked funcs from
|
||||||
enable_modules=enable_modules,
|
enable_modules=enable_modules,
|
||||||
loglevel=loglevel,
|
loglevel=loglevel,
|
||||||
|
|
||||||
# verbatim relay this actor's registrar addresses
|
# verbatim relay this actor's registrar addresses
|
||||||
registry_addrs=current_actor().registry_addrs,
|
registry_addrs=current_actor().reg_addrs,
|
||||||
)
|
)
|
||||||
parent_addr: UnwrappedAddress = self._actor.accept_addr
|
parent_addr = self._actor.accept_addr
|
||||||
assert parent_addr
|
assert parent_addr
|
||||||
|
|
||||||
# start a task to spawn a process
|
# start a task to spawn a process
|
||||||
|
|
@ -265,7 +204,6 @@ class ActorNursery:
|
||||||
parent_addr,
|
parent_addr,
|
||||||
_rtv, # run time vars
|
_rtv, # run time vars
|
||||||
infect_asyncio=infect_asyncio,
|
infect_asyncio=infect_asyncio,
|
||||||
proc_kwargs=proc_kwargs
|
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -284,12 +222,11 @@ class ActorNursery:
|
||||||
*,
|
*,
|
||||||
|
|
||||||
name: str | None = None,
|
name: str | None = None,
|
||||||
bind_addrs: UnwrappedAddress|None = None,
|
bind_addrs: tuple[str, int] = [_default_bind_addr],
|
||||||
rpc_module_paths: list[str] | None = None,
|
rpc_module_paths: list[str] | None = None,
|
||||||
enable_modules: list[str] | None = None,
|
enable_modules: list[str] | None = None,
|
||||||
loglevel: str | None = None, # set log level per subactor
|
loglevel: str | None = None, # set log level per subactor
|
||||||
infect_asyncio: bool = False,
|
infect_asyncio: bool = False,
|
||||||
proc_kwargs: dict[str, any] = {},
|
|
||||||
|
|
||||||
**kwargs, # explicit args to ``fn``
|
**kwargs, # explicit args to ``fn``
|
||||||
|
|
||||||
|
|
@ -320,7 +257,6 @@ class ActorNursery:
|
||||||
# use the run_in_actor nursery
|
# use the run_in_actor nursery
|
||||||
nursery=self._ria_nursery,
|
nursery=self._ria_nursery,
|
||||||
infect_asyncio=infect_asyncio,
|
infect_asyncio=infect_asyncio,
|
||||||
proc_kwargs=proc_kwargs
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# XXX: don't allow stream funcs
|
# XXX: don't allow stream funcs
|
||||||
|
|
@ -358,21 +294,15 @@ class ActorNursery:
|
||||||
|
|
||||||
'''
|
'''
|
||||||
__runtimeframe__: int = 1 # noqa
|
__runtimeframe__: int = 1 # noqa
|
||||||
self._cancel_called = True
|
self.cancelled = True
|
||||||
|
|
||||||
# TODO: impl a repr for spawn more compact
|
# TODO: impl a repr for spawn more compact
|
||||||
# then `._children`..
|
# then `._children`..
|
||||||
children: dict = self._children
|
children: dict = self._children
|
||||||
child_count: int = len(children)
|
child_count: int = len(children)
|
||||||
msg: str = f'Cancelling actor nursery with {child_count} children\n'
|
msg: str = f'Cancelling actor nursery with {child_count} children\n'
|
||||||
|
|
||||||
server: IPCServer = self._actor.ipc_server
|
|
||||||
|
|
||||||
with trio.move_on_after(3) as cs:
|
with trio.move_on_after(3) as cs:
|
||||||
async with (
|
async with trio.open_nursery() as tn:
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as tn,
|
|
||||||
):
|
|
||||||
|
|
||||||
subactor: Actor
|
subactor: Actor
|
||||||
proc: trio.Process
|
proc: trio.Process
|
||||||
|
|
@ -391,7 +321,7 @@ class ActorNursery:
|
||||||
|
|
||||||
else:
|
else:
|
||||||
if portal is None: # actor hasn't fully spawned yet
|
if portal is None: # actor hasn't fully spawned yet
|
||||||
event: trio.Event = server._peer_connected[subactor.uid]
|
event = self._actor._peer_connected[subactor.uid]
|
||||||
log.warning(
|
log.warning(
|
||||||
f"{subactor.uid} never 't finished spawning?"
|
f"{subactor.uid} never 't finished spawning?"
|
||||||
)
|
)
|
||||||
|
|
@ -407,7 +337,7 @@ class ActorNursery:
|
||||||
if portal is None:
|
if portal is None:
|
||||||
# cancelled while waiting on the event
|
# cancelled while waiting on the event
|
||||||
# to arrive
|
# to arrive
|
||||||
chan = server._peers[subactor.uid][-1]
|
chan = self._actor._peers[subactor.uid][-1]
|
||||||
if chan:
|
if chan:
|
||||||
portal = Portal(chan)
|
portal = Portal(chan)
|
||||||
else: # there's no other choice left
|
else: # there's no other choice left
|
||||||
|
|
@ -436,8 +366,6 @@ class ActorNursery:
|
||||||
) in children.values():
|
) in children.values():
|
||||||
log.warning(f"Hard killing process {proc}")
|
log.warning(f"Hard killing process {proc}")
|
||||||
proc.terminate()
|
proc.terminate()
|
||||||
else:
|
|
||||||
self._cancelled_caught
|
|
||||||
|
|
||||||
# mark ourselves as having (tried to have) cancelled all subactors
|
# mark ourselves as having (tried to have) cancelled all subactors
|
||||||
self._join_procs.set()
|
self._join_procs.set()
|
||||||
|
|
@ -446,12 +374,12 @@ class ActorNursery:
|
||||||
@acm
|
@acm
|
||||||
async def _open_and_supervise_one_cancels_all_nursery(
|
async def _open_and_supervise_one_cancels_all_nursery(
|
||||||
actor: Actor,
|
actor: Actor,
|
||||||
hide_tb: bool = True,
|
tb_hide: bool = False,
|
||||||
|
|
||||||
) -> typing.AsyncGenerator[ActorNursery, None]:
|
) -> typing.AsyncGenerator[ActorNursery, None]:
|
||||||
|
|
||||||
# normally don't need to show user by default
|
# normally don't need to show user by default
|
||||||
__tracebackhide__: bool = hide_tb
|
__tracebackhide__: bool = tb_hide
|
||||||
|
|
||||||
outer_err: BaseException|None = None
|
outer_err: BaseException|None = None
|
||||||
inner_err: BaseException|None = None
|
inner_err: BaseException|None = None
|
||||||
|
|
@ -467,23 +395,18 @@ async def _open_and_supervise_one_cancels_all_nursery(
|
||||||
# `ActorNursery.start_actor()`).
|
# `ActorNursery.start_actor()`).
|
||||||
|
|
||||||
# errors from this daemon actor nursery bubble up to caller
|
# errors from this daemon actor nursery bubble up to caller
|
||||||
async with (
|
async with trio.open_nursery() as da_nursery:
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as da_nursery,
|
|
||||||
):
|
|
||||||
try:
|
try:
|
||||||
# This is the inner level "run in actor" nursery. It is
|
# This is the inner level "run in actor" nursery. It is
|
||||||
# awaited first since actors spawned in this way (using
|
# awaited first since actors spawned in this way (using
|
||||||
# `ActorNusery.run_in_actor()`) are expected to only
|
# ``ActorNusery.run_in_actor()``) are expected to only
|
||||||
# return a single result and then complete (i.e. be canclled
|
# return a single result and then complete (i.e. be canclled
|
||||||
# gracefully). Errors collected from these actors are
|
# gracefully). Errors collected from these actors are
|
||||||
# immediately raised for handling by a supervisor strategy.
|
# immediately raised for handling by a supervisor strategy.
|
||||||
# As such if the strategy propagates any error(s) upwards
|
# As such if the strategy propagates any error(s) upwards
|
||||||
# the above "daemon actor" nursery will be notified.
|
# the above "daemon actor" nursery will be notified.
|
||||||
async with (
|
async with trio.open_nursery() as ria_nursery:
|
||||||
collapse_eg(),
|
|
||||||
trio.open_nursery() as ria_nursery,
|
|
||||||
):
|
|
||||||
an = ActorNursery(
|
an = ActorNursery(
|
||||||
actor,
|
actor,
|
||||||
ria_nursery,
|
ria_nursery,
|
||||||
|
|
@ -500,7 +423,7 @@ async def _open_and_supervise_one_cancels_all_nursery(
|
||||||
# the "hard join phase".
|
# the "hard join phase".
|
||||||
log.runtime(
|
log.runtime(
|
||||||
'Waiting on subactors to complete:\n'
|
'Waiting on subactors to complete:\n'
|
||||||
f'>}} {len(an._children)}\n'
|
f'{pformat(an._children)}\n'
|
||||||
)
|
)
|
||||||
an._join_procs.set()
|
an._join_procs.set()
|
||||||
|
|
||||||
|
|
@ -514,7 +437,7 @@ async def _open_and_supervise_one_cancels_all_nursery(
|
||||||
# will make the pdb repl unusable.
|
# will make the pdb repl unusable.
|
||||||
# Instead try to wait for pdb to be released before
|
# Instead try to wait for pdb to be released before
|
||||||
# tearing down.
|
# tearing down.
|
||||||
await debug.maybe_wait_for_debugger(
|
await maybe_wait_for_debugger(
|
||||||
child_in_debug=an._at_least_one_child_in_debug
|
child_in_debug=an._at_least_one_child_in_debug
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -549,8 +472,8 @@ async def _open_and_supervise_one_cancels_all_nursery(
|
||||||
ContextCancelled,
|
ContextCancelled,
|
||||||
}:
|
}:
|
||||||
log.cancel(
|
log.cancel(
|
||||||
'Actor-nursery caught remote cancellation\n'
|
'Actor-nursery caught remote cancellation\n\n'
|
||||||
'\n'
|
|
||||||
f'{inner_err.tb_str}'
|
f'{inner_err.tb_str}'
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
|
|
@ -590,7 +513,7 @@ async def _open_and_supervise_one_cancels_all_nursery(
|
||||||
|
|
||||||
# XXX: yet another guard before allowing the cancel
|
# XXX: yet another guard before allowing the cancel
|
||||||
# sequence in case a (single) child is in debug.
|
# sequence in case a (single) child is in debug.
|
||||||
await debug.maybe_wait_for_debugger(
|
await maybe_wait_for_debugger(
|
||||||
child_in_debug=an._at_least_one_child_in_debug
|
child_in_debug=an._at_least_one_child_in_debug
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -639,18 +562,10 @@ async def _open_and_supervise_one_cancels_all_nursery(
|
||||||
# final exit
|
# final exit
|
||||||
|
|
||||||
|
|
||||||
_shutdown_msg: str = (
|
|
||||||
'Actor-runtime-shutdown'
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
@acm
|
||||||
# @api_frame
|
# @api_frame
|
||||||
async def open_nursery(
|
async def open_nursery(
|
||||||
*, # named params only!
|
|
||||||
hide_tb: bool = True,
|
|
||||||
**kwargs,
|
**kwargs,
|
||||||
# ^TODO, paramspec for `open_root_actor()`
|
|
||||||
|
|
||||||
) -> typing.AsyncGenerator[ActorNursery, None]:
|
) -> typing.AsyncGenerator[ActorNursery, None]:
|
||||||
'''
|
'''
|
||||||
|
|
@ -668,7 +583,7 @@ async def open_nursery(
|
||||||
which cancellation scopes correspond to each spawned subactor set.
|
which cancellation scopes correspond to each spawned subactor set.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
__tracebackhide__: bool = hide_tb
|
__tracebackhide__: bool = True
|
||||||
implicit_runtime: bool = False
|
implicit_runtime: bool = False
|
||||||
actor: Actor = current_actor(err_on_no_runtime=False)
|
actor: Actor = current_actor(err_on_no_runtime=False)
|
||||||
an: ActorNursery|None = None
|
an: ActorNursery|None = None
|
||||||
|
|
@ -684,10 +599,7 @@ async def open_nursery(
|
||||||
# mark us for teardown on exit
|
# mark us for teardown on exit
|
||||||
implicit_runtime: bool = True
|
implicit_runtime: bool = True
|
||||||
|
|
||||||
async with open_root_actor(
|
async with open_root_actor(**kwargs) as actor:
|
||||||
hide_tb=hide_tb,
|
|
||||||
**kwargs,
|
|
||||||
) as actor:
|
|
||||||
assert actor is current_actor()
|
assert actor is current_actor()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
|
@ -725,33 +637,22 @@ async def open_nursery(
|
||||||
# show frame on any internal runtime-scope error
|
# show frame on any internal runtime-scope error
|
||||||
if (
|
if (
|
||||||
an
|
an
|
||||||
and
|
and not an.cancelled
|
||||||
not an.cancelled
|
and an._scope_error
|
||||||
and
|
|
||||||
an._scope_error
|
|
||||||
):
|
):
|
||||||
__tracebackhide__: bool = False
|
__tracebackhide__: bool = False
|
||||||
|
|
||||||
|
msg: str = (
|
||||||
op_nested_an_repr: str = _pformat.nest_from_op(
|
'Actor-nursery exited\n'
|
||||||
input_op=')>',
|
f'|_{an}\n'
|
||||||
text=f'{an}',
|
|
||||||
# nest_prefix='|_',
|
|
||||||
nest_indent=1, # under >
|
|
||||||
)
|
)
|
||||||
an_msg: str = (
|
|
||||||
f'Actor-nursery exited\n'
|
|
||||||
f'{op_nested_an_repr}\n'
|
|
||||||
)
|
|
||||||
# keep noise low during std operation.
|
|
||||||
log.runtime(an_msg)
|
|
||||||
|
|
||||||
if implicit_runtime:
|
if implicit_runtime:
|
||||||
# shutdown runtime if it was started and report noisly
|
# shutdown runtime if it was started and report noisly
|
||||||
# that we're did so.
|
# that we're did so.
|
||||||
msg: str = (
|
msg += '=> Shutting down actor runtime <=\n'
|
||||||
'\n'
|
|
||||||
'\n'
|
|
||||||
f'{_shutdown_msg} )>\n'
|
|
||||||
)
|
|
||||||
log.info(msg)
|
log.info(msg)
|
||||||
|
|
||||||
|
else:
|
||||||
|
# keep noise low during std operation.
|
||||||
|
log.runtime(msg)
|
||||||
|
|
|
||||||
|
|
@ -19,16 +19,10 @@ Various helpers/utils for auditing your `tractor` app and/or the
|
||||||
core runtime.
|
core runtime.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from contextlib import (
|
from contextlib import asynccontextmanager as acm
|
||||||
asynccontextmanager as acm,
|
|
||||||
)
|
|
||||||
import os
|
|
||||||
import pathlib
|
import pathlib
|
||||||
|
|
||||||
import tractor
|
import tractor
|
||||||
from tractor.devx.debug import (
|
|
||||||
BoxedMaybeException,
|
|
||||||
)
|
|
||||||
from .pytest import (
|
from .pytest import (
|
||||||
tractor_test as tractor_test
|
tractor_test as tractor_test
|
||||||
)
|
)
|
||||||
|
|
@ -37,9 +31,6 @@ from .fault_simulation import (
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# TODO, use dulwhich for this instead?
|
|
||||||
# -> we're going to likely need it (or something similar)
|
|
||||||
# for supporting hot-coad reload feats eventually anyway!
|
|
||||||
def repodir() -> pathlib.Path:
|
def repodir() -> pathlib.Path:
|
||||||
'''
|
'''
|
||||||
Return the abspath to the repo directory.
|
Return the abspath to the repo directory.
|
||||||
|
|
@ -68,12 +59,7 @@ def mk_cmd(
|
||||||
exs_subpath: str = 'debugging',
|
exs_subpath: str = 'debugging',
|
||||||
) -> str:
|
) -> str:
|
||||||
'''
|
'''
|
||||||
Generate a shell command suitable to pass to `pexpect.spawn()`
|
Generate a shell command suitable to pass to ``pexpect.spawn()``.
|
||||||
which runs the script as a python program's entrypoint.
|
|
||||||
|
|
||||||
In particular ensure we disable the new tb coloring via unsetting
|
|
||||||
`$PYTHON_COLORS` so that `pexpect` can pattern match without
|
|
||||||
color-escape-codes.
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
script_path: pathlib.Path = (
|
script_path: pathlib.Path = (
|
||||||
|
|
@ -81,15 +67,10 @@ def mk_cmd(
|
||||||
/ exs_subpath
|
/ exs_subpath
|
||||||
/ f'{ex_name}.py'
|
/ f'{ex_name}.py'
|
||||||
)
|
)
|
||||||
py_cmd: str = ' '.join([
|
return ' '.join([
|
||||||
'python',
|
'python',
|
||||||
str(script_path)
|
str(script_path)
|
||||||
])
|
])
|
||||||
# XXX, required for py 3.13+
|
|
||||||
# https://docs.python.org/3/using/cmdline.html#using-on-controlling-color
|
|
||||||
# https://docs.python.org/3/using/cmdline.html#envvar-PYTHON_COLORS
|
|
||||||
os.environ['PYTHON_COLORS'] = '0'
|
|
||||||
return py_cmd
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
@acm
|
||||||
|
|
@ -104,13 +85,12 @@ async def expect_ctxc(
|
||||||
'''
|
'''
|
||||||
if yay:
|
if yay:
|
||||||
try:
|
try:
|
||||||
yield (maybe_exc := BoxedMaybeException())
|
yield
|
||||||
raise RuntimeError('Never raised ctxc?')
|
raise RuntimeError('Never raised ctxc?')
|
||||||
except tractor.ContextCancelled as ctxc:
|
except tractor.ContextCancelled:
|
||||||
maybe_exc.value = ctxc
|
|
||||||
if reraise:
|
if reraise:
|
||||||
raise
|
raise
|
||||||
else:
|
else:
|
||||||
return
|
return
|
||||||
else:
|
else:
|
||||||
yield (maybe_exc := BoxedMaybeException())
|
yield
|
||||||
|
|
|
||||||
|
|
@ -1,70 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
|
||||||
# it under the terms of the GNU Affero General Public License as published by
|
|
||||||
# the Free Software Foundation, either version 3 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful,
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
# GNU Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public License
|
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
'''
|
|
||||||
Random IPC addr generation for isolating
|
|
||||||
the discovery space between test sessions.
|
|
||||||
|
|
||||||
Might be eventually useful to expose as a util set from
|
|
||||||
our `tractor.discovery` subsys?
|
|
||||||
|
|
||||||
'''
|
|
||||||
import random
|
|
||||||
from typing import (
|
|
||||||
Type,
|
|
||||||
)
|
|
||||||
from tractor import (
|
|
||||||
_addr,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def get_rando_addr(
|
|
||||||
tpt_proto: str,
|
|
||||||
*,
|
|
||||||
|
|
||||||
# choose random port at import time
|
|
||||||
_rando_port: str = random.randint(1000, 9999)
|
|
||||||
|
|
||||||
) -> tuple[str, str|int]:
|
|
||||||
'''
|
|
||||||
Used to globally override the runtime to the
|
|
||||||
per-test-session-dynamic addr so that all tests never conflict
|
|
||||||
with any other actor tree using the default.
|
|
||||||
|
|
||||||
'''
|
|
||||||
addr_type: Type[_addr.Addres] = _addr._address_types[tpt_proto]
|
|
||||||
def_reg_addr: tuple[str, int] = _addr._default_lo_addrs[tpt_proto]
|
|
||||||
|
|
||||||
# this is the "unwrapped" form expected to be passed to
|
|
||||||
# `.open_root_actor()` by test body.
|
|
||||||
testrun_reg_addr: tuple[str, int|str]
|
|
||||||
match tpt_proto:
|
|
||||||
case 'tcp':
|
|
||||||
testrun_reg_addr = (
|
|
||||||
addr_type.def_bindspace,
|
|
||||||
_rando_port,
|
|
||||||
)
|
|
||||||
|
|
||||||
# NOTE, file-name uniqueness (no-collisions) will be based on
|
|
||||||
# the runtime-directory and root (pytest-proc's) pid.
|
|
||||||
case 'uds':
|
|
||||||
testrun_reg_addr = addr_type.get_random().unwrap()
|
|
||||||
|
|
||||||
# XXX, as sanity it should never the same as the default for the
|
|
||||||
# host-singleton registry actor.
|
|
||||||
assert def_reg_addr != testrun_reg_addr
|
|
||||||
|
|
||||||
return testrun_reg_addr
|
|
||||||
|
|
@ -26,46 +26,29 @@ from functools import (
|
||||||
import inspect
|
import inspect
|
||||||
import platform
|
import platform
|
||||||
|
|
||||||
import pytest
|
|
||||||
import tractor
|
import tractor
|
||||||
import trio
|
import trio
|
||||||
|
|
||||||
|
|
||||||
def tractor_test(fn):
|
def tractor_test(fn):
|
||||||
'''
|
'''
|
||||||
Decorator for async test fns to decorator-wrap them as "native"
|
Decorator for async test funcs to present them as "native"
|
||||||
looking sync funcs runnable by `pytest` and auto invoked with
|
looking sync funcs runnable by `pytest` using `trio.run()`.
|
||||||
`trio.run()` (much like the `pytest-trio` plugin's approach).
|
|
||||||
|
|
||||||
Further the test fn body will be invoked AFTER booting the actor
|
Use:
|
||||||
runtime, i.e. from inside a `tractor.open_root_actor()` block AND
|
|
||||||
with various runtime and tooling parameters implicitly passed as
|
|
||||||
requested by by the test session's config; see immediately below.
|
|
||||||
|
|
||||||
Basic deco use:
|
|
||||||
---------------
|
|
||||||
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_whatever():
|
async def test_whatever():
|
||||||
await ...
|
await ...
|
||||||
|
|
||||||
|
If fixtures:
|
||||||
|
|
||||||
Runtime config via special fixtures:
|
- ``reg_addr`` (a socket addr tuple where arbiter is listening)
|
||||||
------------------------------------
|
- ``loglevel`` (logging level passed to tractor internals)
|
||||||
If any of the following fixture are requested by the wrapped test
|
- ``start_method`` (subprocess spawning backend)
|
||||||
fn (via normal func-args declaration),
|
|
||||||
|
|
||||||
- `reg_addr` (a socket addr tuple where arbiter is listening)
|
|
||||||
- `loglevel` (logging level passed to tractor internals)
|
|
||||||
- `start_method` (subprocess spawning backend)
|
|
||||||
|
|
||||||
(TODO support)
|
|
||||||
- `tpt_proto` (IPC transport protocol key)
|
|
||||||
|
|
||||||
they will be automatically injected to each test as normally
|
|
||||||
expected as well as passed to the initial
|
|
||||||
`tractor.open_root_actor()` funcargs.
|
|
||||||
|
|
||||||
|
are defined in the `pytest` fixture space they will be automatically
|
||||||
|
injected to tests declaring these funcargs.
|
||||||
'''
|
'''
|
||||||
@wraps(fn)
|
@wraps(fn)
|
||||||
def wrapper(
|
def wrapper(
|
||||||
|
|
@ -128,164 +111,3 @@ def tractor_test(fn):
|
||||||
return trio.run(main)
|
return trio.run(main)
|
||||||
|
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
def pytest_addoption(
|
|
||||||
parser: pytest.Parser,
|
|
||||||
):
|
|
||||||
# parser.addoption(
|
|
||||||
# "--ll",
|
|
||||||
# action="store",
|
|
||||||
# dest='loglevel',
|
|
||||||
# default='ERROR', help="logging level to set when testing"
|
|
||||||
# )
|
|
||||||
|
|
||||||
parser.addoption(
|
|
||||||
"--spawn-backend",
|
|
||||||
action="store",
|
|
||||||
dest='spawn_backend',
|
|
||||||
default='trio',
|
|
||||||
help="Processing spawning backend to use for test run",
|
|
||||||
)
|
|
||||||
|
|
||||||
parser.addoption(
|
|
||||||
"--tpdb",
|
|
||||||
"--debug-mode",
|
|
||||||
action="store_true",
|
|
||||||
dest='tractor_debug_mode',
|
|
||||||
# default=False,
|
|
||||||
help=(
|
|
||||||
'Enable a flag that can be used by tests to to set the '
|
|
||||||
'`debug_mode: bool` for engaging the internal '
|
|
||||||
'multi-proc debugger sys.'
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
# provide which IPC transport protocols opting-in test suites
|
|
||||||
# should accumulatively run against.
|
|
||||||
parser.addoption(
|
|
||||||
"--tpt-proto",
|
|
||||||
nargs='+', # accumulate-multiple-args
|
|
||||||
action="store",
|
|
||||||
dest='tpt_protos',
|
|
||||||
default=['tcp'],
|
|
||||||
help="Transport protocol to use under the `tractor.ipc.Channel`",
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def pytest_configure(config):
|
|
||||||
backend = config.option.spawn_backend
|
|
||||||
tractor._spawn.try_set_start_method(backend)
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='session')
|
|
||||||
def debug_mode(request) -> bool:
|
|
||||||
'''
|
|
||||||
Flag state for whether `--tpdb` (for `tractor`-py-debugger)
|
|
||||||
was passed to the test run.
|
|
||||||
|
|
||||||
Normally tests should pass this directly to `.open_root_actor()`
|
|
||||||
to allow the user to opt into suite-wide crash handling.
|
|
||||||
|
|
||||||
'''
|
|
||||||
debug_mode: bool = request.config.option.tractor_debug_mode
|
|
||||||
return debug_mode
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='session')
|
|
||||||
def spawn_backend(request) -> str:
|
|
||||||
return request.config.option.spawn_backend
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='session')
|
|
||||||
def tpt_protos(request) -> list[str]:
|
|
||||||
|
|
||||||
# allow quoting on CLI
|
|
||||||
proto_keys: list[str] = [
|
|
||||||
proto_key.replace('"', '').replace("'", "")
|
|
||||||
for proto_key in request.config.option.tpt_protos
|
|
||||||
]
|
|
||||||
|
|
||||||
# ?TODO, eventually support multiple protos per test-sesh?
|
|
||||||
if len(proto_keys) > 1:
|
|
||||||
pytest.fail(
|
|
||||||
'We only support one `--tpt-proto <key>` atm!\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# XXX ensure we support the protocol by name via lookup!
|
|
||||||
for proto_key in proto_keys:
|
|
||||||
addr_type = tractor._addr._address_types[proto_key]
|
|
||||||
assert addr_type.proto_key == proto_key
|
|
||||||
|
|
||||||
yield proto_keys
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(
|
|
||||||
scope='session',
|
|
||||||
autouse=True,
|
|
||||||
)
|
|
||||||
def tpt_proto(
|
|
||||||
tpt_protos: list[str],
|
|
||||||
) -> str:
|
|
||||||
proto_key: str = tpt_protos[0]
|
|
||||||
|
|
||||||
from tractor import _state
|
|
||||||
if _state._def_tpt_proto != proto_key:
|
|
||||||
_state._def_tpt_proto = proto_key
|
|
||||||
|
|
||||||
yield proto_key
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='session')
|
|
||||||
def reg_addr(
|
|
||||||
tpt_proto: str,
|
|
||||||
) -> tuple[str, int|str]:
|
|
||||||
'''
|
|
||||||
Deliver a test-sesh unique registry address such
|
|
||||||
that each run's (tests which use this fixture) will
|
|
||||||
have no conflicts/cross-talk when running simultaneously
|
|
||||||
nor will interfere with other live `tractor` apps active
|
|
||||||
on the same network-host (namespace).
|
|
||||||
|
|
||||||
'''
|
|
||||||
from tractor._testing.addr import get_rando_addr
|
|
||||||
return get_rando_addr(
|
|
||||||
tpt_proto=tpt_proto,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def pytest_generate_tests(
|
|
||||||
metafunc: pytest.Metafunc,
|
|
||||||
):
|
|
||||||
spawn_backend: str = metafunc.config.option.spawn_backend
|
|
||||||
|
|
||||||
if not spawn_backend:
|
|
||||||
# XXX some weird windows bug with `pytest`?
|
|
||||||
spawn_backend = 'trio'
|
|
||||||
|
|
||||||
# TODO: maybe just use the literal `._spawn.SpawnMethodKey`?
|
|
||||||
assert spawn_backend in (
|
|
||||||
'mp_spawn',
|
|
||||||
'mp_forkserver',
|
|
||||||
'trio',
|
|
||||||
)
|
|
||||||
|
|
||||||
# NOTE: used-to-be-used-to dyanmically parametrize tests for when
|
|
||||||
# you just passed --spawn-backend=`mp` on the cli, but now we expect
|
|
||||||
# that cli input to be manually specified, BUT, maybe we'll do
|
|
||||||
# something like this again in the future?
|
|
||||||
if 'start_method' in metafunc.fixturenames:
|
|
||||||
metafunc.parametrize(
|
|
||||||
"start_method",
|
|
||||||
[spawn_backend],
|
|
||||||
scope='module',
|
|
||||||
)
|
|
||||||
|
|
||||||
# TODO, parametrize any `tpt_proto: str` declaring tests!
|
|
||||||
# proto_tpts: list[str] = metafunc.config.option.proto_tpts
|
|
||||||
# if 'tpt_proto' in metafunc.fixturenames:
|
|
||||||
# metafunc.parametrize(
|
|
||||||
# 'tpt_proto',
|
|
||||||
# proto_tpts, # TODO, double check this list usage!
|
|
||||||
# scope='module',
|
|
||||||
# )
|
|
||||||
|
|
|
||||||
|
|
@ -1,35 +0,0 @@
|
||||||
import os
|
|
||||||
import random
|
|
||||||
|
|
||||||
|
|
||||||
def generate_sample_messages(
|
|
||||||
amount: int,
|
|
||||||
rand_min: int = 0,
|
|
||||||
rand_max: int = 0,
|
|
||||||
silent: bool = False
|
|
||||||
) -> tuple[list[bytes], int]:
|
|
||||||
|
|
||||||
msgs = []
|
|
||||||
size = 0
|
|
||||||
|
|
||||||
if not silent:
|
|
||||||
print(f'\ngenerating {amount} messages...')
|
|
||||||
|
|
||||||
for i in range(amount):
|
|
||||||
msg = f'[{i:08}]'.encode('utf-8')
|
|
||||||
|
|
||||||
if rand_max > 0:
|
|
||||||
msg += os.urandom(
|
|
||||||
random.randint(rand_min, rand_max))
|
|
||||||
|
|
||||||
size += len(msg)
|
|
||||||
|
|
||||||
msgs.append(msg)
|
|
||||||
|
|
||||||
if not silent and i and i % 10_000 == 0:
|
|
||||||
print(f'{i} generated')
|
|
||||||
|
|
||||||
if not silent:
|
|
||||||
print(f'done, {size:,} bytes in total')
|
|
||||||
|
|
||||||
return msgs, size
|
|
||||||
|
|
@ -20,7 +20,7 @@ Runtime "developer experience" utils and addons to aid our
|
||||||
and working with/on the actor runtime.
|
and working with/on the actor runtime.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from .debug import (
|
from ._debug import (
|
||||||
maybe_wait_for_debugger as maybe_wait_for_debugger,
|
maybe_wait_for_debugger as maybe_wait_for_debugger,
|
||||||
acquire_debug_lock as acquire_debug_lock,
|
acquire_debug_lock as acquire_debug_lock,
|
||||||
breakpoint as breakpoint,
|
breakpoint as breakpoint,
|
||||||
|
|
@ -41,38 +41,3 @@ from .pformat import (
|
||||||
pformat_caller_frame as pformat_caller_frame,
|
pformat_caller_frame as pformat_caller_frame,
|
||||||
pformat_boxed_tb as pformat_boxed_tb,
|
pformat_boxed_tb as pformat_boxed_tb,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# TODO, move this to a new `.devx._pdbp` mod?
|
|
||||||
def _enable_readline_feats() -> str:
|
|
||||||
'''
|
|
||||||
Handle `readline` when compiled with `libedit` to avoid breaking
|
|
||||||
tab completion in `pdbp` (and its dep `tabcompleter`)
|
|
||||||
particularly since `uv` cpython distis are compiled this way..
|
|
||||||
|
|
||||||
See docs for deats,
|
|
||||||
https://docs.python.org/3/library/readline.html#module-readline
|
|
||||||
|
|
||||||
Originally discovered soln via SO answer,
|
|
||||||
https://stackoverflow.com/q/49287102
|
|
||||||
|
|
||||||
'''
|
|
||||||
import readline
|
|
||||||
if (
|
|
||||||
# 3.13+ attr
|
|
||||||
# https://docs.python.org/3/library/readline.html#readline.backend
|
|
||||||
(getattr(readline, 'backend', False) == 'libedit')
|
|
||||||
or
|
|
||||||
'libedit' in readline.__doc__
|
|
||||||
):
|
|
||||||
readline.parse_and_bind("python:bind -v")
|
|
||||||
readline.parse_and_bind("python:bind ^I rl_complete")
|
|
||||||
return 'libedit'
|
|
||||||
else:
|
|
||||||
readline.parse_and_bind("tab: complete")
|
|
||||||
readline.parse_and_bind("set editing-mode vi")
|
|
||||||
readline.parse_and_bind("set keymap vi")
|
|
||||||
return 'readline'
|
|
||||||
|
|
||||||
|
|
||||||
_enable_readline_feats()
|
|
||||||
|
|
|
||||||
File diff suppressed because it is too large
Load Diff
|
|
@ -20,18 +20,13 @@ as it pertains to improving the grok-ability of our runtime!
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
from contextlib import (
|
|
||||||
_GeneratorContextManager,
|
|
||||||
_AsyncGeneratorContextManager,
|
|
||||||
)
|
|
||||||
from functools import partial
|
from functools import partial
|
||||||
import inspect
|
import inspect
|
||||||
import textwrap
|
|
||||||
from types import (
|
from types import (
|
||||||
FrameType,
|
FrameType,
|
||||||
FunctionType,
|
FunctionType,
|
||||||
MethodType,
|
MethodType,
|
||||||
CodeType,
|
# CodeType,
|
||||||
)
|
)
|
||||||
from typing import (
|
from typing import (
|
||||||
Any,
|
Any,
|
||||||
|
|
@ -39,9 +34,6 @@ from typing import (
|
||||||
Type,
|
Type,
|
||||||
)
|
)
|
||||||
|
|
||||||
import pdbp
|
|
||||||
from tractor.log import get_logger
|
|
||||||
import trio
|
|
||||||
from tractor.msg import (
|
from tractor.msg import (
|
||||||
pretty_struct,
|
pretty_struct,
|
||||||
NamespacePath,
|
NamespacePath,
|
||||||
|
|
@ -49,8 +41,6 @@ from tractor.msg import (
|
||||||
import wrapt
|
import wrapt
|
||||||
|
|
||||||
|
|
||||||
log = get_logger(__name__)
|
|
||||||
|
|
||||||
# TODO: yeah, i don't love this and we should prolly just
|
# TODO: yeah, i don't love this and we should prolly just
|
||||||
# write a decorator that actually keeps a stupid ref to the func
|
# write a decorator that actually keeps a stupid ref to the func
|
||||||
# obj..
|
# obj..
|
||||||
|
|
@ -311,70 +301,3 @@ def api_frame(
|
||||||
# error_set: set[BaseException],
|
# error_set: set[BaseException],
|
||||||
# ) -> TracebackType:
|
# ) -> TracebackType:
|
||||||
# ...
|
# ...
|
||||||
|
|
||||||
|
|
||||||
def hide_runtime_frames() -> dict[FunctionType, CodeType]:
|
|
||||||
'''
|
|
||||||
Hide call-stack frames for various std-lib and `trio`-API primitives
|
|
||||||
such that the tracebacks presented from our runtime are as minimized
|
|
||||||
as possible, particularly from inside a `PdbREPL`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
# XXX HACKZONE XXX
|
|
||||||
# hide exit stack frames on nurseries and cancel-scopes!
|
|
||||||
# |_ so avoid seeing it when the `pdbp` REPL is first engaged from
|
|
||||||
# inside a `trio.open_nursery()` scope (with no line after it
|
|
||||||
# in before the block end??).
|
|
||||||
#
|
|
||||||
# TODO: FINALLY got this workin originally with
|
|
||||||
# `@pdbp.hideframe` around the `wrapper()` def embedded inside
|
|
||||||
# `_ki_protection_decoratior()`.. which is in the module:
|
|
||||||
# /home/goodboy/.virtualenvs/tractor311/lib/python3.11/site-packages/trio/_core/_ki.py
|
|
||||||
#
|
|
||||||
# -[ ] make an issue and patch for `trio` core? maybe linked
|
|
||||||
# to the long outstanding `pdb` one below?
|
|
||||||
# |_ it's funny that there's frame hiding throughout `._run.py`
|
|
||||||
# but not where it matters on the below exit funcs..
|
|
||||||
#
|
|
||||||
# -[ ] provide a patchset for the lonstanding
|
|
||||||
# |_ https://github.com/python-trio/trio/issues/1155
|
|
||||||
#
|
|
||||||
# -[ ] make a linked issue to ^ and propose allowing all the
|
|
||||||
# `._core._run` code to have their `__tracebackhide__` value
|
|
||||||
# configurable by a `RunVar` to allow getting scheduler frames
|
|
||||||
# if desired through configuration?
|
|
||||||
#
|
|
||||||
# -[ ] maybe dig into the core `pdb` issue why the extra frame is shown
|
|
||||||
# at all?
|
|
||||||
#
|
|
||||||
funcs: list[FunctionType] = [
|
|
||||||
trio._core._run.NurseryManager.__aexit__,
|
|
||||||
trio._core._run.CancelScope.__exit__,
|
|
||||||
_GeneratorContextManager.__exit__,
|
|
||||||
_AsyncGeneratorContextManager.__aexit__,
|
|
||||||
_AsyncGeneratorContextManager.__aenter__,
|
|
||||||
trio.Event.wait,
|
|
||||||
]
|
|
||||||
func_list_str: str = textwrap.indent(
|
|
||||||
"\n".join(f.__qualname__ for f in funcs),
|
|
||||||
prefix=' |_ ',
|
|
||||||
)
|
|
||||||
log.devx(
|
|
||||||
'Hiding the following runtime frames by default:\n'
|
|
||||||
f'{func_list_str}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
codes: dict[FunctionType, CodeType] = {}
|
|
||||||
for ref in funcs:
|
|
||||||
# stash a pre-modified version of each ref's code-obj
|
|
||||||
# so it can be reverted later if needed.
|
|
||||||
codes[ref] = ref.__code__
|
|
||||||
pdbp.hideframe(ref)
|
|
||||||
#
|
|
||||||
# pdbp.hideframe(trio._core._run.NurseryManager.__aexit__)
|
|
||||||
# pdbp.hideframe(trio._core._run.CancelScope.__exit__)
|
|
||||||
# pdbp.hideframe(_GeneratorContextManager.__exit__)
|
|
||||||
# pdbp.hideframe(_AsyncGeneratorContextManager.__aexit__)
|
|
||||||
# pdbp.hideframe(_AsyncGeneratorContextManager.__aenter__)
|
|
||||||
# pdbp.hideframe(trio.Event.wait)
|
|
||||||
return codes
|
|
||||||
|
|
|
||||||
|
|
@ -35,7 +35,6 @@ from signal import (
|
||||||
signal,
|
signal,
|
||||||
getsignal,
|
getsignal,
|
||||||
SIGUSR1,
|
SIGUSR1,
|
||||||
SIGINT,
|
|
||||||
)
|
)
|
||||||
# import traceback
|
# import traceback
|
||||||
from types import ModuleType
|
from types import ModuleType
|
||||||
|
|
@ -49,7 +48,6 @@ from tractor import (
|
||||||
_state,
|
_state,
|
||||||
log as logmod,
|
log as logmod,
|
||||||
)
|
)
|
||||||
from tractor.devx import debug
|
|
||||||
|
|
||||||
log = logmod.get_logger(__name__)
|
log = logmod.get_logger(__name__)
|
||||||
|
|
||||||
|
|
@ -78,45 +76,22 @@ def dump_task_tree() -> None:
|
||||||
)
|
)
|
||||||
actor: Actor = _state.current_actor()
|
actor: Actor = _state.current_actor()
|
||||||
thr: Thread = current_thread()
|
thr: Thread = current_thread()
|
||||||
current_sigint_handler: Callable = getsignal(SIGINT)
|
|
||||||
if (
|
|
||||||
current_sigint_handler
|
|
||||||
is not
|
|
||||||
debug.DebugStatus._trio_handler
|
|
||||||
):
|
|
||||||
sigint_handler_report: str = (
|
|
||||||
'The default `trio` SIGINT handler was replaced?!'
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
sigint_handler_report: str = (
|
|
||||||
'The default `trio` SIGINT handler is in use?!'
|
|
||||||
)
|
|
||||||
|
|
||||||
# sclang symbology
|
|
||||||
# |_<object>
|
|
||||||
# |_(Task/Thread/Process/Actor
|
|
||||||
# |_{Supervisor/Scope
|
|
||||||
# |_[Storage/Memory/IPC-Stream/Data-Struct
|
|
||||||
|
|
||||||
log.devx(
|
log.devx(
|
||||||
f'Dumping `stackscope` tree for actor\n'
|
f'Dumping `stackscope` tree for actor\n'
|
||||||
f'(>: {actor.uid!r}\n'
|
f'{actor.uid}:\n'
|
||||||
f'|_{mp.current_process()}\n'
|
f'|_{mp.current_process()}\n'
|
||||||
f' |_{thr}\n'
|
f' |_{thr}\n'
|
||||||
f' |_{actor}\n'
|
f' |_{actor}\n\n'
|
||||||
f'\n'
|
|
||||||
f'{sigint_handler_report}\n'
|
|
||||||
f'signal.getsignal(SIGINT) -> {current_sigint_handler!r}\n'
|
|
||||||
# f'\n'
|
|
||||||
# start-of-trace-tree delimiter (mostly for testing)
|
# start-of-trace-tree delimiter (mostly for testing)
|
||||||
# f'------ {actor.uid!r} ------\n'
|
'------ - ------\n'
|
||||||
f'\n'
|
'\n'
|
||||||
f'------ start-of-{actor.uid!r} ------\n'
|
+
|
||||||
f'|\n'
|
f'{tree_str}\n'
|
||||||
f'{tree_str}'
|
+
|
||||||
# end-of-trace-tree delimiter (mostly for testing)
|
# end-of-trace-tree delimiter (mostly for testing)
|
||||||
f'|\n'
|
f'\n'
|
||||||
f'|_____ end-of-{actor.uid!r} ______\n'
|
f'------ {actor.uid!r} ------\n'
|
||||||
)
|
)
|
||||||
# TODO: can remove this right?
|
# TODO: can remove this right?
|
||||||
# -[ ] was original code from author
|
# -[ ] was original code from author
|
||||||
|
|
@ -148,11 +123,11 @@ def dump_tree_on_sig(
|
||||||
) -> None:
|
) -> None:
|
||||||
global _tree_dumped, _handler_lock
|
global _tree_dumped, _handler_lock
|
||||||
with _handler_lock:
|
with _handler_lock:
|
||||||
# if _tree_dumped:
|
if _tree_dumped:
|
||||||
# log.warning(
|
log.warning(
|
||||||
# 'Already dumped for this actor...??'
|
'Already dumped for this actor...??'
|
||||||
# )
|
)
|
||||||
# return
|
return
|
||||||
|
|
||||||
_tree_dumped = True
|
_tree_dumped = True
|
||||||
|
|
||||||
|
|
@ -186,9 +161,9 @@ def dump_tree_on_sig(
|
||||||
)
|
)
|
||||||
raise
|
raise
|
||||||
|
|
||||||
# log.devx(
|
log.devx(
|
||||||
# 'Supposedly we dumped just fine..?'
|
'Supposedly we dumped just fine..?'
|
||||||
# )
|
)
|
||||||
|
|
||||||
if not relay_to_subs:
|
if not relay_to_subs:
|
||||||
return
|
return
|
||||||
|
|
@ -227,19 +202,18 @@ def enable_stack_on_sig(
|
||||||
(https://www.gnu.org/software/bash/manual/bash.html#Command-Substitution)
|
(https://www.gnu.org/software/bash/manual/bash.html#Command-Substitution)
|
||||||
you could use:
|
you could use:
|
||||||
|
|
||||||
>> kill -SIGUSR1 $(pgrep -f <part-of-cmd: str>)
|
>> kill -SIGUSR1 $(pgrep -f '<cmd>')
|
||||||
|
|
||||||
OR without a sub-shell,
|
Or with with `xonsh` (which has diff capture-from-subproc syntax)
|
||||||
|
|
||||||
>> pkill --signal SIGUSR1 -f <part-of-cmd: str>
|
>> kill -SIGUSR1 @$(pgrep -f '<cmd>')
|
||||||
|
|
||||||
'''
|
'''
|
||||||
try:
|
try:
|
||||||
import stackscope
|
import stackscope
|
||||||
except ImportError:
|
except ImportError:
|
||||||
log.warning(
|
log.warning(
|
||||||
'The `stackscope` lib is not installed!\n'
|
'`stackscope` not installed for use in debug mode!'
|
||||||
'`Ignoring enable_stack_on_sig() call!\n'
|
|
||||||
)
|
)
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
@ -256,8 +230,8 @@ def enable_stack_on_sig(
|
||||||
dump_tree_on_sig,
|
dump_tree_on_sig,
|
||||||
)
|
)
|
||||||
log.devx(
|
log.devx(
|
||||||
f'Enabling trace-trees on `SIGUSR1` '
|
'Enabling trace-trees on `SIGUSR1` '
|
||||||
f'since `stackscope` is installed @ \n'
|
'since `stackscope` is installed @ \n'
|
||||||
f'{stackscope!r}\n\n'
|
f'{stackscope!r}\n\n'
|
||||||
f'With `SIGUSR1` handler\n'
|
f'With `SIGUSR1` handler\n'
|
||||||
f'|_{dump_tree_on_sig}\n'
|
f'|_{dump_tree_on_sig}\n'
|
||||||
|
|
|
||||||
|
|
@ -1,100 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or
|
|
||||||
# modify it under the terms of the GNU Affero General Public License
|
|
||||||
# as published by the Free Software Foundation, either version 3 of
|
|
||||||
# the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful, but
|
|
||||||
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
# Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public
|
|
||||||
# License along with this program. If not, see
|
|
||||||
# <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
'''
|
|
||||||
Multi-actor debugging for da peeps!
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
from tractor.log import get_logger
|
|
||||||
from ._repl import (
|
|
||||||
PdbREPL as PdbREPL,
|
|
||||||
mk_pdb as mk_pdb,
|
|
||||||
TractorConfig as TractorConfig,
|
|
||||||
)
|
|
||||||
from ._tty_lock import (
|
|
||||||
DebugStatus as DebugStatus,
|
|
||||||
DebugStateError as DebugStateError,
|
|
||||||
)
|
|
||||||
from ._trace import (
|
|
||||||
Lock as Lock,
|
|
||||||
_pause_msg as _pause_msg,
|
|
||||||
_repl_fail_msg as _repl_fail_msg,
|
|
||||||
_set_trace as _set_trace,
|
|
||||||
_sync_pause_from_builtin as _sync_pause_from_builtin,
|
|
||||||
breakpoint as breakpoint,
|
|
||||||
maybe_init_greenback as maybe_init_greenback,
|
|
||||||
maybe_import_greenback as maybe_import_greenback,
|
|
||||||
pause as pause,
|
|
||||||
pause_from_sync as pause_from_sync,
|
|
||||||
)
|
|
||||||
from ._post_mortem import (
|
|
||||||
BoxedMaybeException as BoxedMaybeException,
|
|
||||||
maybe_open_crash_handler as maybe_open_crash_handler,
|
|
||||||
open_crash_handler as open_crash_handler,
|
|
||||||
post_mortem as post_mortem,
|
|
||||||
_crash_msg as _crash_msg,
|
|
||||||
_maybe_enter_pm as _maybe_enter_pm,
|
|
||||||
)
|
|
||||||
from ._sync import (
|
|
||||||
maybe_wait_for_debugger as maybe_wait_for_debugger,
|
|
||||||
acquire_debug_lock as acquire_debug_lock,
|
|
||||||
)
|
|
||||||
from ._sigint import (
|
|
||||||
sigint_shield as sigint_shield,
|
|
||||||
_ctlc_ignore_header as _ctlc_ignore_header
|
|
||||||
)
|
|
||||||
|
|
||||||
log = get_logger(__name__)
|
|
||||||
|
|
||||||
# ----------------
|
|
||||||
# XXX PKG TODO XXX
|
|
||||||
# ----------------
|
|
||||||
# refine the internal impl and APIs!
|
|
||||||
#
|
|
||||||
# -[ ] rework `._pause()` and it's branch-cases for root vs.
|
|
||||||
# subactor:
|
|
||||||
# -[ ] `._pause_from_root()` + `_pause_from_subactor()`?
|
|
||||||
# -[ ] do the de-factor based on bg-thread usage in
|
|
||||||
# `.pause_from_sync()` & `_pause_from_bg_root_thread()`.
|
|
||||||
# -[ ] drop `debug_func == None` case which is confusing af..
|
|
||||||
# -[ ] factor out `_enter_repl_sync()` into a util func for calling
|
|
||||||
# the `_set_trace()` / `_post_mortem()` APIs?
|
|
||||||
#
|
|
||||||
# -[ ] figure out if we need `acquire_debug_lock()` and/or re-implement
|
|
||||||
# it as part of the `.pause_from_sync()` rework per above?
|
|
||||||
#
|
|
||||||
# -[ ] pair the `._pause_from_subactor()` impl with a "debug nursery"
|
|
||||||
# that's dynamically allocated inside the `._rpc` task thus
|
|
||||||
# avoiding the `._service_n.start()` usage for the IPC request?
|
|
||||||
# -[ ] see the TODO inside `._rpc._errors_relayed_via_ipc()`
|
|
||||||
#
|
|
||||||
# -[ ] impl a `open_debug_request()` which encaps all
|
|
||||||
# `request_root_stdio_lock()` task scheduling deats
|
|
||||||
# + `DebugStatus` state mgmt; which should prolly be re-branded as
|
|
||||||
# a `DebugRequest` type anyway AND with suppoort for bg-thread
|
|
||||||
# (from root actor) usage?
|
|
||||||
#
|
|
||||||
# -[ ] handle the `xonsh` case for bg-root-threads in the SIGINT
|
|
||||||
# handler!
|
|
||||||
# -[ ] do we need to do the same for subactors?
|
|
||||||
# -[ ] make the failing tests finally pass XD
|
|
||||||
#
|
|
||||||
# -[ ] simplify `maybe_wait_for_debugger()` to be a root-task only
|
|
||||||
# API?
|
|
||||||
# -[ ] currently it's implemented as that so might as well make it
|
|
||||||
# formal?
|
|
||||||
|
|
@ -1,412 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or
|
|
||||||
# modify it under the terms of the GNU Affero General Public License
|
|
||||||
# as published by the Free Software Foundation, either version 3 of
|
|
||||||
# the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful, but
|
|
||||||
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
# Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public
|
|
||||||
# License along with this program. If not, see
|
|
||||||
# <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
'''
|
|
||||||
Post-mortem debugging APIs and surrounding machinery for both
|
|
||||||
sync and async contexts.
|
|
||||||
|
|
||||||
Generally we maintain the same semantics a `pdb.post.mortem()` but
|
|
||||||
with actor-tree-wide sync/cooperation around any (sub)actor's use of
|
|
||||||
the root's TTY.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
import bdb
|
|
||||||
from contextlib import (
|
|
||||||
AbstractContextManager,
|
|
||||||
contextmanager as cm,
|
|
||||||
nullcontext,
|
|
||||||
)
|
|
||||||
from functools import (
|
|
||||||
partial,
|
|
||||||
)
|
|
||||||
import inspect
|
|
||||||
import sys
|
|
||||||
import traceback
|
|
||||||
from typing import (
|
|
||||||
Callable,
|
|
||||||
Sequence,
|
|
||||||
Type,
|
|
||||||
TYPE_CHECKING,
|
|
||||||
)
|
|
||||||
from types import (
|
|
||||||
TracebackType,
|
|
||||||
FrameType,
|
|
||||||
)
|
|
||||||
|
|
||||||
from msgspec import Struct
|
|
||||||
import trio
|
|
||||||
from tractor._exceptions import (
|
|
||||||
NoRuntime,
|
|
||||||
)
|
|
||||||
from tractor import _state
|
|
||||||
from tractor._state import (
|
|
||||||
current_actor,
|
|
||||||
debug_mode,
|
|
||||||
)
|
|
||||||
from tractor.log import get_logger
|
|
||||||
from tractor.trionics import (
|
|
||||||
is_multi_cancelled,
|
|
||||||
)
|
|
||||||
from ._trace import (
|
|
||||||
_pause,
|
|
||||||
)
|
|
||||||
from ._tty_lock import (
|
|
||||||
DebugStatus,
|
|
||||||
)
|
|
||||||
from ._repl import (
|
|
||||||
PdbREPL,
|
|
||||||
mk_pdb,
|
|
||||||
TractorConfig as TractorConfig,
|
|
||||||
)
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from trio.lowlevel import Task
|
|
||||||
from tractor._runtime import (
|
|
||||||
Actor,
|
|
||||||
)
|
|
||||||
|
|
||||||
_crash_msg: str = (
|
|
||||||
'Opening a pdb REPL in crashed actor'
|
|
||||||
)
|
|
||||||
|
|
||||||
log = get_logger(__package__)
|
|
||||||
|
|
||||||
|
|
||||||
class BoxedMaybeException(Struct):
|
|
||||||
'''
|
|
||||||
Box a maybe-exception for post-crash introspection usage
|
|
||||||
from the body of a `open_crash_handler()` scope.
|
|
||||||
|
|
||||||
'''
|
|
||||||
value: BaseException|None = None
|
|
||||||
|
|
||||||
# handler can suppress crashes dynamically
|
|
||||||
raise_on_exit: bool|Sequence[Type[BaseException]] = True
|
|
||||||
|
|
||||||
def pformat(self) -> str:
|
|
||||||
'''
|
|
||||||
Repr the boxed `.value` error in more-than-string
|
|
||||||
repr form.
|
|
||||||
|
|
||||||
'''
|
|
||||||
if not self.value:
|
|
||||||
return f'<{type(self).__name__}( .value=None )>'
|
|
||||||
|
|
||||||
return (
|
|
||||||
f'<{type(self.value).__name__}(\n'
|
|
||||||
f' |_.value = {self.value}\n'
|
|
||||||
f')>\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
__repr__ = pformat
|
|
||||||
|
|
||||||
|
|
||||||
def _post_mortem(
|
|
||||||
repl: PdbREPL, # normally passed by `_pause()`
|
|
||||||
|
|
||||||
# XXX all `partial`-ed in by `post_mortem()` below!
|
|
||||||
tb: TracebackType,
|
|
||||||
api_frame: FrameType,
|
|
||||||
|
|
||||||
shield: bool = False,
|
|
||||||
hide_tb: bool = True,
|
|
||||||
|
|
||||||
# maybe pre/post REPL entry
|
|
||||||
repl_fixture: (
|
|
||||||
AbstractContextManager[bool]
|
|
||||||
|None
|
|
||||||
) = None,
|
|
||||||
|
|
||||||
boxed_maybe_exc: BoxedMaybeException|None = None,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
Enter the ``pdbpp`` port mortem entrypoint using our custom
|
|
||||||
debugger instance.
|
|
||||||
|
|
||||||
'''
|
|
||||||
__tracebackhide__: bool = hide_tb
|
|
||||||
|
|
||||||
# maybe enter any user fixture
|
|
||||||
enter_repl: bool = DebugStatus.maybe_enter_repl_fixture(
|
|
||||||
repl=repl,
|
|
||||||
repl_fixture=repl_fixture,
|
|
||||||
boxed_maybe_exc=boxed_maybe_exc,
|
|
||||||
)
|
|
||||||
try:
|
|
||||||
if not enter_repl:
|
|
||||||
# XXX, trigger `.release()` below immediately!
|
|
||||||
return
|
|
||||||
try:
|
|
||||||
actor: Actor = current_actor()
|
|
||||||
actor_repr: str = str(actor.uid)
|
|
||||||
# ^TODO, instead a nice runtime-info + maddr + uid?
|
|
||||||
# -[ ] impl a `Actor.__repr()__`??
|
|
||||||
# |_ <task>:<thread> @ <actor>
|
|
||||||
|
|
||||||
except NoRuntime:
|
|
||||||
actor_repr: str = '<no-actor-runtime?>'
|
|
||||||
|
|
||||||
try:
|
|
||||||
task_repr: Task = trio.lowlevel.current_task()
|
|
||||||
except RuntimeError:
|
|
||||||
task_repr: str = '<unknown-Task>'
|
|
||||||
|
|
||||||
# TODO: print the actor supervion tree up to the root
|
|
||||||
# here! Bo
|
|
||||||
log.pdb(
|
|
||||||
f'{_crash_msg}\n'
|
|
||||||
f'x>(\n'
|
|
||||||
f' |_ {task_repr} @ {actor_repr}\n'
|
|
||||||
|
|
||||||
)
|
|
||||||
|
|
||||||
# XXX NOTE(s) on `pdbp.xpm()` version..
|
|
||||||
#
|
|
||||||
# - seems to lose the up-stack tb-info?
|
|
||||||
# - currently we're (only) replacing this from `pdbp.xpm()`
|
|
||||||
# to add the `end=''` to the print XD
|
|
||||||
#
|
|
||||||
print(traceback.format_exc(), end='')
|
|
||||||
caller_frame: FrameType = api_frame.f_back
|
|
||||||
|
|
||||||
# NOTE, see the impl details of these in the lib to
|
|
||||||
# understand usage:
|
|
||||||
# - `pdbp.post_mortem()`
|
|
||||||
# - `pdbp.xps()`
|
|
||||||
# - `bdb.interaction()`
|
|
||||||
repl.reset()
|
|
||||||
repl.interaction(
|
|
||||||
frame=caller_frame,
|
|
||||||
# frame=None,
|
|
||||||
traceback=tb,
|
|
||||||
)
|
|
||||||
finally:
|
|
||||||
# XXX NOTE XXX: this is abs required to avoid hangs!
|
|
||||||
#
|
|
||||||
# Since we presume the post-mortem was enaged to
|
|
||||||
# a task-ending error, we MUST release the local REPL request
|
|
||||||
# so that not other local task nor the root remains blocked!
|
|
||||||
DebugStatus.release()
|
|
||||||
|
|
||||||
|
|
||||||
async def post_mortem(
|
|
||||||
*,
|
|
||||||
tb: TracebackType|None = None,
|
|
||||||
api_frame: FrameType|None = None,
|
|
||||||
hide_tb: bool = False,
|
|
||||||
|
|
||||||
# TODO: support shield here just like in `pause()`?
|
|
||||||
# shield: bool = False,
|
|
||||||
|
|
||||||
**_pause_kwargs,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
Our builtin async equivalient of `pdb.post_mortem()` which can be
|
|
||||||
used inside exception handlers.
|
|
||||||
|
|
||||||
It's also used for the crash handler when `debug_mode == True` ;)
|
|
||||||
|
|
||||||
'''
|
|
||||||
__tracebackhide__: bool = hide_tb
|
|
||||||
|
|
||||||
tb: TracebackType = tb or sys.exc_info()[2]
|
|
||||||
|
|
||||||
# TODO: do upward stack scan for highest @api_frame and
|
|
||||||
# use its parent frame as the expected user-app code
|
|
||||||
# interact point.
|
|
||||||
api_frame: FrameType = api_frame or inspect.currentframe()
|
|
||||||
|
|
||||||
# TODO, move to submod `._pausing` or ._api? _trace
|
|
||||||
await _pause(
|
|
||||||
debug_func=partial(
|
|
||||||
_post_mortem,
|
|
||||||
api_frame=api_frame,
|
|
||||||
tb=tb,
|
|
||||||
),
|
|
||||||
hide_tb=hide_tb,
|
|
||||||
**_pause_kwargs
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def _maybe_enter_pm(
|
|
||||||
err: BaseException,
|
|
||||||
*,
|
|
||||||
tb: TracebackType|None = None,
|
|
||||||
api_frame: FrameType|None = None,
|
|
||||||
hide_tb: bool = True,
|
|
||||||
|
|
||||||
# only enter debugger REPL when returns `True`
|
|
||||||
debug_filter: Callable[
|
|
||||||
[BaseException|BaseExceptionGroup],
|
|
||||||
bool,
|
|
||||||
] = lambda err: not is_multi_cancelled(err),
|
|
||||||
**_pause_kws,
|
|
||||||
):
|
|
||||||
if (
|
|
||||||
debug_mode()
|
|
||||||
|
|
||||||
# NOTE: don't enter debug mode recursively after quitting pdb
|
|
||||||
# Iow, don't re-enter the repl if the `quit` command was issued
|
|
||||||
# by the user.
|
|
||||||
and not isinstance(err, bdb.BdbQuit)
|
|
||||||
|
|
||||||
# XXX: if the error is the likely result of runtime-wide
|
|
||||||
# cancellation, we don't want to enter the debugger since
|
|
||||||
# there's races between when the parent actor has killed all
|
|
||||||
# comms and when the child tries to contact said parent to
|
|
||||||
# acquire the tty lock.
|
|
||||||
|
|
||||||
# Really we just want to mostly avoid catching KBIs here so there
|
|
||||||
# might be a simpler check we can do?
|
|
||||||
and
|
|
||||||
debug_filter(err)
|
|
||||||
):
|
|
||||||
api_frame: FrameType = api_frame or inspect.currentframe()
|
|
||||||
tb: TracebackType = tb or sys.exc_info()[2]
|
|
||||||
await post_mortem(
|
|
||||||
api_frame=api_frame,
|
|
||||||
tb=tb,
|
|
||||||
**_pause_kws,
|
|
||||||
)
|
|
||||||
return True
|
|
||||||
|
|
||||||
else:
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
# TODO: better naming and what additionals?
|
|
||||||
# - [ ] optional runtime plugging?
|
|
||||||
# - [ ] detection for sync vs. async code?
|
|
||||||
# - [ ] specialized REPL entry when in distributed mode?
|
|
||||||
# -[x] hide tb by def
|
|
||||||
# - [x] allow ignoring kbi Bo
|
|
||||||
@cm
|
|
||||||
def open_crash_handler(
|
|
||||||
catch: set[BaseException] = {
|
|
||||||
BaseException,
|
|
||||||
},
|
|
||||||
ignore: set[BaseException] = {
|
|
||||||
KeyboardInterrupt,
|
|
||||||
trio.Cancelled,
|
|
||||||
},
|
|
||||||
hide_tb: bool = True,
|
|
||||||
|
|
||||||
repl_fixture: (
|
|
||||||
AbstractContextManager[bool] # pre/post REPL entry
|
|
||||||
|None
|
|
||||||
) = None,
|
|
||||||
raise_on_exit: bool|Sequence[Type[BaseException]] = True,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Generic "post mortem" crash handler using `pdbp` REPL debugger.
|
|
||||||
|
|
||||||
We expose this as a CLI framework addon to both `click` and
|
|
||||||
`typer` users so they can quickly wrap cmd endpoints which get
|
|
||||||
automatically wrapped to use the runtime's `debug_mode: bool`
|
|
||||||
AND `pdbp.pm()` around any code that is PRE-runtime entry
|
|
||||||
- any sync code which runs BEFORE the main call to
|
|
||||||
`trio.run()`.
|
|
||||||
|
|
||||||
'''
|
|
||||||
__tracebackhide__: bool = hide_tb
|
|
||||||
|
|
||||||
# TODO, yield a `outcome.Error`-like boxed type?
|
|
||||||
# -[~] use `outcome.Value/Error` X-> frozen!
|
|
||||||
# -[x] write our own..?
|
|
||||||
# -[ ] consider just wtv is used by `pytest.raises()`?
|
|
||||||
#
|
|
||||||
boxed_maybe_exc = BoxedMaybeException(
|
|
||||||
raise_on_exit=raise_on_exit,
|
|
||||||
)
|
|
||||||
err: BaseException
|
|
||||||
try:
|
|
||||||
yield boxed_maybe_exc
|
|
||||||
except tuple(catch) as err:
|
|
||||||
boxed_maybe_exc.value = err
|
|
||||||
if (
|
|
||||||
type(err) not in ignore
|
|
||||||
and
|
|
||||||
not is_multi_cancelled(
|
|
||||||
err,
|
|
||||||
ignore_nested=ignore
|
|
||||||
)
|
|
||||||
):
|
|
||||||
try:
|
|
||||||
# use our re-impl-ed version of `pdbp.xpm()`
|
|
||||||
_post_mortem(
|
|
||||||
repl=mk_pdb(),
|
|
||||||
tb=sys.exc_info()[2],
|
|
||||||
api_frame=inspect.currentframe().f_back,
|
|
||||||
hide_tb=hide_tb,
|
|
||||||
|
|
||||||
repl_fixture=repl_fixture,
|
|
||||||
boxed_maybe_exc=boxed_maybe_exc,
|
|
||||||
)
|
|
||||||
except bdb.BdbQuit:
|
|
||||||
__tracebackhide__: bool = False
|
|
||||||
raise err
|
|
||||||
|
|
||||||
if (
|
|
||||||
raise_on_exit is True
|
|
||||||
or (
|
|
||||||
raise_on_exit is not False
|
|
||||||
and (
|
|
||||||
set(raise_on_exit)
|
|
||||||
and
|
|
||||||
type(err) in raise_on_exit
|
|
||||||
)
|
|
||||||
)
|
|
||||||
and
|
|
||||||
boxed_maybe_exc.raise_on_exit == raise_on_exit
|
|
||||||
):
|
|
||||||
raise err
|
|
||||||
|
|
||||||
|
|
||||||
@cm
|
|
||||||
def maybe_open_crash_handler(
|
|
||||||
pdb: bool|None = None,
|
|
||||||
hide_tb: bool = True,
|
|
||||||
|
|
||||||
**kwargs,
|
|
||||||
):
|
|
||||||
'''
|
|
||||||
Same as `open_crash_handler()` but with bool input flag
|
|
||||||
to allow conditional handling.
|
|
||||||
|
|
||||||
Normally this is used with CLI endpoints such that if the --pdb
|
|
||||||
flag is passed the pdb REPL is engaed on any crashes B)
|
|
||||||
|
|
||||||
'''
|
|
||||||
__tracebackhide__: bool = hide_tb
|
|
||||||
|
|
||||||
if pdb is None:
|
|
||||||
pdb: bool = _state.is_debug_mode()
|
|
||||||
|
|
||||||
rtctx = nullcontext(
|
|
||||||
enter_result=BoxedMaybeException()
|
|
||||||
)
|
|
||||||
if pdb:
|
|
||||||
rtctx = open_crash_handler(
|
|
||||||
hide_tb=hide_tb,
|
|
||||||
**kwargs,
|
|
||||||
)
|
|
||||||
|
|
||||||
with rtctx as boxed_maybe_exc:
|
|
||||||
yield boxed_maybe_exc
|
|
||||||
|
|
@ -1,207 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or
|
|
||||||
# modify it under the terms of the GNU Affero General Public License
|
|
||||||
# as published by the Free Software Foundation, either version 3 of
|
|
||||||
# the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful, but
|
|
||||||
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
# Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public
|
|
||||||
# License along with this program. If not, see
|
|
||||||
# <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
'''
|
|
||||||
`pdpp.Pdb` extentions/customization and other delegate usage.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from functools import (
|
|
||||||
cached_property,
|
|
||||||
)
|
|
||||||
import os
|
|
||||||
|
|
||||||
import pdbp
|
|
||||||
from tractor._state import (
|
|
||||||
is_root_process,
|
|
||||||
)
|
|
||||||
|
|
||||||
from ._tty_lock import (
|
|
||||||
Lock,
|
|
||||||
DebugStatus,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class TractorConfig(pdbp.DefaultConfig):
|
|
||||||
'''
|
|
||||||
Custom `pdbp` config which tries to use the best tradeoff
|
|
||||||
between pretty and minimal.
|
|
||||||
|
|
||||||
'''
|
|
||||||
use_pygments: bool = True
|
|
||||||
sticky_by_default: bool = False
|
|
||||||
enable_hidden_frames: bool = True
|
|
||||||
|
|
||||||
# much thanks @mdmintz for the hot tip!
|
|
||||||
# fixes line spacing issue when resizing terminal B)
|
|
||||||
truncate_long_lines: bool = False
|
|
||||||
|
|
||||||
# ------ - ------
|
|
||||||
# our own custom config vars mostly
|
|
||||||
# for syncing with the actor tree's singleton
|
|
||||||
# TTY `Lock`.
|
|
||||||
|
|
||||||
|
|
||||||
class PdbREPL(pdbp.Pdb):
|
|
||||||
'''
|
|
||||||
Add teardown hooks and local state describing any
|
|
||||||
ongoing TTY `Lock` request dialog.
|
|
||||||
|
|
||||||
'''
|
|
||||||
# override the pdbp config with our coolio one
|
|
||||||
# NOTE: this is only loaded when no `~/.pdbrc` exists
|
|
||||||
# so we should prolly pass it into the .__init__() instead?
|
|
||||||
# i dunno, see the `DefaultFactory` and `pdb.Pdb` impls.
|
|
||||||
DefaultConfig = TractorConfig
|
|
||||||
|
|
||||||
status = DebugStatus
|
|
||||||
|
|
||||||
# NOTE: see details in stdlib's `bdb.py`
|
|
||||||
# def user_exception(self, frame, exc_info):
|
|
||||||
# '''
|
|
||||||
# Called when we stop on an exception.
|
|
||||||
# '''
|
|
||||||
# log.warning(
|
|
||||||
# 'Exception during REPL sesh\n\n'
|
|
||||||
# f'{frame}\n\n'
|
|
||||||
# f'{exc_info}\n\n'
|
|
||||||
# )
|
|
||||||
|
|
||||||
# NOTE: this actually hooks but i don't see anyway to detect
|
|
||||||
# if an error was caught.. this is why currently we just always
|
|
||||||
# call `DebugStatus.release` inside `_post_mortem()`.
|
|
||||||
# def preloop(self):
|
|
||||||
# print('IN PRELOOP')
|
|
||||||
# super().preloop()
|
|
||||||
|
|
||||||
# TODO: cleaner re-wrapping of all this?
|
|
||||||
# -[ ] figure out how to disallow recursive .set_trace() entry
|
|
||||||
# since that'll cause deadlock for us.
|
|
||||||
# -[ ] maybe a `@cm` to call `super().<same_meth_name>()`?
|
|
||||||
# -[ ] look at hooking into the `pp` hook specially with our
|
|
||||||
# own set of pretty-printers?
|
|
||||||
# * `.pretty_struct.Struct.pformat()`
|
|
||||||
# * `.pformat(MsgType.pld)`
|
|
||||||
# * `.pformat(Error.tb_str)`?
|
|
||||||
# * .. maybe more?
|
|
||||||
#
|
|
||||||
def set_continue(self):
|
|
||||||
try:
|
|
||||||
super().set_continue()
|
|
||||||
finally:
|
|
||||||
# NOTE: for subactors the stdio lock is released via the
|
|
||||||
# allocated RPC locker task, so for root we have to do it
|
|
||||||
# manually.
|
|
||||||
if (
|
|
||||||
is_root_process()
|
|
||||||
and
|
|
||||||
Lock._debug_lock.locked()
|
|
||||||
and
|
|
||||||
DebugStatus.is_main_trio_thread()
|
|
||||||
):
|
|
||||||
# Lock.release(raise_on_thread=False)
|
|
||||||
Lock.release()
|
|
||||||
|
|
||||||
# XXX AFTER `Lock.release()` for root local repl usage
|
|
||||||
DebugStatus.release()
|
|
||||||
|
|
||||||
def set_quit(self):
|
|
||||||
try:
|
|
||||||
super().set_quit()
|
|
||||||
finally:
|
|
||||||
if (
|
|
||||||
is_root_process()
|
|
||||||
and
|
|
||||||
Lock._debug_lock.locked()
|
|
||||||
and
|
|
||||||
DebugStatus.is_main_trio_thread()
|
|
||||||
):
|
|
||||||
# Lock.release(raise_on_thread=False)
|
|
||||||
Lock.release()
|
|
||||||
|
|
||||||
# XXX after `Lock.release()` for root local repl usage
|
|
||||||
DebugStatus.release()
|
|
||||||
|
|
||||||
# XXX NOTE: we only override this because apparently the stdlib pdb
|
|
||||||
# bois likes to touch the SIGINT handler as much as i like to touch
|
|
||||||
# my d$%&.
|
|
||||||
def _cmdloop(self):
|
|
||||||
self.cmdloop()
|
|
||||||
|
|
||||||
@cached_property
|
|
||||||
def shname(self) -> str | None:
|
|
||||||
'''
|
|
||||||
Attempt to return the login shell name with a special check for
|
|
||||||
the infamous `xonsh` since it seems to have some issues much
|
|
||||||
different from std shells when it comes to flushing the prompt?
|
|
||||||
|
|
||||||
'''
|
|
||||||
# SUPER HACKY and only really works if `xonsh` is not used
|
|
||||||
# before spawning further sub-shells..
|
|
||||||
shpath = os.getenv('SHELL', None)
|
|
||||||
|
|
||||||
if shpath:
|
|
||||||
if (
|
|
||||||
os.getenv('XONSH_LOGIN', default=False)
|
|
||||||
or 'xonsh' in shpath
|
|
||||||
):
|
|
||||||
return 'xonsh'
|
|
||||||
|
|
||||||
return os.path.basename(shpath)
|
|
||||||
|
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def mk_pdb() -> PdbREPL:
|
|
||||||
'''
|
|
||||||
Deliver a new `PdbREPL`: a multi-process safe `pdbp.Pdb`-variant
|
|
||||||
using the magic of `tractor`'s SC-safe IPC.
|
|
||||||
|
|
||||||
B)
|
|
||||||
|
|
||||||
Our `pdb.Pdb` subtype accomplishes multi-process safe debugging
|
|
||||||
by:
|
|
||||||
|
|
||||||
- mutexing access to the root process' std-streams (& thus parent
|
|
||||||
process TTY) via an IPC managed `Lock` singleton per
|
|
||||||
actor-process tree.
|
|
||||||
|
|
||||||
- temporarily overriding any subactor's SIGINT handler to shield
|
|
||||||
during live REPL sessions in sub-actors such that cancellation
|
|
||||||
is never (mistakenly) triggered by a ctrl-c and instead only by
|
|
||||||
explicit runtime API requests or after the
|
|
||||||
`pdb.Pdb.interaction()` call has returned.
|
|
||||||
|
|
||||||
FURTHER, the `pdbp.Pdb` instance is configured to be `trio`
|
|
||||||
"compatible" from a SIGINT handling perspective; we mask out
|
|
||||||
the default `pdb` handler and instead apply `trio`s default
|
|
||||||
which mostly addresses all issues described in:
|
|
||||||
|
|
||||||
- https://github.com/python-trio/trio/issues/1155
|
|
||||||
|
|
||||||
The instance returned from this factory should always be
|
|
||||||
preferred over the default `pdb[p].set_trace()` whenever using
|
|
||||||
a `pdb` REPL inside a `trio` based runtime.
|
|
||||||
|
|
||||||
'''
|
|
||||||
pdb = PdbREPL()
|
|
||||||
|
|
||||||
# XXX: These are the important flags mentioned in
|
|
||||||
# https://github.com/python-trio/trio/issues/1155
|
|
||||||
# which resolve the traceback spews to console.
|
|
||||||
pdb.allow_kbdint = True
|
|
||||||
pdb.nosigint = True
|
|
||||||
return pdb
|
|
||||||
|
|
@ -1,333 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or
|
|
||||||
# modify it under the terms of the GNU Affero General Public License
|
|
||||||
# as published by the Free Software Foundation, either version 3 of
|
|
||||||
# the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful, but
|
|
||||||
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
# Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public
|
|
||||||
# License along with this program. If not, see
|
|
||||||
# <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
'''
|
|
||||||
A custom SIGINT handler which mainly shields actor (task)
|
|
||||||
cancellation during REPL interaction.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
from typing import (
|
|
||||||
TYPE_CHECKING,
|
|
||||||
)
|
|
||||||
import trio
|
|
||||||
from tractor.log import get_logger
|
|
||||||
from tractor._state import (
|
|
||||||
current_actor,
|
|
||||||
is_root_process,
|
|
||||||
)
|
|
||||||
from ._repl import (
|
|
||||||
PdbREPL,
|
|
||||||
)
|
|
||||||
from ._tty_lock import (
|
|
||||||
any_connected_locker_child,
|
|
||||||
DebugStatus,
|
|
||||||
Lock,
|
|
||||||
)
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from tractor.ipc import (
|
|
||||||
Channel,
|
|
||||||
)
|
|
||||||
from tractor._runtime import (
|
|
||||||
Actor,
|
|
||||||
)
|
|
||||||
|
|
||||||
log = get_logger(__name__)
|
|
||||||
|
|
||||||
_ctlc_ignore_header: str = (
|
|
||||||
'Ignoring SIGINT while debug REPL in use'
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def sigint_shield(
|
|
||||||
signum: int,
|
|
||||||
frame: 'frame', # type: ignore # noqa
|
|
||||||
*args,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
Specialized, debugger-aware SIGINT handler.
|
|
||||||
|
|
||||||
In childred we always ignore/shield for SIGINT to avoid
|
|
||||||
deadlocks since cancellation should always be managed by the
|
|
||||||
supervising parent actor. The root actor-proces is always
|
|
||||||
cancelled on ctrl-c.
|
|
||||||
|
|
||||||
'''
|
|
||||||
__tracebackhide__: bool = True
|
|
||||||
actor: Actor = current_actor()
|
|
||||||
|
|
||||||
def do_cancel():
|
|
||||||
# If we haven't tried to cancel the runtime then do that instead
|
|
||||||
# of raising a KBI (which may non-gracefully destroy
|
|
||||||
# a ``trio.run()``).
|
|
||||||
if not actor._cancel_called:
|
|
||||||
actor.cancel_soon()
|
|
||||||
|
|
||||||
# If the runtime is already cancelled it likely means the user
|
|
||||||
# hit ctrl-c again because teardown didn't fully take place in
|
|
||||||
# which case we do the "hard" raising of a local KBI.
|
|
||||||
else:
|
|
||||||
raise KeyboardInterrupt
|
|
||||||
|
|
||||||
# only set in the actor actually running the REPL
|
|
||||||
repl: PdbREPL|None = DebugStatus.repl
|
|
||||||
|
|
||||||
# TODO: maybe we should flatten out all these cases using
|
|
||||||
# a match/case?
|
|
||||||
#
|
|
||||||
# root actor branch that reports whether or not a child
|
|
||||||
# has locked debugger.
|
|
||||||
if is_root_process():
|
|
||||||
# log.warning(
|
|
||||||
log.devx(
|
|
||||||
'Handling SIGINT in root actor\n'
|
|
||||||
f'{Lock.repr()}'
|
|
||||||
f'{DebugStatus.repr()}\n'
|
|
||||||
)
|
|
||||||
# try to see if the supposed (sub)actor in debug still
|
|
||||||
# has an active connection to *this* actor, and if not
|
|
||||||
# it's likely they aren't using the TTY lock / debugger
|
|
||||||
# and we should propagate SIGINT normally.
|
|
||||||
any_connected: bool = any_connected_locker_child()
|
|
||||||
|
|
||||||
problem = (
|
|
||||||
f'root {actor.uid} handling SIGINT\n'
|
|
||||||
f'any_connected: {any_connected}\n\n'
|
|
||||||
|
|
||||||
f'{Lock.repr()}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
if (
|
|
||||||
(ctx := Lock.ctx_in_debug)
|
|
||||||
and
|
|
||||||
(uid_in_debug := ctx.chan.uid) # "someone" is (ostensibly) using debug `Lock`
|
|
||||||
):
|
|
||||||
name_in_debug: str = uid_in_debug[0]
|
|
||||||
assert not repl
|
|
||||||
# if not repl: # but it's NOT us, the root actor.
|
|
||||||
# sanity: since no repl ref is set, we def shouldn't
|
|
||||||
# be the lock owner!
|
|
||||||
assert name_in_debug != 'root'
|
|
||||||
|
|
||||||
# IDEAL CASE: child has REPL as expected
|
|
||||||
if any_connected: # there are subactors we can contact
|
|
||||||
# XXX: only if there is an existing connection to the
|
|
||||||
# (sub-)actor in debug do we ignore SIGINT in this
|
|
||||||
# parent! Otherwise we may hang waiting for an actor
|
|
||||||
# which has already terminated to unlock.
|
|
||||||
#
|
|
||||||
# NOTE: don't emit this with `.pdb()` level in
|
|
||||||
# root without a higher level.
|
|
||||||
log.runtime(
|
|
||||||
_ctlc_ignore_header
|
|
||||||
+
|
|
||||||
f' by child '
|
|
||||||
f'{uid_in_debug}\n'
|
|
||||||
)
|
|
||||||
problem = None
|
|
||||||
|
|
||||||
else:
|
|
||||||
problem += (
|
|
||||||
'\n'
|
|
||||||
f'A `pdb` REPL is SUPPOSEDLY in use by child {uid_in_debug}\n'
|
|
||||||
f'BUT, no child actors are IPC contactable!?!?\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# IDEAL CASE: root has REPL as expected
|
|
||||||
else:
|
|
||||||
# root actor still has this SIGINT handler active without
|
|
||||||
# an actor using the `Lock` (a bug state) ??
|
|
||||||
# => so immediately cancel any stale lock cs and revert
|
|
||||||
# the handler!
|
|
||||||
if not DebugStatus.repl:
|
|
||||||
# TODO: WHEN should we revert back to ``trio``
|
|
||||||
# handler if this one is stale?
|
|
||||||
# -[ ] maybe after a counts work of ctl-c mashes?
|
|
||||||
# -[ ] use a state var like `stale_handler: bool`?
|
|
||||||
problem += (
|
|
||||||
'No subactor is using a `pdb` REPL according `Lock.ctx_in_debug`?\n'
|
|
||||||
'BUT, the root should be using it, WHY this handler ??\n\n'
|
|
||||||
'So either..\n'
|
|
||||||
'- some root-thread is using it but has no `.repl` set?, OR\n'
|
|
||||||
'- something else weird is going on outside the runtime!?\n'
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
# NOTE: since we emit this msg on ctl-c, we should
|
|
||||||
# also always re-print the prompt the tail block!
|
|
||||||
log.pdb(
|
|
||||||
_ctlc_ignore_header
|
|
||||||
+
|
|
||||||
f' by root actor..\n'
|
|
||||||
f'{DebugStatus.repl_task}\n'
|
|
||||||
f' |_{repl}\n'
|
|
||||||
)
|
|
||||||
problem = None
|
|
||||||
|
|
||||||
# XXX if one is set it means we ARE NOT operating an ideal
|
|
||||||
# case where a child subactor or us (the root) has the
|
|
||||||
# lock without any other detected problems.
|
|
||||||
if problem:
|
|
||||||
|
|
||||||
# detect, report and maybe clear a stale lock request
|
|
||||||
# cancel scope.
|
|
||||||
lock_cs: trio.CancelScope = Lock.get_locking_task_cs()
|
|
||||||
maybe_stale_lock_cs: bool = (
|
|
||||||
lock_cs is not None
|
|
||||||
and not lock_cs.cancel_called
|
|
||||||
)
|
|
||||||
if maybe_stale_lock_cs:
|
|
||||||
problem += (
|
|
||||||
'\n'
|
|
||||||
'Stale `Lock.ctx_in_debug._scope: CancelScope` detected?\n'
|
|
||||||
f'{Lock.ctx_in_debug}\n\n'
|
|
||||||
|
|
||||||
'-> Calling ctx._scope.cancel()!\n'
|
|
||||||
)
|
|
||||||
lock_cs.cancel()
|
|
||||||
|
|
||||||
# TODO: wen do we actually want/need this, see above.
|
|
||||||
# DebugStatus.unshield_sigint()
|
|
||||||
log.warning(problem)
|
|
||||||
|
|
||||||
# child actor that has locked the debugger
|
|
||||||
elif not is_root_process():
|
|
||||||
log.debug(
|
|
||||||
f'Subactor {actor.uid} handling SIGINT\n\n'
|
|
||||||
f'{Lock.repr()}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
rent_chan: Channel = actor._parent_chan
|
|
||||||
if (
|
|
||||||
rent_chan is None
|
|
||||||
or
|
|
||||||
not rent_chan.connected()
|
|
||||||
):
|
|
||||||
log.warning(
|
|
||||||
'This sub-actor thinks it is debugging '
|
|
||||||
'but it has no connection to its parent ??\n'
|
|
||||||
f'{actor.uid}\n'
|
|
||||||
'Allowing SIGINT propagation..'
|
|
||||||
)
|
|
||||||
DebugStatus.unshield_sigint()
|
|
||||||
|
|
||||||
repl_task: str|None = DebugStatus.repl_task
|
|
||||||
req_task: str|None = DebugStatus.req_task
|
|
||||||
if (
|
|
||||||
repl_task
|
|
||||||
and
|
|
||||||
repl
|
|
||||||
):
|
|
||||||
log.pdb(
|
|
||||||
_ctlc_ignore_header
|
|
||||||
+
|
|
||||||
f' by local task\n\n'
|
|
||||||
f'{repl_task}\n'
|
|
||||||
f' |_{repl}\n'
|
|
||||||
)
|
|
||||||
elif req_task:
|
|
||||||
log.debug(
|
|
||||||
_ctlc_ignore_header
|
|
||||||
+
|
|
||||||
f' by local request-task and either,\n'
|
|
||||||
f'- someone else is already REPL-in and has the `Lock`, or\n'
|
|
||||||
f'- some other local task already is replin?\n\n'
|
|
||||||
f'{req_task}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# TODO can we remove this now?
|
|
||||||
# -[ ] does this path ever get hit any more?
|
|
||||||
else:
|
|
||||||
msg: str = (
|
|
||||||
'SIGINT shield handler still active BUT, \n\n'
|
|
||||||
)
|
|
||||||
if repl_task is None:
|
|
||||||
msg += (
|
|
||||||
'- No local task claims to be in debug?\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
if repl is None:
|
|
||||||
msg += (
|
|
||||||
'- No local REPL is currently active?\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
if req_task is None:
|
|
||||||
msg += (
|
|
||||||
'- No debug request task is active?\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
log.warning(
|
|
||||||
msg
|
|
||||||
+
|
|
||||||
'Reverting handler to `trio` default!\n'
|
|
||||||
)
|
|
||||||
DebugStatus.unshield_sigint()
|
|
||||||
|
|
||||||
# XXX ensure that the reverted-to-handler actually is
|
|
||||||
# able to rx what should have been **this** KBI ;)
|
|
||||||
do_cancel()
|
|
||||||
|
|
||||||
# TODO: how to handle the case of an intermediary-child actor
|
|
||||||
# that **is not** marked in debug mode? See oustanding issue:
|
|
||||||
# https://github.com/goodboy/tractor/issues/320
|
|
||||||
# elif debug_mode():
|
|
||||||
|
|
||||||
# maybe redraw/print last REPL output to console since
|
|
||||||
# we want to alert the user that more input is expect since
|
|
||||||
# nothing has been done dur to ignoring sigint.
|
|
||||||
if (
|
|
||||||
DebugStatus.repl # only when current actor has a REPL engaged
|
|
||||||
):
|
|
||||||
flush_status: str = (
|
|
||||||
'Flushing stdout to ensure new prompt line!\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# XXX: yah, mega hack, but how else do we catch this madness XD
|
|
||||||
if (
|
|
||||||
repl.shname == 'xonsh'
|
|
||||||
):
|
|
||||||
flush_status += (
|
|
||||||
'-> ALSO re-flushing due to `xonsh`..\n'
|
|
||||||
)
|
|
||||||
repl.stdout.write(repl.prompt)
|
|
||||||
|
|
||||||
# log.warning(
|
|
||||||
log.devx(
|
|
||||||
flush_status
|
|
||||||
)
|
|
||||||
repl.stdout.flush()
|
|
||||||
|
|
||||||
# TODO: better console UX to match the current "mode":
|
|
||||||
# -[ ] for example if in sticky mode where if there is output
|
|
||||||
# detected as written to the tty we redraw this part underneath
|
|
||||||
# and erase the past draw of this same bit above?
|
|
||||||
# repl.sticky = True
|
|
||||||
# repl._print_if_sticky()
|
|
||||||
|
|
||||||
# also see these links for an approach from `ptk`:
|
|
||||||
# https://github.com/goodboy/tractor/issues/130#issuecomment-663752040
|
|
||||||
# https://github.com/prompt-toolkit/python-prompt-toolkit/blob/c2c6af8a0308f9e5d7c0e28cb8a02963fe0ce07a/prompt_toolkit/patch_stdout.py
|
|
||||||
else:
|
|
||||||
log.devx(
|
|
||||||
# log.warning(
|
|
||||||
'Not flushing stdout since not needed?\n'
|
|
||||||
f'|_{repl}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
# XXX only for tracing this handler
|
|
||||||
log.devx('exiting SIGINT')
|
|
||||||
|
|
@ -1,220 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or
|
|
||||||
# modify it under the terms of the GNU Affero General Public License
|
|
||||||
# as published by the Free Software Foundation, either version 3 of
|
|
||||||
# the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful, but
|
|
||||||
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
# Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public
|
|
||||||
# License along with this program. If not, see
|
|
||||||
# <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
'''
|
|
||||||
Debugger synchronization APIs to ensure orderly access and
|
|
||||||
non-TTY-clobbering graceful teardown.
|
|
||||||
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
from contextlib import (
|
|
||||||
asynccontextmanager as acm,
|
|
||||||
)
|
|
||||||
from functools import (
|
|
||||||
partial,
|
|
||||||
)
|
|
||||||
from typing import (
|
|
||||||
AsyncGenerator,
|
|
||||||
Callable,
|
|
||||||
)
|
|
||||||
|
|
||||||
from tractor.log import get_logger
|
|
||||||
import trio
|
|
||||||
from trio.lowlevel import (
|
|
||||||
current_task,
|
|
||||||
Task,
|
|
||||||
)
|
|
||||||
from tractor._context import Context
|
|
||||||
from tractor._state import (
|
|
||||||
current_actor,
|
|
||||||
debug_mode,
|
|
||||||
is_root_process,
|
|
||||||
)
|
|
||||||
from ._repl import (
|
|
||||||
TractorConfig as TractorConfig,
|
|
||||||
)
|
|
||||||
from ._tty_lock import (
|
|
||||||
Lock,
|
|
||||||
request_root_stdio_lock,
|
|
||||||
any_connected_locker_child,
|
|
||||||
)
|
|
||||||
from ._sigint import (
|
|
||||||
sigint_shield as sigint_shield,
|
|
||||||
_ctlc_ignore_header as _ctlc_ignore_header
|
|
||||||
)
|
|
||||||
|
|
||||||
log = get_logger(__package__)
|
|
||||||
|
|
||||||
|
|
||||||
async def maybe_wait_for_debugger(
|
|
||||||
poll_steps: int = 2,
|
|
||||||
poll_delay: float = 0.1,
|
|
||||||
child_in_debug: bool = False,
|
|
||||||
|
|
||||||
header_msg: str = '',
|
|
||||||
_ll: str = 'devx',
|
|
||||||
|
|
||||||
) -> bool: # was locked and we polled?
|
|
||||||
|
|
||||||
if (
|
|
||||||
not debug_mode()
|
|
||||||
and
|
|
||||||
not child_in_debug
|
|
||||||
):
|
|
||||||
return False
|
|
||||||
|
|
||||||
logmeth: Callable = getattr(log, _ll)
|
|
||||||
|
|
||||||
msg: str = header_msg
|
|
||||||
if (
|
|
||||||
is_root_process()
|
|
||||||
):
|
|
||||||
# If we error in the root but the debugger is
|
|
||||||
# engaged we don't want to prematurely kill (and
|
|
||||||
# thus clobber access to) the local tty since it
|
|
||||||
# will make the pdb repl unusable.
|
|
||||||
# Instead try to wait for pdb to be released before
|
|
||||||
# tearing down.
|
|
||||||
ctx_in_debug: Context|None = Lock.ctx_in_debug
|
|
||||||
in_debug: tuple[str, str]|None = (
|
|
||||||
ctx_in_debug.chan.uid
|
|
||||||
if ctx_in_debug
|
|
||||||
else None
|
|
||||||
)
|
|
||||||
if in_debug == current_actor().uid:
|
|
||||||
log.debug(
|
|
||||||
msg
|
|
||||||
+
|
|
||||||
'Root already owns the TTY LOCK'
|
|
||||||
)
|
|
||||||
return True
|
|
||||||
|
|
||||||
elif in_debug:
|
|
||||||
msg += (
|
|
||||||
f'Debug `Lock` in use by subactor\n|\n|_{in_debug}\n'
|
|
||||||
)
|
|
||||||
# TODO: could this make things more deterministic?
|
|
||||||
# wait to see if a sub-actor task will be
|
|
||||||
# scheduled and grab the tty lock on the next
|
|
||||||
# tick?
|
|
||||||
# XXX => but it doesn't seem to work..
|
|
||||||
# await trio.testing.wait_all_tasks_blocked(cushion=0)
|
|
||||||
else:
|
|
||||||
logmeth(
|
|
||||||
msg
|
|
||||||
+
|
|
||||||
'Root immediately acquired debug TTY LOCK'
|
|
||||||
)
|
|
||||||
return False
|
|
||||||
|
|
||||||
for istep in range(poll_steps):
|
|
||||||
if (
|
|
||||||
Lock.req_handler_finished is not None
|
|
||||||
and not Lock.req_handler_finished.is_set()
|
|
||||||
and in_debug is not None
|
|
||||||
):
|
|
||||||
# caller_frame_info: str = pformat_caller_frame()
|
|
||||||
logmeth(
|
|
||||||
msg
|
|
||||||
+
|
|
||||||
'\n^^ Root is waiting on tty lock release.. ^^\n'
|
|
||||||
# f'{caller_frame_info}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
if not any_connected_locker_child():
|
|
||||||
Lock.get_locking_task_cs().cancel()
|
|
||||||
|
|
||||||
with trio.CancelScope(shield=True):
|
|
||||||
await Lock.req_handler_finished.wait()
|
|
||||||
|
|
||||||
log.devx(
|
|
||||||
f'Subactor released debug lock\n'
|
|
||||||
f'|_{in_debug}\n'
|
|
||||||
)
|
|
||||||
break
|
|
||||||
|
|
||||||
# is no subactor locking debugger currently?
|
|
||||||
if (
|
|
||||||
in_debug is None
|
|
||||||
and (
|
|
||||||
Lock.req_handler_finished is None
|
|
||||||
or Lock.req_handler_finished.is_set()
|
|
||||||
)
|
|
||||||
):
|
|
||||||
logmeth(
|
|
||||||
msg
|
|
||||||
+
|
|
||||||
'Root acquired tty lock!'
|
|
||||||
)
|
|
||||||
break
|
|
||||||
|
|
||||||
else:
|
|
||||||
logmeth(
|
|
||||||
'Root polling for debug:\n'
|
|
||||||
f'poll step: {istep}\n'
|
|
||||||
f'poll delya: {poll_delay}\n\n'
|
|
||||||
f'{Lock.repr()}\n'
|
|
||||||
)
|
|
||||||
with trio.CancelScope(shield=True):
|
|
||||||
await trio.sleep(poll_delay)
|
|
||||||
continue
|
|
||||||
|
|
||||||
return True
|
|
||||||
|
|
||||||
# else:
|
|
||||||
# # TODO: non-root call for #320?
|
|
||||||
# this_uid: tuple[str, str] = current_actor().uid
|
|
||||||
# async with acquire_debug_lock(
|
|
||||||
# subactor_uid=this_uid,
|
|
||||||
# ):
|
|
||||||
# pass
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
|
||||||
async def acquire_debug_lock(
|
|
||||||
subactor_uid: tuple[str, str],
|
|
||||||
) -> AsyncGenerator[
|
|
||||||
trio.CancelScope|None,
|
|
||||||
tuple,
|
|
||||||
]:
|
|
||||||
'''
|
|
||||||
Request to acquire the TTY `Lock` in the root actor, release on
|
|
||||||
exit.
|
|
||||||
|
|
||||||
This helper is for actor's who don't actually need to acquired
|
|
||||||
the debugger but want to wait until the lock is free in the
|
|
||||||
process-tree root such that they don't clobber an ongoing pdb
|
|
||||||
REPL session in some peer or child!
|
|
||||||
|
|
||||||
'''
|
|
||||||
if not debug_mode():
|
|
||||||
yield None
|
|
||||||
return
|
|
||||||
|
|
||||||
task: Task = current_task()
|
|
||||||
async with trio.open_nursery() as n:
|
|
||||||
ctx: Context = await n.start(
|
|
||||||
partial(
|
|
||||||
request_root_stdio_lock,
|
|
||||||
actor_uid=subactor_uid,
|
|
||||||
task_uid=(task.name, id(task)),
|
|
||||||
)
|
|
||||||
)
|
|
||||||
yield ctx
|
|
||||||
ctx.cancel()
|
|
||||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
|
|
@ -15,13 +15,10 @@
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
Pretty formatters for use throughout our internals.
|
Pretty formatters for use throughout the code base.
|
||||||
|
Mostly handy for logging and exception message content.
|
||||||
Handy for logging and exception message content but also for `repr()`
|
|
||||||
in REPL(s).
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
import sys
|
|
||||||
import textwrap
|
import textwrap
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
|
|
@ -118,85 +115,6 @@ def pformat_boxed_tb(
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def pformat_exc(
|
|
||||||
exc: Exception,
|
|
||||||
header: str = '',
|
|
||||||
message: str = '',
|
|
||||||
body: str = '',
|
|
||||||
with_type_header: bool = True,
|
|
||||||
) -> str:
|
|
||||||
|
|
||||||
# XXX when the currently raised exception is this instance,
|
|
||||||
# we do not ever use the "type header" style repr.
|
|
||||||
is_being_raised: bool = False
|
|
||||||
if (
|
|
||||||
(curr_exc := sys.exception())
|
|
||||||
and
|
|
||||||
curr_exc is exc
|
|
||||||
):
|
|
||||||
is_being_raised: bool = True
|
|
||||||
|
|
||||||
with_type_header: bool = (
|
|
||||||
with_type_header
|
|
||||||
and
|
|
||||||
not is_being_raised
|
|
||||||
)
|
|
||||||
|
|
||||||
# <RemoteActorError( .. )> style
|
|
||||||
if (
|
|
||||||
with_type_header
|
|
||||||
and
|
|
||||||
not header
|
|
||||||
):
|
|
||||||
header: str = f'<{type(exc).__name__}('
|
|
||||||
|
|
||||||
message: str = (
|
|
||||||
message
|
|
||||||
or
|
|
||||||
exc.message
|
|
||||||
)
|
|
||||||
if message:
|
|
||||||
# split off the first line so, if needed, it isn't
|
|
||||||
# indented the same like the "boxed content" which
|
|
||||||
# since there is no `.tb_str` is just the `.message`.
|
|
||||||
lines: list[str] = message.splitlines()
|
|
||||||
first: str = lines[0]
|
|
||||||
message: str = message.removeprefix(first)
|
|
||||||
|
|
||||||
# with a type-style header we,
|
|
||||||
# - have no special message "first line" extraction/handling
|
|
||||||
# - place the message a space in from the header:
|
|
||||||
# `MsgTypeError( <message> ..`
|
|
||||||
# ^-here
|
|
||||||
# - indent the `.message` inside the type body.
|
|
||||||
if with_type_header:
|
|
||||||
first = f' {first} )>'
|
|
||||||
|
|
||||||
message: str = textwrap.indent(
|
|
||||||
message,
|
|
||||||
prefix=' '*2,
|
|
||||||
)
|
|
||||||
message: str = first + message
|
|
||||||
|
|
||||||
tail: str = ''
|
|
||||||
if (
|
|
||||||
with_type_header
|
|
||||||
and
|
|
||||||
not message
|
|
||||||
):
|
|
||||||
tail: str = '>'
|
|
||||||
|
|
||||||
return (
|
|
||||||
header
|
|
||||||
+
|
|
||||||
message
|
|
||||||
+
|
|
||||||
f'{body}'
|
|
||||||
+
|
|
||||||
tail
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def pformat_caller_frame(
|
def pformat_caller_frame(
|
||||||
stack_limit: int = 1,
|
stack_limit: int = 1,
|
||||||
box_tb: bool = True,
|
box_tb: bool = True,
|
||||||
|
|
@ -226,8 +144,8 @@ def pformat_cs(
|
||||||
field_prefix: str = ' |_',
|
field_prefix: str = ' |_',
|
||||||
) -> str:
|
) -> str:
|
||||||
'''
|
'''
|
||||||
Pretty format info about a `trio.CancelScope` including most of
|
Pretty format info about a `trio.CancelScope` including most
|
||||||
its public state and `._cancel_status`.
|
of its public state and `._cancel_status`.
|
||||||
|
|
||||||
The output can be modified to show a "var name" for the
|
The output can be modified to show a "var name" for the
|
||||||
instance as a field prefix, just a simple str before each
|
instance as a field prefix, just a simple str before each
|
||||||
|
|
@ -249,279 +167,3 @@ def pformat_cs(
|
||||||
+
|
+
|
||||||
fields
|
fields
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def nest_from_op(
|
|
||||||
input_op: str, # TODO, Literal of all op-"symbols" from below?
|
|
||||||
text: str,
|
|
||||||
prefix_op: bool = True, # unset is to suffix the first line
|
|
||||||
# optionally suffix `text`, by def on a newline
|
|
||||||
op_suffix='\n',
|
|
||||||
|
|
||||||
nest_prefix: str = '|_',
|
|
||||||
nest_indent: int|None = None,
|
|
||||||
# XXX indent `next_prefix` "to-the-right-of" `input_op`
|
|
||||||
# by this count of whitespaces (' ').
|
|
||||||
rm_from_first_ln: str|None = None,
|
|
||||||
|
|
||||||
) -> str:
|
|
||||||
'''
|
|
||||||
Depth-increment the input (presumably hierarchy/supervision)
|
|
||||||
input "tree string" below the provided `input_op` execution
|
|
||||||
operator, so injecting a `"\n|_{input_op}\n"`and indenting the
|
|
||||||
`tree_str` to nest content aligned with the ops last char.
|
|
||||||
|
|
||||||
'''
|
|
||||||
# `sclang` "structurred-concurrency-language": an ascii-encoded
|
|
||||||
# symbolic alphabet to describe concurrent systems.
|
|
||||||
#
|
|
||||||
# ?TODO? aa more fomal idea for a syntax to the state of
|
|
||||||
# concurrent systems as a "3-domain" (execution, scope, storage)
|
|
||||||
# model and using a minimal ascii/utf-8 operator-set.
|
|
||||||
#
|
|
||||||
# try not to take any of this seriously yet XD
|
|
||||||
#
|
|
||||||
# > is a "play operator" indicating (CPU bound)
|
|
||||||
# exec/work/ops required at the "lowest level computing"
|
|
||||||
#
|
|
||||||
# execution primititves (tasks, threads, actors..) denote their
|
|
||||||
# lifetime with '(' and ')' since parentheses normally are used
|
|
||||||
# in many langs to denote function calls.
|
|
||||||
#
|
|
||||||
# starting = (
|
|
||||||
# >( opening/starting; beginning of the thread-of-exec (toe?)
|
|
||||||
# (> opened/started, (finished spawning toe)
|
|
||||||
# |_<Task: blah blah..> repr of toe, in py these look like <objs>
|
|
||||||
#
|
|
||||||
# >) closing/exiting/stopping,
|
|
||||||
# )> closed/exited/stopped,
|
|
||||||
# |_<Task: blah blah..>
|
|
||||||
# [OR <), )< ?? ]
|
|
||||||
#
|
|
||||||
# ending = )
|
|
||||||
# >c) cancelling to close/exit
|
|
||||||
# c)> cancelled (caused close), OR?
|
|
||||||
# |_<Actor: ..>
|
|
||||||
# OR maybe "<c)" which better indicates the cancel being
|
|
||||||
# "delivered/returned" / returned" to LHS?
|
|
||||||
#
|
|
||||||
# >x) erroring to eventuall exit
|
|
||||||
# x)> errored and terminated
|
|
||||||
# |_<Actor: ...>
|
|
||||||
#
|
|
||||||
# scopes: supers/nurseries, IPC-ctxs, sessions, perms, etc.
|
|
||||||
# >{ opening
|
|
||||||
# {> opened
|
|
||||||
# }> closed
|
|
||||||
# >} closing
|
|
||||||
#
|
|
||||||
# storage: like queues, shm-buffers, files, etc..
|
|
||||||
# >[ opening
|
|
||||||
# [> opened
|
|
||||||
# |_<FileObj: ..>
|
|
||||||
#
|
|
||||||
# >] closing
|
|
||||||
# ]> closed
|
|
||||||
|
|
||||||
# IPC ops: channels, transports, msging
|
|
||||||
# => req msg
|
|
||||||
# <= resp msg
|
|
||||||
# <=> 2-way streaming (of msgs)
|
|
||||||
# <- recv 1 msg
|
|
||||||
# -> send 1 msg
|
|
||||||
#
|
|
||||||
# TODO: still not sure on R/L-HS approach..?
|
|
||||||
# =>( send-req to exec start (task, actor, thread..)
|
|
||||||
# (<= recv-req to ^
|
|
||||||
#
|
|
||||||
# (<= recv-req ^
|
|
||||||
# <=( recv-resp opened remote exec primitive
|
|
||||||
# <=) recv-resp closed
|
|
||||||
#
|
|
||||||
# )<=c req to stop due to cancel
|
|
||||||
# c=>) req to stop due to cancel
|
|
||||||
#
|
|
||||||
# =>{ recv-req to open
|
|
||||||
# <={ send-status that it closed
|
|
||||||
#
|
|
||||||
if (
|
|
||||||
nest_prefix
|
|
||||||
and
|
|
||||||
nest_indent != 0
|
|
||||||
):
|
|
||||||
if nest_indent is not None:
|
|
||||||
nest_prefix: str = textwrap.indent(
|
|
||||||
nest_prefix,
|
|
||||||
prefix=nest_indent*' ',
|
|
||||||
)
|
|
||||||
nest_indent: int = len(nest_prefix)
|
|
||||||
|
|
||||||
# determine body-text indent either by,
|
|
||||||
# - using wtv explicit indent value is provided,
|
|
||||||
# OR
|
|
||||||
# - auto-calcing the indent to embed `text` under
|
|
||||||
# the `nest_prefix` if provided, **IFF** `nest_indent=None`.
|
|
||||||
tree_str_indent: int = 0
|
|
||||||
if nest_indent not in {0, None}:
|
|
||||||
tree_str_indent = nest_indent
|
|
||||||
elif (
|
|
||||||
nest_prefix
|
|
||||||
and
|
|
||||||
nest_indent != 0
|
|
||||||
):
|
|
||||||
tree_str_indent = len(nest_prefix)
|
|
||||||
|
|
||||||
indented_tree_str: str = text
|
|
||||||
if tree_str_indent:
|
|
||||||
indented_tree_str: str = textwrap.indent(
|
|
||||||
text,
|
|
||||||
prefix=' '*tree_str_indent,
|
|
||||||
)
|
|
||||||
|
|
||||||
# inject any provided nesting-prefix chars
|
|
||||||
# into the head of the first line.
|
|
||||||
if nest_prefix:
|
|
||||||
indented_tree_str: str = (
|
|
||||||
f'{nest_prefix}{indented_tree_str[tree_str_indent:]}'
|
|
||||||
)
|
|
||||||
|
|
||||||
if (
|
|
||||||
not prefix_op
|
|
||||||
or
|
|
||||||
rm_from_first_ln
|
|
||||||
):
|
|
||||||
tree_lns: list[str] = indented_tree_str.splitlines()
|
|
||||||
first: str = tree_lns[0]
|
|
||||||
if rm_from_first_ln:
|
|
||||||
first = first.strip().replace(
|
|
||||||
rm_from_first_ln,
|
|
||||||
'',
|
|
||||||
)
|
|
||||||
indented_tree_str: str = '\n'.join(tree_lns[1:])
|
|
||||||
|
|
||||||
if prefix_op:
|
|
||||||
indented_tree_str = (
|
|
||||||
f'{first}\n'
|
|
||||||
f'{indented_tree_str}'
|
|
||||||
)
|
|
||||||
|
|
||||||
if prefix_op:
|
|
||||||
return (
|
|
||||||
f'{input_op}{op_suffix}'
|
|
||||||
f'{indented_tree_str}'
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
return (
|
|
||||||
f'{first}{input_op}{op_suffix}'
|
|
||||||
f'{indented_tree_str}'
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# ------ modden.repr ------
|
|
||||||
# XXX originally taken verbaatim from `modden.repr`
|
|
||||||
'''
|
|
||||||
More "multi-line" representation then the stdlib's `pprint` equivs.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from inspect import (
|
|
||||||
FrameInfo,
|
|
||||||
stack,
|
|
||||||
)
|
|
||||||
import pprint
|
|
||||||
import reprlib
|
|
||||||
from typing import (
|
|
||||||
Callable,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def mk_repr(
|
|
||||||
**repr_kws,
|
|
||||||
) -> Callable[[str], str]:
|
|
||||||
'''
|
|
||||||
Allocate and deliver a `repr.Repr` instance with provided input
|
|
||||||
settings using the std-lib's `reprlib` mod,
|
|
||||||
* https://docs.python.org/3/library/reprlib.html
|
|
||||||
|
|
||||||
------ Ex. ------
|
|
||||||
An up to 6-layer-nested `dict` as multi-line:
|
|
||||||
- https://stackoverflow.com/a/79102479
|
|
||||||
- https://docs.python.org/3/library/reprlib.html#reprlib.Repr.maxlevel
|
|
||||||
|
|
||||||
'''
|
|
||||||
def_kws: dict[str, int] = dict(
|
|
||||||
indent=3, # indent used for repr of recursive objects
|
|
||||||
maxlevel=616, # recursion levels
|
|
||||||
maxdict=616, # max items shown for `dict`
|
|
||||||
maxlist=616, # max items shown for `dict`
|
|
||||||
maxstring=616, # match editor line-len limit
|
|
||||||
maxtuple=616, # match editor line-len limit
|
|
||||||
maxother=616, # match editor line-len limit
|
|
||||||
)
|
|
||||||
def_kws |= repr_kws
|
|
||||||
reprr = reprlib.Repr(**def_kws)
|
|
||||||
return reprr.repr
|
|
||||||
|
|
||||||
|
|
||||||
def ppfmt(
|
|
||||||
obj: object,
|
|
||||||
do_print: bool = False,
|
|
||||||
) -> str:
|
|
||||||
'''
|
|
||||||
The `pprint.pformat()` version of `pprint.pp()`, namely
|
|
||||||
a default `sort_dicts=False`.. (which i think should be
|
|
||||||
the normal default in the stdlib).
|
|
||||||
|
|
||||||
'''
|
|
||||||
pprepr: Callable = mk_repr()
|
|
||||||
repr_str: str = pprepr(obj)
|
|
||||||
|
|
||||||
if do_print:
|
|
||||||
return pprint.pp(repr_str)
|
|
||||||
|
|
||||||
return repr_str
|
|
||||||
|
|
||||||
|
|
||||||
pformat = ppfmt
|
|
||||||
|
|
||||||
|
|
||||||
def pfmt_frame_info(fi: FrameInfo) -> str:
|
|
||||||
'''
|
|
||||||
Like a std `inspect.FrameInfo.__repr__()` but multi-line..
|
|
||||||
|
|
||||||
'''
|
|
||||||
return (
|
|
||||||
'FrameInfo(\n'
|
|
||||||
' frame={!r},\n'
|
|
||||||
' filename={!r},\n'
|
|
||||||
' lineno={!r},\n'
|
|
||||||
' function={!r},\n'
|
|
||||||
' code_context={!r},\n'
|
|
||||||
' index={!r},\n'
|
|
||||||
' positions={!r})'
|
|
||||||
).format(
|
|
||||||
fi.frame,
|
|
||||||
fi.filename,
|
|
||||||
fi.lineno,
|
|
||||||
fi.function,
|
|
||||||
fi.code_context,
|
|
||||||
fi.index,
|
|
||||||
fi.positions
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def pfmt_callstack(frames: int = 1) -> str:
|
|
||||||
'''
|
|
||||||
Generate a string of nested `inspect.FrameInfo` objects returned
|
|
||||||
from a `inspect.stack()` call such that only the `.frame` field
|
|
||||||
for each layer is pprinted.
|
|
||||||
|
|
||||||
'''
|
|
||||||
caller_frames: list[FrameInfo] = stack()[1:1+frames]
|
|
||||||
frames_str: str = ''
|
|
||||||
for i, frame_info in enumerate(caller_frames):
|
|
||||||
frames_str += textwrap.indent(
|
|
||||||
f'{frame_info.frame!r}\n',
|
|
||||||
prefix=' '*i,
|
|
||||||
|
|
||||||
)
|
|
||||||
return frames_str
|
|
||||||
|
|
|
||||||
|
|
@ -45,8 +45,6 @@ __all__ = ['pub']
|
||||||
log = get_logger('messaging')
|
log = get_logger('messaging')
|
||||||
|
|
||||||
|
|
||||||
# TODO! this needs to reworked to use the modern
|
|
||||||
# `Context`/`MsgStream` APIs!!
|
|
||||||
async def fan_out_to_ctxs(
|
async def fan_out_to_ctxs(
|
||||||
pub_async_gen_func: typing.Callable, # it's an async gen ... gd mypy
|
pub_async_gen_func: typing.Callable, # it's an async gen ... gd mypy
|
||||||
topics2ctxs: dict[str, list],
|
topics2ctxs: dict[str, list],
|
||||||
|
|
|
||||||
|
|
@ -1,5 +1,5 @@
|
||||||
# tractor: structured concurrent "actors".
|
# tractor: structured concurrent "actors".
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
# Copyright 2024-eternity Tyler Goodlet.
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
# This program is free software: you can redistribute it and/or modify
|
||||||
# it under the terms of the GNU Affero General Public License as published by
|
# it under the terms of the GNU Affero General Public License as published by
|
||||||
|
|
@ -15,10 +15,12 @@
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
A modular IPC layer supporting the power of cross-process SC!
|
High level design patterns, APIs and runtime extensions built on top
|
||||||
|
of the `tractor` runtime core.
|
||||||
|
|
||||||
'''
|
'''
|
||||||
from ._chan import (
|
from ._service import (
|
||||||
_connect_chan as _connect_chan,
|
open_service_mngr as open_service_mngr,
|
||||||
Channel as Channel
|
get_service_mngr as get_service_mngr,
|
||||||
|
ServiceMngr as ServiceMngr,
|
||||||
)
|
)
|
||||||
|
|
@ -0,0 +1,592 @@
|
||||||
|
# tractor: structured concurrent "actors".
|
||||||
|
# Copyright 2024-eternity Tyler Goodlet.
|
||||||
|
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU Affero General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU Affero General Public License for more details.
|
||||||
|
|
||||||
|
# You should have received a copy of the GNU Affero General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
'''
|
||||||
|
Daemon subactor as service(s) management and supervision primitives
|
||||||
|
and API.
|
||||||
|
|
||||||
|
'''
|
||||||
|
from __future__ import annotations
|
||||||
|
from contextlib import (
|
||||||
|
asynccontextmanager as acm,
|
||||||
|
# contextmanager as cm,
|
||||||
|
)
|
||||||
|
from collections import defaultdict
|
||||||
|
from dataclasses import (
|
||||||
|
dataclass,
|
||||||
|
field,
|
||||||
|
)
|
||||||
|
import functools
|
||||||
|
import inspect
|
||||||
|
from typing import (
|
||||||
|
Callable,
|
||||||
|
Any,
|
||||||
|
)
|
||||||
|
|
||||||
|
import tractor
|
||||||
|
import trio
|
||||||
|
from trio import TaskStatus
|
||||||
|
from tractor import (
|
||||||
|
log,
|
||||||
|
ActorNursery,
|
||||||
|
current_actor,
|
||||||
|
ContextCancelled,
|
||||||
|
Context,
|
||||||
|
Portal,
|
||||||
|
)
|
||||||
|
|
||||||
|
log = log.get_logger('tractor')
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: implement a `@singleton` deco-API for wrapping the below
|
||||||
|
# factory's impl for general actor-singleton use?
|
||||||
|
#
|
||||||
|
# -[ ] go through the options peeps on SO did?
|
||||||
|
# * https://stackoverflow.com/questions/6760685/what-is-the-best-way-of-implementing-singleton-in-python
|
||||||
|
# * including @mikenerone's answer
|
||||||
|
# |_https://stackoverflow.com/questions/6760685/what-is-the-best-way-of-implementing-singleton-in-python/39186313#39186313
|
||||||
|
#
|
||||||
|
# -[ ] put it in `tractor.lowlevel._globals` ?
|
||||||
|
# * fits with our oustanding actor-local/global feat req?
|
||||||
|
# |_ https://github.com/goodboy/tractor/issues/55
|
||||||
|
# * how can it relate to the `Actor.lifetime_stack` that was
|
||||||
|
# silently patched in?
|
||||||
|
# |_ we could implicitly call both of these in the same
|
||||||
|
# spot in the runtime using the lifetime stack?
|
||||||
|
# - `open_singleton_cm().__exit__()`
|
||||||
|
# -`del_singleton()`
|
||||||
|
# |_ gives SC fixtue semantics to sync code oriented around
|
||||||
|
# sub-process lifetime?
|
||||||
|
# * what about with `trio.RunVar`?
|
||||||
|
# |_https://trio.readthedocs.io/en/stable/reference-lowlevel.html#trio.lowlevel.RunVar
|
||||||
|
# - which we'll need for no-GIL cpython (right?) presuming
|
||||||
|
# multiple `trio.run()` calls in process?
|
||||||
|
#
|
||||||
|
#
|
||||||
|
# @singleton
|
||||||
|
# async def open_service_mngr(
|
||||||
|
# **init_kwargs,
|
||||||
|
# ) -> ServiceMngr:
|
||||||
|
# '''
|
||||||
|
# Note this function body is invoke IFF no existing singleton instance already
|
||||||
|
# exists in this proc's memory.
|
||||||
|
|
||||||
|
# '''
|
||||||
|
# # setup
|
||||||
|
# yield ServiceMngr(**init_kwargs)
|
||||||
|
# # teardown
|
||||||
|
|
||||||
|
|
||||||
|
# a deletion API for explicit instance de-allocation?
|
||||||
|
# @open_service_mngr.deleter
|
||||||
|
# def del_service_mngr() -> None:
|
||||||
|
# mngr = open_service_mngr._singleton[0]
|
||||||
|
# open_service_mngr._singleton[0] = None
|
||||||
|
# del mngr
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: implement a singleton deco-API for wrapping the below
|
||||||
|
# factory's impl for general actor-singleton use?
|
||||||
|
#
|
||||||
|
# @singleton
|
||||||
|
# async def open_service_mngr(
|
||||||
|
# **init_kwargs,
|
||||||
|
# ) -> ServiceMngr:
|
||||||
|
# '''
|
||||||
|
# Note this function body is invoke IFF no existing singleton instance already
|
||||||
|
# exists in this proc's memory.
|
||||||
|
|
||||||
|
# '''
|
||||||
|
# # setup
|
||||||
|
# yield ServiceMngr(**init_kwargs)
|
||||||
|
# # teardown
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: singleton factory API instead of a class API
|
||||||
|
@acm
|
||||||
|
async def open_service_mngr(
|
||||||
|
*,
|
||||||
|
debug_mode: bool = False,
|
||||||
|
|
||||||
|
# NOTE; since default values for keyword-args are effectively
|
||||||
|
# module-vars/globals as per the note from,
|
||||||
|
# https://docs.python.org/3/tutorial/controlflow.html#default-argument-values
|
||||||
|
#
|
||||||
|
# > "The default value is evaluated only once. This makes
|
||||||
|
# a difference when the default is a mutable object such as
|
||||||
|
# a list, dictionary, or instances of most classes"
|
||||||
|
#
|
||||||
|
_singleton: list[ServiceMngr|None] = [None],
|
||||||
|
**init_kwargs,
|
||||||
|
|
||||||
|
) -> ServiceMngr:
|
||||||
|
'''
|
||||||
|
Open an actor-global "service-manager" for supervising a tree
|
||||||
|
of subactors and/or actor-global tasks.
|
||||||
|
|
||||||
|
The delivered `ServiceMngr` is singleton instance for each
|
||||||
|
actor-process, that is, allocated on first open and never
|
||||||
|
de-allocated unless explicitly deleted by al call to
|
||||||
|
`del_service_mngr()`.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# TODO: factor this an allocation into
|
||||||
|
# a `._mngr.open_service_mngr()` and put in the
|
||||||
|
# once-n-only-once setup/`.__aenter__()` part!
|
||||||
|
# -[ ] how to make this only happen on the `mngr == None` case?
|
||||||
|
# |_ use `.trionics.maybe_open_context()` (for generic
|
||||||
|
# async-with-style-only-once of the factory impl, though
|
||||||
|
# what do we do for the allocation case?
|
||||||
|
# / `.maybe_open_nursery()` (since for this specific case
|
||||||
|
# it's simpler?) to activate
|
||||||
|
async with (
|
||||||
|
tractor.open_nursery() as an,
|
||||||
|
trio.open_nursery() as tn,
|
||||||
|
):
|
||||||
|
# impl specific obvi..
|
||||||
|
init_kwargs.update({
|
||||||
|
'an': an,
|
||||||
|
'tn': tn,
|
||||||
|
})
|
||||||
|
|
||||||
|
mngr: ServiceMngr|None
|
||||||
|
if (mngr := _singleton[0]) is None:
|
||||||
|
|
||||||
|
log.info('Allocating a new service mngr!')
|
||||||
|
mngr = _singleton[0] = ServiceMngr(**init_kwargs)
|
||||||
|
|
||||||
|
# TODO: put into `.__aenter__()` section of
|
||||||
|
# eventual `@singleton_acm` API wrapper.
|
||||||
|
#
|
||||||
|
# assign globally for future daemon/task creation
|
||||||
|
mngr.an = an
|
||||||
|
mngr.tn = tn
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert (mngr.an and mngr.tn)
|
||||||
|
log.info(
|
||||||
|
'Using extant service mngr!\n\n'
|
||||||
|
f'{mngr!r}\n' # it has a nice `.__repr__()` of services state
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
# NOTE: this is a singleton factory impl specific detail
|
||||||
|
# which should be supported in the condensed
|
||||||
|
# `@singleton_acm` API?
|
||||||
|
mngr.debug_mode = debug_mode
|
||||||
|
|
||||||
|
yield mngr
|
||||||
|
finally:
|
||||||
|
# TODO: is this more clever/efficient?
|
||||||
|
# if 'samplerd' in mngr.service_ctxs:
|
||||||
|
# await mngr.cancel_service('samplerd')
|
||||||
|
tn.cancel_scope.cancel()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def get_service_mngr() -> ServiceMngr:
|
||||||
|
'''
|
||||||
|
Try to get the singleton service-mngr for this actor presuming it
|
||||||
|
has already been allocated using,
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
async with open_<@singleton_acm(func)>() as mngr`
|
||||||
|
... this block kept open ...
|
||||||
|
|
||||||
|
If not yet allocated raise a `ServiceError`.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# https://stackoverflow.com/a/12627202
|
||||||
|
# https://docs.python.org/3/library/inspect.html#inspect.Signature
|
||||||
|
maybe_mngr: ServiceMngr|None = inspect.signature(
|
||||||
|
open_service_mngr
|
||||||
|
).parameters['_singleton'].default[0]
|
||||||
|
|
||||||
|
if maybe_mngr is None:
|
||||||
|
raise RuntimeError(
|
||||||
|
'Someone must allocate a `ServiceMngr` using\n\n'
|
||||||
|
'`async with open_service_mngr()` beforehand!!\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
return maybe_mngr
|
||||||
|
|
||||||
|
|
||||||
|
async def _open_and_supervise_service_ctx(
|
||||||
|
serman: ServiceMngr,
|
||||||
|
name: str,
|
||||||
|
ctx_fn: Callable, # TODO, type for `@tractor.context` requirement
|
||||||
|
portal: Portal,
|
||||||
|
|
||||||
|
allow_overruns: bool = False,
|
||||||
|
task_status: TaskStatus[
|
||||||
|
tuple[
|
||||||
|
trio.CancelScope,
|
||||||
|
Context,
|
||||||
|
trio.Event,
|
||||||
|
Any,
|
||||||
|
]
|
||||||
|
] = trio.TASK_STATUS_IGNORED,
|
||||||
|
**ctx_kwargs,
|
||||||
|
|
||||||
|
) -> Any:
|
||||||
|
'''
|
||||||
|
Open a remote IPC-context defined by `ctx_fn` in the
|
||||||
|
(service) actor accessed via `portal` and supervise the
|
||||||
|
(local) parent task to termination at which point the remote
|
||||||
|
actor runtime is cancelled alongside it.
|
||||||
|
|
||||||
|
The main application is for allocating long-running
|
||||||
|
"sub-services" in a main daemon and explicitly controlling
|
||||||
|
their lifetimes from an actor-global singleton.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# TODO: use the ctx._scope directly here instead?
|
||||||
|
# -[ ] actually what semantics do we expect for this
|
||||||
|
# usage!?
|
||||||
|
with trio.CancelScope() as cs:
|
||||||
|
try:
|
||||||
|
async with portal.open_context(
|
||||||
|
ctx_fn,
|
||||||
|
allow_overruns=allow_overruns,
|
||||||
|
**ctx_kwargs,
|
||||||
|
|
||||||
|
) as (ctx, started):
|
||||||
|
|
||||||
|
# unblock once the remote context has started
|
||||||
|
complete = trio.Event()
|
||||||
|
task_status.started((
|
||||||
|
cs,
|
||||||
|
ctx,
|
||||||
|
complete,
|
||||||
|
started,
|
||||||
|
))
|
||||||
|
log.info(
|
||||||
|
f'`pikerd` service {name} started with value {started}'
|
||||||
|
)
|
||||||
|
# wait on any context's return value
|
||||||
|
# and any final portal result from the
|
||||||
|
# sub-actor.
|
||||||
|
ctx_res: Any = await ctx.wait_for_result()
|
||||||
|
|
||||||
|
# NOTE: blocks indefinitely until cancelled
|
||||||
|
# either by error from the target context
|
||||||
|
# function or by being cancelled here by the
|
||||||
|
# surrounding cancel scope.
|
||||||
|
return (
|
||||||
|
await portal.wait_for_result(),
|
||||||
|
ctx_res,
|
||||||
|
)
|
||||||
|
|
||||||
|
except ContextCancelled as ctxe:
|
||||||
|
canceller: tuple[str, str] = ctxe.canceller
|
||||||
|
our_uid: tuple[str, str] = current_actor().uid
|
||||||
|
if (
|
||||||
|
canceller != portal.chan.uid
|
||||||
|
and
|
||||||
|
canceller != our_uid
|
||||||
|
):
|
||||||
|
log.cancel(
|
||||||
|
f'Actor-service `{name}` was remotely cancelled by a peer?\n'
|
||||||
|
|
||||||
|
# TODO: this would be a good spot to use
|
||||||
|
# a respawn feature Bo
|
||||||
|
f'-> Keeping `pikerd` service manager alive despite this inter-peer cancel\n\n'
|
||||||
|
|
||||||
|
f'cancellee: {portal.chan.uid}\n'
|
||||||
|
f'canceller: {canceller}\n'
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
raise
|
||||||
|
|
||||||
|
finally:
|
||||||
|
# NOTE: the ctx MUST be cancelled first if we
|
||||||
|
# don't want the above `ctx.wait_for_result()` to
|
||||||
|
# raise a self-ctxc. WHY, well since from the ctx's
|
||||||
|
# perspective the cancel request will have
|
||||||
|
# arrived out-out-of-band at the `Actor.cancel()`
|
||||||
|
# level, thus `Context.cancel_called == False`,
|
||||||
|
# meaning `ctx._is_self_cancelled() == False`.
|
||||||
|
# with trio.CancelScope(shield=True):
|
||||||
|
# await ctx.cancel()
|
||||||
|
await portal.cancel_actor() # terminate (remote) sub-actor
|
||||||
|
complete.set() # signal caller this task is done
|
||||||
|
serman.service_ctxs.pop(name) # remove mngr entry
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: we need remote wrapping and a general soln:
|
||||||
|
# - factor this into a ``tractor.highlevel`` extension # pack for the
|
||||||
|
# library.
|
||||||
|
# - wrap a "remote api" wherein you can get a method proxy
|
||||||
|
# to the pikerd actor for starting services remotely!
|
||||||
|
# - prolly rename this to ActorServicesNursery since it spawns
|
||||||
|
# new actors and supervises them to completion?
|
||||||
|
@dataclass
|
||||||
|
class ServiceMngr:
|
||||||
|
'''
|
||||||
|
A multi-subactor-as-service manager.
|
||||||
|
|
||||||
|
Spawn, supervise and monitor service/daemon subactors in a SC
|
||||||
|
process tree.
|
||||||
|
|
||||||
|
'''
|
||||||
|
an: ActorNursery
|
||||||
|
tn: trio.Nursery
|
||||||
|
debug_mode: bool = False # tractor sub-actor debug mode flag
|
||||||
|
|
||||||
|
service_tasks: dict[
|
||||||
|
str,
|
||||||
|
tuple[
|
||||||
|
trio.CancelScope,
|
||||||
|
trio.Event,
|
||||||
|
]
|
||||||
|
] = field(default_factory=dict)
|
||||||
|
|
||||||
|
service_ctxs: dict[
|
||||||
|
str,
|
||||||
|
tuple[
|
||||||
|
trio.CancelScope,
|
||||||
|
Context,
|
||||||
|
Portal,
|
||||||
|
trio.Event,
|
||||||
|
]
|
||||||
|
] = field(default_factory=dict)
|
||||||
|
|
||||||
|
# internal per-service task mutexs
|
||||||
|
_locks = defaultdict(trio.Lock)
|
||||||
|
|
||||||
|
# TODO, unify this interface with our `TaskManager` PR!
|
||||||
|
#
|
||||||
|
#
|
||||||
|
async def start_service_task(
|
||||||
|
self,
|
||||||
|
name: str,
|
||||||
|
# TODO: typevar for the return type of the target and then
|
||||||
|
# use it below for `ctx_res`?
|
||||||
|
fn: Callable,
|
||||||
|
|
||||||
|
allow_overruns: bool = False,
|
||||||
|
**ctx_kwargs,
|
||||||
|
|
||||||
|
) -> tuple[
|
||||||
|
trio.CancelScope,
|
||||||
|
Any,
|
||||||
|
trio.Event,
|
||||||
|
]:
|
||||||
|
async def _task_manager_start(
|
||||||
|
task_status: TaskStatus[
|
||||||
|
tuple[
|
||||||
|
trio.CancelScope,
|
||||||
|
trio.Event,
|
||||||
|
]
|
||||||
|
] = trio.TASK_STATUS_IGNORED,
|
||||||
|
) -> Any:
|
||||||
|
|
||||||
|
task_cs = trio.CancelScope()
|
||||||
|
task_complete = trio.Event()
|
||||||
|
|
||||||
|
with task_cs as cs:
|
||||||
|
task_status.started((
|
||||||
|
cs,
|
||||||
|
task_complete,
|
||||||
|
))
|
||||||
|
try:
|
||||||
|
await fn()
|
||||||
|
except trio.Cancelled as taskc:
|
||||||
|
log.cancel(
|
||||||
|
f'Service task for `{name}` was cancelled!\n'
|
||||||
|
# TODO: this would be a good spot to use
|
||||||
|
# a respawn feature Bo
|
||||||
|
)
|
||||||
|
raise taskc
|
||||||
|
finally:
|
||||||
|
task_complete.set()
|
||||||
|
(
|
||||||
|
cs,
|
||||||
|
complete,
|
||||||
|
) = await self.tn.start(_task_manager_start)
|
||||||
|
|
||||||
|
# store the cancel scope and portal for later cancellation or
|
||||||
|
# retstart if needed.
|
||||||
|
self.service_tasks[name] = (
|
||||||
|
cs,
|
||||||
|
complete,
|
||||||
|
)
|
||||||
|
return (
|
||||||
|
cs,
|
||||||
|
complete,
|
||||||
|
)
|
||||||
|
|
||||||
|
async def cancel_service_task(
|
||||||
|
self,
|
||||||
|
name: str,
|
||||||
|
|
||||||
|
) -> Any:
|
||||||
|
log.info(f'Cancelling `pikerd` service {name}')
|
||||||
|
cs, complete = self.service_tasks[name]
|
||||||
|
|
||||||
|
cs.cancel()
|
||||||
|
await complete.wait()
|
||||||
|
# TODO, if we use the `TaskMngr` from #346
|
||||||
|
# we can also get the return value from the task!
|
||||||
|
|
||||||
|
if name in self.service_tasks:
|
||||||
|
# TODO: custom err?
|
||||||
|
# raise ServiceError(
|
||||||
|
raise RuntimeError(
|
||||||
|
f'Service task {name!r} not terminated!?\n'
|
||||||
|
)
|
||||||
|
|
||||||
|
async def start_service_ctx(
|
||||||
|
self,
|
||||||
|
name: str,
|
||||||
|
portal: Portal,
|
||||||
|
# TODO: typevar for the return type of the target and then
|
||||||
|
# use it below for `ctx_res`?
|
||||||
|
ctx_fn: Callable,
|
||||||
|
**ctx_kwargs,
|
||||||
|
|
||||||
|
) -> tuple[
|
||||||
|
trio.CancelScope,
|
||||||
|
Context,
|
||||||
|
Any,
|
||||||
|
]:
|
||||||
|
'''
|
||||||
|
Start a remote IPC-context defined by `ctx_fn` in a background
|
||||||
|
task and immediately return supervision primitives to manage it:
|
||||||
|
|
||||||
|
- a `cs: CancelScope` for the newly allocated bg task
|
||||||
|
- the `ipc_ctx: Context` to manage the remotely scheduled
|
||||||
|
`trio.Task`.
|
||||||
|
- the `started: Any` value returned by the remote endpoint
|
||||||
|
task's `Context.started(<value>)` call.
|
||||||
|
|
||||||
|
The bg task supervises the ctx such that when it terminates the supporting
|
||||||
|
actor runtime is also cancelled, see `_open_and_supervise_service_ctx()`
|
||||||
|
for details.
|
||||||
|
|
||||||
|
'''
|
||||||
|
cs, ipc_ctx, complete, started = await self.tn.start(
|
||||||
|
functools.partial(
|
||||||
|
_open_and_supervise_service_ctx,
|
||||||
|
serman=self,
|
||||||
|
name=name,
|
||||||
|
ctx_fn=ctx_fn,
|
||||||
|
portal=portal,
|
||||||
|
**ctx_kwargs,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
# store the cancel scope and portal for later cancellation or
|
||||||
|
# retstart if needed.
|
||||||
|
self.service_ctxs[name] = (cs, ipc_ctx, portal, complete)
|
||||||
|
return (
|
||||||
|
cs,
|
||||||
|
ipc_ctx,
|
||||||
|
started,
|
||||||
|
)
|
||||||
|
|
||||||
|
async def start_service(
|
||||||
|
self,
|
||||||
|
daemon_name: str,
|
||||||
|
ctx_ep: Callable, # kwargs must `partial`-ed in!
|
||||||
|
# ^TODO, type for `@tractor.context` deco-ed funcs!
|
||||||
|
|
||||||
|
debug_mode: bool = False,
|
||||||
|
**start_actor_kwargs,
|
||||||
|
|
||||||
|
) -> Context:
|
||||||
|
'''
|
||||||
|
Start new subactor and schedule a supervising "service task"
|
||||||
|
in it which explicitly defines the sub's lifetime.
|
||||||
|
|
||||||
|
"Service daemon subactors" are cancelled (and thus
|
||||||
|
terminated) using the paired `.cancel_service()`.
|
||||||
|
|
||||||
|
Effectively this API can be used to manage "service daemons"
|
||||||
|
spawned under a single parent actor with supervision
|
||||||
|
semantics equivalent to a one-cancels-one style actor-nursery
|
||||||
|
or "(subactor) task manager" where each subprocess's (and
|
||||||
|
thus its embedded actor runtime) lifetime is synced to that
|
||||||
|
of the remotely spawned task defined by `ctx_ep`.
|
||||||
|
|
||||||
|
The funcionality can be likened to a "daemonized" version of
|
||||||
|
`.hilevel.worker.run_in_actor()` but with supervision
|
||||||
|
controls offered by `tractor.Context` where the main/root
|
||||||
|
remotely scheduled `trio.Task` invoking `ctx_ep` determines
|
||||||
|
the underlying subactor's lifetime.
|
||||||
|
|
||||||
|
'''
|
||||||
|
entry: tuple|None = self.service_ctxs.get(daemon_name)
|
||||||
|
if entry:
|
||||||
|
(cs, sub_ctx, portal, complete) = entry
|
||||||
|
return sub_ctx
|
||||||
|
|
||||||
|
if daemon_name not in self.service_ctxs:
|
||||||
|
portal: Portal = await self.an.start_actor(
|
||||||
|
daemon_name,
|
||||||
|
debug_mode=( # maybe set globally during allocate
|
||||||
|
debug_mode
|
||||||
|
or
|
||||||
|
self.debug_mode
|
||||||
|
),
|
||||||
|
**start_actor_kwargs,
|
||||||
|
)
|
||||||
|
ctx_kwargs: dict[str, Any] = {}
|
||||||
|
if isinstance(ctx_ep, functools.partial):
|
||||||
|
ctx_kwargs: dict[str, Any] = ctx_ep.keywords
|
||||||
|
ctx_ep: Callable = ctx_ep.func
|
||||||
|
|
||||||
|
(
|
||||||
|
cs,
|
||||||
|
sub_ctx,
|
||||||
|
started,
|
||||||
|
) = await self.start_service_ctx(
|
||||||
|
name=daemon_name,
|
||||||
|
portal=portal,
|
||||||
|
ctx_fn=ctx_ep,
|
||||||
|
**ctx_kwargs,
|
||||||
|
)
|
||||||
|
|
||||||
|
return sub_ctx
|
||||||
|
|
||||||
|
async def cancel_service(
|
||||||
|
self,
|
||||||
|
name: str,
|
||||||
|
|
||||||
|
) -> Any:
|
||||||
|
'''
|
||||||
|
Cancel the service task and actor for the given ``name``.
|
||||||
|
|
||||||
|
'''
|
||||||
|
log.info(f'Cancelling `pikerd` service {name}')
|
||||||
|
cs, sub_ctx, portal, complete = self.service_ctxs[name]
|
||||||
|
|
||||||
|
# cs.cancel()
|
||||||
|
await sub_ctx.cancel()
|
||||||
|
await complete.wait()
|
||||||
|
|
||||||
|
if name in self.service_ctxs:
|
||||||
|
# TODO: custom err?
|
||||||
|
# raise ServiceError(
|
||||||
|
raise RuntimeError(
|
||||||
|
f'Service actor for {name} not terminated and/or unknown?'
|
||||||
|
)
|
||||||
|
|
||||||
|
# assert name not in self.service_ctxs, \
|
||||||
|
# f'Serice task for {name} not terminated?'
|
||||||
|
|
@ -1,503 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
|
||||||
# it under the terms of the GNU Affero General Public License as published by
|
|
||||||
# the Free Software Foundation, either version 3 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful,
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
# GNU Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public License
|
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
"""
|
|
||||||
Inter-process comms abstractions
|
|
||||||
|
|
||||||
"""
|
|
||||||
from __future__ import annotations
|
|
||||||
from collections.abc import AsyncGenerator
|
|
||||||
from contextlib import (
|
|
||||||
asynccontextmanager as acm,
|
|
||||||
contextmanager as cm,
|
|
||||||
)
|
|
||||||
import platform
|
|
||||||
from pprint import pformat
|
|
||||||
import typing
|
|
||||||
from typing import (
|
|
||||||
Any,
|
|
||||||
TYPE_CHECKING,
|
|
||||||
)
|
|
||||||
import warnings
|
|
||||||
|
|
||||||
import trio
|
|
||||||
|
|
||||||
from ._types import (
|
|
||||||
transport_from_addr,
|
|
||||||
transport_from_stream,
|
|
||||||
)
|
|
||||||
from tractor._addr import (
|
|
||||||
is_wrapped_addr,
|
|
||||||
wrap_address,
|
|
||||||
Address,
|
|
||||||
UnwrappedAddress,
|
|
||||||
)
|
|
||||||
from tractor.log import get_logger
|
|
||||||
from tractor._exceptions import (
|
|
||||||
MsgTypeError,
|
|
||||||
pack_from_raise,
|
|
||||||
TransportClosed,
|
|
||||||
)
|
|
||||||
from tractor.msg import (
|
|
||||||
Aid,
|
|
||||||
MsgCodec,
|
|
||||||
)
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from ._transport import MsgTransport
|
|
||||||
|
|
||||||
|
|
||||||
log = get_logger(__name__)
|
|
||||||
|
|
||||||
_is_windows = platform.system() == 'Windows'
|
|
||||||
|
|
||||||
|
|
||||||
class Channel:
|
|
||||||
'''
|
|
||||||
An inter-process channel for communication between (remote) actors.
|
|
||||||
|
|
||||||
Wraps a ``MsgStream``: transport + encoding IPC connection.
|
|
||||||
|
|
||||||
Currently we only support ``trio.SocketStream`` for transport
|
|
||||||
(aka TCP) and the ``msgpack`` interchange format via the ``msgspec``
|
|
||||||
codec libary.
|
|
||||||
|
|
||||||
'''
|
|
||||||
def __init__(
|
|
||||||
|
|
||||||
self,
|
|
||||||
transport: MsgTransport|None = None,
|
|
||||||
# TODO: optional reconnection support?
|
|
||||||
# auto_reconnect: bool = False,
|
|
||||||
# on_reconnect: typing.Callable[..., typing.Awaitable] = None,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
|
|
||||||
# self._recon_seq = on_reconnect
|
|
||||||
# self._autorecon = auto_reconnect
|
|
||||||
|
|
||||||
# Either created in ``.connect()`` or passed in by
|
|
||||||
# user in ``.from_stream()``.
|
|
||||||
self._transport: MsgTransport|None = transport
|
|
||||||
|
|
||||||
# set after handshake - always info from peer end
|
|
||||||
self.aid: Aid|None = None
|
|
||||||
|
|
||||||
self._aiter_msgs = self._iter_msgs()
|
|
||||||
self._exc: Exception|None = None
|
|
||||||
# ^XXX! ONLY set if a remote actor sends an `Error`-msg
|
|
||||||
self._closed: bool = False
|
|
||||||
|
|
||||||
# flag set by `Portal.cancel_actor()` indicating remote
|
|
||||||
# (possibly peer) cancellation of the far end actor runtime.
|
|
||||||
self._cancel_called: bool = False
|
|
||||||
|
|
||||||
@property
|
|
||||||
def closed(self) -> bool:
|
|
||||||
'''
|
|
||||||
Was `.aclose()` successfully called?
|
|
||||||
|
|
||||||
'''
|
|
||||||
return self._closed
|
|
||||||
|
|
||||||
@property
|
|
||||||
def cancel_called(self) -> bool:
|
|
||||||
'''
|
|
||||||
Set when `Portal.cancel_actor()` is called on a portal which
|
|
||||||
wraps this IPC channel.
|
|
||||||
|
|
||||||
'''
|
|
||||||
return self._cancel_called
|
|
||||||
|
|
||||||
@property
|
|
||||||
def uid(self) -> tuple[str, str]:
|
|
||||||
'''
|
|
||||||
Peer actor's unique id.
|
|
||||||
|
|
||||||
'''
|
|
||||||
msg: str = (
|
|
||||||
f'`{type(self).__name__}.uid` is now deprecated.\n'
|
|
||||||
'Use the new `.aid: tractor.msg.Aid` (struct) instead '
|
|
||||||
'which also provides additional named (optional) fields '
|
|
||||||
'beyond just the `.name` and `.uuid`.'
|
|
||||||
)
|
|
||||||
warnings.warn(
|
|
||||||
msg,
|
|
||||||
DeprecationWarning,
|
|
||||||
stacklevel=2,
|
|
||||||
)
|
|
||||||
peer_aid: Aid = self.aid
|
|
||||||
return (
|
|
||||||
peer_aid.name,
|
|
||||||
peer_aid.uuid,
|
|
||||||
)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def stream(self) -> trio.abc.Stream | None:
|
|
||||||
return self._transport.stream if self._transport else None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def msgstream(self) -> MsgTransport:
|
|
||||||
log.info(
|
|
||||||
'`Channel.msgstream` is an old name, use `._transport`'
|
|
||||||
)
|
|
||||||
return self._transport
|
|
||||||
|
|
||||||
@property
|
|
||||||
def transport(self) -> MsgTransport:
|
|
||||||
return self._transport
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def from_stream(
|
|
||||||
cls,
|
|
||||||
stream: trio.abc.Stream,
|
|
||||||
) -> Channel:
|
|
||||||
transport_cls = transport_from_stream(stream)
|
|
||||||
return Channel(
|
|
||||||
transport=transport_cls(stream)
|
|
||||||
)
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
async def from_addr(
|
|
||||||
cls,
|
|
||||||
addr: UnwrappedAddress,
|
|
||||||
**kwargs
|
|
||||||
) -> Channel:
|
|
||||||
|
|
||||||
if not is_wrapped_addr(addr):
|
|
||||||
addr: Address = wrap_address(addr)
|
|
||||||
|
|
||||||
transport_cls = transport_from_addr(addr)
|
|
||||||
transport = await transport_cls.connect_to(
|
|
||||||
addr,
|
|
||||||
**kwargs,
|
|
||||||
)
|
|
||||||
# XXX, for UDS *no!* since we recv the peer-pid and build out
|
|
||||||
# a new addr..
|
|
||||||
# assert transport.raddr == addr
|
|
||||||
chan = Channel(transport=transport)
|
|
||||||
|
|
||||||
# ?TODO, compact this into adapter level-methods?
|
|
||||||
# -[ ] would avoid extra repr-calcs if level not active?
|
|
||||||
# |_ how would the `calc_if_level` look though? func?
|
|
||||||
if log.at_least_level('runtime'):
|
|
||||||
from tractor.devx import (
|
|
||||||
pformat as _pformat,
|
|
||||||
)
|
|
||||||
chan_repr: str = _pformat.nest_from_op(
|
|
||||||
input_op='[>',
|
|
||||||
text=chan.pformat(),
|
|
||||||
nest_indent=1,
|
|
||||||
)
|
|
||||||
log.runtime(
|
|
||||||
f'Connected channel IPC transport\n'
|
|
||||||
f'{chan_repr}'
|
|
||||||
)
|
|
||||||
return chan
|
|
||||||
|
|
||||||
@cm
|
|
||||||
def apply_codec(
|
|
||||||
self,
|
|
||||||
codec: MsgCodec,
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
Temporarily override the underlying IPC msg codec for
|
|
||||||
dynamic enforcement of messaging schema.
|
|
||||||
|
|
||||||
'''
|
|
||||||
orig: MsgCodec = self._transport.codec
|
|
||||||
try:
|
|
||||||
self._transport.codec = codec
|
|
||||||
yield
|
|
||||||
finally:
|
|
||||||
self._transport.codec = orig
|
|
||||||
|
|
||||||
# TODO: do a .src/.dst: str for maddrs?
|
|
||||||
def pformat(
|
|
||||||
self,
|
|
||||||
privates: bool = False,
|
|
||||||
) -> str:
|
|
||||||
if not self._transport:
|
|
||||||
return '<Channel( with inactive transport? )>'
|
|
||||||
|
|
||||||
tpt: MsgTransport = self._transport
|
|
||||||
tpt_name: str = type(tpt).__name__
|
|
||||||
tpt_status: str = (
|
|
||||||
'connected' if self.connected()
|
|
||||||
else 'closed'
|
|
||||||
)
|
|
||||||
repr_str: str = (
|
|
||||||
f'<Channel(\n'
|
|
||||||
f' |_status: {tpt_status!r}\n'
|
|
||||||
) + (
|
|
||||||
f' _closed={self._closed}\n'
|
|
||||||
f' _cancel_called={self._cancel_called}\n'
|
|
||||||
if privates else ''
|
|
||||||
) + ( # peer-actor (processs) section
|
|
||||||
f' |_peer: {self.aid.reprol()!r}\n'
|
|
||||||
if self.aid else ' |_peer: <unknown>\n'
|
|
||||||
) + (
|
|
||||||
f' |_msgstream: {tpt_name}\n'
|
|
||||||
f' maddr: {tpt.maddr!r}\n'
|
|
||||||
f' proto: {tpt.laddr.proto_key!r}\n'
|
|
||||||
f' layer: {tpt.layer_key!r}\n'
|
|
||||||
f' codec: {tpt.codec_key!r}\n'
|
|
||||||
f' .laddr={tpt.laddr}\n'
|
|
||||||
f' .raddr={tpt.raddr}\n'
|
|
||||||
) + (
|
|
||||||
f' ._transport.stream={tpt.stream}\n'
|
|
||||||
f' ._transport.drained={tpt.drained}\n'
|
|
||||||
if privates else ''
|
|
||||||
) + (
|
|
||||||
f' _send_lock={tpt._send_lock.statistics()}\n'
|
|
||||||
if privates else ''
|
|
||||||
) + (
|
|
||||||
')>\n'
|
|
||||||
)
|
|
||||||
return repr_str
|
|
||||||
|
|
||||||
# NOTE: making this return a value that can be passed to
|
|
||||||
# `eval()` is entirely **optional** FYI!
|
|
||||||
# https://docs.python.org/3/library/functions.html#repr
|
|
||||||
# https://docs.python.org/3/reference/datamodel.html#object.__repr__
|
|
||||||
#
|
|
||||||
# Currently we target **readability** from a (console)
|
|
||||||
# logging perspective over `eval()`-ability since we do NOT
|
|
||||||
# target serializing non-struct instances!
|
|
||||||
# def __repr__(self) -> str:
|
|
||||||
__str__ = pformat
|
|
||||||
__repr__ = pformat
|
|
||||||
|
|
||||||
@property
|
|
||||||
def laddr(self) -> Address|None:
|
|
||||||
return self._transport.laddr if self._transport else None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def raddr(self) -> Address|None:
|
|
||||||
return self._transport.raddr if self._transport else None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def maddr(self) -> str:
|
|
||||||
return self._transport.maddr if self._transport else '<no-tpt>'
|
|
||||||
|
|
||||||
# TODO: something like,
|
|
||||||
# `pdbp.hideframe_on(errors=[MsgTypeError])`
|
|
||||||
# instead of the `try/except` hack we have rn..
|
|
||||||
# seems like a pretty useful thing to have in general
|
|
||||||
# along with being able to filter certain stack frame(s / sets)
|
|
||||||
# possibly based on the current log-level?
|
|
||||||
async def send(
|
|
||||||
self,
|
|
||||||
payload: Any,
|
|
||||||
|
|
||||||
hide_tb: bool = False,
|
|
||||||
|
|
||||||
) -> None:
|
|
||||||
'''
|
|
||||||
Send a coded msg-blob over the transport.
|
|
||||||
|
|
||||||
'''
|
|
||||||
__tracebackhide__: bool = hide_tb
|
|
||||||
try:
|
|
||||||
log.transport(
|
|
||||||
'=> send IPC msg:\n\n'
|
|
||||||
f'{pformat(payload)}\n'
|
|
||||||
)
|
|
||||||
# assert self._transport # but why typing?
|
|
||||||
await self._transport.send(
|
|
||||||
payload,
|
|
||||||
hide_tb=hide_tb,
|
|
||||||
)
|
|
||||||
except (
|
|
||||||
BaseException,
|
|
||||||
MsgTypeError,
|
|
||||||
TransportClosed,
|
|
||||||
)as _err:
|
|
||||||
err = _err # bind for introspection
|
|
||||||
match err:
|
|
||||||
case MsgTypeError():
|
|
||||||
try:
|
|
||||||
assert err.cid
|
|
||||||
except KeyError:
|
|
||||||
raise err
|
|
||||||
case TransportClosed():
|
|
||||||
log.transport(
|
|
||||||
f'Transport stream closed due to\n'
|
|
||||||
f'{err.repr_src_exc()}\n'
|
|
||||||
)
|
|
||||||
|
|
||||||
case _:
|
|
||||||
# never suppress non-tpt sources
|
|
||||||
__tracebackhide__: bool = False
|
|
||||||
raise
|
|
||||||
|
|
||||||
async def recv(self) -> Any:
|
|
||||||
assert self._transport
|
|
||||||
return await self._transport.recv()
|
|
||||||
|
|
||||||
# TODO: auto-reconnect features like 0mq/nanomsg?
|
|
||||||
# -[ ] implement it manually with nods to SC prot
|
|
||||||
# possibly on multiple transport backends?
|
|
||||||
# -> seems like that might be re-inventing scalability
|
|
||||||
# prots tho no?
|
|
||||||
# try:
|
|
||||||
# return await self._transport.recv()
|
|
||||||
# except trio.BrokenResourceError:
|
|
||||||
# if self._autorecon:
|
|
||||||
# await self._reconnect()
|
|
||||||
# return await self.recv()
|
|
||||||
# raise
|
|
||||||
|
|
||||||
async def aclose(self) -> None:
|
|
||||||
|
|
||||||
log.transport(
|
|
||||||
f'Closing channel to {self.aid} '
|
|
||||||
f'{self.laddr} -> {self.raddr}'
|
|
||||||
)
|
|
||||||
assert self._transport
|
|
||||||
await self._transport.stream.aclose()
|
|
||||||
self._closed = True
|
|
||||||
|
|
||||||
async def __aenter__(self):
|
|
||||||
await self.connect()
|
|
||||||
return self
|
|
||||||
|
|
||||||
async def __aexit__(self, *args):
|
|
||||||
await self.aclose(*args)
|
|
||||||
|
|
||||||
def __aiter__(self):
|
|
||||||
return self._aiter_msgs
|
|
||||||
|
|
||||||
# ?TODO? run any reconnection sequence?
|
|
||||||
# -[ ] prolly should be impl-ed as deco-API?
|
|
||||||
#
|
|
||||||
# async def _reconnect(self) -> None:
|
|
||||||
# """Handle connection failures by polling until a reconnect can be
|
|
||||||
# established.
|
|
||||||
# """
|
|
||||||
# down = False
|
|
||||||
# while True:
|
|
||||||
# try:
|
|
||||||
# with trio.move_on_after(3) as cancel_scope:
|
|
||||||
# await self.connect()
|
|
||||||
# cancelled = cancel_scope.cancelled_caught
|
|
||||||
# if cancelled:
|
|
||||||
# log.transport(
|
|
||||||
# "Reconnect timed out after 3 seconds, retrying...")
|
|
||||||
# continue
|
|
||||||
# else:
|
|
||||||
# log.transport("Stream connection re-established!")
|
|
||||||
|
|
||||||
# # on_recon = self._recon_seq
|
|
||||||
# # if on_recon:
|
|
||||||
# # await on_recon(self)
|
|
||||||
|
|
||||||
# break
|
|
||||||
# except (OSError, ConnectionRefusedError):
|
|
||||||
# if not down:
|
|
||||||
# down = True
|
|
||||||
# log.transport(
|
|
||||||
# f"Connection to {self.raddr} went down, waiting"
|
|
||||||
# " for re-establishment")
|
|
||||||
# await trio.sleep(1)
|
|
||||||
|
|
||||||
async def _iter_msgs(
|
|
||||||
self
|
|
||||||
) -> AsyncGenerator[Any, None]:
|
|
||||||
'''
|
|
||||||
Yield `MsgType` IPC msgs decoded and deliverd from
|
|
||||||
an underlying `MsgTransport` protocol.
|
|
||||||
|
|
||||||
This is a streaming routine alo implemented as an async-gen
|
|
||||||
func (same a `MsgTransport._iter_pkts()`) gets allocated by
|
|
||||||
a `.__call__()` inside `.__init__()` where it is assigned to
|
|
||||||
the `._aiter_msgs` attr.
|
|
||||||
|
|
||||||
'''
|
|
||||||
assert self._transport
|
|
||||||
while True:
|
|
||||||
try:
|
|
||||||
async for msg in self._transport:
|
|
||||||
match msg:
|
|
||||||
# NOTE: if transport/interchange delivers
|
|
||||||
# a type error, we pack it with the far
|
|
||||||
# end peer `Actor.uid` and relay the
|
|
||||||
# `Error`-msg upward to the `._rpc` stack
|
|
||||||
# for normal RAE handling.
|
|
||||||
case MsgTypeError():
|
|
||||||
yield pack_from_raise(
|
|
||||||
local_err=msg,
|
|
||||||
cid=msg.cid,
|
|
||||||
|
|
||||||
# XXX we pack it here bc lower
|
|
||||||
# layers have no notion of an
|
|
||||||
# actor-id ;)
|
|
||||||
src_uid=self.uid,
|
|
||||||
)
|
|
||||||
case _:
|
|
||||||
yield msg
|
|
||||||
|
|
||||||
except trio.BrokenResourceError:
|
|
||||||
|
|
||||||
# if not self._autorecon:
|
|
||||||
raise
|
|
||||||
|
|
||||||
await self.aclose()
|
|
||||||
|
|
||||||
# if self._autorecon: # attempt reconnect
|
|
||||||
# await self._reconnect()
|
|
||||||
# continue
|
|
||||||
|
|
||||||
def connected(self) -> bool:
|
|
||||||
return self._transport.connected() if self._transport else False
|
|
||||||
|
|
||||||
async def _do_handshake(
|
|
||||||
self,
|
|
||||||
aid: Aid,
|
|
||||||
|
|
||||||
) -> Aid:
|
|
||||||
'''
|
|
||||||
Exchange `(name, UUIDs)` identifiers as the first
|
|
||||||
communication step with any (peer) remote `Actor`.
|
|
||||||
|
|
||||||
These are essentially the "mailbox addresses" found in
|
|
||||||
"actor model" parlance.
|
|
||||||
|
|
||||||
'''
|
|
||||||
await self.send(aid)
|
|
||||||
peer_aid: Aid = await self.recv()
|
|
||||||
log.runtime(
|
|
||||||
f'Received hanshake with peer\n'
|
|
||||||
f'<= {peer_aid.reprol(sin_uuid=False)}\n'
|
|
||||||
)
|
|
||||||
# NOTE, we always are referencing the remote peer!
|
|
||||||
self.aid = peer_aid
|
|
||||||
return peer_aid
|
|
||||||
|
|
||||||
|
|
||||||
@acm
|
|
||||||
async def _connect_chan(
|
|
||||||
addr: UnwrappedAddress
|
|
||||||
) -> typing.AsyncGenerator[Channel, None]:
|
|
||||||
'''
|
|
||||||
Create and connect a channel with disconnect on context manager
|
|
||||||
teardown.
|
|
||||||
|
|
||||||
'''
|
|
||||||
chan = await Channel.from_addr(addr)
|
|
||||||
yield chan
|
|
||||||
with trio.CancelScope(shield=True):
|
|
||||||
await chan.aclose()
|
|
||||||
|
|
@ -1,163 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
|
||||||
# it under the terms of the GNU Affero General Public License as published by
|
|
||||||
# the Free Software Foundation, either version 3 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful,
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
# GNU Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public License
|
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
'''
|
|
||||||
File-descriptor-sharing on `linux` by "wilhelm_of_bohemia".
|
|
||||||
|
|
||||||
'''
|
|
||||||
from __future__ import annotations
|
|
||||||
import os
|
|
||||||
import array
|
|
||||||
import socket
|
|
||||||
import tempfile
|
|
||||||
from pathlib import Path
|
|
||||||
from contextlib import ExitStack
|
|
||||||
|
|
||||||
import trio
|
|
||||||
import tractor
|
|
||||||
from tractor.ipc import RBToken
|
|
||||||
|
|
||||||
|
|
||||||
actor_name = 'ringd'
|
|
||||||
|
|
||||||
|
|
||||||
_rings: dict[str, dict] = {}
|
|
||||||
|
|
||||||
|
|
||||||
async def _attach_to_ring(
|
|
||||||
ring_name: str
|
|
||||||
) -> tuple[int, int, int]:
|
|
||||||
actor = tractor.current_actor()
|
|
||||||
|
|
||||||
fd_amount = 3
|
|
||||||
sock_path = (
|
|
||||||
Path(tempfile.gettempdir())
|
|
||||||
/
|
|
||||||
f'{os.getpid()}-pass-ring-fds-{ring_name}-to-{actor.name}.sock'
|
|
||||||
)
|
|
||||||
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
||||||
sock.bind(sock_path)
|
|
||||||
sock.listen(1)
|
|
||||||
|
|
||||||
async with (
|
|
||||||
tractor.find_actor(actor_name) as ringd,
|
|
||||||
ringd.open_context(
|
|
||||||
_pass_fds,
|
|
||||||
name=ring_name,
|
|
||||||
sock_path=sock_path
|
|
||||||
) as (ctx, _sent)
|
|
||||||
):
|
|
||||||
# prepare array to receive FD
|
|
||||||
fds = array.array("i", [0] * fd_amount)
|
|
||||||
|
|
||||||
conn, _ = sock.accept()
|
|
||||||
|
|
||||||
# receive FD
|
|
||||||
msg, ancdata, flags, addr = conn.recvmsg(
|
|
||||||
1024,
|
|
||||||
socket.CMSG_LEN(fds.itemsize * fd_amount)
|
|
||||||
)
|
|
||||||
|
|
||||||
for (
|
|
||||||
cmsg_level,
|
|
||||||
cmsg_type,
|
|
||||||
cmsg_data,
|
|
||||||
) in ancdata:
|
|
||||||
if (
|
|
||||||
cmsg_level == socket.SOL_SOCKET
|
|
||||||
and
|
|
||||||
cmsg_type == socket.SCM_RIGHTS
|
|
||||||
):
|
|
||||||
fds.frombytes(cmsg_data[:fds.itemsize * fd_amount])
|
|
||||||
break
|
|
||||||
else:
|
|
||||||
raise RuntimeError("Receiver: No FDs received")
|
|
||||||
|
|
||||||
conn.close()
|
|
||||||
sock.close()
|
|
||||||
sock_path.unlink()
|
|
||||||
|
|
||||||
return RBToken.from_msg(
|
|
||||||
await ctx.wait_for_result()
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def _pass_fds(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
name: str,
|
|
||||||
sock_path: str
|
|
||||||
) -> RBToken:
|
|
||||||
global _rings
|
|
||||||
token = _rings[name]
|
|
||||||
client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
||||||
client.connect(sock_path)
|
|
||||||
await ctx.started()
|
|
||||||
fds = array.array('i', token.fds)
|
|
||||||
client.sendmsg([b'FDs'], [(socket.SOL_SOCKET, socket.SCM_RIGHTS, fds)])
|
|
||||||
client.close()
|
|
||||||
return token
|
|
||||||
|
|
||||||
|
|
||||||
@tractor.context
|
|
||||||
async def _open_ringbuf(
|
|
||||||
ctx: tractor.Context,
|
|
||||||
name: str,
|
|
||||||
buf_size: int
|
|
||||||
) -> RBToken:
|
|
||||||
global _rings
|
|
||||||
is_owner = False
|
|
||||||
if name not in _rings:
|
|
||||||
stack = ExitStack()
|
|
||||||
token = stack.enter_context(
|
|
||||||
tractor.open_ringbuf(
|
|
||||||
name,
|
|
||||||
buf_size=buf_size
|
|
||||||
)
|
|
||||||
)
|
|
||||||
_rings[name] = {
|
|
||||||
'token': token,
|
|
||||||
'stack': stack,
|
|
||||||
}
|
|
||||||
is_owner = True
|
|
||||||
|
|
||||||
ring = _rings[name]
|
|
||||||
await ctx.started()
|
|
||||||
|
|
||||||
try:
|
|
||||||
await trio.sleep_forever()
|
|
||||||
|
|
||||||
except tractor.ContextCancelled:
|
|
||||||
...
|
|
||||||
|
|
||||||
finally:
|
|
||||||
if is_owner:
|
|
||||||
ring['stack'].close()
|
|
||||||
|
|
||||||
|
|
||||||
async def open_ringbuf(
|
|
||||||
name: str,
|
|
||||||
buf_size: int
|
|
||||||
) -> RBToken:
|
|
||||||
async with (
|
|
||||||
tractor.find_actor(actor_name) as ringd,
|
|
||||||
ringd.open_context(
|
|
||||||
_open_ringbuf,
|
|
||||||
name=name,
|
|
||||||
buf_size=buf_size
|
|
||||||
) as (rd_ctx, _)
|
|
||||||
):
|
|
||||||
yield await _attach_to_ring(name)
|
|
||||||
await rd_ctx.cancel()
|
|
||||||
|
|
@ -1,153 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
|
||||||
# it under the terms of the GNU Affero General Public License as published by
|
|
||||||
# the Free Software Foundation, either version 3 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful,
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
# GNU Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public License
|
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
'''
|
|
||||||
Linux specifics, for now we are only exposing EventFD
|
|
||||||
|
|
||||||
'''
|
|
||||||
import os
|
|
||||||
import errno
|
|
||||||
|
|
||||||
import cffi
|
|
||||||
import trio
|
|
||||||
|
|
||||||
ffi = cffi.FFI()
|
|
||||||
|
|
||||||
# Declare the C functions and types we plan to use.
|
|
||||||
# - eventfd: for creating the event file descriptor
|
|
||||||
# - write: for writing to the file descriptor
|
|
||||||
# - read: for reading from the file descriptor
|
|
||||||
# - close: for closing the file descriptor
|
|
||||||
ffi.cdef(
|
|
||||||
'''
|
|
||||||
int eventfd(unsigned int initval, int flags);
|
|
||||||
|
|
||||||
ssize_t write(int fd, const void *buf, size_t count);
|
|
||||||
ssize_t read(int fd, void *buf, size_t count);
|
|
||||||
|
|
||||||
int close(int fd);
|
|
||||||
'''
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# Open the default dynamic library (essentially 'libc' in most cases)
|
|
||||||
C = ffi.dlopen(None)
|
|
||||||
|
|
||||||
|
|
||||||
# Constants from <sys/eventfd.h>, if needed.
|
|
||||||
EFD_SEMAPHORE = 1
|
|
||||||
EFD_CLOEXEC = 0o2000000
|
|
||||||
EFD_NONBLOCK = 0o4000
|
|
||||||
|
|
||||||
|
|
||||||
def open_eventfd(initval: int = 0, flags: int = 0) -> int:
|
|
||||||
'''
|
|
||||||
Open an eventfd with the given initial value and flags.
|
|
||||||
Returns the file descriptor on success, otherwise raises OSError.
|
|
||||||
|
|
||||||
'''
|
|
||||||
fd = C.eventfd(initval, flags)
|
|
||||||
if fd < 0:
|
|
||||||
raise OSError(errno.errorcode[ffi.errno], 'eventfd failed')
|
|
||||||
return fd
|
|
||||||
|
|
||||||
|
|
||||||
def write_eventfd(fd: int, value: int) -> int:
|
|
||||||
'''
|
|
||||||
Write a 64-bit integer (uint64_t) to the eventfd's counter.
|
|
||||||
|
|
||||||
'''
|
|
||||||
# Create a uint64_t* in C, store `value`
|
|
||||||
data_ptr = ffi.new('uint64_t *', value)
|
|
||||||
|
|
||||||
# Call write(fd, data_ptr, 8)
|
|
||||||
# We expect to write exactly 8 bytes (sizeof(uint64_t))
|
|
||||||
ret = C.write(fd, data_ptr, 8)
|
|
||||||
if ret < 0:
|
|
||||||
raise OSError(errno.errorcode[ffi.errno], 'write to eventfd failed')
|
|
||||||
return ret
|
|
||||||
|
|
||||||
|
|
||||||
def read_eventfd(fd: int) -> int:
|
|
||||||
'''
|
|
||||||
Read a 64-bit integer (uint64_t) from the eventfd, returning the value.
|
|
||||||
Reading resets the counter to 0 (unless using EFD_SEMAPHORE).
|
|
||||||
|
|
||||||
'''
|
|
||||||
# Allocate an 8-byte buffer in C for reading
|
|
||||||
buf = ffi.new('char[]', 8)
|
|
||||||
|
|
||||||
ret = C.read(fd, buf, 8)
|
|
||||||
if ret < 0:
|
|
||||||
raise OSError(errno.errorcode[ffi.errno], 'read from eventfd failed')
|
|
||||||
# Convert the 8 bytes we read into a Python integer
|
|
||||||
data_bytes = ffi.unpack(buf, 8) # returns a Python bytes object of length 8
|
|
||||||
value = int.from_bytes(data_bytes, byteorder='little', signed=False)
|
|
||||||
return value
|
|
||||||
|
|
||||||
|
|
||||||
def close_eventfd(fd: int) -> int:
|
|
||||||
'''
|
|
||||||
Close the eventfd.
|
|
||||||
|
|
||||||
'''
|
|
||||||
ret = C.close(fd)
|
|
||||||
if ret < 0:
|
|
||||||
raise OSError(errno.errorcode[ffi.errno], 'close failed')
|
|
||||||
|
|
||||||
|
|
||||||
class EventFD:
|
|
||||||
'''
|
|
||||||
Use a previously opened eventfd(2), meant to be used in
|
|
||||||
sub-actors after root actor opens the eventfds then passes
|
|
||||||
them through pass_fds
|
|
||||||
|
|
||||||
'''
|
|
||||||
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
fd: int,
|
|
||||||
omode: str
|
|
||||||
):
|
|
||||||
self._fd: int = fd
|
|
||||||
self._omode: str = omode
|
|
||||||
self._fobj = None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def fd(self) -> int | None:
|
|
||||||
return self._fd
|
|
||||||
|
|
||||||
def write(self, value: int) -> int:
|
|
||||||
return write_eventfd(self._fd, value)
|
|
||||||
|
|
||||||
async def read(self) -> int:
|
|
||||||
return await trio.to_thread.run_sync(
|
|
||||||
read_eventfd, self._fd,
|
|
||||||
abandon_on_cancel=True
|
|
||||||
)
|
|
||||||
|
|
||||||
def open(self):
|
|
||||||
self._fobj = os.fdopen(self._fd, self._omode)
|
|
||||||
|
|
||||||
def close(self):
|
|
||||||
if self._fobj:
|
|
||||||
self._fobj.close()
|
|
||||||
|
|
||||||
def __enter__(self):
|
|
||||||
self.open()
|
|
||||||
return self
|
|
||||||
|
|
||||||
def __exit__(self, exc_type, exc_value, traceback):
|
|
||||||
self.close()
|
|
||||||
|
|
@ -1,75 +0,0 @@
|
||||||
# tractor: structured concurrent "actors".
|
|
||||||
# Copyright 2018-eternity Tyler Goodlet.
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
|
||||||
# it under the terms of the GNU Affero General Public License as published by
|
|
||||||
# the Free Software Foundation, either version 3 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful,
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
# GNU Affero General Public License for more details.
|
|
||||||
|
|
||||||
# You should have received a copy of the GNU Affero General Public License
|
|
||||||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
'''
|
|
||||||
Utils to tame mp non-SC madeness
|
|
||||||
|
|
||||||
'''
|
|
||||||
import platform
|
|
||||||
|
|
||||||
|
|
||||||
def disable_mantracker():
|
|
||||||
'''
|
|
||||||
Disable all `multiprocessing` "resource tracking" machinery since
|
|
||||||
it's an absolute multi-threaded mess of non-SC madness.
|
|
||||||
|
|
||||||
'''
|
|
||||||
from multiprocessing.shared_memory import SharedMemory
|
|
||||||
|
|
||||||
|
|
||||||
# 3.13+ only.. can pass `track=False` to disable
|
|
||||||
# all the resource tracker bs.
|
|
||||||
# https://docs.python.org/3/library/multiprocessing.shared_memory.html
|
|
||||||
if (_py_313 := (
|
|
||||||
platform.python_version_tuple()[:-1]
|
|
||||||
>=
|
|
||||||
('3', '13')
|
|
||||||
)
|
|
||||||
):
|
|
||||||
from functools import partial
|
|
||||||
return partial(
|
|
||||||
SharedMemory,
|
|
||||||
track=False,
|
|
||||||
)
|
|
||||||
|
|
||||||
# !TODO, once we drop 3.12- we can obvi remove all this!
|
|
||||||
else:
|
|
||||||
from multiprocessing import (
|
|
||||||
resource_tracker as mantracker,
|
|
||||||
)
|
|
||||||
|
|
||||||
# Tell the "resource tracker" thing to fuck off.
|
|
||||||
class ManTracker(mantracker.ResourceTracker):
|
|
||||||
def register(self, name, rtype):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def unregister(self, name, rtype):
|
|
||||||
pass
|
|
||||||
|
|
||||||
def ensure_running(self):
|
|
||||||
pass
|
|
||||||
|
|
||||||
# "know your land and know your prey"
|
|
||||||
# https://www.dailymotion.com/video/x6ozzco
|
|
||||||
mantracker._resource_tracker = ManTracker()
|
|
||||||
mantracker.register = mantracker._resource_tracker.register
|
|
||||||
mantracker.ensure_running = mantracker._resource_tracker.ensure_running
|
|
||||||
mantracker.unregister = mantracker._resource_tracker.unregister
|
|
||||||
mantracker.getfd = mantracker._resource_tracker.getfd
|
|
||||||
|
|
||||||
# use std type verbatim
|
|
||||||
shmT = SharedMemory
|
|
||||||
|
|
||||||
return shmT
|
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue