forked from goodboy/tractor
Compare commits
933 Commits
func_refs_
...
master
Author | SHA1 | Date |
---|---|---|
goodboy | e5ee2e3de8 | |
Tyler Goodlet | 41aa91c8eb | |
Tyler Goodlet | 6758e4487c | |
Tyler Goodlet | 1c3893a383 | |
Tyler Goodlet | 73befac9bc | |
Tyler Goodlet | 79622bbeea | |
Tyler Goodlet | 95535b2226 | |
Tyler Goodlet | 87c6e09d6b | |
Tyler Goodlet | 9ccd3a74b6 | |
Tyler Goodlet | ae4ff5dc8d | |
Tyler Goodlet | 705538398f | |
Tyler Goodlet | 86aef5238d | |
Tyler Goodlet | cc82447db6 | |
Tyler Goodlet | 23cffbd940 | |
Tyler Goodlet | 3d202272c4 | |
Tyler Goodlet | 63cdb0891f | |
goodboy | 0f7db27b68 | |
Tyler Goodlet | c53d62d2f7 | |
Tyler Goodlet | f667d16d66 | |
Tyler Goodlet | 24a062341e | |
goodboy | e714bec8db | |
Igor | 009cd6552e | |
goodboy | 649c5e7504 | |
Tyler Goodlet | 203f95615c | |
Tyler Goodlet | efb8bec828 | |
Tyler Goodlet | 8637778739 | |
Tyler Goodlet | 47166e45f0 | |
Tyler Goodlet | 4ce2dcd12b | |
Tyler Goodlet | 80f983818f | |
Tyler Goodlet | 6ba29f8d56 | |
Tyler Goodlet | 2707a0e971 | |
Tyler Goodlet | c8efcdd0d3 | |
Tyler Goodlet | 9f9907271b | |
Tyler Goodlet | c2367c1c5e | |
goodboy | a777217674 | |
Tyler Goodlet | 13c9eadc8f | |
Tyler Goodlet | af6c325072 | |
Tyler Goodlet | 195d2f0ed4 | |
Tyler Goodlet | aa4871b13d | |
Tyler Goodlet | 556f4626db | |
Tyler Goodlet | 3967c0ed9e | |
Tyler Goodlet | e34823aab4 | |
Tyler Goodlet | 6c35ba2cb6 | |
Tyler Goodlet | 3a0817ff55 | |
Tyler Goodlet | 7fddb4416b | |
Tyler Goodlet | 1d92f2552a | |
Tyler Goodlet | 4f8586a928 | |
Tyler Goodlet | fb9ff45745 | |
Tyler Goodlet | 36a83cb306 | |
Tyler Goodlet | 7394a187e0 | |
Tyler Goodlet | df01294bb2 | |
Tyler Goodlet | ddf3d0d1b3 | |
Tyler Goodlet | 158569adae | |
Tyler Goodlet | 97d5f7233b | |
Tyler Goodlet | d27c081a15 | |
Tyler Goodlet | a4874a3227 | |
Tyler Goodlet | de04bbb2bb | |
Tyler Goodlet | 4f977189c0 | |
goodboy | 9fd62cf71f | |
Tyler Goodlet | 606efa5bb7 | |
Tyler Goodlet | 121a8cc891 | |
Tyler Goodlet | c54b8ca4ba | |
goodboy | de93c8257c | |
Tyler Goodlet | 5b8a87d0f6 | |
Tyler Goodlet | 9e5c8ce6f6 | |
Tyler Goodlet | 965cd406a2 | |
Tyler Goodlet | 2e278ceb74 | |
Tyler Goodlet | 6d124db7c9 | |
Tyler Goodlet | dba8118553 | |
Tyler Goodlet | fca2e7c10e | |
Tyler Goodlet | 5ed62c5c54 | |
goodboy | 588b7ca7bf | |
Tyler Goodlet | d8214735b9 | |
Tyler Goodlet | 48f6d514ef | |
Tyler Goodlet | 6c8cacc9d1 | |
Tyler Goodlet | 38326e8c15 | |
Tyler Goodlet | b5192cca8e | |
Tyler Goodlet | c606be8c64 | |
Tyler Goodlet | d8e48e29ba | |
goodboy | a0f6668ce8 | |
Tyler Goodlet | 274c66cf9d | |
Tyler Goodlet | f2641c8964 | |
Tyler Goodlet | c47575997a | |
Tyler Goodlet | f39414ce12 | |
Tyler Goodlet | 0a1bf8e57d | |
Tyler Goodlet | e298b70edf | |
Tyler Goodlet | c0dd5d7ffc | |
Tyler Goodlet | 347591c348 | |
Tyler Goodlet | 38f9d35dee | |
Tyler Goodlet | 88448f7281 | |
Tyler Goodlet | 0956d5f461 | |
Tyler Goodlet | c646c79a82 | |
Tyler Goodlet | 33f2234baf | |
Tyler Goodlet | 7521bded3d | |
Tyler Goodlet | 0f523b65fb | |
Tyler Goodlet | 50fe098e06 | |
Tyler Goodlet | d87d6af7e1 | |
Tyler Goodlet | df69aedcd5 | |
Tyler Goodlet | b15e4ed9ce | |
Tyler Goodlet | 98056f6ed7 | |
goodboy | 247d3448ae | |
Tyler Goodlet | fc17f6790e | |
Tyler Goodlet | b81b6be98a | |
Tyler Goodlet | 72fbda4cef | |
Tyler Goodlet | fb721f36ef | |
Tyler Goodlet | 734d8dd663 | |
Tyler Goodlet | 30ea7a06b0 | |
Tyler Goodlet | 3398153c52 | |
Tyler Goodlet | 1c480e6c92 | |
goodboy | dfdad4d1fa | |
Tyler Goodlet | b892bc74f6 | |
Tyler Goodlet | 44b59f3338 | |
Tyler Goodlet | 7a719ac2a7 | |
goodboy | 9e6266dda3 | |
Tyler Goodlet | b1abec543f | |
Tyler Goodlet | 93b9d2dc2d | |
Tyler Goodlet | 4d808757a6 | |
Tyler Goodlet | 7e5bb0437e | |
Tyler Goodlet | b19f08d9f0 | |
Tyler Goodlet | 2c20b2d64f | |
Tyler Goodlet | 023b6fc845 | |
Tyler Goodlet | d24fae8381 | |
Tyler Goodlet | 5ab98513b7 | |
Tyler Goodlet | 90f4912580 | |
goodboy | 6e24e16068 | |
Tyler Goodlet | 15047341bd | |
Tyler Goodlet | dc295ab227 | |
goodboy | 6a0337b69d | |
Tyler Goodlet | e609183242 | |
goodboy | 368e9f3f7c | |
Tyler Goodlet | 10eeda2d2b | |
Tyler Goodlet | a113e22bb9 | |
Tyler Goodlet | ad19bf2cf1 | |
Tyler Goodlet | 9aef03772a | |
Tyler Goodlet | 7548dba8f2 | |
Tyler Goodlet | ba4d4e9af3 | |
Tyler Goodlet | 208d56af2c | |
Tyler Goodlet | a3a5bc267e | |
Tyler Goodlet | d4084b2032 | |
Tyler Goodlet | 1e6b4d5dd4 | |
Tyler Goodlet | c613acfe5c | |
goodboy | fea9dc7065 | |
goodboy | e558c427de | |
Tyler Goodlet | f07c3aa4a1 | |
Tyler Goodlet | bafd10a260 | |
Tyler Goodlet | 5ad540c417 | |
Tyler Goodlet | 83b44cf469 | |
Tyler Goodlet | 1f2001020e | |
Tyler Goodlet | 71f9881a60 | |
Tyler Goodlet | e24645eec8 | |
Tyler Goodlet | c3cdeeb3ba | |
Tyler Goodlet | 9bd534df83 | |
goodboy | c1d700f257 | |
Tyler Goodlet | 14c6e34658 | |
Tyler Goodlet | 3393bc23e4 | |
Tyler Goodlet | 171f1bc243 | |
Tyler Goodlet | ee02cd2496 | |
Tyler Goodlet | 4c5d435aac | |
Tyler Goodlet | a9b4a61620 | |
goodboy | 641ed7a32a | |
Tyler Goodlet | cc5f60bba0 | |
Tyler Goodlet | 8f1fe2376a | |
Tyler Goodlet | 65540f3e2a | |
Tyler Goodlet | 650313dfef | |
Tyler Goodlet | e4006da6f4 | |
Tyler Goodlet | 7f6169a050 | |
Tyler Goodlet | 2d387f2610 | |
Tyler Goodlet | 8115759984 | |
Tyler Goodlet | 02c3b9a672 | |
Tyler Goodlet | fa4388835c | |
Tyler Goodlet | 54de72d8df | |
Tyler Goodlet | c5c7a9027c | |
Tyler Goodlet | e4771eec16 | |
Tyler Goodlet | a9aaee9dbd | |
Tyler Goodlet | acfbae4b95 | |
Tyler Goodlet | aca9a6b99a | |
Tyler Goodlet | 8896ba2bf8 | |
Tyler Goodlet | 87b2ccb86a | |
Tyler Goodlet | 937ed99e39 | |
Tyler Goodlet | 91f034a136 | |
Tyler Goodlet | 08cf03cd9e | |
Tyler Goodlet | 5e23b3ca0d | |
Tyler Goodlet | 6f01c78122 | |
Tyler Goodlet | 457499bc2e | |
Tyler Goodlet | a4bac135d9 | |
Tyler Goodlet | 20c660faa7 | |
Tyler Goodlet | 1d4d55f5cd | |
Tyler Goodlet | c0cd99e374 | |
Tyler Goodlet | a4538a3d84 | |
Tyler Goodlet | b01daa5319 | |
Tyler Goodlet | bd362a05f0 | |
Tyler Goodlet | cb0c47c42a | |
Tyler Goodlet | 808d7ae2c6 | |
Tyler Goodlet | b21f2e16ad | |
Tyler Goodlet | 4779badd96 | |
Tyler Goodlet | 6bdcbdb96f | |
Tyler Goodlet | adbebd3f06 | |
Tyler Goodlet | a2e90194bc | |
Tyler Goodlet | ba7b355d9c | |
Tyler Goodlet | 617d57dc35 | |
Tyler Goodlet | dadd5e6148 | |
Tyler Goodlet | a72350118c | |
Tyler Goodlet | ef8dc0204c | |
Tyler Goodlet | a101971027 | |
Tyler Goodlet | 835836123b | |
Tyler Goodlet | 70ad0f6b8e | |
Tyler Goodlet | 56b30a9a53 | |
Tyler Goodlet | 925d5c1ceb | |
Tyler Goodlet | b9eb601265 | |
Tyler Goodlet | 4dcc21234e | |
Tyler Goodlet | 64909e676e | |
Tyler Goodlet | 19fb77f698 | |
Tyler Goodlet | 8b9f342eef | |
Tyler Goodlet | bd7d507153 | |
Tyler Goodlet | 9bc38cbf04 | |
Tyler Goodlet | a90ca4b384 | |
Tyler Goodlet | d0dcd55f47 | |
Tyler Goodlet | 4e08605b0d | |
Tyler Goodlet | 519f4c300b | |
Tyler Goodlet | 56c19093bb | |
Tyler Goodlet | ff3f5959e9 | |
Tyler Goodlet | abb00531d3 | |
Tyler Goodlet | 439d320a25 | |
Tyler Goodlet | 18c525d2f1 | |
Tyler Goodlet | 201c026284 | |
Tyler Goodlet | 2a61aa099b | |
Tyler Goodlet | e2453fd3da | |
Tyler Goodlet | b29def8b5d | |
Tyler Goodlet | f07e9dbb2f | |
Tyler Goodlet | 2f5a6049a4 | |
Tyler Goodlet | 418e74eee7 | |
Tyler Goodlet | c7035be2fc | |
Tyler Goodlet | deaca7d6cc | |
Tyler Goodlet | d47d0e7c37 | |
Tyler Goodlet | 0062c96a3c | |
Tyler Goodlet | 4be13b7387 | |
Tyler Goodlet | 7bb5addd4c | |
Tyler Goodlet | 4fd924cfd2 | |
Tyler Goodlet | fe0fd1a1c1 | |
Tyler Goodlet | dd23e78de1 | |
Tyler Goodlet | 89b44f8163 | |
Tyler Goodlet | 2819b6a5b2 | |
Tyler Goodlet | f2671ed026 | |
Tyler Goodlet | 41924c86a6 | |
Tyler Goodlet | 206c7c0720 | |
Tyler Goodlet | bf0ac3116c | |
Tyler Goodlet | bb732cefd0 | |
Tyler Goodlet | 74b819a857 | |
Tyler Goodlet | 8892204c84 | |
Tyler Goodlet | 8f4bbf1cbf | |
Tyler Goodlet | 21dccb2e79 | |
Tyler Goodlet | aea8f63bae | |
Tyler Goodlet | 7964a9f6f8 | |
Tyler Goodlet | 99c4319940 | |
Tyler Goodlet | e5195264a1 | |
Tyler Goodlet | 42f9d10252 | |
Tyler Goodlet | 345573e602 | |
Tyler Goodlet | 4e60c17375 | |
Tyler Goodlet | 6b7b58346f | |
Tyler Goodlet | 3cac323421 | |
goodboy | 4902e184e9 | |
Tyler Goodlet | 05790a20c1 | |
Tyler Goodlet | 565c603300 | |
Tyler Goodlet | f0d78e1a6e | |
Tyler Goodlet | ce01f6b21c | |
Tyler Goodlet | 0906559ed9 | |
Tyler Goodlet | 38d03858d7 | |
Tyler Goodlet | 98de2fab31 | |
goodboy | 80121ed211 | |
Tyler Goodlet | 41983edc43 | |
Tyler Goodlet | 5168700fbf | |
Tyler Goodlet | 673c4a8c66 | |
Tyler Goodlet | 932b841176 | |
Tyler Goodlet | f594f1bdda | |
Tyler Goodlet | 53e3648eca | |
Tyler Goodlet | fc36503f4f | |
Tyler Goodlet | 4e7ab54452 | |
goodboy | 86d020d309 | |
Tyler Goodlet | bb3f35cdd0 | |
Tyler Goodlet | f94b7cd991 | |
Tyler Goodlet | f6af5c7bf8 | |
Tyler Goodlet | 9740a585d3 | |
Tyler Goodlet | b700dc34a8 | |
Tyler Goodlet | 9bc1c6f385 | |
goodboy | f4973e90e9 | |
Tyler Goodlet | 780e3dd13d | |
Tyler Goodlet | e0419b24ec | |
goodboy | 71f19f217d | |
Tyler Goodlet | 8901272854 | |
Tyler Goodlet | 7c151bed48 | |
Tyler Goodlet | 80897a8f2b | |
goodboy | 62983684d1 | |
Tyler Goodlet | 1c63bb6130 | |
goodboy | bfe99f29b8 | |
Tyler Goodlet | 9c27858aaf | |
Tyler Goodlet | 597ae4b690 | |
Tyler Goodlet | fa354ffe2b | |
Tyler Goodlet | 333fad8819 | |
Tyler Goodlet | 90593611bb | |
Tyler Goodlet | 9c43bb28f1 | |
Tyler Goodlet | e45251db56 | |
Tyler Goodlet | faf751acac | |
Tyler Goodlet | 20d281f619 | |
Tyler Goodlet | f3606d5bd8 | |
Tyler Goodlet | 032e14e326 | |
Tyler Goodlet | c322a193f2 | |
Tyler Goodlet | 46963c2e63 | |
Tyler Goodlet | 9b77b8c9ee | |
Tyler Goodlet | 13c8300226 | |
goodboy | 1109d96263 | |
Tyler Goodlet | 65b4bc8888 | |
Tyler Goodlet | bef9946f91 | |
Tyler Goodlet | c30cece37a | |
Tyler Goodlet | 509082c935 | |
Tyler Goodlet | 75bb1added | |
goodboy | 6e5590dad6 | |
Tyler Goodlet | 76a0492028 | |
Tyler Goodlet | 4eab4a0213 | |
Tyler Goodlet | 0edc6a26bc | |
Tyler Goodlet | c5acc3b969 | |
Tyler Goodlet | 17e195aacf | |
Tyler Goodlet | c65756ed80 | |
Tyler Goodlet | 927decc88d | |
Tyler Goodlet | 17bfa120cc | |
Tyler Goodlet | 77ddc073e8 | |
goodboy | 26bebc42b7 | |
Tyler Goodlet | 87de28fd88 | |
Tyler Goodlet | 56b29c27de | |
Tyler Goodlet | adf9a1d0aa | |
Tyler Goodlet | 25a27e780d | |
Tyler Goodlet | c265f3f94e | |
Tyler Goodlet | 2900ceb003 | |
Tyler Goodlet | b6ae77b5ac | |
Tyler Goodlet | 949cb2c9fe | |
goodboy | 094206ee9d | |
Tyler Goodlet | debbf64d58 | |
Tyler Goodlet | 070e6ba459 | |
Tyler Goodlet | 7e004c0688 | |
Tyler Goodlet | ffe88de53b | |
Tyler Goodlet | d29a915d48 | |
Tyler Goodlet | be87caa99b | |
goodboy | 0b51ebfe11 | |
Tyler Goodlet | 4bf7992200 | |
Tyler Goodlet | 41296448e8 | |
Tyler Goodlet | 9650055519 | |
Tyler Goodlet | 532974fb90 | |
Tyler Goodlet | b1d72b77c9 | |
Tyler Goodlet | a2171c7e71 | |
goodboy | 30986d6b64 | |
Tyler Goodlet | 9ab04b1f6b | |
Tyler Goodlet | b3ff4b7804 | |
goodboy | d27bdbd40e | |
Tyler Goodlet | a95b0dc05e | |
goodboy | 909c996346 | |
Giacomo Camporini | adc8e5c009 | |
Giacomo Camporini | 5eed85d5dd | |
Giacomo Camporini | 137fed790f | |
goodboy | 96123f21d2 | |
overclockworked64 | 338aa2b74a | |
goodboy | 89551ef371 | |
Tyler Goodlet | 884bdf2d57 | |
goodboy | 343b2803b5 | |
Tyler Goodlet | cdef579d22 | |
Tyler Goodlet | f9400b4beb | |
Tyler Goodlet | 45cdf25f14 | |
Tyler Goodlet | 0a5818fe05 | |
Tyler Goodlet | fd6e18eba4 | |
goodboy | bbcdbaaba4 | |
Tyler Goodlet | 9b4cdb00e6 | |
Tyler Goodlet | 9b14d82086 | |
Tyler Goodlet | 4d1a48a47b | |
Tyler Goodlet | 4c0cfa68ac | |
Tyler Goodlet | 73d252e09e | |
Tyler Goodlet | 1fdcaf36f3 | |
Tyler Goodlet | 6952c7defa | |
Tyler Goodlet | 7237d696ce | |
Tyler Goodlet | b463841019 | |
Tyler Goodlet | d65912e1ae | |
Tyler Goodlet | 24078f2d6e | |
Tyler Goodlet | 56cc98375e | |
Tyler Goodlet | 9a2de90de6 | |
Tyler Goodlet | 2b9b29eb71 | |
Tyler Goodlet | b69412a903 | |
Tyler Goodlet | c4b3bb354e | |
Tyler Goodlet | 6803891bd7 | |
Tyler Goodlet | 5f4094691d | |
Tyler Goodlet | c48c68c0bc | |
Tyler Goodlet | ad2567dd73 | |
Tyler Goodlet | 44d0e9fc32 | |
Tyler Goodlet | d27ddb7bbb | |
Tyler Goodlet | 9bc94b5ccc | |
Tyler Goodlet | e6687bcdc4 | |
Tyler Goodlet | e815f766f6 | |
Tyler Goodlet | c19123b588 | |
Tyler Goodlet | 8704664719 | |
Tyler Goodlet | 04c0eda69d | |
Tyler Goodlet | 1114b6980e | |
Tyler Goodlet | 56357242e9 | |
Tyler Goodlet | 0ab5e5cadd | |
Tyler Goodlet | 06fa650ed0 | |
Tyler Goodlet | 446feff172 | |
Tyler Goodlet | 299e4192b0 | |
Tyler Goodlet | 41eddffc2c | |
Tyler Goodlet | 7a65165279 | |
Tyler Goodlet | b376b7cd32 | |
Tyler Goodlet | c262b1a3e8 | |
Tyler Goodlet | d9dac3f36c | |
Tyler Goodlet | 325c0cdb1b | |
Tyler Goodlet | 55e210fec6 | |
Tyler Goodlet | aa24bbc11c | |
Tyler Goodlet | 793bcfb7d4 | |
Tyler Goodlet | d80f8d7a39 | |
Tyler Goodlet | 340effae11 | |
Tyler Goodlet | 509ae132ec | |
Tyler Goodlet | 80f47dece2 | |
Tyler Goodlet | 2cf87146a3 | |
Tyler Goodlet | 8070b16bd0 | |
Tyler Goodlet | 1406ddc5ee | |
Tyler Goodlet | 055788cf16 | |
Tyler Goodlet | 1825b21d2c | |
Tyler Goodlet | acd63d0c89 | |
goodboy | cdf1f8c2f7 | |
Tyler Goodlet | 8eff788d2d | |
Tyler Goodlet | 916e27eedc | |
Tyler Goodlet | 98a830ccba | |
Tyler Goodlet | 8c004c1f36 | |
Tyler Goodlet | e2139c2bf0 | |
Tyler Goodlet | 9650b010de | |
Tyler Goodlet | 5d424e3703 | |
Tyler Goodlet | c38d0f826e | |
goodboy | 4001d2c3fc | |
Tyler Goodlet | 953d15b67d | |
Tyler Goodlet | da5e36bf0c | |
Tyler Goodlet | 21a9c47496 | |
Tyler Goodlet | 67dc0d014c | |
Tyler Goodlet | 9b1d8bf7b0 | |
Tyler Goodlet | 26394dd8df | |
Tyler Goodlet | 11e64426f6 | |
Tyler Goodlet | 213447008b | |
Tyler Goodlet | f617da6ff1 | |
Tyler Goodlet | 52627a6326 | |
Tyler Goodlet | 3826bc9972 | |
Tyler Goodlet | b210278e2f | |
Tyler Goodlet | 4a0252baf2 | |
Tyler Goodlet | ac22b4a875 | |
Tyler Goodlet | 5f41dbf34f | |
goodboy | 2d6fbd5437 | |
Tyler Goodlet | 325e550ff3 | |
Tyler Goodlet | b5d62909ff | |
Tyler Goodlet | 57f2aca18c | |
Tyler Goodlet | 1652716574 | |
Tyler Goodlet | f2ba961e81 | |
Tyler Goodlet | 79d63585b0 | |
Tyler Goodlet | 3deb1b91e6 | |
Tyler Goodlet | 61e134dc5d | |
goodboy | cfdc95fe7f | |
Tyler Goodlet | 6f94ffc304 | |
goodboy | 56297cf25c | |
Tyler Goodlet | 94f098e5f7 | |
Tyler Goodlet | 949aa9c405 | |
Tyler Goodlet | a38a983225 | |
Tyler Goodlet | 4f411d6926 | |
Tyler Goodlet | 9bee513136 | |
Tyler Goodlet | 5d9e3d1163 | |
Tyler Goodlet | 95c52436e5 | |
Tyler Goodlet | e51c0e17a2 | |
Tyler Goodlet | 92c6ec1882 | |
Tyler Goodlet | 72eef2a4a1 | |
Tyler Goodlet | 205e254072 | |
Tyler Goodlet | 9bd5226e76 | |
Tyler Goodlet | e899cc42bf | |
goodboy | f7c9056419 | |
Tyler Goodlet | faaecbf810 | |
Tyler Goodlet | 703dee8a59 | |
Tyler Goodlet | df59071747 | |
Tyler Goodlet | 4856285dee | |
Tyler Goodlet | efba5229fc | |
Tyler Goodlet | fd6f4574ce | |
Tyler Goodlet | 52a2b7a5ed | |
Tyler Goodlet | 63ecae70c4 | |
Tyler Goodlet | 4b40599c48 | |
Tyler Goodlet | a79cdc7b44 | |
Tyler Goodlet | c9132de7dc | |
Tyler Goodlet | 1f8e1cccbb | |
Tyler Goodlet | 58805a0430 | |
Tyler Goodlet | 142083d81b | |
Tyler Goodlet | 318027ebd1 | |
Tyler Goodlet | b826ec8103 | |
Tyler Goodlet | 4ea5c9b5db | |
Tyler Goodlet | f3432bd8fb | |
Tyler Goodlet | 41a3e6a9ca | |
Tyler Goodlet | 7b9d410c4d | |
Tyler Goodlet | 2b05ffcc23 | |
Tyler Goodlet | 185dbc7e3f | |
Tyler Goodlet | 2680a9473d | |
Tyler Goodlet | 92b540d518 | |
Tyler Goodlet | 6751349987 | |
Tyler Goodlet | d307eab118 | |
Tyler Goodlet | 872b24aedd | |
Tyler Goodlet | c5c3f7e789 | |
Tyler Goodlet | 3f6099f161 | |
Tyler Goodlet | 568902a5a9 | |
Tyler Goodlet | f4793af2b9 | |
goodboy | ae6d751d71 | |
Tyler Goodlet | 94a3cc532c | |
Tyler Goodlet | 08e9593306 | |
Tyler Goodlet | 14f84571fb | |
Tyler Goodlet | e561a4908f | |
Tyler Goodlet | a29924f330 | |
Tyler Goodlet | 46070f99de | |
Tyler Goodlet | d81eb1a51e | |
Tyler Goodlet | d817f1a658 | |
Tyler Goodlet | a23afb0bb8 | |
Tyler Goodlet | 1976e61d1a | |
Tyler Goodlet | 0ac3397dbb | |
Tyler Goodlet | 62b2867e07 | |
Tyler Goodlet | bf6958cdbe | |
goodboy | d05885d650 | |
Tyler Goodlet | 77fc705b1f | |
Tyler Goodlet | 16a3321a38 | |
Tyler Goodlet | 7eb465a699 | |
Tyler Goodlet | 121f7fd844 | |
goodboy | ac821bdd94 | |
Tyler Goodlet | f6de7e0afd | |
Tyler Goodlet | 0e7234aa68 | |
Tyler Goodlet | 83da92d4cb | |
Tyler Goodlet | 57e98b25e7 | |
Tyler Goodlet | 095c94b1d2 | |
Tyler Goodlet | f32ccd76aa | |
goodboy | b527fdbe1a | |
Tyler Goodlet | 6b0366fe04 | |
Tyler Goodlet | dbe5d96d66 | |
goodboy | 08fa55a8c3 | |
Tyler Goodlet | 546e1b2fa3 | |
Tyler Goodlet | 94a6fefede | |
Tyler Goodlet | 74f460eba7 | |
Tyler Goodlet | 4cbb8641de | |
Tyler Goodlet | 7efb7da300 | |
goodboy | 2c12d39617 | |
Tyler Goodlet | 6a063b3814 | |
Tyler Goodlet | 9da1abeecd | |
Tyler Goodlet | 3452e18e6d | |
Tyler Goodlet | 8fdc548676 | |
goodboy | 5dbe8e4b14 | |
goodboy | 9c13827a14 | |
overclockworked64 | 6da76949fd | |
overclockworked64 | 49dd230b4f | |
overclockworked64 | c7f59bd483 | |
Tyler Goodlet | 083b73ad4a | |
goodboy | 925af28092 | |
Tyler Goodlet | d0f5c7a5e2 | |
goodboy | ebf080b8a2 | |
overclockworked64 | 50400359b8 | |
goodboy | 71b8f9f1ea | |
overclockworked64 | b91adcf38d | |
overclockworked64 | 87e3d32992 | |
overclockworked64 | 04895b9d5e | |
overclockworked64 | b7a4641674 | |
overclockworked64 | c1089dbd95 | |
overclockworked64 | 3130a04c61 | |
overclockworked64 | 6f9229cd09 | |
overclockworked64 | 6e6baf250b | |
overclockworked64 | 73cbb2388a | |
overclockworked64 | 2815f1c343 | |
overclockworked64 | 21afc69ac7 | |
overclockworked64 | 7d502cef74 | |
overclockworked64 | 76767a3d7e | |
Tyler Goodlet | c372367cc2 | |
Tyler Goodlet | 9ddd75733c | |
Tyler Goodlet | 8ba10315c1 | |
Tyler Goodlet | 97006c904c | |
Tyler Goodlet | 79fb1d0ebc | |
Tyler Goodlet | 1e917fdb1d | |
Tyler Goodlet | 4114eb1d25 | |
Tyler Goodlet | 680a841282 | |
Tyler Goodlet | 340ddba4ae | |
goodboy | be5582aae3 | |
overclockworked64 | 43cb117bf7 | |
goodboy | 2cf56a5f8b | |
overclockworked64 | 39c8447dfb | |
overclockworked64 | 63ddf119fd | |
goodboy | 828754dbb5 | |
Tyler Goodlet | b3c4851ffb | |
Tyler Goodlet | 5cfac58873 | |
Tyler Goodlet | 5d827f78e2 | |
Tyler Goodlet | 4f222a5f9c | |
Tyler Goodlet | e4ed0fd2b3 | |
Tyler Goodlet | a42ec1f571 | |
Tyler Goodlet | 533457c64d | |
Tyler Goodlet | 51259c4809 | |
Tyler Goodlet | 7ee121aeaf | |
Tyler Goodlet | 9d83ef82b2 | |
Tyler Goodlet | fa317d1600 | |
Tyler Goodlet | 6f5c35dd1b | |
Tyler Goodlet | b14699d40b | |
Tyler Goodlet | daa28ea0e9 | |
Tyler Goodlet | 4b2710b8a5 | |
Tyler Goodlet | d30ce96740 | |
Tyler Goodlet | f3a6ab62af | |
Tyler Goodlet | 62035078ce | |
Tyler Goodlet | 893bad72d5 | |
Tyler Goodlet | 77ec29008d | |
Tyler Goodlet | 2df16c1557 | |
Tyler Goodlet | 46ff558556 | |
Tyler Goodlet | bb9d9c74b1 | |
Tyler Goodlet | 41f0992445 | |
Tyler Goodlet | 7643bbf183 | |
goodboy | dfeebd6382 | |
Tyler Goodlet | 6cda17436a | |
Tyler Goodlet | 1f0cc15675 | |
Tyler Goodlet | 10f66e5141 | |
Tyler Goodlet | 4d5a5c147a | |
Tyler Goodlet | d2f0843041 | |
Tyler Goodlet | 3f6d4d6af4 | |
goodboy | 21e60554cc | |
Tyler Goodlet | b496e790fe | |
Tyler Goodlet | c6dc96b08c | |
Tyler Goodlet | 135459ca25 | |
Tyler Goodlet | ef75883b62 | |
Tyler Goodlet | f7fc464ce8 | |
Tyler Goodlet | 07e8821cd5 | |
Tyler Goodlet | 5b23a3bc35 | |
Tyler Goodlet | 1382ad653d | |
Tyler Goodlet | 076f37c589 | |
Tyler Goodlet | 19d6885243 | |
Tyler Goodlet | dbc4e3dd46 | |
Tyler Goodlet | 93726f1392 | |
Tyler Goodlet | 486e983964 | |
Tyler Goodlet | 1ab495a64d | |
Tyler Goodlet | 562419c907 | |
Tyler Goodlet | 3facfb6d4c | |
Tyler Goodlet | 3eb4c6dce1 | |
Tyler Goodlet | aa080543d0 | |
Tyler Goodlet | 8375002b40 | |
Tyler Goodlet | b64396f708 | |
Tyler Goodlet | 96b3f94c72 | |
Tyler Goodlet | ecd8c4bc7e | |
Tyler Goodlet | 112117c1fc | |
Tyler Goodlet | 95e35f3d60 | |
Tyler Goodlet | e39ee3a9cc | |
Tyler Goodlet | 3771734311 | |
Tyler Goodlet | dda0b22870 | |
Tyler Goodlet | 4079f02acf | |
goodboy | e6763d4daf | |
Tyler Goodlet | 4f831abe25 | |
Tyler Goodlet | 8fd515c7b9 | |
Tyler Goodlet | b1235442fb | |
Tyler Goodlet | d734dcede4 | |
Tyler Goodlet | 518a0d5e14 | |
Tyler Goodlet | 8b416e6bba | |
Tyler Goodlet | bd31f47d5f | |
Tyler Goodlet | 8d79d83ac2 | |
Tyler Goodlet | c1727ce05e | |
goodboy | a868196d13 | |
Tyler Goodlet | 088597ba50 | |
Tyler Goodlet | 91f222983c | |
Tyler Goodlet | 4259738864 | |
goodboy | 2d0e35b316 | |
Tyler Goodlet | 95f2f10b64 | |
Tyler Goodlet | 2b2c73905c | |
Tyler Goodlet | 02307d2656 | |
goodboy | 3f1bc37143 | |
Tyler Goodlet | 1137a9e7ac | |
Tyler Goodlet | bcf5b9fd18 | |
Tyler Goodlet | 2745a2b1dc | |
Tyler Goodlet | 5881a82d2a | |
Tyler Goodlet | b7b489dd07 | |
Tyler Goodlet | d9bb52fe7b | |
Tyler Goodlet | 9258f79510 | |
Tyler Goodlet | 5c6355062c | |
Tyler Goodlet | 44ef26bb18 | |
Tyler Goodlet | d9e793d4ba | |
Tyler Goodlet | 7857a9ac6d | |
Tyler Goodlet | 5182ee7782 | |
Tyler Goodlet | 39cf9af9fc | |
Tyler Goodlet | 63ec740e27 | |
Tyler Goodlet | 0d70e3081a | |
Tyler Goodlet | 093e7d921c | |
Tyler Goodlet | d7ad8982ff | |
Tyler Goodlet | bec3f5999d | |
Tyler Goodlet | 2bad2bac50 | |
Tyler Goodlet | a4cb0ef21f | |
Tyler Goodlet | 236ed0b0dd | |
Tyler Goodlet | 346b5d2eda | |
Tyler Goodlet | 6c17c7367a | |
Tyler Goodlet | 2d1c24112b | |
Tyler Goodlet | a12b1fc631 | |
Tyler Goodlet | ceed96aa3f | |
Tyler Goodlet | 6e78bcf898 | |
Tyler Goodlet | 4ad75a3287 | |
Tyler Goodlet | 64358f6525 | |
Tyler Goodlet | 1af7dbb732 | |
Tyler Goodlet | 6a2c3da1bb | |
Tyler Goodlet | 3817b4fb5e | |
goodboy | 87ce6c8eb3 | |
Tyler Goodlet | 76342ed0c5 | |
Tyler Goodlet | 3208b67f57 | |
Tyler Goodlet | 61d2307e52 | |
Tyler Goodlet | 79f0d6fda0 | |
Tyler Goodlet | 4f166500d0 | |
Tyler Goodlet | d906c81f14 | |
Tyler Goodlet | 68d56d5df0 | |
Tyler Goodlet | 497fa72c96 | |
goodboy | e5845b5d36 | |
goodboy | 7e98afa685 | |
Tyler Goodlet | 22a79219a1 | |
Tyler Goodlet | 3919c9739e | |
Tyler Goodlet | 558c44fdbe | |
Tyler Goodlet | b4d95e9543 | |
Tyler Goodlet | af85d35685 | |
Tyler Goodlet | b176628206 | |
Tyler Goodlet | 47a469484d | |
goodboy | 07e43f88bf | |
Tyler Goodlet | a221949e8f | |
Tyler Goodlet | fc76e97a45 | |
goodboy | a105e32e34 | |
Tyler Goodlet | ace1b1312c | |
Tyler Goodlet | 7431e8ea01 | |
Tyler Goodlet | 82999801a6 | |
Tyler Goodlet | c5c7e694ec | |
goodboy | b01f594025 | |
Tyler Goodlet | a84a27c6d3 | |
goodboy | 9cfec2d3b5 | |
Tyler Goodlet | 8a4a11b885 | |
goodboy | 14379a0f46 | |
Tyler Goodlet | 674fbbc6b3 | |
Tyler Goodlet | f173012fea | |
Tyler Goodlet | 6006adc0de | |
Tyler Goodlet | 0afa7f0f8e | |
Tyler Goodlet | b3d28a1ee4 | |
Tyler Goodlet | 13b76c9439 | |
Tyler Goodlet | 632c666a7d | |
Tyler Goodlet | 09f00a5a00 | |
Tyler Goodlet | 44bfacc0c2 | |
Tyler Goodlet | 551816e80d | |
Tyler Goodlet | be1fcb2a5b | |
Tyler Goodlet | ef89ed947a | |
Tyler Goodlet | 5b3894827f | |
Tyler Goodlet | 0fdcfa0ba1 | |
Tyler Goodlet | 37a1897c47 | |
Tyler Goodlet | 0f2a39a311 | |
Tyler Goodlet | 23a1622256 | |
Tyler Goodlet | 63bdddd0c9 | |
Tyler Goodlet | 49d439b681 | |
Tyler Goodlet | 6f05f5d5e6 | |
Tyler Goodlet | b369b91357 | |
Tyler Goodlet | 969bce3aa4 | |
goodboy | 54d8c93f1b | |
Tyler Goodlet | 240f591234 | |
Tyler Goodlet | 69bbf6a957 | |
Tyler Goodlet | 443ebea165 | |
Tyler Goodlet | 25779d48a8 | |
Tyler Goodlet | fde52d2464 | |
Tyler Goodlet | 8c927d708d | |
Tyler Goodlet | 31590e82a3 | |
Tyler Goodlet | 929b6dcc83 | |
Tyler Goodlet | 2513c652c1 | |
Tyler Goodlet | 9ddb654783 | |
Tyler Goodlet | 7f86d63e77 | |
Tyler Goodlet | 12f987514d | |
Tyler Goodlet | 98bbf8e0df | |
Tyler Goodlet | b1cd7fdedf | |
Tyler Goodlet | ef725c5972 | |
Tyler Goodlet | a134bc490f | |
Tyler Goodlet | 0623de0b47 | |
Tyler Goodlet | b21e2a6caa | |
Tyler Goodlet | c6cdaf9c31 | |
Tyler Goodlet | 91640facbc | |
Tyler Goodlet | c2484e88a1 | |
Tyler Goodlet | 3423ea4011 | |
Tyler Goodlet | af701c16ee | |
Tyler Goodlet | 1703171bea | |
Tyler Goodlet | 3d633408fc | |
Tyler Goodlet | 8eb889a745 | |
Tyler Goodlet | 349d82d182 | |
Tyler Goodlet | 7c5fd8ce9f | |
Tyler Goodlet | 196dea80db | |
Tyler Goodlet | 54916be601 | |
Tyler Goodlet | 1a69727b75 | |
Tyler Goodlet | 348148ff1e | |
Tyler Goodlet | 73302d9d16 | |
Tyler Goodlet | 409f7f0d5a | |
Tyler Goodlet | eb3662f981 | |
Tyler Goodlet | 3999849b03 | |
Tyler Goodlet | 39b9896a62 | |
Tyler Goodlet | 5b8b7d374a | |
Tyler Goodlet | 9a4244b9a6 | |
Tyler Goodlet | a2e2f7e7a8 | |
Tyler Goodlet | 6559fb72aa | |
Tyler Goodlet | e5bc07f355 | |
Tyler Goodlet | e311430d25 | |
Tyler Goodlet | 08eb6bd019 | |
Tyler Goodlet | 98133a984e | |
Tyler Goodlet | 1f8966ba64 | |
Tyler Goodlet | 4240efc7e3 | |
Tyler Goodlet | 4846c6d498 | |
Tyler Goodlet | 14114547e8 | |
Tyler Goodlet | e3955bb62b | |
goodboy | 4d530deac3 | |
Tyler Goodlet | 55760b3fe0 | |
Tyler Goodlet | 6aab16f877 | |
Tyler Goodlet | caa70245e0 | |
Tyler Goodlet | 9c9309faf8 | |
Tyler Goodlet | 3f75732b02 | |
Tyler Goodlet | 1edf5c2f06 | |
Tyler Goodlet | a2d400583f | |
Tyler Goodlet | 85246d2df3 | |
Tyler Goodlet | b372f4c92b | |
Tyler Goodlet | 32b4ae0603 | |
Tyler Goodlet | 80e100f818 | |
Tyler Goodlet | 44d7988204 | |
goodboy | e98302212a | |
Tyler Goodlet | 0301d105dd | |
Tyler Goodlet | 4ee7db338d | |
Tyler Goodlet | 558d546c8f | |
Tyler Goodlet | 5528a4eb45 | |
Tyler Goodlet | fb23a9d8d4 | |
goodboy | f48548ab94 | |
Tyler Goodlet | 73e123bac7 | |
Tyler Goodlet | 2b4cf59ee1 | |
Tyler Goodlet | 0551756e22 | |
Tyler Goodlet | 5ca963148e | |
Tyler Goodlet | 4798d3b5db | |
Tyler Goodlet | 247483ee93 | |
Tyler Goodlet | 9e64161538 | |
Tyler Goodlet | 3bddf9a94b | |
Tyler Goodlet | 1eedd463cb | |
Tyler Goodlet | b46e60ab9d | |
Tyler Goodlet | cb527c2562 | |
Tyler Goodlet | 1584c547cd | |
Tyler Goodlet | 2efd8ed167 | |
Tyler Goodlet | a54260a67e | |
Tyler Goodlet | 98a0594c26 | |
goodboy | ffd10e193e | |
Tyler Goodlet | 87971de1d9 | |
Tyler Goodlet | 9f38406e85 | |
Tyler Goodlet | c4b42000eb | |
Tyler Goodlet | 607c48f1ac | |
goodboy | af93b8532a | |
Tyler Goodlet | fc36e73628 | |
Tyler Goodlet | b1f657e246 | |
Tyler Goodlet | 2498a4963b | |
Tyler Goodlet | 5a5e6baad1 | |
Tyler Goodlet | f59346d854 | |
Tyler Goodlet | 86fc418050 | |
Tyler Goodlet | 83af295b45 | |
Tyler Goodlet | ad9256bcdb | |
Tyler Goodlet | 3e19fd311b | |
Tyler Goodlet | 80c96cab01 | |
Tyler Goodlet | 36251357b3 | |
Tyler Goodlet | 81f3558494 | |
Tyler Goodlet | 897ab79946 | |
Tyler Goodlet | 7f38b7225d | |
goodboy | a5a88e2f64 | |
Tyler Goodlet | d0eacc3fd6 | |
Tyler Goodlet | 89ce1a63e4 | |
Tyler Goodlet | 1f1619c730 | |
Tyler Goodlet | 5798ef6796 | |
Tyler Goodlet | c2a1612bf5 | |
Tyler Goodlet | be22a2526a | |
Tyler Goodlet | a13b3fe0a5 | |
goodboy | 0f4f7f05cb | |
Tyler Goodlet | 34064cd2c7 | |
goodboy | ac30374a15 | |
Tyler Goodlet | e80ab60e0b | |
Tyler Goodlet | c07a5f6e11 | |
Tyler Goodlet | 3766731de5 | |
Tyler Goodlet | 2f8dd0199d | |
goodboy | a0981be1fa | |
Tyler Goodlet | 0679eb5646 | |
Tyler Goodlet | ab192741ce | |
Tyler Goodlet | 8ee9007798 | |
Tyler Goodlet | f342c3a0c4 | |
Tyler Goodlet | 0c8f9dbce0 | |
Tyler Goodlet | 71a35aadef | |
Tyler Goodlet | 4a512bc879 | |
Tyler Goodlet | 0e7db46631 | |
Tyler Goodlet | 92f4b402ad | |
Tyler Goodlet | 90c987d0ae | |
Tyler Goodlet | 0a5a4d8487 | |
goodboy | 49a02e6700 | |
Tyler Goodlet | 47565cfbf3 | |
Tyler Goodlet | cd636b270e | |
Tyler Goodlet | b7b2436bc1 | |
Tyler Goodlet | 8fabd27dbe | |
Tyler Goodlet | 983e66b31b | |
Tyler Goodlet | b285db4c58 | |
goodboy | 35775c6763 | |
Tyler Goodlet | 2b3beac4b4 | |
goodboy | 35dc56d2c5 | |
Tyler Goodlet | a93321e48e | |
Tyler Goodlet | 5ffd2d2ab3 | |
Tyler Goodlet | f7e1c526c5 | |
Tyler Goodlet | 07653bc02e | |
Tyler Goodlet | a90a2b8787 | |
Tyler Goodlet | da8c8c1773 | |
Tyler Goodlet | 57a24cdcf8 | |
Tyler Goodlet | 9b07e9ad7c | |
Tyler Goodlet | 3c320f467f | |
Tyler Goodlet | 2555765882 | |
Tyler Goodlet | 7db5739143 | |
Tyler Goodlet | 7888ef6f01 | |
Tyler Goodlet | 109066dda9 | |
goodboy | d8b6c0093c | |
goodboy | 8fdab8e0be | |
Tyler Goodlet | 9f4e497b9c | |
Tyler Goodlet | 14d60147fa | |
Tyler Goodlet | e546ead2ff | |
Tyler Goodlet | 3df001f3a9 | |
Tyler Goodlet | 5ed5d18ccb | |
goodboy | dfaf1e3631 | |
Tyler Goodlet | 32b10681a1 | |
Tyler Goodlet | 41a4de5af2 | |
Tyler Goodlet | 59421d9f3a | |
Tyler Goodlet | 333ddcf93f | |
Tyler Goodlet | 0bb2163b0c | |
Tyler Goodlet | bd3059f01b | |
Tyler Goodlet | 803152ead5 | |
Tyler Goodlet | e6245671b0 | |
goodboy | bfe500060f | |
goodboy | 3a5daa5b7a | |
Tyler Goodlet | 723fb17394 | |
Tyler Goodlet | f05534e472 | |
Tyler Goodlet | b040cdc0c9 | |
Tyler Goodlet | 02ac20a43c | |
goodboy | f427c98cf6 | |
Tyler Goodlet | 5127effd88 | |
Tyler Goodlet | 6b650c0fe6 | |
Tyler Goodlet | 0d05a727b6 | |
Tyler Goodlet | c28ffd8b1c | |
Tyler Goodlet | 5d7a4e2b12 | |
Tyler Goodlet | 8522f90000 | |
goodboy | f4f39c29f3 |
|
@ -1,41 +1,131 @@
|
||||||
name: CI
|
name: CI
|
||||||
|
|
||||||
on: push
|
on:
|
||||||
|
# any time someone pushes a new branch to origin
|
||||||
|
push:
|
||||||
|
|
||||||
|
# Allows you to run this workflow manually from the Actions tab
|
||||||
|
workflow_dispatch:
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
|
|
||||||
mypy:
|
mypy:
|
||||||
name: 'MyPy'
|
name: 'MyPy'
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v2
|
uses: actions/checkout@v2
|
||||||
|
|
||||||
- name: Setup python
|
- name: Setup python
|
||||||
uses: actions/setup-python@v2
|
uses: actions/setup-python@v2
|
||||||
with:
|
with:
|
||||||
python-version: '3.8'
|
python-version: '3.10'
|
||||||
- name: Install dependencies
|
|
||||||
run: pip install -U . --upgrade-strategy eager
|
- name: Install dependencies
|
||||||
- name: Run MyPy check
|
run: pip install -U . --upgrade-strategy eager -r requirements-test.txt
|
||||||
run: mypy tractor/ --ignore-missing-imports
|
|
||||||
|
- name: Run MyPy check
|
||||||
|
run: mypy tractor/ --ignore-missing-imports --show-traceback
|
||||||
|
|
||||||
|
# test that we can generate a software distribution and install it
|
||||||
|
# thus avoid missing file issues after packaging.
|
||||||
|
sdist-linux:
|
||||||
|
name: 'sdist'
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
testing:
|
|
||||||
name: '${{ matrix.os }} Python ${{ matrix.python }} - ${{ matrix.spawn_backend }}'
|
|
||||||
timeout-minutes: 10
|
|
||||||
runs-on: ${{ matrix.os }}
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
os: [ubuntu-latest, windows-latest]
|
|
||||||
python: ['3.7', '3.8']
|
|
||||||
spawn_backend: ['trio', 'mp']
|
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v2
|
uses: actions/checkout@v2
|
||||||
|
|
||||||
|
- name: Setup python
|
||||||
|
uses: actions/setup-python@v2
|
||||||
|
with:
|
||||||
|
python-version: '3.10'
|
||||||
|
|
||||||
|
- name: Build sdist
|
||||||
|
run: python setup.py sdist --formats=zip
|
||||||
|
|
||||||
|
- name: Install sdist from .zips
|
||||||
|
run: python -m pip install dist/*.zip
|
||||||
|
|
||||||
|
|
||||||
|
testing-linux:
|
||||||
|
name: '${{ matrix.os }} Python ${{ matrix.python }} - ${{ matrix.spawn_backend }}'
|
||||||
|
timeout-minutes: 10
|
||||||
|
runs-on: ${{ matrix.os }}
|
||||||
|
|
||||||
|
strategy:
|
||||||
|
fail-fast: false
|
||||||
|
matrix:
|
||||||
|
os: [ubuntu-latest]
|
||||||
|
python: ['3.10']
|
||||||
|
spawn_backend: [
|
||||||
|
'trio',
|
||||||
|
'mp_spawn',
|
||||||
|
'mp_forkserver',
|
||||||
|
]
|
||||||
|
|
||||||
|
steps:
|
||||||
|
|
||||||
|
- name: Checkout
|
||||||
|
uses: actions/checkout@v2
|
||||||
|
|
||||||
- name: Setup python
|
- name: Setup python
|
||||||
uses: actions/setup-python@v2
|
uses: actions/setup-python@v2
|
||||||
with:
|
with:
|
||||||
python-version: '${{ matrix.python }}'
|
python-version: '${{ matrix.python }}'
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: pip install -U . -r requirements-test.txt -r requirements-docs.txt --upgrade-strategy eager
|
run: pip install -U . -r requirements-test.txt -r requirements-docs.txt --upgrade-strategy eager
|
||||||
|
|
||||||
|
- name: List dependencies
|
||||||
|
run: pip list
|
||||||
|
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: pytest tests/ --spawn-backend=${{ matrix.spawn_backend }} -rs
|
run: pytest tests/ --spawn-backend=${{ matrix.spawn_backend }} -rsx
|
||||||
|
|
||||||
|
# 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
|
||||||
|
# welcome, but our primary user base is not using that OS.
|
||||||
|
|
||||||
|
# TODO: use job filtering to accomplish instead of repeated
|
||||||
|
# boilerplate as is above XD:
|
||||||
|
# - https://docs.github.com/en/actions/learn-github-actions/managing-complex-workflows
|
||||||
|
# - https://docs.github.com/en/actions/learn-github-actions/managing-complex-workflows#using-a-build-matrix
|
||||||
|
# - https://docs.github.com/en/actions/learn-github-actions/workflow-syntax-for-github-actions#jobsjob_idif
|
||||||
|
# testing-windows:
|
||||||
|
# name: '${{ matrix.os }} Python ${{ matrix.python }} - ${{ matrix.spawn_backend }}'
|
||||||
|
# timeout-minutes: 12
|
||||||
|
# runs-on: ${{ matrix.os }}
|
||||||
|
|
||||||
|
# strategy:
|
||||||
|
# fail-fast: false
|
||||||
|
# matrix:
|
||||||
|
# os: [windows-latest]
|
||||||
|
# python: ['3.10']
|
||||||
|
# spawn_backend: ['trio', 'mp']
|
||||||
|
|
||||||
|
# steps:
|
||||||
|
|
||||||
|
# - name: Checkout
|
||||||
|
# uses: actions/checkout@v2
|
||||||
|
|
||||||
|
# - name: Setup python
|
||||||
|
# uses: actions/setup-python@v2
|
||||||
|
# with:
|
||||||
|
# python-version: '${{ matrix.python }}'
|
||||||
|
|
||||||
|
# - name: Install dependencies
|
||||||
|
# run: pip install -U . -r requirements-test.txt -r requirements-docs.txt --upgrade-strategy eager
|
||||||
|
|
||||||
|
# # TODO: pretty sure this solves debugger deps-issues on windows, but it needs to
|
||||||
|
# # be verified by someone with a native setup.
|
||||||
|
# # - name: Force pyreadline3
|
||||||
|
# # run: pip uninstall pyreadline; pip install -U pyreadline3
|
||||||
|
|
||||||
|
# - name: List dependencies
|
||||||
|
# run: pip list
|
||||||
|
|
||||||
|
# - name: Run tests
|
||||||
|
# run: pytest tests/ --spawn-backend=${{ matrix.spawn_backend }} -rsx
|
||||||
|
|
147
LICENSE
147
LICENSE
|
@ -1,23 +1,21 @@
|
||||||
GNU GENERAL PUBLIC LICENSE
|
GNU AFFERO GENERAL PUBLIC LICENSE
|
||||||
Version 3, 29 June 2007
|
Version 3, 19 November 2007
|
||||||
|
|
||||||
Copyright (C) 2007 Free Software Foundation, Inc. <http://fsf.org/>
|
Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
|
||||||
Everyone is permitted to copy and distribute verbatim copies
|
Everyone is permitted to copy and distribute verbatim copies
|
||||||
of this license document, but changing it is not allowed.
|
of this license document, but changing it is not allowed.
|
||||||
|
|
||||||
Preamble
|
Preamble
|
||||||
|
|
||||||
The GNU General Public License is a free, copyleft license for
|
The GNU Affero General Public License is a free, copyleft license for
|
||||||
software and other kinds of works.
|
software and other kinds of works, specifically designed to ensure
|
||||||
|
cooperation with the community in the case of network server software.
|
||||||
|
|
||||||
The licenses for most software and other practical works are designed
|
The licenses for most software and other practical works are designed
|
||||||
to take away your freedom to share and change the works. By contrast,
|
to take away your freedom to share and change the works. By contrast,
|
||||||
the GNU General Public License is intended to guarantee your freedom to
|
our General Public Licenses are intended to guarantee your freedom to
|
||||||
share and change all versions of a program--to make sure it remains free
|
share and change all versions of a program--to make sure it remains free
|
||||||
software for all its users. We, the Free Software Foundation, use the
|
software for all its users.
|
||||||
GNU General Public License for most of our software; it applies also to
|
|
||||||
any other work released this way by its authors. You can apply it to
|
|
||||||
your programs, too.
|
|
||||||
|
|
||||||
When we speak of free software, we are referring to freedom, not
|
When we speak of free software, we are referring to freedom, not
|
||||||
price. Our General Public Licenses are designed to make sure that you
|
price. Our General Public Licenses are designed to make sure that you
|
||||||
|
@ -26,44 +24,34 @@ them if you wish), that you receive source code or can get it if you
|
||||||
want it, that you can change the software or use pieces of it in new
|
want it, that you can change the software or use pieces of it in new
|
||||||
free programs, and that you know you can do these things.
|
free programs, and that you know you can do these things.
|
||||||
|
|
||||||
To protect your rights, we need to prevent others from denying you
|
Developers that use our General Public Licenses protect your rights
|
||||||
these rights or asking you to surrender the rights. Therefore, you have
|
with two steps: (1) assert copyright on the software, and (2) offer
|
||||||
certain responsibilities if you distribute copies of the software, or if
|
you this License which gives you legal permission to copy, distribute
|
||||||
you modify it: responsibilities to respect the freedom of others.
|
and/or modify the software.
|
||||||
|
|
||||||
For example, if you distribute copies of such a program, whether
|
A secondary benefit of defending all users' freedom is that
|
||||||
gratis or for a fee, you must pass on to the recipients the same
|
improvements made in alternate versions of the program, if they
|
||||||
freedoms that you received. You must make sure that they, too, receive
|
receive widespread use, become available for other developers to
|
||||||
or can get the source code. And you must show them these terms so they
|
incorporate. Many developers of free software are heartened and
|
||||||
know their rights.
|
encouraged by the resulting cooperation. However, in the case of
|
||||||
|
software used on network servers, this result may fail to come about.
|
||||||
|
The GNU General Public License permits making a modified version and
|
||||||
|
letting the public access it on a server without ever releasing its
|
||||||
|
source code to the public.
|
||||||
|
|
||||||
Developers that use the GNU GPL protect your rights with two steps:
|
The GNU Affero General Public License is designed specifically to
|
||||||
(1) assert copyright on the software, and (2) offer you this License
|
ensure that, in such cases, the modified source code becomes available
|
||||||
giving you legal permission to copy, distribute and/or modify it.
|
to the community. It requires the operator of a network server to
|
||||||
|
provide the source code of the modified version running there to the
|
||||||
|
users of that server. Therefore, public use of a modified version, on
|
||||||
|
a publicly accessible server, gives the public access to the source
|
||||||
|
code of the modified version.
|
||||||
|
|
||||||
For the developers' and authors' protection, the GPL clearly explains
|
An older license, called the Affero General Public License and
|
||||||
that there is no warranty for this free software. For both users' and
|
published by Affero, was designed to accomplish similar goals. This is
|
||||||
authors' sake, the GPL requires that modified versions be marked as
|
a different license, not a version of the Affero GPL, but Affero has
|
||||||
changed, so that their problems will not be attributed erroneously to
|
released a new version of the Affero GPL which permits relicensing under
|
||||||
authors of previous versions.
|
this license.
|
||||||
|
|
||||||
Some devices are designed to deny users access to install or run
|
|
||||||
modified versions of the software inside them, although the manufacturer
|
|
||||||
can do so. This is fundamentally incompatible with the aim of
|
|
||||||
protecting users' freedom to change the software. The systematic
|
|
||||||
pattern of such abuse occurs in the area of products for individuals to
|
|
||||||
use, which is precisely where it is most unacceptable. Therefore, we
|
|
||||||
have designed this version of the GPL to prohibit the practice for those
|
|
||||||
products. If such problems arise substantially in other domains, we
|
|
||||||
stand ready to extend this provision to those domains in future versions
|
|
||||||
of the GPL, as needed to protect the freedom of users.
|
|
||||||
|
|
||||||
Finally, every program is threatened constantly by software patents.
|
|
||||||
States should not allow patents to restrict development and use of
|
|
||||||
software on general-purpose computers, but in those that do, we wish to
|
|
||||||
avoid the special danger that patents applied to a free program could
|
|
||||||
make it effectively proprietary. To prevent this, the GPL assures that
|
|
||||||
patents cannot be used to render the program non-free.
|
|
||||||
|
|
||||||
The precise terms and conditions for copying, distribution and
|
The precise terms and conditions for copying, distribution and
|
||||||
modification follow.
|
modification follow.
|
||||||
|
@ -72,7 +60,7 @@ modification follow.
|
||||||
|
|
||||||
0. Definitions.
|
0. Definitions.
|
||||||
|
|
||||||
"This License" refers to version 3 of the GNU General Public License.
|
"This License" refers to version 3 of the GNU Affero General Public License.
|
||||||
|
|
||||||
"Copyright" also means copyright-like laws that apply to other kinds of
|
"Copyright" also means copyright-like laws that apply to other kinds of
|
||||||
works, such as semiconductor masks.
|
works, such as semiconductor masks.
|
||||||
|
@ -549,35 +537,45 @@ to collect a royalty for further conveying from those to whom you convey
|
||||||
the Program, the only way you could satisfy both those terms and this
|
the Program, the only way you could satisfy both those terms and this
|
||||||
License would be to refrain entirely from conveying the Program.
|
License would be to refrain entirely from conveying the Program.
|
||||||
|
|
||||||
13. Use with the GNU Affero General Public License.
|
13. Remote Network Interaction; Use with the GNU General Public License.
|
||||||
|
|
||||||
|
Notwithstanding any other provision of this License, if you modify the
|
||||||
|
Program, your modified version must prominently offer all users
|
||||||
|
interacting with it remotely through a computer network (if your version
|
||||||
|
supports such interaction) an opportunity to receive the Corresponding
|
||||||
|
Source of your version by providing access to the Corresponding Source
|
||||||
|
from a network server at no charge, through some standard or customary
|
||||||
|
means of facilitating copying of software. This Corresponding Source
|
||||||
|
shall include the Corresponding Source for any work covered by version 3
|
||||||
|
of the GNU General Public License that is incorporated pursuant to the
|
||||||
|
following paragraph.
|
||||||
|
|
||||||
Notwithstanding any other provision of this License, you have
|
Notwithstanding any other provision of this License, you have
|
||||||
permission to link or combine any covered work with a work licensed
|
permission to link or combine any covered work with a work licensed
|
||||||
under version 3 of the GNU Affero General Public License into a single
|
under version 3 of the GNU General Public License into a single
|
||||||
combined work, and to convey the resulting work. The terms of this
|
combined work, and to convey the resulting work. The terms of this
|
||||||
License will continue to apply to the part which is the covered work,
|
License will continue to apply to the part which is the covered work,
|
||||||
but the special requirements of the GNU Affero General Public License,
|
but the work with which it is combined will remain governed by version
|
||||||
section 13, concerning interaction through a network will apply to the
|
3 of the GNU General Public License.
|
||||||
combination as such.
|
|
||||||
|
|
||||||
14. Revised Versions of this License.
|
14. Revised Versions of this License.
|
||||||
|
|
||||||
The Free Software Foundation may publish revised and/or new versions of
|
The Free Software Foundation may publish revised and/or new versions of
|
||||||
the GNU General Public License from time to time. Such new versions will
|
the GNU Affero General Public License from time to time. Such new versions
|
||||||
be similar in spirit to the present version, but may differ in detail to
|
will be similar in spirit to the present version, but may differ in detail to
|
||||||
address new problems or concerns.
|
address new problems or concerns.
|
||||||
|
|
||||||
Each version is given a distinguishing version number. If the
|
Each version is given a distinguishing version number. If the
|
||||||
Program specifies that a certain numbered version of the GNU General
|
Program specifies that a certain numbered version of the GNU Affero General
|
||||||
Public License "or any later version" applies to it, you have the
|
Public License "or any later version" applies to it, you have the
|
||||||
option of following the terms and conditions either of that numbered
|
option of following the terms and conditions either of that numbered
|
||||||
version or of any later version published by the Free Software
|
version or of any later version published by the Free Software
|
||||||
Foundation. If the Program does not specify a version number of the
|
Foundation. If the Program does not specify a version number of the
|
||||||
GNU General Public License, you may choose any version ever published
|
GNU Affero General Public License, you may choose any version ever published
|
||||||
by the Free Software Foundation.
|
by the Free Software Foundation.
|
||||||
|
|
||||||
If the Program specifies that a proxy can decide which future
|
If the Program specifies that a proxy can decide which future
|
||||||
versions of the GNU General Public License can be used, that proxy's
|
versions of the GNU Affero General Public License can be used, that proxy's
|
||||||
public statement of acceptance of a version permanently authorizes you
|
public statement of acceptance of a version permanently authorizes you
|
||||||
to choose that version for the Program.
|
to choose that version for the Program.
|
||||||
|
|
||||||
|
@ -635,40 +633,29 @@ the "copyright" line and a pointer to where the full notice is found.
|
||||||
Copyright (C) <year> <name of author>
|
Copyright (C) <year> <name of author>
|
||||||
|
|
||||||
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 General Public License as published by
|
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
|
the Free Software Foundation, either version 3 of the License, or
|
||||||
(at your option) any later version.
|
(at your option) any later version.
|
||||||
|
|
||||||
This program is distributed in the hope that it will be useful,
|
This program is distributed in the hope that it will be useful,
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
GNU General Public License for more details.
|
GNU Affero General Public License for more details.
|
||||||
|
|
||||||
You should have received a copy of the GNU General Public License
|
You should have received a copy of the GNU Affero General Public License
|
||||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
Also add information on how to contact you by electronic and paper mail.
|
Also add information on how to contact you by electronic and paper mail.
|
||||||
|
|
||||||
If the program does terminal interaction, make it output a short
|
If your software can interact with users remotely through a computer
|
||||||
notice like this when it starts in an interactive mode:
|
network, you should also make sure that it provides a way for users to
|
||||||
|
get its source. For example, if your program is a web application, its
|
||||||
<program> Copyright (C) <year> <name of author>
|
interface could display a "Source" link that leads users to an archive
|
||||||
This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
|
of the code. There are many ways you could offer source, and different
|
||||||
This is free software, and you are welcome to redistribute it
|
solutions will be better for different programs; see section 13 for the
|
||||||
under certain conditions; type `show c' for details.
|
specific requirements.
|
||||||
|
|
||||||
The hypothetical commands `show w' and `show c' should show the appropriate
|
|
||||||
parts of the General Public License. Of course, your program's commands
|
|
||||||
might be different; for a GUI interface, you would use an "about box".
|
|
||||||
|
|
||||||
You should also get your employer (if you work as a programmer) or school,
|
You should also get your employer (if you work as a programmer) or school,
|
||||||
if any, to sign a "copyright disclaimer" for the program, if necessary.
|
if any, to sign a "copyright disclaimer" for the program, if necessary.
|
||||||
For more information on this, and how to apply and follow the GNU GPL, see
|
For more information on this, and how to apply and follow the GNU AGPL, see
|
||||||
<http://www.gnu.org/licenses/>.
|
<https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
The GNU General Public License does not permit incorporating your program
|
|
||||||
into proprietary programs. If your program is a subroutine library, you
|
|
||||||
may consider it more useful to permit linking proprietary applications with
|
|
||||||
the library. If this is what you want to do, use the GNU Lesser General
|
|
||||||
Public License instead of this License. But first, please read
|
|
||||||
<http://www.gnu.org/philosophy/why-not-lgpl.html>.
|
|
||||||
|
|
|
@ -0,0 +1,2 @@
|
||||||
|
# https://packaging.python.org/en/latest/guides/using-manifest-in/#using-manifest-in
|
||||||
|
include docs/README.rst
|
|
@ -0,0 +1,528 @@
|
||||||
|
=========
|
||||||
|
Changelog
|
||||||
|
=========
|
||||||
|
|
||||||
|
.. towncrier release notes start
|
||||||
|
|
||||||
|
tractor 0.1.0a5 (2022-08-03)
|
||||||
|
============================
|
||||||
|
|
||||||
|
This is our final release supporting Python 3.9 since we will be moving
|
||||||
|
internals to the new `match:` syntax from 3.10 going forward and
|
||||||
|
further, we have officially dropped usage of the `msgpack` library and
|
||||||
|
happily adopted `msgspec`.
|
||||||
|
|
||||||
|
Features
|
||||||
|
--------
|
||||||
|
|
||||||
|
- `#165 <https://github.com/goodboy/tractor/issues/165>`_: Add SIGINT
|
||||||
|
protection to our `pdbpp` based debugger subystem such that for
|
||||||
|
(single-depth) actor trees in debug mode we ignore interrupts in any
|
||||||
|
actor currently holding the TTY lock thus avoiding clobbering IPC
|
||||||
|
connections and/or task and process state when working in the REPL.
|
||||||
|
|
||||||
|
As a big note currently so called "nested" actor trees (trees with
|
||||||
|
actors having more then one parent/ancestor) are not fully supported
|
||||||
|
since we don't yet have a mechanism to relay the debug mode knowledge
|
||||||
|
"up" the actor tree (for eg. when handling a crash in a leaf actor).
|
||||||
|
As such currently there is a set of tests and known scenarios which will
|
||||||
|
result in process cloberring by the zombie repaing machinery and these
|
||||||
|
have been documented in https://github.com/goodboy/tractor/issues/320.
|
||||||
|
|
||||||
|
The implementation details include:
|
||||||
|
|
||||||
|
- utilizing a custom SIGINT handler which we apply whenever an actor's
|
||||||
|
runtime enters the debug machinery, which we also make sure the
|
||||||
|
stdlib's `pdb` configuration doesn't override (which it does by
|
||||||
|
default without special instance config).
|
||||||
|
- litter the runtime with `maybe_wait_for_debugger()` mostly in spots
|
||||||
|
where the root actor should block before doing embedded nursery
|
||||||
|
teardown ops which both cancel potential-children-in-deubg as well
|
||||||
|
as eventually trigger zombie reaping machinery.
|
||||||
|
- hardening of the TTY locking semantics/API both in terms of IPC
|
||||||
|
terminations and cancellation and lock release determinism from
|
||||||
|
sync debugger instance methods.
|
||||||
|
- factoring of locking infrastructure into a new `._debug.Lock` global
|
||||||
|
which encapsulates all details of the ``trio`` sync primitives and
|
||||||
|
task/actor uid management and tracking.
|
||||||
|
|
||||||
|
We also add `ctrl-c` cases throughout the test suite though these are
|
||||||
|
disabled for py3.9 (`pdbpp` UX differences that don't seem worth
|
||||||
|
compensating for, especially since this will be our last 3.9 supported
|
||||||
|
release) and there are a slew of marked cases that aren't expected to
|
||||||
|
work in CI more generally (as mentioned in the "nested" tree note
|
||||||
|
above) despite seemingly working when run manually on linux.
|
||||||
|
|
||||||
|
- `#304 <https://github.com/goodboy/tractor/issues/304>`_: Add a new
|
||||||
|
``to_asyncio.LinkedTaskChannel.subscribe()`` which gives task-oriented
|
||||||
|
broadcast functionality semantically equivalent to
|
||||||
|
``tractor.MsgStream.subscribe()`` this makes it possible for multiple
|
||||||
|
``trio``-side tasks to consume ``asyncio``-side task msgs in tandem.
|
||||||
|
|
||||||
|
Further Improvements to the test suite were added in this patch set
|
||||||
|
including a new scenario test for a sub-actor managed "service nursery"
|
||||||
|
(implementing the basics of a "service manager") including use of
|
||||||
|
*infected asyncio* mode. Further we added a lower level
|
||||||
|
``test_trioisms.py`` to start to track issues we need to work around in
|
||||||
|
``trio`` itself which in this case included a bug we were trying to
|
||||||
|
solve related to https://github.com/python-trio/trio/issues/2258.
|
||||||
|
|
||||||
|
|
||||||
|
Bug Fixes
|
||||||
|
---------
|
||||||
|
|
||||||
|
- `#318 <https://github.com/goodboy/tractor/issues/318>`_: Fix
|
||||||
|
a previously undetected ``trio``-``asyncio`` task lifetime linking
|
||||||
|
issue with the ``to_asyncio.open_channel_from()`` api where both sides
|
||||||
|
where not properly waiting/signalling termination and it was possible
|
||||||
|
for ``asyncio``-side errors to not propagate due to a race condition.
|
||||||
|
|
||||||
|
The implementation fix summary is:
|
||||||
|
- add state to signal the end of the ``trio`` side task to be
|
||||||
|
read by the ``asyncio`` side and always cancel any ongoing
|
||||||
|
task in such cases.
|
||||||
|
- always wait on the ``asyncio`` task termination from the ``trio``
|
||||||
|
side on error before maybe raising said error.
|
||||||
|
- always close the ``trio`` mem chan on exit to ensure the other
|
||||||
|
side can detect it and follow.
|
||||||
|
|
||||||
|
|
||||||
|
Trivial/Internal Changes
|
||||||
|
------------------------
|
||||||
|
|
||||||
|
- `#248 <https://github.com/goodboy/tractor/issues/248>`_: Adjust the
|
||||||
|
`tractor._spawn.soft_wait()` strategy to avoid sending an actor cancel
|
||||||
|
request (via `Portal.cancel_actor()`) if either the child process is
|
||||||
|
detected as having terminated or the IPC channel is detected to be
|
||||||
|
closed.
|
||||||
|
|
||||||
|
This ensures (even) more deterministic inter-actor cancellation by
|
||||||
|
avoiding the timeout condition where possible when a whild never
|
||||||
|
sucessfully spawned, crashed, or became un-contactable over IPC.
|
||||||
|
|
||||||
|
- `#295 <https://github.com/goodboy/tractor/issues/295>`_: Add an
|
||||||
|
experimental ``tractor.msg.NamespacePath`` type for passing Python
|
||||||
|
objects by "reference" through a ``str``-subtype message and using the
|
||||||
|
new ``pkgutil.resolve_name()`` for reference loading.
|
||||||
|
|
||||||
|
- `#298 <https://github.com/goodboy/tractor/issues/298>`_: Add a new
|
||||||
|
`tractor.experimental` subpackage for staging new high level APIs and
|
||||||
|
subystems that we might eventually make built-ins.
|
||||||
|
|
||||||
|
- `#300 <https://github.com/goodboy/tractor/issues/300>`_: Update to and
|
||||||
|
pin latest ``msgpack`` (1.0.3) and ``msgspec`` (0.4.0) both of which
|
||||||
|
required adjustments for backwards imcompatible API tweaks.
|
||||||
|
|
||||||
|
- `#303 <https://github.com/goodboy/tractor/issues/303>`_: Fence off
|
||||||
|
``multiprocessing`` imports until absolutely necessary in an effort to
|
||||||
|
avoid "resource tracker" spawning side effects that seem to have
|
||||||
|
varying degrees of unreliability per Python release. Port to new
|
||||||
|
``msgspec.DecodeError``.
|
||||||
|
|
||||||
|
- `#305 <https://github.com/goodboy/tractor/issues/305>`_: Add
|
||||||
|
``tractor.query_actor()`` an addr looker-upper which doesn't deliver
|
||||||
|
a ``Portal`` instance and instead just a socket address ``tuple``.
|
||||||
|
|
||||||
|
Sometimes it's handy to just have a simple way to figure out if
|
||||||
|
a "service" actor is up, so add this discovery helper for that. We'll
|
||||||
|
prolly just leave it undocumented for now until we figure out
|
||||||
|
a longer-term/better discovery system.
|
||||||
|
|
||||||
|
- `#316 <https://github.com/goodboy/tractor/issues/316>`_: Run windows
|
||||||
|
CI jobs on python 3.10 after some hacks for ``pdbpp`` dependency
|
||||||
|
issues.
|
||||||
|
|
||||||
|
Issue was to do with the now deprecated `pyreadline` project which
|
||||||
|
should be changed over to `pyreadline3`.
|
||||||
|
|
||||||
|
- `#317 <https://github.com/goodboy/tractor/issues/317>`_: Drop use of
|
||||||
|
the ``msgpack`` package and instead move fully to the ``msgspec``
|
||||||
|
codec library.
|
||||||
|
|
||||||
|
We've now used ``msgspec`` extensively in production and there's no
|
||||||
|
reason to not use it as default. Further this change preps us for the up
|
||||||
|
and coming typed messaging semantics (#196), dialog-unprotocol system
|
||||||
|
(#297), and caps-based messaging-protocols (#299) planned before our
|
||||||
|
first beta.
|
||||||
|
|
||||||
|
|
||||||
|
tractor 0.1.0a4 (2021-12-18)
|
||||||
|
============================
|
||||||
|
|
||||||
|
Features
|
||||||
|
--------
|
||||||
|
- `#275 <https://github.com/goodboy/tractor/issues/275>`_: Re-license
|
||||||
|
code base under AGPLv3. Also see `#274
|
||||||
|
<https://github.com/goodboy/tractor/pull/274>`_ for majority
|
||||||
|
contributor consensus on this decision.
|
||||||
|
|
||||||
|
- `#121 <https://github.com/goodboy/tractor/issues/121>`_: Add
|
||||||
|
"infected ``asyncio`` mode; a sub-system to spawn and control
|
||||||
|
``asyncio`` actors using ``trio``'s guest-mode.
|
||||||
|
|
||||||
|
This gets us the following very interesting functionality:
|
||||||
|
|
||||||
|
- ability to spawn an actor that has a process entry point of
|
||||||
|
``asyncio.run()`` by passing ``infect_asyncio=True`` to
|
||||||
|
``Portal.start_actor()`` (and friends).
|
||||||
|
- the ``asyncio`` actor embeds ``trio`` using guest-mode and starts
|
||||||
|
a main ``trio`` task which runs the ``tractor.Actor._async_main()``
|
||||||
|
entry point engages all the normal ``tractor`` runtime IPC/messaging
|
||||||
|
machinery; for all purposes the actor is now running normally on
|
||||||
|
a ``trio.run()``.
|
||||||
|
- the actor can now make one-to-one task spawning requests to the
|
||||||
|
underlying ``asyncio`` event loop using either of:
|
||||||
|
|
||||||
|
* ``to_asyncio.run_task()`` to spawn and run an ``asyncio`` task to
|
||||||
|
completion and block until a return value is delivered.
|
||||||
|
* ``async with to_asyncio.open_channel_from():`` which spawns a task
|
||||||
|
and hands it a pair of "memory channels" to allow for bi-directional
|
||||||
|
streaming between the now SC-linked ``trio`` and ``asyncio`` tasks.
|
||||||
|
|
||||||
|
The output from any call(s) to ``asyncio`` can be handled as normal in
|
||||||
|
``trio``/``tractor`` task operation with the caveat of the overhead due
|
||||||
|
to guest-mode use.
|
||||||
|
|
||||||
|
For more details see the `original PR
|
||||||
|
<https://github.com/goodboy/tractor/pull/121>`_ and `issue
|
||||||
|
<https://github.com/goodboy/tractor/issues/120>`_.
|
||||||
|
|
||||||
|
- `#257 <https://github.com/goodboy/tractor/issues/257>`_: Add
|
||||||
|
``trionics.maybe_open_context()`` an actor-scoped async multi-task
|
||||||
|
context manager resource caching API.
|
||||||
|
|
||||||
|
Adds an SC-safe cacheing async context manager api that only enters on
|
||||||
|
the *first* task entry and only exits on the *last* task exit while in
|
||||||
|
between delivering the same cached value per input key. Keys can be
|
||||||
|
either an explicit ``key`` named arg provided by the user or a
|
||||||
|
hashable ``kwargs`` dict (will be converted to a ``list[tuple]``) which
|
||||||
|
is passed to the underlying manager function as input.
|
||||||
|
|
||||||
|
- `#261 <https://github.com/goodboy/tractor/issues/261>`_: Add
|
||||||
|
cross-actor-task ``Context`` oriented error relay, a new stream
|
||||||
|
overrun error-signal ``StreamOverrun``, and support disabling
|
||||||
|
``MsgStream`` backpressure as the default before a stream is opened or
|
||||||
|
by choice of the user.
|
||||||
|
|
||||||
|
We added stricter semantics around ``tractor.Context.open_stream():``
|
||||||
|
particularly to do with streams which are only opened at one end.
|
||||||
|
Previously, if only one end opened a stream there was no way for that
|
||||||
|
sender to know if msgs are being received until first, the feeder mem
|
||||||
|
chan on the receiver side hit a backpressure state and then that
|
||||||
|
condition delayed its msg loop processing task to eventually create
|
||||||
|
backpressure on the associated IPC transport. This is non-ideal in the
|
||||||
|
case where the receiver side never opened a stream by mistake since it
|
||||||
|
results in silent block of the sender and no adherence to the underlying
|
||||||
|
mem chan buffer size settings (which is still unsolved btw).
|
||||||
|
|
||||||
|
To solve this we add non-backpressure style message pushing inside
|
||||||
|
``Actor._push_result()`` by default and only use the backpressure
|
||||||
|
``trio.MemorySendChannel.send()`` call **iff** the local end of the
|
||||||
|
context has entered ``Context.open_stream():``. This way if the stream
|
||||||
|
was never opened but the mem chan is overrun, we relay back to the
|
||||||
|
sender a (new exception) ``SteamOverrun`` error which is raised in the
|
||||||
|
sender's scope with a special error message about the stream never
|
||||||
|
having been opened. Further, this behaviour (non-backpressure style
|
||||||
|
where senders can expect an error on overruns) can now be enabled with
|
||||||
|
``.open_stream(backpressure=False)`` and the underlying mem chan size
|
||||||
|
can be specified with a kwarg ``msg_buffer_size: int``.
|
||||||
|
|
||||||
|
Further bug fixes and enhancements in this changeset include:
|
||||||
|
|
||||||
|
- fix a race we were ignoring where if the callee task opened a context
|
||||||
|
it could enter ``Context.open_stream()`` before calling
|
||||||
|
``.started()``.
|
||||||
|
- Disallow calling ``Context.started()`` more then once.
|
||||||
|
- Enable ``Context`` linked tasks error relaying via the new
|
||||||
|
``Context._maybe_raise_from_remote_msg()`` which (for now) uses
|
||||||
|
a simple ``trio.Nursery.start_soon()`` to raise the error via closure
|
||||||
|
in the local scope.
|
||||||
|
|
||||||
|
- `#267 <https://github.com/goodboy/tractor/issues/267>`_: This
|
||||||
|
(finally) adds fully acknowledged remote cancellation messaging
|
||||||
|
support for both explicit ``Portal.cancel_actor()`` calls as well as
|
||||||
|
when there is a "runtime-wide" cancellations (eg. during KBI or
|
||||||
|
general actor nursery exception handling which causes a full actor
|
||||||
|
"crash"/termination).
|
||||||
|
|
||||||
|
You can think of this as the most ideal case in 2-generals where the
|
||||||
|
actor requesting the cancel of its child is able to always receive back
|
||||||
|
the ACK to that request. This leads to a more deterministic shutdown of
|
||||||
|
the child where the parent is able to wait for the child to fully
|
||||||
|
respond to the request. On a localhost setup, where the parent can
|
||||||
|
monitor the state of the child through process or other OS APIs instead
|
||||||
|
of solely through IPC messaging, the parent can know whether or not the
|
||||||
|
child decided to cancel with more certainty. In the case of separate
|
||||||
|
hosts, we still rely on a simple timeout approach until such a time
|
||||||
|
where we prefer to get "fancier".
|
||||||
|
|
||||||
|
- `#271 <https://github.com/goodboy/tractor/issues/271>`_: Add a per
|
||||||
|
actor ``debug_mode: bool`` control to our nursery.
|
||||||
|
|
||||||
|
This allows spawning actors via ``ActorNursery.start_actor()`` (and
|
||||||
|
other dependent methods) with a ``debug_mode=True`` flag much like
|
||||||
|
``tractor.open_nursery():`` such that per process crash handling
|
||||||
|
can be toggled for cases where a user does not need/want all child actors
|
||||||
|
to drop into the debugger on error. This is often useful when you have
|
||||||
|
actor-tasks which are expected to error often (and be re-run) but want
|
||||||
|
to specifically interact with some (problematic) child.
|
||||||
|
|
||||||
|
|
||||||
|
Bugfixes
|
||||||
|
--------
|
||||||
|
|
||||||
|
- `#239 <https://github.com/goodboy/tractor/issues/239>`_: Fix
|
||||||
|
keyboard interrupt handling in ``Portal.open_context()`` blocks.
|
||||||
|
|
||||||
|
Previously this was not triggering cancellation of the remote task
|
||||||
|
context and could result in hangs if a stream was also opened. This
|
||||||
|
fix is to accept `BaseException` since it is likely any other top
|
||||||
|
level exception other then KBI (even though not expected) should also
|
||||||
|
get this result.
|
||||||
|
|
||||||
|
- `#264 <https://github.com/goodboy/tractor/issues/264>`_: Fix
|
||||||
|
``Portal.run_in_actor()`` returns ``None`` result.
|
||||||
|
|
||||||
|
``None`` was being used as the cached result flag and obviously breaks
|
||||||
|
on a ``None`` returned from the remote target task. This would cause an
|
||||||
|
infinite hang if user code ever called ``Portal.result()`` *before* the
|
||||||
|
nursery exit. The simple fix is to use the *return message* as the
|
||||||
|
initial "no-result-received-yet" flag value and, once received, the
|
||||||
|
return value is read from the message to avoid the cache logic error.
|
||||||
|
|
||||||
|
- `#266 <https://github.com/goodboy/tractor/issues/266>`_: Fix
|
||||||
|
graceful cancellation of daemon actors
|
||||||
|
|
||||||
|
Previously, his was a bug where if the soft wait on a sub-process (the
|
||||||
|
``await .proc.wait()``) in the reaper task teardown was cancelled we
|
||||||
|
would fail over to the hard reaping sequence (meant for culling off any
|
||||||
|
potential zombies via system kill signals). The hard reap has a timeout
|
||||||
|
of 3s (currently though in theory we could make it shorter?) before
|
||||||
|
system signalling kicks in. This means that any daemon actor still
|
||||||
|
running during nursery exit would get hard reaped (3s later) instead of
|
||||||
|
cancelled via IPC message. Now we catch the ``trio.Cancelled``, call
|
||||||
|
``Portal.cancel_actor()`` on the daemon and expect the child to
|
||||||
|
self-terminate after the runtime cancels and shuts down the process.
|
||||||
|
|
||||||
|
- `#278 <https://github.com/goodboy/tractor/issues/278>`_: Repair
|
||||||
|
inter-actor stream closure semantics to work correctly with
|
||||||
|
``tractor.trionics.BroadcastReceiver`` task fan out usage.
|
||||||
|
|
||||||
|
A set of previously unknown bugs discovered in `#257
|
||||||
|
<https://github.com/goodboy/tractor/pull/257>`_ let graceful stream
|
||||||
|
closure result in hanging consumer tasks that use the broadcast APIs.
|
||||||
|
This adds better internal closure state tracking to the broadcast
|
||||||
|
receiver and message stream APIs and in particular ensures that when an
|
||||||
|
underlying stream/receive-channel (a broadcast receiver is receiving
|
||||||
|
from) is closed, all consumer tasks waiting on that underlying channel
|
||||||
|
are woken so they can receive the ``trio.EndOfChannel`` signal and
|
||||||
|
promptly terminate.
|
||||||
|
|
||||||
|
|
||||||
|
tractor 0.1.0a3 (2021-11-02)
|
||||||
|
============================
|
||||||
|
|
||||||
|
Features
|
||||||
|
--------
|
||||||
|
|
||||||
|
- Switch to using the ``trio`` process spawner by default on windows. (#166)
|
||||||
|
|
||||||
|
This gets windows users debugger support (manually tested) and in
|
||||||
|
general a more resilient (nested) actor tree implementation.
|
||||||
|
|
||||||
|
- Add optional `msgspec <https://jcristharif.com/msgspec/>`_ support
|
||||||
|
as an alernative, faster MessagePack codec. (#214)
|
||||||
|
|
||||||
|
Provides us with a path toward supporting typed IPC message contracts. Further,
|
||||||
|
``msgspec`` structs may be a valid tool to start for formalizing our
|
||||||
|
"SC dialog un-protocol" messages as described in `#36
|
||||||
|
<https://github.com/goodboy/tractor/issues/36>`_.
|
||||||
|
|
||||||
|
- Introduce a new ``tractor.trionics`` `sub-package`_ that exposes
|
||||||
|
a selection of our relevant high(er) level trio primitives and
|
||||||
|
goodies. (#241)
|
||||||
|
|
||||||
|
At outset we offer a ``gather_contexts()`` context manager for
|
||||||
|
concurrently entering a sequence of async context managers (much like
|
||||||
|
a version of ``asyncio.gather()`` but for context managers) and use it
|
||||||
|
in a new ``tractor.open_actor_cluster()`` manager-helper that can be
|
||||||
|
entered to concurrently spawn a flat actor pool. We also now publicly
|
||||||
|
expose our "broadcast channel" APIs (``open_broadcast_receiver()``)
|
||||||
|
from here.
|
||||||
|
|
||||||
|
.. _sub-package: ../tractor/trionics
|
||||||
|
|
||||||
|
- Change the core message loop to handle task and actor-runtime cancel
|
||||||
|
requests immediately instead of scheduling them as is done for rpc-task
|
||||||
|
requests. (#245)
|
||||||
|
|
||||||
|
In order to obtain more reliable teardown mechanics for (complex) actor
|
||||||
|
trees it's important that we specially treat cancel requests as having
|
||||||
|
higher priority. Previously, it was possible that task cancel requests
|
||||||
|
could actually also themselves be cancelled if a "actor-runtime" cancel
|
||||||
|
request was received (can happen during messy multi actor crashes that
|
||||||
|
propagate). Instead cancels now block the msg loop until serviced and
|
||||||
|
a response is relayed back to the requester. This also allows for
|
||||||
|
improved debugger support since we have determinism guarantees about
|
||||||
|
which processes must wait before hard killing their children.
|
||||||
|
|
||||||
|
- (`#248 <https://github.com/goodboy/tractor/pull/248>`_) Drop Python
|
||||||
|
3.8 support in favour of rolling with two latest releases for the time
|
||||||
|
being.
|
||||||
|
|
||||||
|
|
||||||
|
Misc
|
||||||
|
----
|
||||||
|
|
||||||
|
- (`#243 <https://github.com/goodboy/tractor/pull/243>`_) add a distinct
|
||||||
|
``'CANCEL'`` log level to allow the runtime to emit details about
|
||||||
|
cancellation machinery statuses.
|
||||||
|
|
||||||
|
|
||||||
|
tractor 0.1.0a2 (2021-09-07)
|
||||||
|
============================
|
||||||
|
|
||||||
|
Features
|
||||||
|
--------
|
||||||
|
|
||||||
|
- Add `tokio-style broadcast channels
|
||||||
|
<https://docs.rs/tokio/1.11.0/tokio/sync/broadcast/index.html>`_ as
|
||||||
|
a solution for `#204 <https://github.com/goodboy/tractor/pull/204>`_ and
|
||||||
|
discussed thoroughly in `trio/#987
|
||||||
|
<https://github.com/python-trio/trio/issues/987>`_.
|
||||||
|
|
||||||
|
This gives us local task broadcast functionality using a new
|
||||||
|
``BroadcastReceiver`` type which can wrap ``trio.ReceiveChannel`` and
|
||||||
|
provide fan-out copies of a stream of data to every subscribed consumer.
|
||||||
|
We use this new machinery to provide a ``ReceiveMsgStream.subscribe()``
|
||||||
|
async context manager which can be used by actor-local concumers tasks
|
||||||
|
to easily pull from a shared and dynamic IPC stream. (`#229
|
||||||
|
<https://github.com/goodboy/tractor/pull/229>`_)
|
||||||
|
|
||||||
|
|
||||||
|
Bugfixes
|
||||||
|
--------
|
||||||
|
|
||||||
|
- Handle broken channel/stream faults where the root's tty lock is left
|
||||||
|
acquired by some child actor who went MIA and the root ends up hanging
|
||||||
|
indefinitely. (`#234 <https://github.com/goodboy/tractor/pull/234>`_)
|
||||||
|
|
||||||
|
There's two parts here: we no longer shield wait on the lock and,
|
||||||
|
now always do our best to release the lock on the expected worst
|
||||||
|
case connection faults.
|
||||||
|
|
||||||
|
|
||||||
|
Deprecations and Removals
|
||||||
|
-------------------------
|
||||||
|
|
||||||
|
- Drop stream "shielding" support which was originally added to sidestep
|
||||||
|
a cancelled call to ``.receive()``
|
||||||
|
|
||||||
|
In the original api design a stream instance was returned directly from
|
||||||
|
a call to ``Portal.run()`` and thus there was no "exit phase" to handle
|
||||||
|
cancellations and errors which would trigger implicit closure. Now that
|
||||||
|
we have said enter/exit semantics with ``Portal.open_stream_from()`` and
|
||||||
|
``Context.open_stream()`` we can drop this implicit (and arguably
|
||||||
|
confusing) behavior. (`#230 <https://github.com/goodboy/tractor/pull/230>`_)
|
||||||
|
|
||||||
|
- Drop Python 3.7 support in preparation for supporting 3.9+ syntax.
|
||||||
|
(`#232 <https://github.com/goodboy/tractor/pull/232>`_)
|
||||||
|
|
||||||
|
|
||||||
|
tractor 0.1.0a1 (2021-08-01)
|
||||||
|
============================
|
||||||
|
|
||||||
|
Features
|
||||||
|
--------
|
||||||
|
- Updated our uni-directional streaming API (`#206
|
||||||
|
<https://github.com/goodboy/tractor/pull/206>`_) to require a context
|
||||||
|
manager style ``async with Portal.open_stream_from(target) as stream:``
|
||||||
|
which explicitly determines when to stop a stream in the calling (aka
|
||||||
|
portal opening) actor much like ``async_generator.aclosing()``
|
||||||
|
enforcement.
|
||||||
|
|
||||||
|
- Improved the ``multiprocessing`` backend sub-actor reaping (`#208
|
||||||
|
<https://github.com/goodboy/tractor/pull/208>`_) during actor nursery
|
||||||
|
exit, particularly during cancellation scenarios that previously might
|
||||||
|
result in hard to debug hangs.
|
||||||
|
|
||||||
|
- Added initial bi-directional streaming support in `#219
|
||||||
|
<https://github.com/goodboy/tractor/pull/219>`_ with follow up debugger
|
||||||
|
improvements via `#220 <https://github.com/goodboy/tractor/pull/220>`_
|
||||||
|
using the new ``tractor.Context`` cross-actor task syncing system.
|
||||||
|
The debugger upgrades add an edge triggered last-in-tty-lock semaphore
|
||||||
|
which allows the root process for a tree to avoid clobbering children
|
||||||
|
who have queued to acquire the ``pdb`` repl by waiting to cancel
|
||||||
|
sub-actors until the lock is known to be released **and** has no
|
||||||
|
pending waiters.
|
||||||
|
|
||||||
|
|
||||||
|
Experiments and WIPs
|
||||||
|
--------------------
|
||||||
|
- Initial optional ``msgspec`` serialization support in `#214
|
||||||
|
<https://github.com/goodboy/tractor/pull/214>`_ which should hopefully
|
||||||
|
land by next release.
|
||||||
|
|
||||||
|
- Improved "infect ``asyncio``" cross-loop task cancellation and error
|
||||||
|
propagation by vastly simplifying the cross-loop-task streaming approach.
|
||||||
|
We may end up just going with a use of ``anyio`` in the medium term to
|
||||||
|
avoid re-doing work done by their cross-event-loop portals. See the
|
||||||
|
``infect_asyncio`` for details.
|
||||||
|
|
||||||
|
|
||||||
|
Improved Documentation
|
||||||
|
----------------------
|
||||||
|
- `Updated our readme <https://github.com/goodboy/tractor/pull/211>`_ to
|
||||||
|
include more (and better) `examples
|
||||||
|
<https://github.com/goodboy/tractor#run-a-func-in-a-process>`_ (with
|
||||||
|
matching multi-terminal process monitoring shell commands) as well as
|
||||||
|
added many more examples to the `repo set
|
||||||
|
<https://github.com/goodboy/tractor/tree/master/examples>`_.
|
||||||
|
|
||||||
|
- Added a readme `"actors under the hood" section
|
||||||
|
<https://github.com/goodboy/tractor#under-the-hood>`_ in an effort to
|
||||||
|
guard against suggestions for changing the API away from ``trio``'s
|
||||||
|
*tasks-as-functions* style.
|
||||||
|
|
||||||
|
- Moved to using the `sphinx book theme
|
||||||
|
<https://sphinx-book-theme.readthedocs.io/en/latest/index.html>`_
|
||||||
|
though it needs some heavy tweaking and doesn't seem to show our logo
|
||||||
|
on rtd :(
|
||||||
|
|
||||||
|
|
||||||
|
Trivial/Internal Changes
|
||||||
|
------------------------
|
||||||
|
- Added a new ``TransportClosed`` internal exception/signal (`#215
|
||||||
|
<https://github.com/goodboy/tractor/pull/215>`_ for catching TCP
|
||||||
|
channel gentle closes instead of silently falling through the message
|
||||||
|
handler loop via an async generator ``return``.
|
||||||
|
|
||||||
|
|
||||||
|
Deprecations and Removals
|
||||||
|
-------------------------
|
||||||
|
- Dropped support for invoking sync functions (`#205
|
||||||
|
<https://github.com/goodboy/tractor/pull/205>`_) in other
|
||||||
|
actors/processes since you can always wrap a sync function from an
|
||||||
|
async one. Users can instead consider using ``trio-parallel`` which
|
||||||
|
is a project specifically geared for purely synchronous calls in
|
||||||
|
sub-processes.
|
||||||
|
|
||||||
|
- Deprecated our ``tractor.run()`` entrypoint `#197
|
||||||
|
<https://github.com/goodboy/tractor/pull/197>`_; the runtime is now
|
||||||
|
either started implicitly in first actor nursery use or via an
|
||||||
|
explicit call to ``tractor.open_root_actor()``. Full removal of
|
||||||
|
``tractor.run()`` will come by beta release.
|
||||||
|
|
||||||
|
|
||||||
|
tractor 0.1.0a0 (2021-02-28)
|
||||||
|
============================
|
||||||
|
|
||||||
|
..
|
||||||
|
TODO: fill out more of the details of the initial feature set in some TLDR form
|
||||||
|
|
||||||
|
Summary
|
||||||
|
-------
|
||||||
|
- ``trio`` based process spawner (using ``subprocess``)
|
||||||
|
- initial multi-process debugging with ``pdb++``
|
||||||
|
- windows support using both ``trio`` and ``multiprocessing`` spawners
|
||||||
|
- "portal" api for cross-process, structured concurrent, (streaming) IPC
|
65
README.rst
65
README.rst
|
@ -1,65 +0,0 @@
|
||||||
tractor
|
|
||||||
=======
|
|
||||||
A `structured concurrent`_, async-native "`actor model`_" built on trio_ and multiprocessing_.
|
|
||||||
|
|
||||||
|gh_actions|
|
|
||||||
|docs|
|
|
||||||
|
|
||||||
.. _actor model: https://en.wikipedia.org/wiki/Actor_model
|
|
||||||
.. _trio: https://github.com/python-trio/trio
|
|
||||||
.. _multiprocessing: https://en.wikipedia.org/wiki/Multiprocessing
|
|
||||||
.. _trionic: https://trio.readthedocs.io/en/latest/design.html#high-level-design-principles
|
|
||||||
.. _async sandwich: https://trio.readthedocs.io/en/latest/tutorial.html#async-sandwich
|
|
||||||
.. _structured concurrent: https://trio.discourse.group/t/concise-definition-of-structured-concurrency/228
|
|
||||||
|
|
||||||
|
|
||||||
``tractor`` is an attempt to bring trionic_ `structured concurrency`_ to
|
|
||||||
distributed multi-core Python; it aims to be the Python multi-processing
|
|
||||||
framework *you always wanted*.
|
|
||||||
|
|
||||||
``tractor`` lets you spawn ``trio`` *"actors"*: processes which each run
|
|
||||||
a ``trio`` scheduled task tree (also known as an `async sandwich`_).
|
|
||||||
*Actors* communicate by exchanging asynchronous messages_ and avoid
|
|
||||||
sharing any state. This model allows for highly distributed software
|
|
||||||
architecture which works just as well on multiple cores as it does over
|
|
||||||
many hosts.
|
|
||||||
|
|
||||||
The first step to grok ``tractor`` is to get the basics of ``trio`` down.
|
|
||||||
A great place to start is the `trio docs`_ and this `blog post`_.
|
|
||||||
|
|
||||||
.. _messages: https://en.wikipedia.org/wiki/Message_passing
|
|
||||||
.. _trio docs: https://trio.readthedocs.io/en/latest/
|
|
||||||
.. _blog post: https://vorpus.org/blog/notes-on-structured-concurrency-or-go-statement-considered-harmful/
|
|
||||||
.. _structured concurrency: https://vorpus.org/blog/notes-on-structured-concurrency-or-go-statement-considered-harmful/
|
|
||||||
.. _3 axioms: https://en.wikipedia.org/wiki/Actor_model#Fundamental_concepts
|
|
||||||
.. _unrequirements: https://en.wikipedia.org/wiki/Actor_model#Direct_communication_and_asynchrony
|
|
||||||
.. _async generators: https://www.python.org/dev/peps/pep-0525/
|
|
||||||
|
|
||||||
|
|
||||||
Install
|
|
||||||
-------
|
|
||||||
No PyPi release yet!
|
|
||||||
|
|
||||||
::
|
|
||||||
|
|
||||||
pip install git+git://github.com/goodboy/tractor.git
|
|
||||||
|
|
||||||
|
|
||||||
Feel like saying hi?
|
|
||||||
--------------------
|
|
||||||
This project is very much coupled to the ongoing development of
|
|
||||||
``trio`` (i.e. ``tractor`` gets most of its ideas from that brilliant
|
|
||||||
community). If you want to help, have suggestions or just want to
|
|
||||||
say hi, please feel free to reach us on in our `matrix channel`_. If
|
|
||||||
matrix seems too hip, we're also mostly all in the the `trio gitter
|
|
||||||
channel`_!
|
|
||||||
|
|
||||||
.. _trio gitter channel: https://gitter.im/python-trio/general
|
|
||||||
.. _matrix channel: https://matrix.to/#/!tractor:matrix.org
|
|
||||||
|
|
||||||
|
|
||||||
.. |gh_actions| image:: https://img.shields.io/endpoint.svg?url=https%3A%2F%2Factions-badge.atrox.dev%2Fgoodboy%2Ftractor%2Fbadge&style=popout-square
|
|
||||||
:target: https://actions-badge.atrox.dev/goodboy/tractor/goto
|
|
||||||
.. |docs| image:: https://readthedocs.org/projects/tractor/badge/?version=latest
|
|
||||||
:target: https://tractor.readthedocs.io/en/latest/?badge=latest
|
|
||||||
:alt: Documentation Status
|
|
|
@ -0,0 +1,632 @@
|
||||||
|
|logo| ``tractor``: next-gen Python parallelism
|
||||||
|
|
||||||
|
|gh_actions|
|
||||||
|
|docs|
|
||||||
|
|
||||||
|
``tractor`` is a `structured concurrent`_, multi-processing_ runtime
|
||||||
|
built on trio_.
|
||||||
|
|
||||||
|
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
|
||||||
|
combined use of our "actor nurseries_" and an "SC-transitive IPC
|
||||||
|
protocol" constructed on top of multiple Pythons each running a ``trio``
|
||||||
|
scheduled runtime - a call to ``trio.run()``.
|
||||||
|
|
||||||
|
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
|
||||||
|
model" looks like, and that's *intentional*.
|
||||||
|
|
||||||
|
The first step to grok ``tractor`` is to get the basics of ``trio`` down.
|
||||||
|
A great place to start is the `trio docs`_ and this `blog post`_.
|
||||||
|
|
||||||
|
|
||||||
|
Features
|
||||||
|
--------
|
||||||
|
- **It's just** a ``trio`` API
|
||||||
|
- *Infinitely nesteable* process trees
|
||||||
|
- Builtin IPC streaming APIs with task fan-out broadcasting
|
||||||
|
- A "native" multi-core debugger REPL using `pdbp`_ (a fork & fix of
|
||||||
|
`pdb++`_ thanks to @mdmintz!)
|
||||||
|
- Support for a swappable, OS specific, process spawning layer
|
||||||
|
- A modular transport stack, allowing for custom serialization (eg. with
|
||||||
|
`msgspec`_), communications protocols, and environment specific IPC
|
||||||
|
primitives
|
||||||
|
- Support for spawning process-level-SC, inter-loop one-to-one-task oriented
|
||||||
|
``asyncio`` actors via "infected ``asyncio``" mode
|
||||||
|
- `structured chadcurrency`_ from the ground up
|
||||||
|
|
||||||
|
|
||||||
|
Run a func in a process
|
||||||
|
-----------------------
|
||||||
|
Use ``trio``'s style of focussing on *tasks as functions*:
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
"""
|
||||||
|
Run with a process monitor from a terminal using::
|
||||||
|
|
||||||
|
$TERM -e watch -n 0.1 "pstree -a $$" \
|
||||||
|
& python examples/parallelism/single_func.py \
|
||||||
|
&& kill $!
|
||||||
|
|
||||||
|
"""
|
||||||
|
import os
|
||||||
|
|
||||||
|
import tractor
|
||||||
|
import trio
|
||||||
|
|
||||||
|
|
||||||
|
async def burn_cpu():
|
||||||
|
|
||||||
|
pid = os.getpid()
|
||||||
|
|
||||||
|
# burn a core @ ~ 50kHz
|
||||||
|
for _ in range(50000):
|
||||||
|
await trio.sleep(1/50000/50)
|
||||||
|
|
||||||
|
return os.getpid()
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.run_in_actor(burn_cpu)
|
||||||
|
|
||||||
|
# burn rubber in the parent too
|
||||||
|
await burn_cpu()
|
||||||
|
|
||||||
|
# wait on result from target function
|
||||||
|
pid = await portal.result()
|
||||||
|
|
||||||
|
# end of nursery block
|
||||||
|
print(f"Collected subproc {pid}")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
This runs ``burn_cpu()`` in a new process and reaps it on completion
|
||||||
|
of the nursery block.
|
||||||
|
|
||||||
|
If you only need to run a sync function and retreive a single result, you
|
||||||
|
might want to check out `trio-parallel`_.
|
||||||
|
|
||||||
|
|
||||||
|
Zombie safe: self-destruct a process tree
|
||||||
|
-----------------------------------------
|
||||||
|
``tractor`` tries to protect you from zombies, no matter what.
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
"""
|
||||||
|
Run with a process monitor from a terminal using::
|
||||||
|
|
||||||
|
$TERM -e watch -n 0.1 "pstree -a $$" \
|
||||||
|
& python examples/parallelism/we_are_processes.py \
|
||||||
|
&& kill $!
|
||||||
|
|
||||||
|
"""
|
||||||
|
from multiprocessing import cpu_count
|
||||||
|
import os
|
||||||
|
|
||||||
|
import tractor
|
||||||
|
import trio
|
||||||
|
|
||||||
|
|
||||||
|
async def target():
|
||||||
|
print(
|
||||||
|
f"Yo, i'm '{tractor.current_actor().name}' "
|
||||||
|
f"running in pid {os.getpid()}"
|
||||||
|
)
|
||||||
|
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
for i in range(cpu_count()):
|
||||||
|
await n.run_in_actor(target, name=f'worker_{i}')
|
||||||
|
|
||||||
|
print('This process tree will self-destruct in 1 sec...')
|
||||||
|
await trio.sleep(1)
|
||||||
|
|
||||||
|
# raise an error in root actor/process and trigger
|
||||||
|
# reaping of all minions
|
||||||
|
raise Exception('Self Destructed')
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
try:
|
||||||
|
trio.run(main)
|
||||||
|
except Exception:
|
||||||
|
print('Zombies Contained')
|
||||||
|
|
||||||
|
|
||||||
|
If you can create zombie child processes (without using a system signal)
|
||||||
|
it **is a bug**.
|
||||||
|
|
||||||
|
|
||||||
|
"Native" multi-process debugging
|
||||||
|
--------------------------------
|
||||||
|
Using the magic of `pdbp`_ and our internal IPC, we've
|
||||||
|
been able to create a native feeling debugging experience for
|
||||||
|
any (sub-)process in your ``tractor`` tree.
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
from os import getpid
|
||||||
|
|
||||||
|
import tractor
|
||||||
|
import trio
|
||||||
|
|
||||||
|
|
||||||
|
async def breakpoint_forever():
|
||||||
|
"Indefinitely re-enter debugger in child actor."
|
||||||
|
while True:
|
||||||
|
yield 'yo'
|
||||||
|
await tractor.breakpoint()
|
||||||
|
|
||||||
|
|
||||||
|
async def name_error():
|
||||||
|
"Raise a ``NameError``"
|
||||||
|
getattr(doggypants)
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
"""Test breakpoint in a streaming actor.
|
||||||
|
"""
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
loglevel='error',
|
||||||
|
) as n:
|
||||||
|
|
||||||
|
p0 = await n.start_actor('bp_forever', enable_modules=[__name__])
|
||||||
|
p1 = await n.start_actor('name_error', enable_modules=[__name__])
|
||||||
|
|
||||||
|
# retreive results
|
||||||
|
stream = await p0.run(breakpoint_forever)
|
||||||
|
await p1.run(name_error)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
You can run this with::
|
||||||
|
|
||||||
|
>>> python examples/debugging/multi_daemon_subactors.py
|
||||||
|
|
||||||
|
And, yes, there's a built-in crash handling mode B)
|
||||||
|
|
||||||
|
We're hoping to add a respawn-from-repl system soon!
|
||||||
|
|
||||||
|
|
||||||
|
SC compatible bi-directional streaming
|
||||||
|
--------------------------------------
|
||||||
|
Yes, you saw it here first; we provide 2-way streams
|
||||||
|
with reliable, transitive setup/teardown semantics.
|
||||||
|
|
||||||
|
Our nascent api is remniscent of ``trio.Nursery.start()``
|
||||||
|
style invocation:
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def simple_rpc(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
data: int,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''Test a small ping-pong 2-way streaming server.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# signal to parent that we're up much like
|
||||||
|
# ``trio_typing.TaskStatus.started()``
|
||||||
|
await ctx.started(data + 1)
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
async for msg in stream:
|
||||||
|
|
||||||
|
assert msg == 'ping'
|
||||||
|
await stream.send('pong')
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert count == 10
|
||||||
|
|
||||||
|
|
||||||
|
async def main() -> None:
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'rpc_server',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
# XXX: this syntax requires py3.9
|
||||||
|
async with (
|
||||||
|
|
||||||
|
portal.open_context(
|
||||||
|
simple_rpc,
|
||||||
|
data=10,
|
||||||
|
) as (ctx, sent),
|
||||||
|
|
||||||
|
ctx.open_stream() as stream,
|
||||||
|
):
|
||||||
|
|
||||||
|
assert sent == 11
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
# receive msgs using async for style
|
||||||
|
await stream.send('ping')
|
||||||
|
|
||||||
|
async for msg in stream:
|
||||||
|
assert msg == 'pong'
|
||||||
|
await stream.send('ping')
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
if count >= 9:
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
|
# explicitly teardown the daemon-actor
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
See original proposal and discussion in `#53`_ as well
|
||||||
|
as follow up improvements in `#223`_ that we'd love to
|
||||||
|
hear your thoughts on!
|
||||||
|
|
||||||
|
.. _#53: https://github.com/goodboy/tractor/issues/53
|
||||||
|
.. _#223: https://github.com/goodboy/tractor/issues/223
|
||||||
|
|
||||||
|
|
||||||
|
Worker poolz are easy peasy
|
||||||
|
---------------------------
|
||||||
|
The initial ask from most new users is *"how do I make a worker
|
||||||
|
pool thing?"*.
|
||||||
|
|
||||||
|
``tractor`` is built to handle any SC (structured concurrent) process
|
||||||
|
tree you can imagine; a "worker pool" pattern is a trivial special
|
||||||
|
case.
|
||||||
|
|
||||||
|
We have a `full worker pool re-implementation`_ of the std-lib's
|
||||||
|
``concurrent.futures.ProcessPoolExecutor`` example for reference.
|
||||||
|
|
||||||
|
You can run it like so (from this dir) to see the process tree in
|
||||||
|
real time::
|
||||||
|
|
||||||
|
$TERM -e watch -n 0.1 "pstree -a $$" \
|
||||||
|
& python examples/parallelism/concurrent_actors_primes.py \
|
||||||
|
&& kill $!
|
||||||
|
|
||||||
|
This uses no extra threads, fancy semaphores or futures; all we need
|
||||||
|
is ``tractor``'s IPC!
|
||||||
|
|
||||||
|
"Infected ``asyncio``" mode
|
||||||
|
---------------------------
|
||||||
|
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
|
||||||
|
``asyncio`` actors:
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
from statistics import mean
|
||||||
|
import time
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
async def aio_echo_server(
|
||||||
|
to_trio: trio.MemorySendChannel,
|
||||||
|
from_trio: asyncio.Queue,
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
# a first message must be sent **from** this ``asyncio``
|
||||||
|
# task or the ``trio`` side will never unblock from
|
||||||
|
# ``tractor.to_asyncio.open_channel_from():``
|
||||||
|
to_trio.send_nowait('start')
|
||||||
|
|
||||||
|
# XXX: this uses an ``from_trio: asyncio.Queue`` currently but we
|
||||||
|
# should probably offer something better.
|
||||||
|
while True:
|
||||||
|
# echo the msg back
|
||||||
|
to_trio.send_nowait(await from_trio.get())
|
||||||
|
await asyncio.sleep(0)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def trio_to_aio_echo_server(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
):
|
||||||
|
# this will block until the ``asyncio`` task sends a "first"
|
||||||
|
# message.
|
||||||
|
async with tractor.to_asyncio.open_channel_from(
|
||||||
|
aio_echo_server,
|
||||||
|
) as (first, chan):
|
||||||
|
|
||||||
|
assert first == 'start'
|
||||||
|
await ctx.started(first)
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
async for msg in stream:
|
||||||
|
await chan.send(msg)
|
||||||
|
|
||||||
|
out = await chan.receive()
|
||||||
|
# echo back to parent actor-task
|
||||||
|
await stream.send(out)
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
p = await n.start_actor(
|
||||||
|
'aio_server',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
async with p.open_context(
|
||||||
|
trio_to_aio_echo_server,
|
||||||
|
) as (ctx, first):
|
||||||
|
|
||||||
|
assert first == 'start'
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
delays = []
|
||||||
|
send = time.time()
|
||||||
|
|
||||||
|
await stream.send(count)
|
||||||
|
async for msg in stream:
|
||||||
|
recv = time.time()
|
||||||
|
delays.append(recv - send)
|
||||||
|
assert msg == count
|
||||||
|
count += 1
|
||||||
|
send = time.time()
|
||||||
|
await stream.send(count)
|
||||||
|
|
||||||
|
if count >= 1e3:
|
||||||
|
break
|
||||||
|
|
||||||
|
print(f'mean round trip rate (Hz): {1/mean(delays)}')
|
||||||
|
await p.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
Yes, we spawn a python process, run ``asyncio``, start ``trio`` on the
|
||||||
|
``asyncio`` loop, then send commands to the ``trio`` scheduled tasks to
|
||||||
|
tell ``asyncio`` tasks what to do XD
|
||||||
|
|
||||||
|
We need help refining the `asyncio`-side channel API to be more
|
||||||
|
`trio`-like. Feel free to sling your opinion in `#273`_!
|
||||||
|
|
||||||
|
|
||||||
|
.. _#273: https://github.com/goodboy/tractor/issues/273
|
||||||
|
|
||||||
|
|
||||||
|
Higher level "cluster" APIs
|
||||||
|
---------------------------
|
||||||
|
To be extra terse the ``tractor`` devs have started hacking some "higher
|
||||||
|
level" APIs for managing actor trees/clusters. These interfaces should
|
||||||
|
generally be condsidered provisional for now but we encourage you to try
|
||||||
|
them and provide feedback. Here's a new API that let's you quickly
|
||||||
|
spawn a flat cluster:
|
||||||
|
|
||||||
|
.. code:: python
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
async def sleepy_jane():
|
||||||
|
uid = tractor.current_actor().uid
|
||||||
|
print(f'Yo i am actor {uid}')
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
'''
|
||||||
|
Spawn a flat actor cluster, with one process per
|
||||||
|
detected core.
|
||||||
|
|
||||||
|
'''
|
||||||
|
portal_map: dict[str, tractor.Portal]
|
||||||
|
results: dict[str, str]
|
||||||
|
|
||||||
|
# look at this hip new syntax!
|
||||||
|
async with (
|
||||||
|
|
||||||
|
tractor.open_actor_cluster(
|
||||||
|
modules=[__name__]
|
||||||
|
) as portal_map,
|
||||||
|
|
||||||
|
trio.open_nursery() as n,
|
||||||
|
):
|
||||||
|
|
||||||
|
for (name, portal) in portal_map.items():
|
||||||
|
n.start_soon(portal.run, sleepy_jane)
|
||||||
|
|
||||||
|
await trio.sleep(0.5)
|
||||||
|
|
||||||
|
# kill the cluster with a cancel
|
||||||
|
raise KeyboardInterrupt
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
try:
|
||||||
|
trio.run(main)
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
.. _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
|
||||||
|
--------------
|
||||||
|
``tractor`` is an attempt to pair trionic_ `structured concurrency`_ with
|
||||||
|
distributed Python. You can think of it as a ``trio``
|
||||||
|
*-across-processes* or simply as an opinionated replacement for the
|
||||||
|
stdlib's ``multiprocessing`` but built on async programming primitives
|
||||||
|
from the ground up.
|
||||||
|
|
||||||
|
Don't be scared off by this description. ``tractor`` **is just** ``trio``
|
||||||
|
but with nurseries for process management and cancel-able streaming IPC.
|
||||||
|
If you understand how to work with ``trio``, ``tractor`` will give you
|
||||||
|
the parallelism you may have been needing.
|
||||||
|
|
||||||
|
|
||||||
|
Wait, huh?! I thought "actors" have messages, and mailboxes and stuff?!
|
||||||
|
***********************************************************************
|
||||||
|
Let's stop and ask how many canon actor model papers have you actually read ;)
|
||||||
|
|
||||||
|
From our experience many "actor systems" aren't really "actor models"
|
||||||
|
since they **don't adhere** to the `3 axioms`_ and pay even less
|
||||||
|
attention to the problem of *unbounded non-determinism* (which was the
|
||||||
|
whole point for creation of the model in the first place).
|
||||||
|
|
||||||
|
From the author's mouth, **the only thing required** is `adherance to`_
|
||||||
|
the `3 axioms`_, *and that's it*.
|
||||||
|
|
||||||
|
``tractor`` adheres to said base requirements of an "actor model"::
|
||||||
|
|
||||||
|
In response to a message, an actor may:
|
||||||
|
|
||||||
|
- send a finite number of new messages
|
||||||
|
- create a finite number of new actors
|
||||||
|
- designate a new behavior to process subsequent messages
|
||||||
|
|
||||||
|
|
||||||
|
**and** requires *no further api changes* to accomplish this.
|
||||||
|
|
||||||
|
If you want do debate this further please feel free to chime in on our
|
||||||
|
chat or discuss on one of the following issues *after you've read
|
||||||
|
everything in them*:
|
||||||
|
|
||||||
|
- https://github.com/goodboy/tractor/issues/210
|
||||||
|
- https://github.com/goodboy/tractor/issues/18
|
||||||
|
|
||||||
|
|
||||||
|
Let's clarify our parlance
|
||||||
|
**************************
|
||||||
|
Whether or not ``tractor`` has "actors" underneath should be mostly
|
||||||
|
irrelevant to users other then for referring to the interactions of our
|
||||||
|
primary runtime primitives: each Python process + ``trio.run()``
|
||||||
|
+ surrounding IPC machinery. These are our high level, base
|
||||||
|
*runtime-units-of-abstraction* which both *are* (as much as they can
|
||||||
|
be in Python) and will be referred to as our *"actors"*.
|
||||||
|
|
||||||
|
The main goal of ``tractor`` is is to allow for highly distributed
|
||||||
|
software that, through the adherence to *structured concurrency*,
|
||||||
|
results in systems which fail in predictable, recoverable and maybe even
|
||||||
|
understandable ways; being an "actor model" is just one way to describe
|
||||||
|
properties of the system.
|
||||||
|
|
||||||
|
|
||||||
|
What's on the TODO:
|
||||||
|
-------------------
|
||||||
|
Help us push toward the future of distributed `Python`.
|
||||||
|
|
||||||
|
- Erlang-style supervisors via composed context managers (see `#22
|
||||||
|
<https://github.com/goodboy/tractor/issues/22>`_)
|
||||||
|
- Typed messaging protocols (ex. via ``msgspec.Struct``, see `#36
|
||||||
|
<https://github.com/goodboy/tractor/issues/36>`_)
|
||||||
|
- Typed capability-based (dialog) protocols ( see `#196
|
||||||
|
<https://github.com/goodboy/tractor/issues/196>`_ with draft work
|
||||||
|
started in `#311 <https://github.com/goodboy/tractor/pull/311>`_)
|
||||||
|
- We **recently disabled CI-testing on windows** and need help getting
|
||||||
|
it running again! (see `#327
|
||||||
|
<https://github.com/goodboy/tractor/pull/327>`_). **We do have windows
|
||||||
|
support** (and have for quite a while) but since no active hacker
|
||||||
|
exists in the user-base to help test on that OS, for now we're not
|
||||||
|
actively maintaining testing due to the added hassle and general
|
||||||
|
latency..
|
||||||
|
|
||||||
|
|
||||||
|
Feel like saying hi?
|
||||||
|
--------------------
|
||||||
|
This project is very much coupled to the ongoing development of
|
||||||
|
``trio`` (i.e. ``tractor`` gets most of its ideas from that brilliant
|
||||||
|
community). If you want to help, have suggestions or just want to
|
||||||
|
say hi, please feel free to reach us in our `matrix channel`_. If
|
||||||
|
matrix seems too hip, we're also mostly all in the the `trio gitter
|
||||||
|
channel`_!
|
||||||
|
|
||||||
|
.. _structured concurrent: https://trio.discourse.group/t/concise-definition-of-structured-concurrency/228
|
||||||
|
.. _multi-processing: https://en.wikipedia.org/wiki/Multiprocessing
|
||||||
|
.. _trio: https://github.com/python-trio/trio
|
||||||
|
.. _nurseries: https://vorpus.org/blog/notes-on-structured-concurrency-or-go-statement-considered-harmful/#nurseries-a-structured-replacement-for-go-statements
|
||||||
|
.. _actor model: https://en.wikipedia.org/wiki/Actor_model
|
||||||
|
.. _trionic: https://trio.readthedocs.io/en/latest/design.html#high-level-design-principles
|
||||||
|
.. _async sandwich: https://trio.readthedocs.io/en/latest/tutorial.html#async-sandwich
|
||||||
|
.. _3 axioms: https://www.youtube.com/watch?v=7erJ1DV_Tlo&t=162s
|
||||||
|
.. .. _3 axioms: https://en.wikipedia.org/wiki/Actor_model#Fundamental_concepts
|
||||||
|
.. _adherance to: https://www.youtube.com/watch?v=7erJ1DV_Tlo&t=1821s
|
||||||
|
.. _trio gitter channel: https://gitter.im/python-trio/general
|
||||||
|
.. _matrix channel: https://matrix.to/#/!tractor:matrix.org
|
||||||
|
.. _pdbp: https://github.com/mdmintz/pdbp
|
||||||
|
.. _pdb++: https://github.com/pdbpp/pdbpp
|
||||||
|
.. _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
|
||||||
|
.. _messages: https://en.wikipedia.org/wiki/Message_passing
|
||||||
|
.. _trio docs: https://trio.readthedocs.io/en/latest/
|
||||||
|
.. _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 chadcurrency: https://en.wikipedia.org/wiki/Structured_concurrency
|
||||||
|
.. _structured concurrency: https://en.wikipedia.org/wiki/Structured_concurrency
|
||||||
|
.. _unrequirements: https://en.wikipedia.org/wiki/Actor_model#Direct_communication_and_asynchrony
|
||||||
|
.. _async generators: https://www.python.org/dev/peps/pep-0525/
|
||||||
|
.. _trio-parallel: https://github.com/richardsheridan/trio-parallel
|
||||||
|
.. _msgspec: https://jcristharif.com/msgspec/
|
||||||
|
.. _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
|
||||||
|
|
||||||
|
|
||||||
|
.. |gh_actions| image:: https://img.shields.io/endpoint.svg?url=https%3A%2F%2Factions-badge.atrox.dev%2Fgoodboy%2Ftractor%2Fbadge&style=popout-square
|
||||||
|
:target: https://actions-badge.atrox.dev/goodboy/tractor/goto
|
||||||
|
|
||||||
|
.. |docs| image:: https://readthedocs.org/projects/tractor/badge/?version=latest
|
||||||
|
:target: https://tractor.readthedocs.io/en/latest/?badge=latest
|
||||||
|
:alt: Documentation Status
|
||||||
|
|
||||||
|
.. |logo| image:: _static/tractor_logo_side.svg
|
||||||
|
:width: 250
|
||||||
|
:align: middle
|
38
docs/conf.py
38
docs/conf.py
|
@ -54,28 +54,44 @@ exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
|
||||||
# The theme to use for HTML and HTML Help pages. See the documentation for
|
# The theme to use for HTML and HTML Help pages. See the documentation for
|
||||||
# a list of builtin themes.
|
# a list of builtin themes.
|
||||||
#
|
#
|
||||||
html_theme = 'sphinx_typlog_theme'
|
html_theme = 'sphinx_book_theme'
|
||||||
|
|
||||||
pygments_style = 'sphinx'
|
pygments_style = 'algol_nu'
|
||||||
|
|
||||||
# Theme options are theme-specific and customize the look and feel of a theme
|
# Theme options are theme-specific and customize the look and feel of a theme
|
||||||
# further. For a list of options available for each theme, see the
|
# further. For a list of options available for each theme, see the
|
||||||
# documentation.
|
# documentation.
|
||||||
html_theme_options = {
|
html_theme_options = {
|
||||||
'logo': 'tractor_logo_side.svg',
|
# 'logo': 'tractor_logo_side.svg',
|
||||||
'description': 'Structured concurrent "actors"',
|
# 'description': 'Structured concurrent "actors"',
|
||||||
'github_user': 'goodboy',
|
"repository_url": "https://github.com/goodboy/tractor",
|
||||||
'github_repo': 'tractor',
|
"use_repository_button": True,
|
||||||
|
"home_page_in_toc": False,
|
||||||
|
"show_toc_level": 1,
|
||||||
|
"path_to_docs": "docs",
|
||||||
|
|
||||||
}
|
}
|
||||||
html_sidebars = {
|
html_sidebars = {
|
||||||
"**": [
|
"**": [
|
||||||
'logo.html',
|
"sbt-sidebar-nav.html",
|
||||||
'github.html',
|
# "sidebar-search-bs.html",
|
||||||
'relations.html',
|
# 'localtoc.html',
|
||||||
'searchbox.html'
|
],
|
||||||
]
|
# 'logo.html',
|
||||||
|
# 'github.html',
|
||||||
|
# 'relations.html',
|
||||||
|
# 'searchbox.html'
|
||||||
|
# ]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# doesn't seem to work?
|
||||||
|
# extra_navbar = "<p>nextttt-gennnnn</p>"
|
||||||
|
|
||||||
|
html_title = ''
|
||||||
|
html_logo = '_static/tractor_logo_side.svg'
|
||||||
|
html_favicon = '_static/tractor_logo_side.svg'
|
||||||
|
# show_navbar_depth = 1
|
||||||
|
|
||||||
# Add any paths that contain custom static files (such as style sheets) here,
|
# Add any paths that contain custom static files (such as style sheets) here,
|
||||||
# relative to this directory. They are copied after the builtin static files,
|
# relative to this directory. They are copied after the builtin static files,
|
||||||
# so a file named "default.css" will overwrite the builtin "default.css".
|
# so a file named "default.css" will overwrite the builtin "default.css".
|
||||||
|
|
|
@ -0,0 +1,51 @@
|
||||||
|
Hot tips for ``tractor`` hackers
|
||||||
|
================================
|
||||||
|
|
||||||
|
This is a WIP guide for newcomers to the project mostly to do with
|
||||||
|
dev, testing, CI and release gotchas, reminders and best practises.
|
||||||
|
|
||||||
|
``tractor`` is a fairly novel project compared to most since it is
|
||||||
|
effectively a new way of doing distributed computing in Python and is
|
||||||
|
much closer to working with an "application level runtime" (like erlang
|
||||||
|
OTP or scala's akka project) then it is a traditional Python library.
|
||||||
|
As such, having an arsenal of tools and recipes for figuring out the
|
||||||
|
right way to debug problems when they do arise is somewhat of
|
||||||
|
a necessity.
|
||||||
|
|
||||||
|
|
||||||
|
Making a Release
|
||||||
|
----------------
|
||||||
|
We currently do nothing special here except the traditional
|
||||||
|
PyPa release recipe as in `documented by twine`_. I personally
|
||||||
|
create sub-dirs within the generated `dist/` with an explicit
|
||||||
|
release name such as `alpha3/` when there's been a sequence of
|
||||||
|
releases I've made, but it really is up to you how you like to
|
||||||
|
organize generated sdists locally.
|
||||||
|
|
||||||
|
The resulting build cmds are approximately:
|
||||||
|
|
||||||
|
.. code:: bash
|
||||||
|
|
||||||
|
python setup.py sdist -d ./dist/XXX.X/
|
||||||
|
|
||||||
|
twine upload -r testpypi dist/XXX.X/*
|
||||||
|
|
||||||
|
twine upload dist/XXX.X/*
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
.. _documented by twine: https://twine.readthedocs.io/en/latest/#using-twine
|
||||||
|
|
||||||
|
|
||||||
|
Debugging and monitoring actor trees
|
||||||
|
------------------------------------
|
||||||
|
TODO: but there are tips in the readme for some terminal commands
|
||||||
|
which can be used to see the process trees easily on Linux.
|
||||||
|
|
||||||
|
|
||||||
|
Using the log system to trace `trio` task flow
|
||||||
|
----------------------------------------------
|
||||||
|
TODO: the logging system is meant to be oriented around
|
||||||
|
stack "layers" of the runtime such that you can track
|
||||||
|
"logical abstraction layers" in the code such as errors, cancellation,
|
||||||
|
IPC and streaming, and the low level transport and wire protocols.
|
|
@ -0,0 +1,109 @@
|
||||||
|
tractor
|
||||||
|
=======
|
||||||
|
The Python async-native multi-core system *you always wanted*.
|
||||||
|
|
||||||
|
|
||||||
|
|gh_actions|
|
||||||
|
|docs|
|
||||||
|
|
||||||
|
.. _actor model: https://en.wikipedia.org/wiki/Actor_model
|
||||||
|
.. _trio: https://github.com/python-trio/trio
|
||||||
|
.. _multi-processing: https://en.wikipedia.org/wiki/Multiprocessing
|
||||||
|
.. _trionic: https://trio.readthedocs.io/en/latest/design.html#high-level-design-principles
|
||||||
|
.. _async sandwich: https://trio.readthedocs.io/en/latest/tutorial.html#async-sandwich
|
||||||
|
.. _structured concurrent: https://trio.discourse.group/t/concise-definition-of-structured-concurrency/228
|
||||||
|
|
||||||
|
|
||||||
|
``tractor`` is a `structured concurrent`_ "`actor model`_" built on trio_ and multi-processing_.
|
||||||
|
|
||||||
|
It is an attempt to pair trionic_ `structured concurrency`_ with
|
||||||
|
distributed Python. You can think of it as a ``trio``
|
||||||
|
*-across-processes* or simply as an opinionated replacement for the
|
||||||
|
stdlib's ``multiprocessing`` but built on async programming primitives
|
||||||
|
from the ground up.
|
||||||
|
|
||||||
|
Don't be scared off by this description. ``tractor`` **is just ``trio``**
|
||||||
|
but with nurseries for process management and cancel-able IPC.
|
||||||
|
If you understand how to work with ``trio``, ``tractor`` will give you
|
||||||
|
the parallelism you've been missing.
|
||||||
|
|
||||||
|
``tractor``'s nurseries let you spawn ``trio`` *"actors"*: new Python
|
||||||
|
processes which each run a ``trio`` scheduled task tree (also known as
|
||||||
|
an `async sandwich`_ - a call to ``trio.run()``). That is, each
|
||||||
|
"*Actor*" is a new process plus a ``trio`` runtime.
|
||||||
|
|
||||||
|
"Actors" communicate by exchanging asynchronous messages_ and avoid
|
||||||
|
sharing state. The intention of this model is to allow for highly
|
||||||
|
distributed software that, through the adherence to *structured
|
||||||
|
concurrency*, results in systems which fail in predictable and
|
||||||
|
recoverable ways.
|
||||||
|
|
||||||
|
The first step to grok ``tractor`` is to get the basics of ``trio`` down.
|
||||||
|
A great place to start is the `trio docs`_ and this `blog post`_.
|
||||||
|
|
||||||
|
.. _messages: https://en.wikipedia.org/wiki/Message_passing
|
||||||
|
.. _trio docs: https://trio.readthedocs.io/en/latest/
|
||||||
|
.. _blog post: https://vorpus.org/blog/notes-on-structured-concurrency-or-go-statement-considered-harmful/
|
||||||
|
.. _structured concurrency: https://vorpus.org/blog/notes-on-structured-concurrency-or-go-statement-considered-harmful/
|
||||||
|
.. _3 axioms: https://en.wikipedia.org/wiki/Actor_model#Fundamental_concepts
|
||||||
|
.. _unrequirements: https://en.wikipedia.org/wiki/Actor_model#Direct_communication_and_asynchrony
|
||||||
|
.. _async generators: https://www.python.org/dev/peps/pep-0525/
|
||||||
|
|
||||||
|
|
||||||
|
Install
|
||||||
|
-------
|
||||||
|
No PyPi release yet!
|
||||||
|
|
||||||
|
::
|
||||||
|
|
||||||
|
pip install git+git://github.com/goodboy/tractor.git
|
||||||
|
|
||||||
|
|
||||||
|
Alluring Features
|
||||||
|
-----------------
|
||||||
|
- **It's just** ``trio``, but with SC applied to processes (aka "actors")
|
||||||
|
- Infinitely nesteable process trees
|
||||||
|
- Built-in API for inter-process streaming
|
||||||
|
- A (first ever?) "native" multi-core debugger for Python using `pdb++`_
|
||||||
|
- (Soon to land) ``asyncio`` support allowing for "infected" actors where
|
||||||
|
`trio` drives the `asyncio` scheduler via the astounding "`guest mode`_"
|
||||||
|
|
||||||
|
|
||||||
|
Example: self-destruct a process tree
|
||||||
|
-------------------------------------
|
||||||
|
.. literalinclude:: ../../examples/parallelism/we_are_processes.py
|
||||||
|
:language: python
|
||||||
|
|
||||||
|
|
||||||
|
The example you're probably after...
|
||||||
|
------------------------------------
|
||||||
|
It seems the initial query from most new users is "how do I make a worker
|
||||||
|
pool thing?".
|
||||||
|
|
||||||
|
``tractor`` is built to handle any SC process tree you can
|
||||||
|
imagine; the "worker pool" pattern is a trivial special case:
|
||||||
|
|
||||||
|
.. literalinclude:: ../../examples/parallelism/concurrent_actors_primes.py
|
||||||
|
:language: python
|
||||||
|
|
||||||
|
|
||||||
|
Feel like saying hi?
|
||||||
|
--------------------
|
||||||
|
This project is very much coupled to the ongoing development of
|
||||||
|
``trio`` (i.e. ``tractor`` gets most of its ideas from that brilliant
|
||||||
|
community). If you want to help, have suggestions or just want to
|
||||||
|
say hi, please feel free to reach us in our `matrix channel`_. If
|
||||||
|
matrix seems too hip, we're also mostly all in the the `trio gitter
|
||||||
|
channel`_!
|
||||||
|
|
||||||
|
.. _trio gitter channel: https://gitter.im/python-trio/general
|
||||||
|
.. _matrix channel: https://matrix.to/#/!tractor:matrix.org
|
||||||
|
.. _pdb++: https://github.com/pdbpp/pdbpp
|
||||||
|
.. _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
|
||||||
|
|
||||||
|
|
||||||
|
.. |gh_actions| image:: https://img.shields.io/endpoint.svg?url=https%3A%2F%2Factions-badge.atrox.dev%2Fgoodboy%2Ftractor%2Fbadge&style=popout-square
|
||||||
|
:target: https://actions-badge.atrox.dev/goodboy/tractor/goto
|
||||||
|
.. |docs| image:: https://readthedocs.org/projects/tractor/badge/?version=latest
|
||||||
|
:target: https://tractor.readthedocs.io/en/latest/?badge=latest
|
||||||
|
:alt: Documentation Status
|
|
@ -0,0 +1,51 @@
|
||||||
|
# Configuration file for the Sphinx documentation builder.
|
||||||
|
|
||||||
|
# this config is for the rst generation extension and thus
|
||||||
|
# requires only basic settings:
|
||||||
|
# https://github.com/sphinx-contrib/restbuilder
|
||||||
|
|
||||||
|
# -- Path setup --------------------------------------------------------------
|
||||||
|
|
||||||
|
# If extensions (or modules to document with autodoc) are in another directory,
|
||||||
|
# add these directories to sys.path here. If the directory is relative to the
|
||||||
|
# documentation root, use os.path.abspath to make it absolute, like shown here.
|
||||||
|
#
|
||||||
|
# import os
|
||||||
|
# import sys
|
||||||
|
# sys.path.insert(0, os.path.abspath('.'))
|
||||||
|
|
||||||
|
# Warn about all references to unknown targets
|
||||||
|
nitpicky = True
|
||||||
|
|
||||||
|
# The master toctree document.
|
||||||
|
master_doc = '_sphinx_readme'
|
||||||
|
|
||||||
|
# -- Project information -----------------------------------------------------
|
||||||
|
|
||||||
|
project = 'tractor'
|
||||||
|
copyright = '2018, Tyler Goodlet'
|
||||||
|
author = 'Tyler Goodlet'
|
||||||
|
|
||||||
|
# The full version, including alpha/beta/rc tags
|
||||||
|
release = '0.0.0a0.dev0'
|
||||||
|
|
||||||
|
# -- General configuration ---------------------------------------------------
|
||||||
|
|
||||||
|
# Add any Sphinx extension module names here, as strings. They can be
|
||||||
|
# extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
|
||||||
|
# ones.
|
||||||
|
extensions = [
|
||||||
|
'sphinx.ext.autodoc',
|
||||||
|
'sphinx.ext.intersphinx',
|
||||||
|
'sphinx.ext.todo',
|
||||||
|
'sphinxcontrib.restbuilder',
|
||||||
|
|
||||||
|
]
|
||||||
|
|
||||||
|
# Add any paths that contain templates here, relative to this directory.
|
||||||
|
templates_path = ['_templates']
|
||||||
|
|
||||||
|
# List of patterns, relative to source directory, that match files and
|
||||||
|
# directories to ignore when looking for source files.
|
||||||
|
# This pattern also affects html_static_path and html_extra_path.
|
||||||
|
exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
|
|
@ -3,12 +3,13 @@
|
||||||
You can adapt this file completely to your liking, but it should at least
|
You can adapt this file completely to your liking, but it should at least
|
||||||
contain the root `toctree` directive.
|
contain the root `toctree` directive.
|
||||||
|
|
||||||
tractor
|
``tractor``
|
||||||
=======
|
===========
|
||||||
|
|
||||||
A `structured concurrent`_, async-native "`actor model`_" built on trio_ and multiprocessing_.
|
A `structured concurrent`_, async-native "`actor model`_" built on trio_ and multiprocessing_.
|
||||||
|
|
||||||
.. toctree::
|
.. toctree::
|
||||||
:maxdepth: 2
|
:maxdepth: 1
|
||||||
:caption: Contents:
|
:caption: Contents:
|
||||||
|
|
||||||
.. _actor model: https://en.wikipedia.org/wiki/Actor_model
|
.. _actor model: https://en.wikipedia.org/wiki/Actor_model
|
||||||
|
@ -58,8 +59,6 @@ say hi, please feel free to ping me on the `trio gitter channel`_!
|
||||||
.. _trio gitter channel: https://gitter.im/python-trio/general
|
.. _trio gitter channel: https://gitter.im/python-trio/general
|
||||||
|
|
||||||
|
|
||||||
.. contents::
|
|
||||||
|
|
||||||
|
|
||||||
Philosophy
|
Philosophy
|
||||||
----------
|
----------
|
||||||
|
@ -145,7 +144,7 @@ and use the ``run_in_actor()`` method:
|
||||||
|
|
||||||
What's going on?
|
What's going on?
|
||||||
|
|
||||||
- an initial *actor* is started with ``tractor.run()`` and told to execute
|
- an initial *actor* is started with ``trio.run()`` and told to execute
|
||||||
its main task_: ``main()``
|
its main task_: ``main()``
|
||||||
|
|
||||||
- inside ``main()`` an actor is *spawned* using an ``ActorNusery`` and is told
|
- inside ``main()`` an actor is *spawned* using an ``ActorNusery`` and is told
|
||||||
|
@ -182,7 +181,7 @@ Here is a similar example using the latter method:
|
||||||
|
|
||||||
.. literalinclude:: ../examples/actor_spawning_and_causality_with_daemon.py
|
.. literalinclude:: ../examples/actor_spawning_and_causality_with_daemon.py
|
||||||
|
|
||||||
The ``rpc_module_paths`` `kwarg` above is a list of module path
|
The ``enable_modules`` `kwarg` above is a list of module path
|
||||||
strings that will be loaded and made accessible for execution in the
|
strings that will be loaded and made accessible for execution in the
|
||||||
remote actor through a call to ``Portal.run()``. For now this is
|
remote actor through a call to ``Portal.run()``. For now this is
|
||||||
a simple mechanism to restrict the functionality of the remote
|
a simple mechanism to restrict the functionality of the remote
|
||||||
|
@ -397,7 +396,7 @@ tasks spawned via multiple RPC calls to an actor can modify
|
||||||
|
|
||||||
|
|
||||||
# a per process cache
|
# a per process cache
|
||||||
_actor_cache: Dict[str, bool] = {}
|
_actor_cache: dict[str, bool] = {}
|
||||||
|
|
||||||
|
|
||||||
def ping_endpoints(endpoints: List[str]):
|
def ping_endpoints(endpoints: List[str]):
|
||||||
|
@ -458,7 +457,7 @@ find an actor's socket address by name use the ``find_actor()`` function:
|
||||||
.. literalinclude:: ../examples/service_discovery.py
|
.. literalinclude:: ../examples/service_discovery.py
|
||||||
|
|
||||||
The ``name`` value you should pass to ``find_actor()`` is the one you passed as the
|
The ``name`` value you should pass to ``find_actor()`` is the one you passed as the
|
||||||
*first* argument to either ``tractor.run()`` or ``ActorNursery.start_actor()``.
|
*first* argument to either ``trio.run()`` or ``ActorNursery.start_actor()``.
|
||||||
|
|
||||||
|
|
||||||
Running actors standalone
|
Running actors standalone
|
||||||
|
@ -472,7 +471,17 @@ need to hop into a debugger. You just need to pass the existing
|
||||||
|
|
||||||
.. code:: python
|
.. code:: python
|
||||||
|
|
||||||
tractor.run(main, arbiter_addr=('192.168.0.10', 1616))
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
arbiter_addr=('192.168.0.10', 1616)
|
||||||
|
):
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
Choosing a process spawning backend
|
Choosing a process spawning backend
|
||||||
|
@ -480,7 +489,7 @@ Choosing a process spawning backend
|
||||||
``tractor`` is architected to support multiple actor (sub-process)
|
``tractor`` is architected to support multiple actor (sub-process)
|
||||||
spawning backends. Specific defaults are chosen based on your system
|
spawning backends. Specific defaults are chosen based on your system
|
||||||
but you can also explicitly select a backend of choice at startup
|
but you can also explicitly select a backend of choice at startup
|
||||||
via a ``start_method`` kwarg to ``tractor.run()``.
|
via a ``start_method`` kwarg to ``tractor.open_nursery()``.
|
||||||
|
|
||||||
Currently the options available are:
|
Currently the options available are:
|
||||||
|
|
||||||
|
@ -536,13 +545,14 @@ main python module of the program:
|
||||||
.. code:: python
|
.. code:: python
|
||||||
|
|
||||||
# application/__main__.py
|
# application/__main__.py
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
import multiprocessing
|
import multiprocessing
|
||||||
from . import tractor_app
|
from . import tractor_app
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
multiprocessing.freeze_support()
|
multiprocessing.freeze_support()
|
||||||
tractor.run(tractor_app.main)
|
trio.run(tractor_app.main)
|
||||||
|
|
||||||
And execute as::
|
And execute as::
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,4 @@
|
||||||
|
#!/bin/bash
|
||||||
|
sphinx-build -b rst ./github_readme ./
|
||||||
|
|
||||||
|
mv _sphinx_readme.rst _README.rst
|
|
@ -16,4 +16,4 @@ if __name__ == '__main__':
|
||||||
# temporary dir and name it test_example.py. We import that script
|
# temporary dir and name it test_example.py. We import that script
|
||||||
# module here and invoke it's ``main()``.
|
# module here and invoke it's ``main()``.
|
||||||
from . import test_example
|
from . import test_example
|
||||||
test_example.tractor.run(test_example.main, start_method='spawn')
|
test_example.trio.run(test_example.main)
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
_this_module = __name__
|
_this_module = __name__
|
||||||
|
@ -40,4 +41,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main)
|
trio.run(main)
|
||||||
|
|
|
@ -1,7 +1,8 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
def cellar_door():
|
async def cellar_door():
|
||||||
assert not tractor.is_root_process()
|
assert not tractor.is_root_process()
|
||||||
return "Dang that's beautiful"
|
return "Dang that's beautiful"
|
||||||
|
|
||||||
|
@ -23,4 +24,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main)
|
trio.run(main)
|
||||||
|
|
|
@ -1,7 +1,8 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
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.
|
||||||
"""
|
"""
|
||||||
|
@ -16,7 +17,7 @@ async def main():
|
||||||
portal = await n.start_actor(
|
portal = await n.start_actor(
|
||||||
'frank',
|
'frank',
|
||||||
# enable the actor to run funcs from this current module
|
# enable the actor to run funcs from this current module
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
print(await portal.run(movie_theatre_question))
|
print(await portal.run(movie_theatre_question))
|
||||||
|
@ -30,4 +31,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main)
|
trio.run(main)
|
||||||
|
|
|
@ -0,0 +1,151 @@
|
||||||
|
'''
|
||||||
|
Complex edge case where during real-time streaming the IPC tranport
|
||||||
|
channels are wiped out (purposely in this example though it could have
|
||||||
|
been an outage) and we want to ensure that despite being in debug mode
|
||||||
|
(or not) the user can sent SIGINT once they notice the hang and the
|
||||||
|
actor tree will eventually be cancelled without leaving any zombies.
|
||||||
|
|
||||||
|
'''
|
||||||
|
import trio
|
||||||
|
from tractor import (
|
||||||
|
open_nursery,
|
||||||
|
context,
|
||||||
|
Context,
|
||||||
|
MsgStream,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def break_channel_silently_then_error(
|
||||||
|
stream: MsgStream,
|
||||||
|
):
|
||||||
|
async for msg in stream:
|
||||||
|
await stream.send(msg)
|
||||||
|
|
||||||
|
# XXX: close the channel right after an error is raised
|
||||||
|
# purposely breaking the IPC transport to make sure the parent
|
||||||
|
# doesn't get stuck in debug or hang on the connection join.
|
||||||
|
# this more or less simulates an infinite msg-receive hang on
|
||||||
|
# the other end.
|
||||||
|
await stream._ctx.chan.send(None)
|
||||||
|
assert 0
|
||||||
|
|
||||||
|
|
||||||
|
async def close_stream_and_error(
|
||||||
|
stream: MsgStream,
|
||||||
|
):
|
||||||
|
async for msg in stream:
|
||||||
|
await stream.send(msg)
|
||||||
|
|
||||||
|
# wipe out channel right before raising
|
||||||
|
await stream._ctx.chan.send(None)
|
||||||
|
await stream.aclose()
|
||||||
|
assert 0
|
||||||
|
|
||||||
|
|
||||||
|
@context
|
||||||
|
async def recv_and_spawn_net_killers(
|
||||||
|
|
||||||
|
ctx: Context,
|
||||||
|
break_ipc_after: bool | int = False,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Receive stream msgs and spawn some IPC killers mid-stream.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
async with (
|
||||||
|
ctx.open_stream() as stream,
|
||||||
|
trio.open_nursery() as n,
|
||||||
|
):
|
||||||
|
async for i in stream:
|
||||||
|
print(f'child echoing {i}')
|
||||||
|
await stream.send(i)
|
||||||
|
if (
|
||||||
|
break_ipc_after
|
||||||
|
and i > break_ipc_after
|
||||||
|
):
|
||||||
|
'#################################\n'
|
||||||
|
'Simulating child-side IPC BREAK!\n'
|
||||||
|
'#################################'
|
||||||
|
n.start_soon(break_channel_silently_then_error, stream)
|
||||||
|
n.start_soon(close_stream_and_error, stream)
|
||||||
|
|
||||||
|
|
||||||
|
async def main(
|
||||||
|
debug_mode: bool = False,
|
||||||
|
start_method: str = 'trio',
|
||||||
|
|
||||||
|
# by default we break the parent IPC first (if configured to break
|
||||||
|
# at all), but this can be changed so the child does first (even if
|
||||||
|
# both are set to break).
|
||||||
|
break_parent_ipc_after: int | bool = False,
|
||||||
|
break_child_ipc_after: int | bool = False,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
async with (
|
||||||
|
open_nursery(
|
||||||
|
start_method=start_method,
|
||||||
|
|
||||||
|
# NOTE: even debugger is used we shouldn't get
|
||||||
|
# a hang since it never engages due to broken IPC
|
||||||
|
debug_mode=debug_mode,
|
||||||
|
loglevel='warning',
|
||||||
|
|
||||||
|
) as an,
|
||||||
|
):
|
||||||
|
portal = await an.start_actor(
|
||||||
|
'chitty_hijo',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
recv_and_spawn_net_killers,
|
||||||
|
break_ipc_after=break_child_ipc_after,
|
||||||
|
|
||||||
|
) as (ctx, sent):
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
for i in range(1000):
|
||||||
|
|
||||||
|
if (
|
||||||
|
break_parent_ipc_after
|
||||||
|
and i > break_parent_ipc_after
|
||||||
|
):
|
||||||
|
print(
|
||||||
|
'#################################\n'
|
||||||
|
'Simulating parent-side IPC BREAK!\n'
|
||||||
|
'#################################'
|
||||||
|
)
|
||||||
|
await stream._ctx.chan.send(None)
|
||||||
|
|
||||||
|
# it actually breaks right here in the
|
||||||
|
# mp_spawn/forkserver backends and thus the zombie
|
||||||
|
# reaper never even kicks in?
|
||||||
|
print(f'parent sending {i}')
|
||||||
|
await stream.send(i)
|
||||||
|
|
||||||
|
with trio.move_on_after(2) as cs:
|
||||||
|
|
||||||
|
# NOTE: in the parent side IPC failure case this
|
||||||
|
# will raise an ``EndOfChannel`` after the child
|
||||||
|
# is killed and sends a stop msg back to it's
|
||||||
|
# caller/this-parent.
|
||||||
|
rx = await stream.receive()
|
||||||
|
|
||||||
|
print(f"I'm a happy user and echoed to me is {rx}")
|
||||||
|
|
||||||
|
if cs.cancelled_caught:
|
||||||
|
# pretend to be a user seeing no streaming action
|
||||||
|
# thinking it's a hang, and then hitting ctl-c..
|
||||||
|
print("YOO i'm a user anddd thingz hangin..")
|
||||||
|
|
||||||
|
print(
|
||||||
|
"YOO i'm mad send side dun but thingz hangin..\n"
|
||||||
|
'MASHING CTlR-C Ctl-c..'
|
||||||
|
)
|
||||||
|
raise KeyboardInterrupt
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -1,36 +1,42 @@
|
||||||
|
from typing import AsyncIterator
|
||||||
from itertools import repeat
|
from itertools import repeat
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
tractor.log.get_console_log("INFO")
|
|
||||||
|
|
||||||
|
async def stream_forever() -> AsyncIterator[int]:
|
||||||
|
|
||||||
async def stream_forever():
|
|
||||||
for i in repeat("I can see these little future bubble things"):
|
for i in repeat("I can see these little future bubble things"):
|
||||||
# each yielded value is sent over the ``Channel`` to the
|
# each yielded value is sent over the ``Channel`` to the parent actor
|
||||||
# parent actor
|
|
||||||
yield i
|
yield i
|
||||||
await trio.sleep(0.01)
|
await trio.sleep(0.01)
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
# stream for at most 1 seconds
|
|
||||||
with 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(
|
||||||
f'donny',
|
'donny',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
# this async for loop streams values from the above
|
# this async for loop streams values from the above
|
||||||
# async generator running in a separate process
|
# async generator running in a separate process
|
||||||
async for letter in await portal.run(stream_forever):
|
async with portal.open_stream_from(stream_forever) as stream:
|
||||||
|
count = 0
|
||||||
|
async for letter in stream:
|
||||||
print(letter)
|
print(letter)
|
||||||
|
count += 1
|
||||||
|
|
||||||
# we support trio's cancellation system
|
if count > 50:
|
||||||
assert cancel_scope.cancelled_caught
|
break
|
||||||
assert n.cancelled
|
|
||||||
|
print('stream terminated')
|
||||||
|
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main)
|
trio.run(main)
|
||||||
|
|
|
@ -0,0 +1,54 @@
|
||||||
|
'''
|
||||||
|
Fast fail test with a context.
|
||||||
|
|
||||||
|
Ensure the partially initialized sub-actor process
|
||||||
|
doesn't cause a hang on error/cancel of the parent
|
||||||
|
nursery.
|
||||||
|
|
||||||
|
'''
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def sleep(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
):
|
||||||
|
await trio.sleep(0.5)
|
||||||
|
await ctx.started()
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
async def open_ctx(
|
||||||
|
n: tractor._supervise.ActorNursery
|
||||||
|
):
|
||||||
|
|
||||||
|
# spawn both actors
|
||||||
|
portal = await n.start_actor(
|
||||||
|
name='sleeper',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
sleep,
|
||||||
|
) as (ctx, first):
|
||||||
|
assert first is None
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
loglevel='runtime',
|
||||||
|
) as an:
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
n.start_soon(open_ctx, an)
|
||||||
|
|
||||||
|
await trio.sleep(0.2)
|
||||||
|
await trio.sleep(0.1)
|
||||||
|
assert 0
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -11,21 +11,35 @@ async def breakpoint_forever():
|
||||||
|
|
||||||
async def name_error():
|
async def name_error():
|
||||||
"Raise a ``NameError``"
|
"Raise a ``NameError``"
|
||||||
getattr(doggypants)
|
getattr(doggypants) # noqa
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
"""Test breakpoint in a streaming actor.
|
"""Test breakpoint in a streaming actor.
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
loglevel='error',
|
||||||
|
) as n:
|
||||||
|
|
||||||
p0 = await n.start_actor('bp_forever', rpc_module_paths=[__name__])
|
p0 = await n.start_actor('bp_forever', enable_modules=[__name__])
|
||||||
p1 = await n.start_actor('name_error', rpc_module_paths=[__name__])
|
p1 = await n.start_actor('name_error', enable_modules=[__name__])
|
||||||
|
|
||||||
# retreive results
|
# retreive results
|
||||||
stream = await p0.run(breakpoint_forever)
|
async with p0.open_stream_from(breakpoint_forever) as stream:
|
||||||
|
|
||||||
|
# triggers the first name error
|
||||||
|
try:
|
||||||
|
await p1.run(name_error)
|
||||||
|
except tractor.RemoteActorError as rae:
|
||||||
|
assert rae.type is NameError
|
||||||
|
|
||||||
|
async for i in stream:
|
||||||
|
|
||||||
|
# a second time try the failing subactor and this tie
|
||||||
|
# let error propagate up to the parent/nursery.
|
||||||
await p1.run(name_error)
|
await p1.run(name_error)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True, loglevel='error')
|
trio.run(main)
|
||||||
|
|
|
@ -1,9 +1,10 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
async def name_error():
|
async def name_error():
|
||||||
"Raise a ``NameError``"
|
"Raise a ``NameError``"
|
||||||
getattr(doggypants)
|
getattr(doggypants) # noqa
|
||||||
|
|
||||||
|
|
||||||
async def breakpoint_forever():
|
async def breakpoint_forever():
|
||||||
|
@ -11,18 +12,31 @@ async def breakpoint_forever():
|
||||||
while True:
|
while True:
|
||||||
await tractor.breakpoint()
|
await tractor.breakpoint()
|
||||||
|
|
||||||
|
# NOTE: if the test never sent 'q'/'quit' commands
|
||||||
|
# on the pdb repl, without this checkpoint line the
|
||||||
|
# repl would spin in this actor forever.
|
||||||
|
# await trio.sleep(0)
|
||||||
|
|
||||||
|
|
||||||
async def spawn_until(depth=0):
|
async def spawn_until(depth=0):
|
||||||
""""A nested nursery that triggers another ``NameError``.
|
""""A nested nursery that triggers another ``NameError``.
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery() as n:
|
||||||
if depth < 1:
|
if depth < 1:
|
||||||
# await n.run_in_actor('breakpoint_forever', breakpoint_forever)
|
|
||||||
await n.run_in_actor(
|
await n.run_in_actor(breakpoint_forever)
|
||||||
|
|
||||||
|
p = await n.run_in_actor(
|
||||||
name_error,
|
name_error,
|
||||||
name='name_error'
|
name='name_error'
|
||||||
)
|
)
|
||||||
|
await trio.sleep(0.5)
|
||||||
|
# rx and propagate error from child
|
||||||
|
await p.result()
|
||||||
|
|
||||||
else:
|
else:
|
||||||
|
# recusrive call to spawn another process branching layer of
|
||||||
|
# the tree
|
||||||
depth -= 1
|
depth -= 1
|
||||||
await n.run_in_actor(
|
await n.run_in_actor(
|
||||||
spawn_until,
|
spawn_until,
|
||||||
|
@ -50,7 +64,10 @@ async def main():
|
||||||
└─ python -m tractor._child --uid ('spawn_until_0', 'de918e6d ...)
|
└─ python -m tractor._child --uid ('spawn_until_0', 'de918e6d ...)
|
||||||
|
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
# loglevel='cancel',
|
||||||
|
) as n:
|
||||||
|
|
||||||
# spawn both actors
|
# spawn both actors
|
||||||
portal = await n.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
|
@ -64,10 +81,18 @@ async def main():
|
||||||
name='spawner1',
|
name='spawner1',
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# TODO: test this case as well where the parent don't see
|
||||||
|
# the sub-actor errors by default and instead expect a user
|
||||||
|
# ctrl-c to kill the root.
|
||||||
|
with trio.move_on_after(3):
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
# gah still an issue here.
|
# gah still an issue here.
|
||||||
await portal.result()
|
await portal.result()
|
||||||
|
|
||||||
|
# should never get here
|
||||||
await portal1.result()
|
await portal1.result()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True)
|
trio.run(main)
|
||||||
|
|
|
@ -1,9 +1,15 @@
|
||||||
|
'''
|
||||||
|
Test that a nested nursery will avoid clobbering
|
||||||
|
the debugger latched by a broken child.
|
||||||
|
|
||||||
|
'''
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
async def name_error():
|
async def name_error():
|
||||||
"Raise a ``NameError``"
|
"Raise a ``NameError``"
|
||||||
getattr(doggypants)
|
getattr(doggypants) # noqa
|
||||||
|
|
||||||
|
|
||||||
async def spawn_error():
|
async def spawn_error():
|
||||||
|
@ -26,8 +32,16 @@ async def main():
|
||||||
├─ python -m tractor._child --uid ('name_error', 'a7caf490 ...)
|
├─ python -m tractor._child --uid ('name_error', 'a7caf490 ...)
|
||||||
`-python -m tractor._child --uid ('spawn_error', '52ee14a5 ...)
|
`-python -m tractor._child --uid ('spawn_error', '52ee14a5 ...)
|
||||||
`-python -m tractor._child --uid ('name_error', '3391222c ...)
|
`-python -m tractor._child --uid ('name_error', '3391222c ...)
|
||||||
|
|
||||||
|
Order of failure:
|
||||||
|
- nested name_error sub-sub-actor
|
||||||
|
- root actor should then fail on assert
|
||||||
|
- program termination
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
# loglevel='cancel',
|
||||||
|
) as n:
|
||||||
|
|
||||||
# spawn both actors
|
# spawn both actors
|
||||||
portal = await n.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
|
@ -49,4 +63,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True)
|
trio.run(main)
|
||||||
|
|
|
@ -11,7 +11,7 @@ async def breakpoint_forever():
|
||||||
|
|
||||||
async def name_error():
|
async def name_error():
|
||||||
"Raise a ``NameError``"
|
"Raise a ``NameError``"
|
||||||
getattr(doggypants)
|
getattr(doggypants) # noqa
|
||||||
|
|
||||||
|
|
||||||
async def spawn_error():
|
async def spawn_error():
|
||||||
|
@ -36,7 +36,9 @@ async def main():
|
||||||
`-python -m tractor._child --uid ('spawn_error', '52ee14a5 ...)
|
`-python -m tractor._child --uid ('spawn_error', '52ee14a5 ...)
|
||||||
`-python -m tractor._child --uid ('name_error', '3391222c ...)
|
`-python -m tractor._child --uid ('name_error', '3391222c ...)
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
) as n:
|
||||||
|
|
||||||
# Spawn both actors, don't bother with collecting results
|
# Spawn both actors, don't bother with collecting results
|
||||||
# (would result in a different debugger outcome due to parent's
|
# (would result in a different debugger outcome due to parent's
|
||||||
|
@ -47,4 +49,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True)
|
trio.run(main)
|
||||||
|
|
|
@ -0,0 +1,40 @@
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def just_sleep(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
**kwargs,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Start and sleep.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
async def main() -> None:
|
||||||
|
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
) as n:
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'ctx_child',
|
||||||
|
|
||||||
|
# XXX: we don't enable the current module in order
|
||||||
|
# to trigger `ModuleNotFound`.
|
||||||
|
enable_modules=[],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
just_sleep, # taken from pytest parameterization
|
||||||
|
) as (ctx, sent):
|
||||||
|
raise KeyboardInterrupt
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,27 @@
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
async def die():
|
||||||
|
raise RuntimeError
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as tn:
|
||||||
|
|
||||||
|
debug_actor = await tn.start_actor(
|
||||||
|
'debugged_boi',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
debug_mode=True,
|
||||||
|
)
|
||||||
|
crash_boi = await tn.start_actor(
|
||||||
|
'crash_boi',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
# debug_mode=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
n.start_soon(debug_actor.run, die)
|
||||||
|
n.start_soon(crash_boi.run, die)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,24 @@
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
async def main() -> None:
|
||||||
|
async with tractor.open_nursery(debug_mode=True) as an:
|
||||||
|
|
||||||
|
assert os.environ['PYTHONBREAKPOINT'] == 'tractor._debug._set_trace'
|
||||||
|
|
||||||
|
# TODO: an assert that verifies the hook has indeed been, hooked
|
||||||
|
# XD
|
||||||
|
assert sys.breakpointhook is not tractor._debug._set_trace
|
||||||
|
|
||||||
|
breakpoint()
|
||||||
|
|
||||||
|
# TODO: an assert that verifies the hook is unhooked..
|
||||||
|
assert sys.breakpointhook
|
||||||
|
breakpoint()
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -4,6 +4,10 @@ import tractor
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
debug_mode=True,
|
||||||
|
):
|
||||||
|
|
||||||
await trio.sleep(0.1)
|
await trio.sleep(0.1)
|
||||||
|
|
||||||
await tractor.breakpoint()
|
await tractor.breakpoint()
|
||||||
|
@ -12,4 +16,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True)
|
trio.run(main)
|
||||||
|
|
|
@ -1,11 +1,15 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
debug_mode=True,
|
||||||
|
):
|
||||||
while True:
|
while True:
|
||||||
await tractor.breakpoint()
|
await tractor.breakpoint()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True)
|
trio.run(main)
|
||||||
|
|
|
@ -1,9 +1,13 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
debug_mode=True,
|
||||||
|
):
|
||||||
assert 0
|
assert 0
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True)
|
trio.run(main)
|
||||||
|
|
|
@ -1,9 +1,10 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
async def name_error():
|
async def name_error():
|
||||||
"Raise a ``NameError``"
|
"Raise a ``NameError``"
|
||||||
getattr(doggypants)
|
getattr(doggypants) # noqa
|
||||||
|
|
||||||
|
|
||||||
async def spawn_until(depth=0):
|
async def spawn_until(depth=0):
|
||||||
|
@ -37,7 +38,10 @@ async def main():
|
||||||
└─ python -m tractor._child --uid ('name_error', '6c2733b8 ...)
|
└─ python -m tractor._child --uid ('name_error', '6c2733b8 ...)
|
||||||
|
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
loglevel='warning'
|
||||||
|
) as n:
|
||||||
|
|
||||||
# spawn both actors
|
# spawn both actors
|
||||||
portal = await n.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
|
@ -58,4 +62,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True, loglevel='warning')
|
trio.run(main)
|
||||||
|
|
|
@ -0,0 +1,31 @@
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
async def key_error():
|
||||||
|
"Raise a ``NameError``"
|
||||||
|
return {}['doggy']
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
"""Root dies
|
||||||
|
|
||||||
|
"""
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
loglevel='debug'
|
||||||
|
) as n:
|
||||||
|
|
||||||
|
# spawn both actors
|
||||||
|
portal = await n.run_in_actor(key_error)
|
||||||
|
|
||||||
|
# XXX: originally a bug caused by this is where root would enter
|
||||||
|
# the debugger and clobber the tty used by the repl even though
|
||||||
|
# child should have it locked.
|
||||||
|
with trio.fail_after(1):
|
||||||
|
await trio.Event().wait()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,50 @@
|
||||||
|
import tractor
|
||||||
|
import trio
|
||||||
|
|
||||||
|
|
||||||
|
async def gen():
|
||||||
|
yield 'yo'
|
||||||
|
await tractor.breakpoint()
|
||||||
|
yield 'yo'
|
||||||
|
await tractor.breakpoint()
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def just_bp(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
await ctx.started()
|
||||||
|
await tractor.breakpoint()
|
||||||
|
|
||||||
|
# TODO: bps and errors in this call..
|
||||||
|
async for val in gen():
|
||||||
|
print(val)
|
||||||
|
|
||||||
|
# await trio.sleep(0.5)
|
||||||
|
|
||||||
|
# prematurely destroy the connection
|
||||||
|
await ctx.chan.aclose()
|
||||||
|
|
||||||
|
# THIS CAUSES AN UNRECOVERABLE HANG
|
||||||
|
# without latest ``pdbpp``:
|
||||||
|
assert 0
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
) as n:
|
||||||
|
p = await n.start_actor(
|
||||||
|
'bp_boi',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
async with p.open_context(
|
||||||
|
just_bp,
|
||||||
|
) as (ctx, first):
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -12,7 +12,9 @@ async def breakpoint_forever():
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
) as n:
|
||||||
|
|
||||||
portal = await n.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
breakpoint_forever,
|
breakpoint_forever,
|
||||||
|
@ -21,4 +23,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True)
|
trio.run(main)
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
@ -6,11 +7,13 @@ async def name_error():
|
||||||
|
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
debug_mode=True,
|
||||||
|
) as n:
|
||||||
|
|
||||||
portal = await n.run_in_actor(name_error)
|
portal = await n.run_in_actor(name_error)
|
||||||
await portal.result()
|
await portal.result()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, debug_mode=True)
|
trio.run(main)
|
||||||
|
|
|
@ -7,7 +7,7 @@ import tractor
|
||||||
async def stream_data(seed):
|
async def stream_data(seed):
|
||||||
for i in range(seed):
|
for i in range(seed):
|
||||||
yield i
|
yield i
|
||||||
await trio.sleep(0) # trigger scheduler
|
await trio.sleep(0.0001) # trigger scheduler
|
||||||
|
|
||||||
|
|
||||||
# this is the third actor; the aggregator
|
# this is the third actor; the aggregator
|
||||||
|
@ -21,7 +21,7 @@ async def aggregate(seed):
|
||||||
# fork point
|
# fork point
|
||||||
portal = await nursery.start_actor(
|
portal = await nursery.start_actor(
|
||||||
name=f'streamer_{i}',
|
name=f'streamer_{i}',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
portals.append(portal)
|
portals.append(portal)
|
||||||
|
@ -29,8 +29,11 @@ async def aggregate(seed):
|
||||||
send_chan, recv_chan = trio.open_memory_channel(500)
|
send_chan, recv_chan = trio.open_memory_channel(500)
|
||||||
|
|
||||||
async def push_to_chan(portal, send_chan):
|
async def push_to_chan(portal, send_chan):
|
||||||
|
|
||||||
|
# TODO: https://github.com/goodboy/tractor/issues/207
|
||||||
async with send_chan:
|
async with send_chan:
|
||||||
async for value in await portal.run(stream_data, seed=seed):
|
async with portal.open_stream_from(stream_data, seed=seed) as stream:
|
||||||
|
async for value in stream:
|
||||||
# leverage trio's built-in backpressure
|
# leverage trio's built-in backpressure
|
||||||
await send_chan.send(value)
|
await send_chan.send(value)
|
||||||
|
|
||||||
|
@ -65,25 +68,32 @@ async def aggregate(seed):
|
||||||
# this is the main actor and *arbiter*
|
# this is the main actor and *arbiter*
|
||||||
async def main():
|
async def main():
|
||||||
# a nursery which spawns "actors"
|
# a nursery which spawns "actors"
|
||||||
async with tractor.open_nursery() as nursery:
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=('127.0.0.1', 1616)
|
||||||
|
) as nursery:
|
||||||
|
|
||||||
seed = int(1e3)
|
seed = int(1e3)
|
||||||
import time
|
|
||||||
pre_start = time.time()
|
pre_start = time.time()
|
||||||
|
|
||||||
portal = await nursery.run_in_actor(
|
portal = await nursery.start_actor(
|
||||||
aggregate,
|
|
||||||
name='aggregator',
|
name='aggregator',
|
||||||
seed=seed,
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
async with portal.open_stream_from(
|
||||||
|
aggregate,
|
||||||
|
seed=seed,
|
||||||
|
) as stream:
|
||||||
|
|
||||||
start = time.time()
|
start = time.time()
|
||||||
# the portal call returns exactly what you'd expect
|
# the portal call returns exactly what you'd expect
|
||||||
# as if the remote "aggregate" function was called locally
|
# as if the remote "aggregate" function was called locally
|
||||||
result_stream = []
|
result_stream = []
|
||||||
async for value in await portal.result():
|
async for value in stream:
|
||||||
result_stream.append(value)
|
result_stream.append(value)
|
||||||
|
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
print(f"STREAM TIME = {time.time() - start}")
|
print(f"STREAM TIME = {time.time() - start}")
|
||||||
print(f"STREAM + SPAWN TIME = {time.time() - pre_start}")
|
print(f"STREAM + SPAWN TIME = {time.time() - pre_start}")
|
||||||
assert result_stream == list(range(seed))
|
assert result_stream == list(range(seed))
|
||||||
|
@ -91,4 +101,4 @@ async def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
final_stream = tractor.run(main, arbiter_addr=('127.0.0.1', 1616))
|
final_stream = trio.run(main)
|
||||||
|
|
|
@ -0,0 +1,92 @@
|
||||||
|
'''
|
||||||
|
An SC compliant infected ``asyncio`` echo server.
|
||||||
|
|
||||||
|
'''
|
||||||
|
import asyncio
|
||||||
|
from statistics import mean
|
||||||
|
import time
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
async def aio_echo_server(
|
||||||
|
to_trio: trio.MemorySendChannel,
|
||||||
|
from_trio: asyncio.Queue,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
# a first message must be sent **from** this ``asyncio``
|
||||||
|
# task or the ``trio`` side will never unblock from
|
||||||
|
# ``tractor.to_asyncio.open_channel_from():``
|
||||||
|
to_trio.send_nowait('start')
|
||||||
|
|
||||||
|
# XXX: this uses an ``from_trio: asyncio.Queue`` currently but we
|
||||||
|
# should probably offer something better.
|
||||||
|
while True:
|
||||||
|
# echo the msg back
|
||||||
|
to_trio.send_nowait(await from_trio.get())
|
||||||
|
await asyncio.sleep(0)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def trio_to_aio_echo_server(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
):
|
||||||
|
# this will block until the ``asyncio`` task sends a "first"
|
||||||
|
# message.
|
||||||
|
async with tractor.to_asyncio.open_channel_from(
|
||||||
|
aio_echo_server,
|
||||||
|
) as (first, chan):
|
||||||
|
|
||||||
|
assert first == 'start'
|
||||||
|
await ctx.started(first)
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
async for msg in stream:
|
||||||
|
await chan.send(msg)
|
||||||
|
|
||||||
|
out = await chan.receive()
|
||||||
|
# echo back to parent actor-task
|
||||||
|
await stream.send(out)
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
p = await n.start_actor(
|
||||||
|
'aio_server',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
async with p.open_context(
|
||||||
|
trio_to_aio_echo_server,
|
||||||
|
) as (ctx, first):
|
||||||
|
|
||||||
|
assert first == 'start'
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
delays = []
|
||||||
|
send = time.time()
|
||||||
|
|
||||||
|
await stream.send(count)
|
||||||
|
async for msg in stream:
|
||||||
|
recv = time.time()
|
||||||
|
delays.append(recv - send)
|
||||||
|
assert msg == count
|
||||||
|
count += 1
|
||||||
|
send = time.time()
|
||||||
|
await stream.send(count)
|
||||||
|
|
||||||
|
if count >= 1e3:
|
||||||
|
break
|
||||||
|
|
||||||
|
print(f'mean round trip rate (Hz): {1/mean(delays)}')
|
||||||
|
await p.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,49 @@
|
||||||
|
import trio
|
||||||
|
import click
|
||||||
|
import tractor
|
||||||
|
import pydantic
|
||||||
|
# from multiprocessing import shared_memory
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def just_sleep(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
**kwargs,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Test a small ping-pong 2-way streaming server.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
async def main() -> None:
|
||||||
|
|
||||||
|
proc = await trio.open_process( (
|
||||||
|
'python',
|
||||||
|
'-c',
|
||||||
|
'import trio; trio.run(trio.sleep_forever)',
|
||||||
|
))
|
||||||
|
await proc.wait()
|
||||||
|
# await trio.sleep_forever()
|
||||||
|
# async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
# portal = await n.start_actor(
|
||||||
|
# 'rpc_server',
|
||||||
|
# enable_modules=[__name__],
|
||||||
|
# )
|
||||||
|
|
||||||
|
# async with portal.open_context(
|
||||||
|
# just_sleep, # taken from pytest parameterization
|
||||||
|
# ) as (ctx, sent):
|
||||||
|
# await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import time
|
||||||
|
# time.sleep(999)
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,46 @@
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
log = tractor.log.get_logger('multiportal')
|
||||||
|
|
||||||
|
|
||||||
|
async def stream_data(seed=10):
|
||||||
|
log.info("Starting stream task")
|
||||||
|
|
||||||
|
for i in range(seed):
|
||||||
|
yield i
|
||||||
|
await trio.sleep(0) # trigger scheduler
|
||||||
|
|
||||||
|
|
||||||
|
async def stream_from_portal(p, consumed):
|
||||||
|
|
||||||
|
async with p.open_stream_from(stream_data) as stream:
|
||||||
|
async for item in stream:
|
||||||
|
if item in consumed:
|
||||||
|
consumed.remove(item)
|
||||||
|
else:
|
||||||
|
consumed.append(item)
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery(loglevel='info') as an:
|
||||||
|
|
||||||
|
p = await an.start_actor('stream_boi', enable_modules=[__name__])
|
||||||
|
|
||||||
|
consumed = []
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
for i in range(2):
|
||||||
|
n.start_soon(stream_from_portal, p, consumed)
|
||||||
|
|
||||||
|
# both streaming consumer tasks have completed and so we should
|
||||||
|
# have nothing in our list thanks to single threadedness
|
||||||
|
assert not consumed
|
||||||
|
|
||||||
|
await an.cancel()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,43 @@
|
||||||
|
import time
|
||||||
|
import concurrent.futures
|
||||||
|
import math
|
||||||
|
|
||||||
|
PRIMES = [
|
||||||
|
112272535095293,
|
||||||
|
112582705942171,
|
||||||
|
112272535095293,
|
||||||
|
115280095190773,
|
||||||
|
115797848077099,
|
||||||
|
1099726899285419]
|
||||||
|
|
||||||
|
|
||||||
|
def is_prime(n):
|
||||||
|
if n < 2:
|
||||||
|
return False
|
||||||
|
if n == 2:
|
||||||
|
return True
|
||||||
|
if n % 2 == 0:
|
||||||
|
return False
|
||||||
|
|
||||||
|
sqrt_n = int(math.floor(math.sqrt(n)))
|
||||||
|
for i in range(3, sqrt_n + 1, 2):
|
||||||
|
if n % i == 0:
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
with concurrent.futures.ProcessPoolExecutor() as executor:
|
||||||
|
start = time.time()
|
||||||
|
|
||||||
|
for number, prime in zip(PRIMES, executor.map(is_prime, PRIMES)):
|
||||||
|
print('%d is prime: %s' % (number, prime))
|
||||||
|
|
||||||
|
print(f'processing took {time.time() - start} seconds')
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
main()
|
||||||
|
print(f'script took {time.time() - start} seconds')
|
|
@ -0,0 +1,119 @@
|
||||||
|
"""
|
||||||
|
Demonstration of the prime number detector example from the
|
||||||
|
``concurrent.futures`` docs:
|
||||||
|
|
||||||
|
https://docs.python.org/3/library/concurrent.futures.html#processpoolexecutor-example
|
||||||
|
|
||||||
|
This uses no extra threads, fancy semaphores or futures; all we need
|
||||||
|
is ``tractor``'s channels.
|
||||||
|
|
||||||
|
"""
|
||||||
|
from contextlib import asynccontextmanager
|
||||||
|
from typing import Callable
|
||||||
|
import itertools
|
||||||
|
import math
|
||||||
|
import time
|
||||||
|
|
||||||
|
import tractor
|
||||||
|
import trio
|
||||||
|
from async_generator import aclosing
|
||||||
|
|
||||||
|
|
||||||
|
PRIMES = [
|
||||||
|
112272535095293,
|
||||||
|
112582705942171,
|
||||||
|
112272535095293,
|
||||||
|
115280095190773,
|
||||||
|
115797848077099,
|
||||||
|
1099726899285419,
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
async def is_prime(n):
|
||||||
|
if n < 2:
|
||||||
|
return False
|
||||||
|
if n == 2:
|
||||||
|
return True
|
||||||
|
if n % 2 == 0:
|
||||||
|
return False
|
||||||
|
|
||||||
|
sqrt_n = int(math.floor(math.sqrt(n)))
|
||||||
|
for i in range(3, sqrt_n + 1, 2):
|
||||||
|
if n % i == 0:
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
@asynccontextmanager
|
||||||
|
async def worker_pool(workers=4):
|
||||||
|
"""Though it's a trivial special case for ``tractor``, the well
|
||||||
|
known "worker pool" seems to be the defacto "but, I want this
|
||||||
|
process pattern!" for most parallelism pilgrims.
|
||||||
|
|
||||||
|
Yes, the workers stay alive (and ready for work) until you close
|
||||||
|
the context.
|
||||||
|
"""
|
||||||
|
async with tractor.open_nursery() as tn:
|
||||||
|
|
||||||
|
portals = []
|
||||||
|
snd_chan, recv_chan = trio.open_memory_channel(len(PRIMES))
|
||||||
|
|
||||||
|
for i in range(workers):
|
||||||
|
|
||||||
|
# this starts a new sub-actor (process + trio runtime) and
|
||||||
|
# stores it's "portal" for later use to "submit jobs" (ugh).
|
||||||
|
portals.append(
|
||||||
|
await tn.start_actor(
|
||||||
|
f'worker_{i}',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
async def _map(
|
||||||
|
worker_func: Callable[[int], bool],
|
||||||
|
sequence: list[int]
|
||||||
|
) -> list[bool]:
|
||||||
|
|
||||||
|
# define an async (local) task to collect results from workers
|
||||||
|
async def send_result(func, value, portal):
|
||||||
|
await snd_chan.send((value, await portal.run(func, n=value)))
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
|
||||||
|
for value, portal in zip(sequence, itertools.cycle(portals)):
|
||||||
|
n.start_soon(
|
||||||
|
send_result,
|
||||||
|
worker_func,
|
||||||
|
value,
|
||||||
|
portal
|
||||||
|
)
|
||||||
|
|
||||||
|
# deliver results as they arrive
|
||||||
|
for _ in range(len(sequence)):
|
||||||
|
yield await recv_chan.receive()
|
||||||
|
|
||||||
|
# deliver the parallel "worker mapper" to user code
|
||||||
|
yield _map
|
||||||
|
|
||||||
|
# tear down all "workers" on pool close
|
||||||
|
await tn.cancel()
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with worker_pool() as actor_map:
|
||||||
|
|
||||||
|
start = time.time()
|
||||||
|
|
||||||
|
async with aclosing(actor_map(is_prime, PRIMES)) as results:
|
||||||
|
async for number, prime in results:
|
||||||
|
|
||||||
|
print(f'{number} is prime: {prime}')
|
||||||
|
|
||||||
|
print(f'processing took {time.time() - start} seconds')
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
start = time.time()
|
||||||
|
trio.run(main)
|
||||||
|
print(f'script took {time.time() - start} seconds')
|
|
@ -0,0 +1,43 @@
|
||||||
|
"""
|
||||||
|
Run with a process monitor from a terminal using::
|
||||||
|
|
||||||
|
$TERM -e watch -n 0.1 "pstree -a $$" \
|
||||||
|
& python examples/parallelism/single_func.py \
|
||||||
|
&& kill $!
|
||||||
|
|
||||||
|
"""
|
||||||
|
import os
|
||||||
|
|
||||||
|
import tractor
|
||||||
|
import trio
|
||||||
|
|
||||||
|
|
||||||
|
async def burn_cpu():
|
||||||
|
|
||||||
|
pid = os.getpid()
|
||||||
|
|
||||||
|
# burn a core @ ~ 50kHz
|
||||||
|
for _ in range(50000):
|
||||||
|
await trio.sleep(1/50000/50)
|
||||||
|
|
||||||
|
return os.getpid()
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.run_in_actor(burn_cpu)
|
||||||
|
|
||||||
|
# burn rubber in the parent too
|
||||||
|
await burn_cpu()
|
||||||
|
|
||||||
|
# wait on result from target function
|
||||||
|
pid = await portal.result()
|
||||||
|
|
||||||
|
# end of nursery block
|
||||||
|
print(f"Collected subproc {pid}")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,43 @@
|
||||||
|
"""
|
||||||
|
Run with a process monitor from a terminal using::
|
||||||
|
|
||||||
|
$TERM -e watch -n 0.1 "pstree -a $$" \
|
||||||
|
& python examples/parallelism/we_are_processes.py \
|
||||||
|
&& kill $!
|
||||||
|
|
||||||
|
"""
|
||||||
|
from multiprocessing import cpu_count
|
||||||
|
import os
|
||||||
|
|
||||||
|
import tractor
|
||||||
|
import trio
|
||||||
|
|
||||||
|
|
||||||
|
async def target():
|
||||||
|
print(
|
||||||
|
f"Yo, i'm '{tractor.current_actor().name}' "
|
||||||
|
f"running in pid {os.getpid()}"
|
||||||
|
)
|
||||||
|
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
for i in range(cpu_count()):
|
||||||
|
await n.run_in_actor(target, name=f'worker_{i}')
|
||||||
|
|
||||||
|
print('This process tree will self-destruct in 1 sec...')
|
||||||
|
await trio.sleep(1)
|
||||||
|
|
||||||
|
# you could have done this yourself
|
||||||
|
raise Exception('Self Destructed')
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
try:
|
||||||
|
trio.run(main)
|
||||||
|
except Exception:
|
||||||
|
print('Zombies Contained')
|
|
@ -0,0 +1,44 @@
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
async def sleepy_jane():
|
||||||
|
uid = tractor.current_actor().uid
|
||||||
|
print(f'Yo i am actor {uid}')
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
'''
|
||||||
|
Spawn a flat actor cluster, with one process per
|
||||||
|
detected core.
|
||||||
|
|
||||||
|
'''
|
||||||
|
portal_map: dict[str, tractor.Portal]
|
||||||
|
results: dict[str, str]
|
||||||
|
|
||||||
|
# look at this hip new syntax!
|
||||||
|
async with (
|
||||||
|
|
||||||
|
tractor.open_actor_cluster(
|
||||||
|
modules=[__name__]
|
||||||
|
) as portal_map,
|
||||||
|
|
||||||
|
trio.open_nursery() as n,
|
||||||
|
):
|
||||||
|
|
||||||
|
for (name, portal) in portal_map.items():
|
||||||
|
n.start_soon(portal.run, sleepy_jane)
|
||||||
|
|
||||||
|
await trio.sleep(0.5)
|
||||||
|
|
||||||
|
# kill the cluster with a cancel
|
||||||
|
raise KeyboardInterrupt
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
try:
|
||||||
|
trio.run(main)
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
pass
|
|
@ -1,3 +1,4 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
@ -11,7 +12,7 @@ async def main():
|
||||||
for i in range(3):
|
for i in range(3):
|
||||||
real_actors.append(await n.start_actor(
|
real_actors.append(await n.start_actor(
|
||||||
f'actor_{i}',
|
f'actor_{i}',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
))
|
))
|
||||||
|
|
||||||
# start one actor that will fail immediately
|
# start one actor that will fail immediately
|
||||||
|
@ -24,6 +25,6 @@ async def main():
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
try:
|
try:
|
||||||
# also raises
|
# also raises
|
||||||
tractor.run(main)
|
trio.run(main)
|
||||||
except tractor.RemoteActorError:
|
except tractor.RemoteActorError:
|
||||||
print("Look Maa that actor failed hard, hehhh!")
|
print("Look Maa that actor failed hard, hehhh!")
|
||||||
|
|
|
@ -0,0 +1,72 @@
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def simple_rpc(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
data: int,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''Test a small ping-pong 2-way streaming server.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# signal to parent that we're up much like
|
||||||
|
# ``trio_typing.TaskStatus.started()``
|
||||||
|
await ctx.started(data + 1)
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
async for msg in stream:
|
||||||
|
|
||||||
|
assert msg == 'ping'
|
||||||
|
await stream.send('pong')
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert count == 10
|
||||||
|
|
||||||
|
|
||||||
|
async def main() -> None:
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'rpc_server',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
# XXX: syntax requires py3.9
|
||||||
|
async with (
|
||||||
|
|
||||||
|
portal.open_context(
|
||||||
|
simple_rpc, # taken from pytest parameterization
|
||||||
|
data=10,
|
||||||
|
|
||||||
|
) as (ctx, sent),
|
||||||
|
|
||||||
|
ctx.open_stream() as stream,
|
||||||
|
):
|
||||||
|
|
||||||
|
assert sent == 11
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
# receive msgs using async for style
|
||||||
|
await stream.send('ping')
|
||||||
|
|
||||||
|
async for msg in stream:
|
||||||
|
assert msg == 'pong'
|
||||||
|
await stream.send('ping')
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
if count >= 9:
|
||||||
|
break
|
||||||
|
|
||||||
|
# explicitly teardown the daemon-actor
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
trio.run(main)
|
|
@ -1,7 +1,9 @@
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
tractor.log.get_console_log("INFO")
|
tractor.log.get_console_log("INFO")
|
||||||
|
|
||||||
|
|
||||||
async def main(service_name):
|
async def main(service_name):
|
||||||
|
|
||||||
async with tractor.open_nursery() as an:
|
async with tractor.open_nursery() as an:
|
||||||
|
@ -17,4 +19,4 @@ async def main(service_name):
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
tractor.run(main, 'some_actor_name')
|
trio.run(main, 'some_actor_name')
|
||||||
|
|
|
@ -0,0 +1 @@
|
||||||
|
!.gitignore
|
|
@ -0,0 +1,16 @@
|
||||||
|
Strictly support Python 3.10+, start runtime machinery reorg
|
||||||
|
|
||||||
|
Since we want to push forward using the new `match:` syntax for our
|
||||||
|
internal RPC-msg loops, we officially drop 3.9 support for the next
|
||||||
|
release which should coincide well with the first release of 3.11.
|
||||||
|
|
||||||
|
This patch set also officially removes the ``tractor.run()`` API (which
|
||||||
|
has been deprecated for some time) as well as starts an initial re-org
|
||||||
|
of the internal runtime core by:
|
||||||
|
- renaming ``tractor._actor`` -> ``._runtime``
|
||||||
|
- moving the ``._runtime.ActorActor._process_messages()`` and
|
||||||
|
``._async_main()`` to be module level singleton-task-functions since
|
||||||
|
they are only started once for each connection and actor spawn
|
||||||
|
respectively; this internal API thus looks more similar to (at the
|
||||||
|
time of writing) the ``trio``-internals in ``trio._core._run``.
|
||||||
|
- officially remove ``tractor.run()``, now deprecated for some time.
|
|
@ -0,0 +1,4 @@
|
||||||
|
Only set `._debug.Lock.local_pdb_complete` if has been created.
|
||||||
|
|
||||||
|
This can be triggered by a very rare race condition (and thus we have no
|
||||||
|
working test yet) but it is known to exist in (a) consumer project(s).
|
|
@ -0,0 +1,25 @@
|
||||||
|
Add support for ``trio >= 0.22`` and support for the new Python 3.11
|
||||||
|
``[Base]ExceptionGroup`` from `pep 654`_ via the backported
|
||||||
|
`exceptiongroup`_ package and some final fixes to the debug mode
|
||||||
|
subsystem.
|
||||||
|
|
||||||
|
This port ended up driving some (hopefully) final fixes to our debugger
|
||||||
|
subsystem including the solution to all lingering stdstreams locking
|
||||||
|
race-conditions and deadlock scenarios. This includes extending the
|
||||||
|
debugger tests suite as well as cancellation and ``asyncio`` mode cases.
|
||||||
|
Some of the notable details:
|
||||||
|
|
||||||
|
- always reverting to the ``trio`` SIGINT handler when leaving debug
|
||||||
|
mode.
|
||||||
|
- bypassing child attempts to acquire the debug lock when detected
|
||||||
|
to be amdist actor-runtime-cancellation.
|
||||||
|
- allowing the root actor to cancel local but IPC-stale subactor
|
||||||
|
requests-tasks for the debug lock when in a "no IPC peers" state.
|
||||||
|
|
||||||
|
Further we refined our ``ActorNursery`` semantics to be more similar to
|
||||||
|
``trio`` in the sense that parent task errors are always packed into the
|
||||||
|
actor-nursery emitted exception group and adjusted all tests and
|
||||||
|
examples accordingly.
|
||||||
|
|
||||||
|
.. _pep 654: https://peps.python.org/pep-0654/#handling-exception-groups
|
||||||
|
.. _exceptiongroup: https://github.com/python-trio/exceptiongroup
|
|
@ -0,0 +1,5 @@
|
||||||
|
Establish an explicit "backend spawning" method table; use it from CI
|
||||||
|
|
||||||
|
More clearly lays out the current set of (3) backends: ``['trio',
|
||||||
|
'mp_spawn', 'mp_forkserver']`` and adjusts the ``._spawn.py`` internals
|
||||||
|
as well as the test suite to accommodate.
|
|
@ -0,0 +1,4 @@
|
||||||
|
Add ``key: Callable[..., Hashable]`` support to ``.trionics.maybe_open_context()``
|
||||||
|
|
||||||
|
Gives users finer grained control over cache hit behaviour using
|
||||||
|
a callable which receives the input ``kwargs: dict``.
|
|
@ -0,0 +1,41 @@
|
||||||
|
Add support for debug-lock blocking using a ``._debug.Lock._blocked:
|
||||||
|
set[tuple]`` and add ids when no-more IPC connections with the
|
||||||
|
root actor are detected.
|
||||||
|
|
||||||
|
This is an enhancement which (mostly) solves a lingering debugger
|
||||||
|
locking race case we needed to handle:
|
||||||
|
|
||||||
|
- child crashes acquires TTY lock in root and attaches to ``pdb``
|
||||||
|
- child IPC goes down such that all channels to the root are broken
|
||||||
|
/ non-functional.
|
||||||
|
- root is stuck thinking the child is still in debug even though it
|
||||||
|
can't be contacted and the child actor machinery hasn't been
|
||||||
|
cancelled by its parent.
|
||||||
|
- root get's stuck in deadlock with child since it won't send a cancel
|
||||||
|
request until the child is finished debugging (to avoid clobbering
|
||||||
|
a child that is actually using the debugger), but the child can't
|
||||||
|
unlock the debugger bc IPC is down and it can't contact the root.
|
||||||
|
|
||||||
|
To avoid this scenario add debug lock blocking list via
|
||||||
|
`._debug.Lock._blocked: set[tuple]` which holds actor uids for any actor
|
||||||
|
that is detected by the root as having no transport channel connections
|
||||||
|
(of which at least one should exist if this sub-actor at some point
|
||||||
|
acquired the debug lock). The root consequently checks this list for any
|
||||||
|
actor that tries to (re)acquire the lock and blocks with
|
||||||
|
a ``ContextCancelled``. Further, when a debug condition is tested in
|
||||||
|
``._runtime._invoke``, the context's ``._enter_debugger_on_cancel`` is
|
||||||
|
set to `False` if the actor was put on the block list then all
|
||||||
|
post-mortem / crash handling will be bypassed for that task.
|
||||||
|
|
||||||
|
In theory this approach to block list management may cause problems
|
||||||
|
where some nested child actor acquires and releases the lock multiple
|
||||||
|
times and it gets stuck on the block list after the first use? If this
|
||||||
|
turns out to be an issue we can try changing the strat so blocks are
|
||||||
|
only added when the root has zero IPC peers left?
|
||||||
|
|
||||||
|
Further, this adds a root-locking-task side cancel scope,
|
||||||
|
``Lock._root_local_task_cs_in_debug``, which can be ``.cancel()``-ed by the root
|
||||||
|
runtime when a stale lock is detected during the IPC channel testing.
|
||||||
|
However, right now we're NOT using this since it seems to cause test
|
||||||
|
failures likely due to causing pre-mature cancellation and maybe needs
|
||||||
|
a bit more experimenting?
|
|
@ -0,0 +1,19 @@
|
||||||
|
Rework our ``.trionics.BroadcastReceiver`` internals to avoid method
|
||||||
|
recursion and approach a design and interface closer to ``trio``'s
|
||||||
|
``MemoryReceiveChannel``.
|
||||||
|
|
||||||
|
The details of the internal changes include:
|
||||||
|
|
||||||
|
- implementing a ``BroadcastReceiver.receive_nowait()`` and using it
|
||||||
|
within the async ``.receive()`` thus avoiding recursion from
|
||||||
|
``.receive()``.
|
||||||
|
- failing over to an internal ``._receive_from_underlying()`` when the
|
||||||
|
``_nowait()`` call raises ``trio.WouldBlock``
|
||||||
|
- adding ``BroadcastState.statistics()`` for debugging and testing both
|
||||||
|
internals and by users.
|
||||||
|
- add an internal ``BroadcastReceiver._raise_on_lag: bool`` which can be
|
||||||
|
set to avoid ``Lagged`` raising for possible use cases where a user
|
||||||
|
wants to choose between a [cheap or nasty
|
||||||
|
pattern](https://zguide.zeromq.org/docs/chapter7/#The-Cheap-or-Nasty-Pattern)
|
||||||
|
the the particular stream (we use this in ``piker``'s dark clearing
|
||||||
|
engine to avoid fast feeds breaking during HFT periods).
|
|
@ -0,0 +1,11 @@
|
||||||
|
Always ``list``-cast the ``mngrs`` input to
|
||||||
|
``.trionics.gather_contexts()`` and ensure its size otherwise raise
|
||||||
|
a ``ValueError``.
|
||||||
|
|
||||||
|
Turns out that trying to pass an inline-style generator comprehension
|
||||||
|
doesn't seem to work inside the ``async with`` expression? Further, in
|
||||||
|
such a case we can get a hang waiting on the all-entered event
|
||||||
|
completion when the internal mngrs iteration is a noop. Instead we
|
||||||
|
always greedily check a size and error on empty input; the lazy
|
||||||
|
iteration of a generator input is not beneficial anyway since we're
|
||||||
|
entering all manager instances in concurrent tasks.
|
|
@ -0,0 +1,15 @@
|
||||||
|
Fixes to ensure IPC (channel) breakage doesn't result in hung actor
|
||||||
|
trees; the zombie reaping and general supervision machinery will always
|
||||||
|
clean up and terminate.
|
||||||
|
|
||||||
|
This includes not only the (mostly minor) fixes to solve these cases but
|
||||||
|
also a new extensive test suite in `test_advanced_faults.py` with an
|
||||||
|
accompanying highly configurable example module-script in
|
||||||
|
`examples/advanced_faults/ipc_failure_during_stream.py`. Tests ensure we
|
||||||
|
never get hang or zombies despite operating in debug mode and attempt to
|
||||||
|
simulate all possible IPC transport failure cases for a local-host actor
|
||||||
|
tree.
|
||||||
|
|
||||||
|
Further we simplify `Context.open_stream.__aexit__()` to just call
|
||||||
|
`MsgStream.aclose()` directly more or less avoiding a pure duplicate
|
||||||
|
code path.
|
|
@ -0,0 +1,10 @@
|
||||||
|
Always redraw the `pdbpp` prompt on `SIGINT` during REPL use.
|
||||||
|
|
||||||
|
There was recent changes todo with Python 3.10 that required us to pin
|
||||||
|
to a specific commit in `pdbpp` which have recently been fixed minus
|
||||||
|
this last issue with `SIGINT` shielding: not clobbering or not
|
||||||
|
showing the `(Pdb++)` prompt on ctlr-c by the user. This repairs all
|
||||||
|
that by firstly removing the standard KBI intercepting of the std lib's
|
||||||
|
`pdb.Pdb._cmdloop()` as well as ensuring that only the actor with REPL
|
||||||
|
control ever reports `SIGINT` handler log msgs and prompt redraws. With
|
||||||
|
this we move back to using pypi `pdbpp` release.
|
|
@ -0,0 +1,7 @@
|
||||||
|
Drop `trio.Process.aclose()` usage, copy into our spawning code.
|
||||||
|
|
||||||
|
The details are laid out in https://github.com/goodboy/tractor/issues/330.
|
||||||
|
`trio` changed is process running quite some time ago, this just copies
|
||||||
|
out the small bit we needed (from the old `.aclose()`) for hard kills
|
||||||
|
where a soft runtime cancel request fails and our "zombie killer"
|
||||||
|
implementation kicks in.
|
|
@ -0,0 +1,15 @@
|
||||||
|
Switch to using the fork & fix of `pdb++`, `pdbp`:
|
||||||
|
https://github.com/mdmintz/pdbp
|
||||||
|
|
||||||
|
Allows us to sidestep a variety of issues that aren't being maintained
|
||||||
|
in the upstream project thanks to the hard work of @mdmintz!
|
||||||
|
|
||||||
|
We also include some default settings adjustments as per recent
|
||||||
|
development on the fork:
|
||||||
|
|
||||||
|
- sticky mode is still turned on by default but now activates when
|
||||||
|
a using the `ll` repl command.
|
||||||
|
- turn off line truncation by default to avoid inter-line gaps when
|
||||||
|
resizing the terimnal during use.
|
||||||
|
- when using the backtrace cmd either by `w` or `bt`, the config
|
||||||
|
automatically switches to non-sticky mode.
|
|
@ -0,0 +1,8 @@
|
||||||
|
See both the `towncrier docs`_ and the `pluggy release readme`_ for hot
|
||||||
|
tips. We basically have the most minimal setup and release process right
|
||||||
|
now and use the default `fragment set`_.
|
||||||
|
|
||||||
|
|
||||||
|
.. _towncrier docs: https://github.com/twisted/towncrier#quick-start
|
||||||
|
.. _pluggy release readme: https://github.com/pytest-dev/pluggy/blob/main/changelog/README.rst
|
||||||
|
.. _fragment set: https://github.com/twisted/towncrier#news-fragments
|
|
@ -0,0 +1,37 @@
|
||||||
|
{% for section in sections %}
|
||||||
|
{% set underline = "-" %}
|
||||||
|
{% if section %}
|
||||||
|
{{section}}
|
||||||
|
{{ underline * section|length }}{% set underline = "~" %}
|
||||||
|
|
||||||
|
{% endif %}
|
||||||
|
{% if sections[section] %}
|
||||||
|
{% for category, val in definitions.items() if category in sections[section] %}
|
||||||
|
|
||||||
|
{{ definitions[category]['name'] }}
|
||||||
|
{{ underline * definitions[category]['name']|length }}
|
||||||
|
|
||||||
|
{% if definitions[category]['showcontent'] %}
|
||||||
|
{% for text, values in sections[section][category]|dictsort(by='value') %}
|
||||||
|
{% set issue_joiner = joiner(', ') %}
|
||||||
|
- {% for value in values|sort %}{{ issue_joiner() }}`{{ value }} <https://github.com/goodboy/tractor/issues/{{ value[1:] }}>`_{% endfor %}: {{ text }}
|
||||||
|
|
||||||
|
{% endfor %}
|
||||||
|
{% else %}
|
||||||
|
- {{ sections[section][category]['']|sort|join(', ') }}
|
||||||
|
|
||||||
|
|
||||||
|
{% endif %}
|
||||||
|
{% if sections[section][category]|length == 0 %}
|
||||||
|
|
||||||
|
No significant changes.
|
||||||
|
|
||||||
|
{% else %}
|
||||||
|
{% endif %}
|
||||||
|
{% endfor %}
|
||||||
|
{% else %}
|
||||||
|
|
||||||
|
No significant changes.
|
||||||
|
|
||||||
|
{% endif %}
|
||||||
|
{% endfor %}
|
|
@ -0,0 +1,28 @@
|
||||||
|
[tool.towncrier]
|
||||||
|
package = "tractor"
|
||||||
|
filename = "NEWS.rst"
|
||||||
|
directory = "nooz/"
|
||||||
|
version = "0.1.0a6"
|
||||||
|
title_format = "tractor {version} ({project_date})"
|
||||||
|
template = "nooz/_template.rst"
|
||||||
|
all_bullets = true
|
||||||
|
|
||||||
|
[[tool.towncrier.type]]
|
||||||
|
directory = "feature"
|
||||||
|
name = "Features"
|
||||||
|
showcontent = true
|
||||||
|
|
||||||
|
[[tool.towncrier.type]]
|
||||||
|
directory = "bugfix"
|
||||||
|
name = "Bug Fixes"
|
||||||
|
showcontent = true
|
||||||
|
|
||||||
|
[[tool.towncrier.type]]
|
||||||
|
directory = "doc"
|
||||||
|
name = "Improved Documentation"
|
||||||
|
showcontent = true
|
||||||
|
|
||||||
|
[[tool.towncrier.type]]
|
||||||
|
directory = "trivial"
|
||||||
|
name = "Trivial/Internal Changes"
|
||||||
|
showcontent = true
|
|
@ -1,2 +1,2 @@
|
||||||
sphinx
|
sphinx
|
||||||
sphinx_typlog_theme
|
sphinx_book_theme
|
||||||
|
|
|
@ -1,6 +1,8 @@
|
||||||
pytest
|
pytest
|
||||||
pytest-trio
|
pytest-trio
|
||||||
pdbpp
|
pytest-timeout
|
||||||
|
pdbp
|
||||||
mypy
|
mypy
|
||||||
trio_typing
|
trio_typing
|
||||||
pexpect
|
pexpect
|
||||||
|
towncrier
|
||||||
|
|
81
setup.py
81
setup.py
|
@ -1,62 +1,97 @@
|
||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
#
|
#
|
||||||
# tractor: a trionic actor model built on `multiprocessing` and `trio`
|
# tractor: structured concurrent "actors".
|
||||||
#
|
#
|
||||||
# Copyright (C) 2018-2020 Tyler Goodlet
|
# Copyright 2018-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 General Public License as published by
|
# 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
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
# (at your option) any later version.
|
# (at your option) any later version.
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful,
|
# This program is distributed in the hope that it will be useful,
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
# GNU General Public License for more details.
|
# 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/>.
|
||||||
|
|
||||||
# You should have received a copy of the GNU General Public License
|
|
||||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
from setuptools import setup
|
from setuptools import setup
|
||||||
|
|
||||||
with open('README.rst', encoding='utf-8') as f:
|
with open('docs/README.rst', encoding='utf-8') as f:
|
||||||
readme = f.read()
|
readme = f.read()
|
||||||
|
|
||||||
|
|
||||||
setup(
|
setup(
|
||||||
name="tractor",
|
name="tractor",
|
||||||
version='0.1.0.alpha0',
|
version='0.1.0a6dev0', # alpha zone
|
||||||
description='A trionic actor model built on `multiprocessing` and `trio`',
|
description='structured concurrrent `trio`-"actors"',
|
||||||
long_description=readme,
|
long_description=readme,
|
||||||
license='GPLv3',
|
license='AGPLv3',
|
||||||
author='Tyler Goodlet',
|
author='Tyler Goodlet',
|
||||||
maintainer='Tyler Goodlet',
|
maintainer='Tyler Goodlet',
|
||||||
maintainer_email='jgbt@protonmail.com',
|
maintainer_email='goodboy_foss@protonmail.com',
|
||||||
url='https://github.com/goodboy/tractor',
|
url='https://github.com/goodboy/tractor',
|
||||||
platforms=['linux', 'windows'],
|
platforms=['linux', 'windows'],
|
||||||
packages=[
|
packages=[
|
||||||
'tractor',
|
'tractor',
|
||||||
'tractor.testing',
|
'tractor.experimental',
|
||||||
|
'tractor.trionics',
|
||||||
],
|
],
|
||||||
install_requires=[
|
install_requires=[
|
||||||
'msgpack', 'trio>0.8', 'async_generator', 'colorlog', 'wrapt',
|
|
||||||
'trio_typing', 'pdbpp',
|
# trio related
|
||||||
|
# proper range spec:
|
||||||
|
# https://packaging.python.org/en/latest/discussions/install-requires-vs-requirements/#id5
|
||||||
|
'trio >= 0.22',
|
||||||
|
'async_generator',
|
||||||
|
'trio_typing',
|
||||||
|
'exceptiongroup',
|
||||||
|
|
||||||
|
# tooling
|
||||||
|
'tricycle',
|
||||||
|
'trio_typing',
|
||||||
|
'colorlog',
|
||||||
|
'wrapt',
|
||||||
|
|
||||||
|
# IPC serialization
|
||||||
|
'msgspec',
|
||||||
|
|
||||||
|
# debug mode REPL
|
||||||
|
'pdbp',
|
||||||
|
|
||||||
|
# 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
|
||||||
|
|
||||||
|
# windows deps workaround for ``pdbpp``
|
||||||
|
# https://github.com/pdbpp/pdbpp/issues/498
|
||||||
|
# https://github.com/pdbpp/fancycompleter/issues/37
|
||||||
|
'pyreadline3 ; platform_system == "Windows"',
|
||||||
|
|
||||||
],
|
],
|
||||||
tests_require=['pytest'],
|
tests_require=['pytest'],
|
||||||
python_requires=">=3.7",
|
python_requires=">=3.10",
|
||||||
keywords=[
|
keywords=[
|
||||||
"async", "concurrency", "actor model", "distributed",
|
'trio',
|
||||||
'trio', 'multiprocessing'
|
'async',
|
||||||
|
'concurrency',
|
||||||
|
'structured concurrency',
|
||||||
|
'actor model',
|
||||||
|
'distributed',
|
||||||
|
'multiprocessing'
|
||||||
],
|
],
|
||||||
classifiers=[
|
classifiers=[
|
||||||
'Development Status :: 3 - Alpha',
|
"Development Status :: 3 - Alpha",
|
||||||
'License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)'
|
"Operating System :: POSIX :: Linux",
|
||||||
'Operating System :: POSIX :: Linux',
|
"Operating System :: Microsoft :: Windows",
|
||||||
"Framework :: Trio",
|
"Framework :: Trio",
|
||||||
|
"License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)",
|
||||||
"Programming Language :: Python :: Implementation :: CPython",
|
"Programming Language :: Python :: Implementation :: CPython",
|
||||||
"Programming Language :: Python :: Implementation :: PyPy",
|
|
||||||
"Programming Language :: Python :: 3 :: Only",
|
"Programming Language :: Python :: 3 :: Only",
|
||||||
"Programming Language :: Python :: 3.7",
|
"Programming Language :: Python :: 3.10",
|
||||||
"Programming Language :: Python :: 3.8",
|
|
||||||
"Intended Audience :: Science/Research",
|
"Intended Audience :: Science/Research",
|
||||||
"Intended Audience :: Developers",
|
"Intended Audience :: Developers",
|
||||||
"Topic :: System :: Distributed Computing",
|
"Topic :: System :: Distributed Computing",
|
||||||
|
|
|
@ -7,16 +7,91 @@ import os
|
||||||
import random
|
import random
|
||||||
import signal
|
import signal
|
||||||
import platform
|
import platform
|
||||||
|
import pathlib
|
||||||
import time
|
import time
|
||||||
|
import inspect
|
||||||
|
from functools import partial, wraps
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
# export for tests
|
|
||||||
from tractor.testing import tractor_test # noqa
|
|
||||||
|
|
||||||
|
|
||||||
pytest_plugins = ['pytester']
|
pytest_plugins = ['pytester']
|
||||||
|
|
||||||
|
|
||||||
|
def tractor_test(fn):
|
||||||
|
"""
|
||||||
|
Use:
|
||||||
|
|
||||||
|
@tractor_test
|
||||||
|
async def test_whatever():
|
||||||
|
await ...
|
||||||
|
|
||||||
|
If fixtures:
|
||||||
|
|
||||||
|
- ``arb_addr`` (a socket addr tuple where arbiter is listening)
|
||||||
|
- ``loglevel`` (logging level passed to tractor internals)
|
||||||
|
- ``start_method`` (subprocess spawning backend)
|
||||||
|
|
||||||
|
are defined in the `pytest` fixture space they will be automatically
|
||||||
|
injected to tests declaring these funcargs.
|
||||||
|
"""
|
||||||
|
@wraps(fn)
|
||||||
|
def wrapper(
|
||||||
|
*args,
|
||||||
|
loglevel=None,
|
||||||
|
arb_addr=None,
|
||||||
|
start_method=None,
|
||||||
|
**kwargs
|
||||||
|
):
|
||||||
|
# __tracebackhide__ = True
|
||||||
|
|
||||||
|
if 'arb_addr' in inspect.signature(fn).parameters:
|
||||||
|
# injects test suite fixture value to test as well
|
||||||
|
# as `run()`
|
||||||
|
kwargs['arb_addr'] = arb_addr
|
||||||
|
|
||||||
|
if 'loglevel' in inspect.signature(fn).parameters:
|
||||||
|
# allows test suites to define a 'loglevel' fixture
|
||||||
|
# that activates the internal logging
|
||||||
|
kwargs['loglevel'] = loglevel
|
||||||
|
|
||||||
|
if start_method is None:
|
||||||
|
if platform.system() == "Windows":
|
||||||
|
start_method = 'trio'
|
||||||
|
|
||||||
|
if 'start_method' in inspect.signature(fn).parameters:
|
||||||
|
# set of subprocess spawning backends
|
||||||
|
kwargs['start_method'] = start_method
|
||||||
|
|
||||||
|
if kwargs:
|
||||||
|
|
||||||
|
# use explicit root actor start
|
||||||
|
|
||||||
|
async def _main():
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
# **kwargs,
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
loglevel=loglevel,
|
||||||
|
start_method=start_method,
|
||||||
|
|
||||||
|
# TODO: only enable when pytest is passed --pdb
|
||||||
|
# debug_mode=True,
|
||||||
|
|
||||||
|
):
|
||||||
|
await fn(*args, **kwargs)
|
||||||
|
|
||||||
|
main = _main
|
||||||
|
|
||||||
|
else:
|
||||||
|
# use implicit root actor start
|
||||||
|
main = partial(fn, *args, **kwargs)
|
||||||
|
|
||||||
|
return trio.run(main)
|
||||||
|
|
||||||
|
return wrapper
|
||||||
|
|
||||||
|
|
||||||
_arb_addr = '127.0.0.1', random.randint(1000, 9999)
|
_arb_addr = '127.0.0.1', random.randint(1000, 9999)
|
||||||
|
|
||||||
|
|
||||||
|
@ -39,20 +114,27 @@ no_windows = pytest.mark.skipif(
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def repodir():
|
def repodir() -> pathlib.Path:
|
||||||
"""Return the abspath to the repo directory.
|
'''
|
||||||
"""
|
Return the abspath to the repo directory.
|
||||||
dirname = os.path.dirname
|
|
||||||
dirpath = os.path.abspath(
|
'''
|
||||||
dirname(dirname(os.path.realpath(__file__)))
|
# 2 parents up to step up through tests/<repo_dir>
|
||||||
)
|
return pathlib.Path(__file__).parent.parent.absolute()
|
||||||
return dirpath
|
|
||||||
|
|
||||||
|
def examples_dir() -> pathlib.Path:
|
||||||
|
'''
|
||||||
|
Return the abspath to the examples directory as `pathlib.Path`.
|
||||||
|
|
||||||
|
'''
|
||||||
|
return repodir() / 'examples'
|
||||||
|
|
||||||
|
|
||||||
def pytest_addoption(parser):
|
def pytest_addoption(parser):
|
||||||
parser.addoption(
|
parser.addoption(
|
||||||
"--ll", action="store", dest='loglevel',
|
"--ll", action="store", dest='loglevel',
|
||||||
default=None, help="logging level to set when testing"
|
default='ERROR', help="logging level to set when testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
parser.addoption(
|
parser.addoption(
|
||||||
|
@ -64,10 +146,6 @@ def pytest_addoption(parser):
|
||||||
|
|
||||||
def pytest_configure(config):
|
def pytest_configure(config):
|
||||||
backend = config.option.spawn_backend
|
backend = config.option.spawn_backend
|
||||||
|
|
||||||
if backend == 'mp':
|
|
||||||
tractor._spawn.try_set_start_method('spawn')
|
|
||||||
elif backend == 'trio':
|
|
||||||
tractor._spawn.try_set_start_method(backend)
|
tractor._spawn.try_set_start_method(backend)
|
||||||
|
|
||||||
|
|
||||||
|
@ -75,20 +153,24 @@ def pytest_configure(config):
|
||||||
def loglevel(request):
|
def loglevel(request):
|
||||||
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)
|
||||||
yield level
|
yield level
|
||||||
tractor.log._default_loglevel = orig
|
tractor.log._default_loglevel = orig
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='session')
|
@pytest.fixture(scope='session')
|
||||||
def spawn_backend(request):
|
def spawn_backend(request) -> str:
|
||||||
return request.config.option.spawn_backend
|
return request.config.option.spawn_backend
|
||||||
|
|
||||||
|
|
||||||
|
_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 envoirment.
|
"""Detect CI envoirment.
|
||||||
"""
|
"""
|
||||||
return os.environ.get('TRAVIS', False) or os.environ.get('CI', False)
|
return _ci_env
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='session')
|
@pytest.fixture(scope='session')
|
||||||
|
@ -98,24 +180,24 @@ def arb_addr():
|
||||||
|
|
||||||
def pytest_generate_tests(metafunc):
|
def pytest_generate_tests(metafunc):
|
||||||
spawn_backend = metafunc.config.option.spawn_backend
|
spawn_backend = metafunc.config.option.spawn_backend
|
||||||
|
|
||||||
if not spawn_backend:
|
if not spawn_backend:
|
||||||
# XXX some weird windows bug with `pytest`?
|
# XXX some weird windows bug with `pytest`?
|
||||||
spawn_backend = 'mp'
|
spawn_backend = 'trio'
|
||||||
assert spawn_backend in ('mp', '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:
|
if 'start_method' in metafunc.fixturenames:
|
||||||
if spawn_backend == 'mp':
|
metafunc.parametrize("start_method", [spawn_backend], scope='module')
|
||||||
from multiprocessing import get_all_start_methods
|
|
||||||
methods = get_all_start_methods()
|
|
||||||
if 'fork' in methods:
|
|
||||||
# fork not available on windows, so check before
|
|
||||||
# removing XXX: the fork method is in general
|
|
||||||
# incompatible with trio's global scheduler state
|
|
||||||
methods.remove('fork')
|
|
||||||
elif spawn_backend == 'trio':
|
|
||||||
methods = ['trio']
|
|
||||||
|
|
||||||
metafunc.parametrize("start_method", methods, scope='module')
|
|
||||||
|
|
||||||
|
|
||||||
def sig_prog(proc, sig):
|
def sig_prog(proc, sig):
|
||||||
|
@ -131,16 +213,22 @@ def sig_prog(proc, sig):
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def daemon(loglevel, testdir, arb_addr):
|
def daemon(
|
||||||
"""Run a daemon actor as a "remote arbiter".
|
loglevel: str,
|
||||||
"""
|
testdir,
|
||||||
|
arb_addr: tuple[str, int],
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Run a daemon actor as a "remote arbiter".
|
||||||
|
|
||||||
|
'''
|
||||||
if loglevel in ('trace', 'debug'):
|
if loglevel in ('trace', 'debug'):
|
||||||
# too much logging will lock up the subproc (smh)
|
# too much logging will lock up the subproc (smh)
|
||||||
loglevel = 'info'
|
loglevel = 'info'
|
||||||
|
|
||||||
cmdargs = [
|
cmdargs = [
|
||||||
sys.executable, '-c',
|
sys.executable, '-c',
|
||||||
"import tractor; tractor.run_daemon([], arbiter_addr={}, loglevel={})"
|
"import tractor; tractor.run_daemon([], registry_addr={}, loglevel={})"
|
||||||
.format(
|
.format(
|
||||||
arb_addr,
|
arb_addr,
|
||||||
"'{}'".format(loglevel) if loglevel else None)
|
"'{}'".format(loglevel) if loglevel else None)
|
||||||
|
|
|
@ -0,0 +1,129 @@
|
||||||
|
"""
|
||||||
|
Bidirectional streaming.
|
||||||
|
|
||||||
|
"""
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def simple_rpc(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
data: int,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Test a small ping-pong server.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# signal to parent that we're up
|
||||||
|
await ctx.started(data + 1)
|
||||||
|
|
||||||
|
print('opening stream in callee')
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
await stream.receive() == 'ping'
|
||||||
|
except trio.EndOfChannel:
|
||||||
|
assert count == 10
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
print('pong')
|
||||||
|
await stream.send('pong')
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def simple_rpc_with_forloop(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
data: int,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
"""Same as previous test but using ``async for`` syntax/api.
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
# signal to parent that we're up
|
||||||
|
await ctx.started(data + 1)
|
||||||
|
|
||||||
|
print('opening stream in callee')
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
async for msg in stream:
|
||||||
|
|
||||||
|
assert msg == 'ping'
|
||||||
|
print('pong')
|
||||||
|
await stream.send('pong')
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert count == 10
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'use_async_for',
|
||||||
|
[True, False],
|
||||||
|
)
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'server_func',
|
||||||
|
[simple_rpc, simple_rpc_with_forloop],
|
||||||
|
)
|
||||||
|
def test_simple_rpc(server_func, use_async_for):
|
||||||
|
'''
|
||||||
|
The simplest request response pattern.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'rpc_server',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
server_func, # taken from pytest parameterization
|
||||||
|
data=10,
|
||||||
|
) as (ctx, sent):
|
||||||
|
|
||||||
|
assert sent == 11
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
if use_async_for:
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
# receive msgs using async for style
|
||||||
|
print('ping')
|
||||||
|
await stream.send('ping')
|
||||||
|
|
||||||
|
async for msg in stream:
|
||||||
|
assert msg == 'pong'
|
||||||
|
print('ping')
|
||||||
|
await stream.send('ping')
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
if count >= 9:
|
||||||
|
break
|
||||||
|
|
||||||
|
else:
|
||||||
|
# classic send/receive style
|
||||||
|
for _ in range(10):
|
||||||
|
|
||||||
|
print('ping')
|
||||||
|
await stream.send('ping')
|
||||||
|
assert await stream.receive() == 'pong'
|
||||||
|
|
||||||
|
# stream should terminate here
|
||||||
|
|
||||||
|
# final context result(s) should be consumed here in __aexit__()
|
||||||
|
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,193 @@
|
||||||
|
'''
|
||||||
|
Sketchy network blackoutz, ugly byzantine gens, puedes eschuchar la
|
||||||
|
cancelacion?..
|
||||||
|
|
||||||
|
'''
|
||||||
|
from functools import partial
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
from _pytest.pathlib import import_path
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
from conftest import (
|
||||||
|
examples_dir,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'debug_mode',
|
||||||
|
[False, True],
|
||||||
|
ids=['no_debug_mode', 'debug_mode'],
|
||||||
|
)
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'ipc_break',
|
||||||
|
[
|
||||||
|
# no breaks
|
||||||
|
{
|
||||||
|
'break_parent_ipc_after': False,
|
||||||
|
'break_child_ipc_after': False,
|
||||||
|
},
|
||||||
|
|
||||||
|
# only parent breaks
|
||||||
|
{
|
||||||
|
'break_parent_ipc_after': 500,
|
||||||
|
'break_child_ipc_after': False,
|
||||||
|
},
|
||||||
|
|
||||||
|
# only child breaks
|
||||||
|
{
|
||||||
|
'break_parent_ipc_after': False,
|
||||||
|
'break_child_ipc_after': 500,
|
||||||
|
},
|
||||||
|
|
||||||
|
# both: break parent first
|
||||||
|
{
|
||||||
|
'break_parent_ipc_after': 500,
|
||||||
|
'break_child_ipc_after': 800,
|
||||||
|
},
|
||||||
|
# both: break child first
|
||||||
|
{
|
||||||
|
'break_parent_ipc_after': 800,
|
||||||
|
'break_child_ipc_after': 500,
|
||||||
|
},
|
||||||
|
|
||||||
|
],
|
||||||
|
ids=[
|
||||||
|
'no_break',
|
||||||
|
'break_parent',
|
||||||
|
'break_child',
|
||||||
|
'break_both_parent_first',
|
||||||
|
'break_both_child_first',
|
||||||
|
],
|
||||||
|
)
|
||||||
|
def test_ipc_channel_break_during_stream(
|
||||||
|
debug_mode: bool,
|
||||||
|
spawn_backend: str,
|
||||||
|
ipc_break: dict | None,
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Ensure we can have an IPC channel break its connection during
|
||||||
|
streaming and it's still possible for the (simulated) user to kill
|
||||||
|
the actor tree using SIGINT.
|
||||||
|
|
||||||
|
We also verify the type of connection error expected in the parent
|
||||||
|
depending on which side if the IPC breaks first.
|
||||||
|
|
||||||
|
'''
|
||||||
|
if spawn_backend != 'trio':
|
||||||
|
if debug_mode:
|
||||||
|
pytest.skip('`debug_mode` only supported on `trio` spawner')
|
||||||
|
|
||||||
|
# non-`trio` spawners should never hit the hang condition that
|
||||||
|
# requires the user to do ctl-c to cancel the actor tree.
|
||||||
|
expect_final_exc = trio.ClosedResourceError
|
||||||
|
|
||||||
|
mod = import_path(
|
||||||
|
examples_dir() / 'advanced_faults' / 'ipc_failure_during_stream.py',
|
||||||
|
root=examples_dir(),
|
||||||
|
)
|
||||||
|
|
||||||
|
expect_final_exc = KeyboardInterrupt
|
||||||
|
|
||||||
|
# when ONLY the child breaks we expect the parent to get a closed
|
||||||
|
# resource error on the next `MsgStream.receive()` and then fail out
|
||||||
|
# and cancel the child from there.
|
||||||
|
if (
|
||||||
|
|
||||||
|
# only child breaks
|
||||||
|
(
|
||||||
|
ipc_break['break_child_ipc_after']
|
||||||
|
and ipc_break['break_parent_ipc_after'] is False
|
||||||
|
)
|
||||||
|
|
||||||
|
# both break but, parent breaks first
|
||||||
|
or (
|
||||||
|
ipc_break['break_child_ipc_after'] is not False
|
||||||
|
and (
|
||||||
|
ipc_break['break_parent_ipc_after']
|
||||||
|
> ipc_break['break_child_ipc_after']
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
):
|
||||||
|
expect_final_exc = trio.ClosedResourceError
|
||||||
|
|
||||||
|
# when the parent IPC side dies (even if the child's does as well
|
||||||
|
# but the child fails BEFORE the parent) we expect the channel to be
|
||||||
|
# sent a stop msg from the child at some point which will signal the
|
||||||
|
# parent that the stream has been terminated.
|
||||||
|
# NOTE: when the parent breaks "after" the child you get this same
|
||||||
|
# case as well, the child breaks the IPC channel with a stop msg
|
||||||
|
# before any closure takes place.
|
||||||
|
elif (
|
||||||
|
# only parent breaks
|
||||||
|
(
|
||||||
|
ipc_break['break_parent_ipc_after']
|
||||||
|
and ipc_break['break_child_ipc_after'] is False
|
||||||
|
)
|
||||||
|
|
||||||
|
# both break but, child breaks first
|
||||||
|
or (
|
||||||
|
ipc_break['break_parent_ipc_after'] is not False
|
||||||
|
and (
|
||||||
|
ipc_break['break_child_ipc_after']
|
||||||
|
> ipc_break['break_parent_ipc_after']
|
||||||
|
)
|
||||||
|
)
|
||||||
|
):
|
||||||
|
expect_final_exc = trio.EndOfChannel
|
||||||
|
|
||||||
|
with pytest.raises(expect_final_exc):
|
||||||
|
trio.run(
|
||||||
|
partial(
|
||||||
|
mod.main,
|
||||||
|
debug_mode=debug_mode,
|
||||||
|
start_method=spawn_backend,
|
||||||
|
**ipc_break,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def break_ipc_after_started(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
) -> None:
|
||||||
|
await ctx.started()
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
await stream.aclose()
|
||||||
|
await trio.sleep(0.2)
|
||||||
|
await ctx.chan.send(None)
|
||||||
|
print('child broke IPC and terminating')
|
||||||
|
|
||||||
|
|
||||||
|
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 a stream
|
||||||
|
the parent can trigger a SIGINT and the child will be reaped out-of-IPC by
|
||||||
|
the localhost process supervision machinery: aka "zombie lord".
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'ipc_breaker',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
with trio.move_on_after(1):
|
||||||
|
async with (
|
||||||
|
portal.open_context(
|
||||||
|
break_ipc_after_started
|
||||||
|
) as (ctx, sent),
|
||||||
|
):
|
||||||
|
async with ctx.open_stream():
|
||||||
|
await trio.sleep(0.5)
|
||||||
|
|
||||||
|
print('parent waiting on context')
|
||||||
|
|
||||||
|
print('parent exited context')
|
||||||
|
raise KeyboardInterrupt
|
||||||
|
|
||||||
|
with pytest.raises(KeyboardInterrupt):
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,380 @@
|
||||||
|
'''
|
||||||
|
Advanced streaming patterns using bidirectional streams and contexts.
|
||||||
|
|
||||||
|
'''
|
||||||
|
from collections import Counter
|
||||||
|
import itertools
|
||||||
|
import platform
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
def is_win():
|
||||||
|
return platform.system() == 'Windows'
|
||||||
|
|
||||||
|
|
||||||
|
_registry: dict[str, set[tractor.MsgStream]] = {
|
||||||
|
'even': set(),
|
||||||
|
'odd': set(),
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
async def publisher(
|
||||||
|
|
||||||
|
seed: int = 0,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
global _registry
|
||||||
|
|
||||||
|
def is_even(i):
|
||||||
|
return i % 2 == 0
|
||||||
|
|
||||||
|
for val in itertools.count(seed):
|
||||||
|
|
||||||
|
sub = 'even' if is_even(val) else 'odd'
|
||||||
|
|
||||||
|
for sub_stream in _registry[sub].copy():
|
||||||
|
await sub_stream.send(val)
|
||||||
|
|
||||||
|
# throttle send rate to ~1kHz
|
||||||
|
# making it readable to a human user
|
||||||
|
await trio.sleep(1/1000)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def subscribe(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
global _registry
|
||||||
|
|
||||||
|
# syn caller
|
||||||
|
await ctx.started(None)
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
# update subs list as consumer requests
|
||||||
|
async for new_subs in stream:
|
||||||
|
|
||||||
|
new_subs = set(new_subs)
|
||||||
|
remove = new_subs - _registry.keys()
|
||||||
|
|
||||||
|
print(f'setting sub to {new_subs} for {ctx.chan.uid}')
|
||||||
|
|
||||||
|
# remove old subs
|
||||||
|
for sub in remove:
|
||||||
|
_registry[sub].remove(stream)
|
||||||
|
|
||||||
|
# add new subs for consumer
|
||||||
|
for sub in new_subs:
|
||||||
|
_registry[sub].add(stream)
|
||||||
|
|
||||||
|
|
||||||
|
async def consumer(
|
||||||
|
|
||||||
|
subs: list[str],
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
uid = tractor.current_actor().uid
|
||||||
|
|
||||||
|
async with tractor.wait_for_actor('publisher') as portal:
|
||||||
|
async with portal.open_context(subscribe) as (ctx, first):
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
# flip between the provided subs dynamically
|
||||||
|
if len(subs) > 1:
|
||||||
|
|
||||||
|
for sub in itertools.cycle(subs):
|
||||||
|
print(f'setting dynamic sub to {sub}')
|
||||||
|
await stream.send([sub])
|
||||||
|
|
||||||
|
count = 0
|
||||||
|
async for value in stream:
|
||||||
|
print(f'{uid} got: {value}')
|
||||||
|
if count > 5:
|
||||||
|
break
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
else: # static sub
|
||||||
|
|
||||||
|
await stream.send(subs)
|
||||||
|
async for value in stream:
|
||||||
|
print(f'{uid} got: {value}')
|
||||||
|
|
||||||
|
|
||||||
|
def test_dynamic_pub_sub():
|
||||||
|
|
||||||
|
global _registry
|
||||||
|
|
||||||
|
from multiprocessing import cpu_count
|
||||||
|
cpus = cpu_count()
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
# name of this actor will be same as target func
|
||||||
|
await n.run_in_actor(publisher)
|
||||||
|
|
||||||
|
for i, sub in zip(
|
||||||
|
range(cpus - 2),
|
||||||
|
itertools.cycle(_registry.keys())
|
||||||
|
):
|
||||||
|
await n.run_in_actor(
|
||||||
|
consumer,
|
||||||
|
name=f'consumer_{sub}',
|
||||||
|
subs=[sub],
|
||||||
|
)
|
||||||
|
|
||||||
|
# make one dynamic subscriber
|
||||||
|
await n.run_in_actor(
|
||||||
|
consumer,
|
||||||
|
name='consumer_dynamic',
|
||||||
|
subs=list(_registry.keys()),
|
||||||
|
)
|
||||||
|
|
||||||
|
# block until cancelled by user
|
||||||
|
with trio.fail_after(3):
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
try:
|
||||||
|
trio.run(main)
|
||||||
|
except trio.TooSlowError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def one_task_streams_and_one_handles_reqresp(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
await ctx.started()
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
async def pingpong():
|
||||||
|
'''Run a simple req/response service.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async for msg in stream:
|
||||||
|
print('rpc server ping')
|
||||||
|
assert msg == 'ping'
|
||||||
|
print('rpc server pong')
|
||||||
|
await stream.send('pong')
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
n.start_soon(pingpong)
|
||||||
|
|
||||||
|
for _ in itertools.count():
|
||||||
|
await stream.send('yo')
|
||||||
|
await trio.sleep(0.01)
|
||||||
|
|
||||||
|
|
||||||
|
def test_reqresp_ontopof_streaming():
|
||||||
|
'''
|
||||||
|
Test a subactor that both streams with one task and
|
||||||
|
spawns another which handles a small requests-response
|
||||||
|
dialogue over the same bidir-stream.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
# flat to make sure we get at least one pong
|
||||||
|
got_pong: bool = False
|
||||||
|
timeout: int = 2
|
||||||
|
|
||||||
|
if is_win(): # smh
|
||||||
|
timeout = 4
|
||||||
|
|
||||||
|
with trio.move_on_after(timeout):
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
# name of this actor will be same as target func
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'dual_tasks',
|
||||||
|
enable_modules=[__name__]
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
one_task_streams_and_one_handles_reqresp,
|
||||||
|
|
||||||
|
) as (ctx, first):
|
||||||
|
|
||||||
|
assert first is None
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
await stream.send('ping')
|
||||||
|
|
||||||
|
async for msg in stream:
|
||||||
|
print(f'client received: {msg}')
|
||||||
|
|
||||||
|
assert msg in {'pong', 'yo'}
|
||||||
|
|
||||||
|
if msg == 'pong':
|
||||||
|
got_pong = True
|
||||||
|
await stream.send('ping')
|
||||||
|
print('client sent ping')
|
||||||
|
|
||||||
|
assert got_pong
|
||||||
|
|
||||||
|
try:
|
||||||
|
trio.run(main)
|
||||||
|
except trio.TooSlowError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
async def async_gen_stream(sequence):
|
||||||
|
for i in sequence:
|
||||||
|
yield i
|
||||||
|
await trio.sleep(0.1)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def echo_ctx_stream(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
) -> None:
|
||||||
|
await ctx.started()
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
async for msg in stream:
|
||||||
|
await stream.send(msg)
|
||||||
|
|
||||||
|
|
||||||
|
def test_sigint_both_stream_types():
|
||||||
|
'''Verify that running a bi-directional and recv only stream
|
||||||
|
side-by-side will cancel correctly from SIGINT.
|
||||||
|
|
||||||
|
'''
|
||||||
|
timeout: float = 2
|
||||||
|
if is_win(): # smh
|
||||||
|
timeout += 1
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
with trio.fail_after(timeout):
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
# name of this actor will be same as target func
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'2_way',
|
||||||
|
enable_modules=[__name__]
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(echo_ctx_stream) as (ctx, _):
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
async with portal.open_stream_from(
|
||||||
|
async_gen_stream,
|
||||||
|
sequence=list(range(1)),
|
||||||
|
) as gen_stream:
|
||||||
|
|
||||||
|
msg = await gen_stream.receive()
|
||||||
|
await stream.send(msg)
|
||||||
|
resp = await stream.receive()
|
||||||
|
assert resp == msg
|
||||||
|
raise KeyboardInterrupt
|
||||||
|
|
||||||
|
try:
|
||||||
|
trio.run(main)
|
||||||
|
assert 0, "Didn't receive KBI!?"
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def inf_streamer(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Stream increasing ints until terminated with a 'done' msg.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
|
||||||
|
async with (
|
||||||
|
ctx.open_stream() as stream,
|
||||||
|
trio.open_nursery() as n,
|
||||||
|
):
|
||||||
|
async def bail_on_sentinel():
|
||||||
|
async for msg in stream:
|
||||||
|
if msg == 'done':
|
||||||
|
await stream.aclose()
|
||||||
|
else:
|
||||||
|
print(f'streamer received {msg}')
|
||||||
|
|
||||||
|
# start termination detector
|
||||||
|
n.start_soon(bail_on_sentinel)
|
||||||
|
|
||||||
|
for val in itertools.count():
|
||||||
|
try:
|
||||||
|
await stream.send(val)
|
||||||
|
except trio.ClosedResourceError:
|
||||||
|
# close out the stream gracefully
|
||||||
|
break
|
||||||
|
|
||||||
|
print('terminating streamer')
|
||||||
|
|
||||||
|
|
||||||
|
def test_local_task_fanout_from_stream():
|
||||||
|
'''
|
||||||
|
Single stream with multiple local consumer tasks using the
|
||||||
|
``MsgStream.subscribe()` api.
|
||||||
|
|
||||||
|
Ensure all tasks receive all values after stream completes sending.
|
||||||
|
|
||||||
|
'''
|
||||||
|
consumers = 22
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
counts = Counter()
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as tn:
|
||||||
|
p = await tn.start_actor(
|
||||||
|
'inf_streamer',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
async with (
|
||||||
|
p.open_context(inf_streamer) as (ctx, _),
|
||||||
|
ctx.open_stream() as stream,
|
||||||
|
):
|
||||||
|
|
||||||
|
async def pull_and_count(name: str):
|
||||||
|
# name = trio.lowlevel.current_task().name
|
||||||
|
async with stream.subscribe() as recver:
|
||||||
|
assert isinstance(
|
||||||
|
recver,
|
||||||
|
tractor.trionics.BroadcastReceiver
|
||||||
|
)
|
||||||
|
async for val in recver:
|
||||||
|
# print(f'{name}: {val}')
|
||||||
|
counts[name] += 1
|
||||||
|
|
||||||
|
print(f'{name} bcaster ended')
|
||||||
|
|
||||||
|
print(f'{name} completed')
|
||||||
|
|
||||||
|
with trio.fail_after(3):
|
||||||
|
async with trio.open_nursery() as nurse:
|
||||||
|
for i in range(consumers):
|
||||||
|
nurse.start_soon(pull_and_count, i)
|
||||||
|
|
||||||
|
await trio.sleep(0.5)
|
||||||
|
print('\nterminating')
|
||||||
|
await stream.send('done')
|
||||||
|
|
||||||
|
print('closed stream connection')
|
||||||
|
|
||||||
|
assert len(counts) == consumers
|
||||||
|
mx = max(counts.values())
|
||||||
|
# make sure each task received all stream values
|
||||||
|
assert all(val == mx for val in counts.values())
|
||||||
|
|
||||||
|
await p.cancel_actor()
|
||||||
|
|
||||||
|
trio.run(main)
|
|
@ -1,5 +1,6 @@
|
||||||
"""
|
"""
|
||||||
Cancellation and error propagation
|
Cancellation and error propagation
|
||||||
|
|
||||||
"""
|
"""
|
||||||
import os
|
import os
|
||||||
import signal
|
import signal
|
||||||
|
@ -7,6 +8,10 @@ import platform
|
||||||
import time
|
import time
|
||||||
from itertools import repeat
|
from itertools import repeat
|
||||||
|
|
||||||
|
from exceptiongroup import (
|
||||||
|
BaseExceptionGroup,
|
||||||
|
ExceptionGroup,
|
||||||
|
)
|
||||||
import pytest
|
import pytest
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
@ -14,6 +19,10 @@ import tractor
|
||||||
from conftest import tractor_test, no_windows
|
from conftest import tractor_test, no_windows
|
||||||
|
|
||||||
|
|
||||||
|
def is_win():
|
||||||
|
return platform.system() == 'Windows'
|
||||||
|
|
||||||
|
|
||||||
async def assert_err(delay=0):
|
async def assert_err(delay=0):
|
||||||
await trio.sleep(delay)
|
await trio.sleep(delay)
|
||||||
assert 0
|
assert 0
|
||||||
|
@ -47,31 +56,57 @@ def test_remote_error(arb_addr, args_err):
|
||||||
args, errtype = args_err
|
args, errtype = args_err
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery() as nursery:
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
) as nursery:
|
||||||
|
|
||||||
portal = await nursery.run_in_actor(assert_err, name='errorer', **args)
|
# on a remote type error caused by bad input args
|
||||||
|
# this should raise directly which means we **don't** get
|
||||||
|
# an exception group outside the nursery since the error
|
||||||
|
# here and the far end task error are one in the same?
|
||||||
|
portal = await nursery.run_in_actor(
|
||||||
|
assert_err, name='errorer', **args
|
||||||
|
)
|
||||||
|
|
||||||
# get result(s) from main task
|
# get result(s) from main task
|
||||||
try:
|
try:
|
||||||
|
# this means the root actor will also raise a local
|
||||||
|
# parent task error and thus an eg will propagate out
|
||||||
|
# of this actor nursery.
|
||||||
await portal.result()
|
await portal.result()
|
||||||
except tractor.RemoteActorError as err:
|
except tractor.RemoteActorError as err:
|
||||||
assert err.type == errtype
|
assert err.type == errtype
|
||||||
print("Look Maa that actor failed hard, hehh")
|
print("Look Maa that actor failed hard, hehh")
|
||||||
raise
|
raise
|
||||||
|
|
||||||
|
# ensure boxed errors
|
||||||
|
if args:
|
||||||
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
||||||
tractor.run(main, arbiter_addr=arb_addr)
|
trio.run(main)
|
||||||
|
|
||||||
# ensure boxed error is correct
|
|
||||||
assert excinfo.value.type == errtype
|
assert excinfo.value.type == errtype
|
||||||
|
|
||||||
|
else:
|
||||||
|
# the root task will also error on the `.result()` call
|
||||||
|
# so we expect an error from there AND the child.
|
||||||
|
with pytest.raises(BaseExceptionGroup) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
# ensure boxed errors
|
||||||
|
for exc in excinfo.value.exceptions:
|
||||||
|
assert exc.type == errtype
|
||||||
|
|
||||||
|
|
||||||
def test_multierror(arb_addr):
|
def test_multierror(arb_addr):
|
||||||
"""Verify we raise a ``trio.MultiError`` out of a nursery where
|
'''
|
||||||
|
Verify we raise a ``BaseExceptionGroup`` out of a nursery where
|
||||||
more then one actor errors.
|
more then one actor errors.
|
||||||
"""
|
|
||||||
|
'''
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery() as nursery:
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
) as nursery:
|
||||||
|
|
||||||
await nursery.run_in_actor(assert_err, name='errorer1')
|
await nursery.run_in_actor(assert_err, name='errorer1')
|
||||||
portal2 = await nursery.run_in_actor(assert_err, name='errorer2')
|
portal2 = await nursery.run_in_actor(assert_err, name='errorer2')
|
||||||
|
@ -84,11 +119,11 @@ def test_multierror(arb_addr):
|
||||||
print("Look Maa that first actor failed hard, hehh")
|
print("Look Maa that first actor failed hard, hehh")
|
||||||
raise
|
raise
|
||||||
|
|
||||||
# here we should get a `trio.MultiError` containing exceptions
|
# here we should get a ``BaseExceptionGroup`` containing exceptions
|
||||||
# from both subactors
|
# from both subactors
|
||||||
|
|
||||||
with pytest.raises(trio.MultiError):
|
with pytest.raises(BaseExceptionGroup):
|
||||||
tractor.run(main, arbiter_addr=arb_addr)
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize('delay', (0, 0.5))
|
@pytest.mark.parametrize('delay', (0, 0.5))
|
||||||
|
@ -96,12 +131,15 @@ def test_multierror(arb_addr):
|
||||||
'num_subactors', range(25, 26),
|
'num_subactors', range(25, 26),
|
||||||
)
|
)
|
||||||
def test_multierror_fast_nursery(arb_addr, start_method, num_subactors, delay):
|
def test_multierror_fast_nursery(arb_addr, start_method, num_subactors, delay):
|
||||||
"""Verify we raise a ``trio.MultiError`` out of a nursery where
|
"""Verify we raise a ``BaseExceptionGroup`` out of a nursery where
|
||||||
more then one actor errors and also with a delay before failure
|
more then one actor errors and also with a delay before failure
|
||||||
to test failure during an ongoing spawning.
|
to test failure during an ongoing spawning.
|
||||||
"""
|
"""
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery() as nursery:
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
) as nursery:
|
||||||
|
|
||||||
for i in range(num_subactors):
|
for i in range(num_subactors):
|
||||||
await nursery.run_in_actor(
|
await nursery.run_in_actor(
|
||||||
assert_err,
|
assert_err,
|
||||||
|
@ -109,18 +147,30 @@ def test_multierror_fast_nursery(arb_addr, start_method, num_subactors, delay):
|
||||||
delay=delay
|
delay=delay
|
||||||
)
|
)
|
||||||
|
|
||||||
with pytest.raises(trio.MultiError) as exc_info:
|
# with pytest.raises(trio.MultiError) as exc_info:
|
||||||
tractor.run(main, arbiter_addr=arb_addr)
|
with pytest.raises(BaseExceptionGroup) as exc_info:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
assert exc_info.type == tractor.MultiError
|
assert exc_info.type == ExceptionGroup
|
||||||
err = exc_info.value
|
err = exc_info.value
|
||||||
assert len(err.exceptions) == num_subactors
|
exceptions = err.exceptions
|
||||||
for exc in err.exceptions:
|
|
||||||
|
if len(exceptions) == 2:
|
||||||
|
# sometimes oddly now there's an embedded BrokenResourceError ?
|
||||||
|
for exc in exceptions:
|
||||||
|
excs = getattr(exc, 'exceptions', None)
|
||||||
|
if excs:
|
||||||
|
exceptions = excs
|
||||||
|
break
|
||||||
|
|
||||||
|
assert len(exceptions) == num_subactors
|
||||||
|
|
||||||
|
for exc in exceptions:
|
||||||
assert isinstance(exc, tractor.RemoteActorError)
|
assert isinstance(exc, tractor.RemoteActorError)
|
||||||
assert exc.type == AssertionError
|
assert exc.type == AssertionError
|
||||||
|
|
||||||
|
|
||||||
def do_nothing():
|
async def do_nothing():
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@ -132,10 +182,12 @@ def test_cancel_single_subactor(arb_addr, mechanism):
|
||||||
async def spawn_actor():
|
async def spawn_actor():
|
||||||
"""Spawn an actor that blocks indefinitely.
|
"""Spawn an actor that blocks indefinitely.
|
||||||
"""
|
"""
|
||||||
async with tractor.open_nursery() as nursery:
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
) as nursery:
|
||||||
|
|
||||||
portal = await nursery.start_actor(
|
portal = await nursery.start_actor(
|
||||||
'nothin', rpc_module_paths=[__name__],
|
'nothin', enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
assert (await portal.run(do_nothing)) is None
|
assert (await portal.run(do_nothing)) is None
|
||||||
|
|
||||||
|
@ -146,10 +198,10 @@ def test_cancel_single_subactor(arb_addr, mechanism):
|
||||||
raise mechanism
|
raise mechanism
|
||||||
|
|
||||||
if mechanism == 'nursery_cancel':
|
if mechanism == 'nursery_cancel':
|
||||||
tractor.run(spawn_actor, arbiter_addr=arb_addr)
|
trio.run(spawn_actor)
|
||||||
else:
|
else:
|
||||||
with pytest.raises(mechanism):
|
with pytest.raises(mechanism):
|
||||||
tractor.run(spawn_actor, arbiter_addr=arb_addr)
|
trio.run(spawn_actor)
|
||||||
|
|
||||||
|
|
||||||
async def stream_forever():
|
async def stream_forever():
|
||||||
|
@ -168,12 +220,13 @@ async def test_cancel_infinite_streamer(start_method):
|
||||||
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',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
# this async for loop streams values from the above
|
# this async for loop streams values from the above
|
||||||
# async generator running in a separate process
|
# async generator running in a separate process
|
||||||
async for letter in await portal.run(stream_forever):
|
async with portal.open_stream_from(stream_forever) as stream:
|
||||||
|
async for letter in stream:
|
||||||
print(letter)
|
print(letter)
|
||||||
|
|
||||||
# we support trio's cancellation system
|
# we support trio's cancellation system
|
||||||
|
@ -186,8 +239,8 @@ async def test_cancel_infinite_streamer(start_method):
|
||||||
[
|
[
|
||||||
# daemon actors sit idle while single task actors error out
|
# daemon actors sit idle while single task actors error out
|
||||||
(1, tractor.RemoteActorError, AssertionError, (assert_err, {}), None),
|
(1, tractor.RemoteActorError, AssertionError, (assert_err, {}), None),
|
||||||
(2, tractor.MultiError, AssertionError, (assert_err, {}), None),
|
(2, BaseExceptionGroup, AssertionError, (assert_err, {}), None),
|
||||||
(3, tractor.MultiError, AssertionError, (assert_err, {}), None),
|
(3, BaseExceptionGroup, AssertionError, (assert_err, {}), None),
|
||||||
|
|
||||||
# 1 daemon actor errors out while single task actors sleep forever
|
# 1 daemon actor errors out while single task actors sleep forever
|
||||||
(3, tractor.RemoteActorError, AssertionError, (sleep_forever, {}),
|
(3, tractor.RemoteActorError, AssertionError, (sleep_forever, {}),
|
||||||
|
@ -198,7 +251,7 @@ async def test_cancel_infinite_streamer(start_method):
|
||||||
(do_nuthin, {}), (assert_err, {'delay': 1}, True)),
|
(do_nuthin, {}), (assert_err, {'delay': 1}, True)),
|
||||||
# daemon complete quickly delay while single task
|
# daemon complete quickly delay while single task
|
||||||
# actors error after brief delay
|
# actors error after brief delay
|
||||||
(3, tractor.MultiError, AssertionError,
|
(3, BaseExceptionGroup, AssertionError,
|
||||||
(assert_err, {'delay': 1}), (do_nuthin, {}, False)),
|
(assert_err, {'delay': 1}), (do_nuthin, {}, False)),
|
||||||
],
|
],
|
||||||
ids=[
|
ids=[
|
||||||
|
@ -226,7 +279,7 @@ async def test_some_cancels_all(num_actors_and_errs, start_method, loglevel):
|
||||||
for i in range(num_actors):
|
for i in range(num_actors):
|
||||||
dactor_portals.append(await n.start_actor(
|
dactor_portals.append(await n.start_actor(
|
||||||
f'deamon_{i}',
|
f'deamon_{i}',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
))
|
))
|
||||||
|
|
||||||
func, kwargs = ria_func
|
func, kwargs = ria_func
|
||||||
|
@ -265,7 +318,7 @@ async def test_some_cancels_all(num_actors_and_errs, start_method, loglevel):
|
||||||
# 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:
|
||||||
if isinstance(err, tractor.MultiError):
|
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:
|
||||||
if isinstance(exc, tractor.RemoteActorError):
|
if isinstance(exc, tractor.RemoteActorError):
|
||||||
|
@ -308,10 +361,12 @@ async def spawn_and_error(breadth, depth) -> None:
|
||||||
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_nested_multierrors(loglevel, start_method):
|
async def test_nested_multierrors(loglevel, start_method):
|
||||||
"""Test that failed actor sets are wrapped in `trio.MultiError`s.
|
'''
|
||||||
This test goes only 2 nurseries deep but we should eventually have tests
|
Test that failed actor sets are wrapped in `BaseExceptionGroup`s. This
|
||||||
|
test goes only 2 nurseries deep but we should eventually have tests
|
||||||
for arbitrary n-depth actor trees.
|
for arbitrary n-depth actor trees.
|
||||||
"""
|
|
||||||
|
'''
|
||||||
if start_method == 'trio':
|
if start_method == 'trio':
|
||||||
depth = 3
|
depth = 3
|
||||||
subactor_breadth = 2
|
subactor_breadth = 2
|
||||||
|
@ -335,24 +390,36 @@ async def test_nested_multierrors(loglevel, start_method):
|
||||||
breadth=subactor_breadth,
|
breadth=subactor_breadth,
|
||||||
depth=depth,
|
depth=depth,
|
||||||
)
|
)
|
||||||
except trio.MultiError as err:
|
except BaseExceptionGroup as err:
|
||||||
assert len(err.exceptions) == subactor_breadth
|
assert len(err.exceptions) == subactor_breadth
|
||||||
for subexc in err.exceptions:
|
for subexc in err.exceptions:
|
||||||
|
|
||||||
# verify first level actor errors are wrapped as remote
|
# verify first level actor errors are wrapped as remote
|
||||||
if platform.system() == 'Windows':
|
if is_win():
|
||||||
|
|
||||||
# windows is often too slow and cancellation seems
|
# windows is often too slow and cancellation seems
|
||||||
# to happen before an actor is spawned
|
# to happen before an actor is spawned
|
||||||
if isinstance(subexc, trio.Cancelled):
|
if isinstance(subexc, trio.Cancelled):
|
||||||
continue
|
continue
|
||||||
else:
|
|
||||||
|
elif isinstance(subexc, tractor.RemoteActorError):
|
||||||
# on windows it seems we can't exactly be sure wtf
|
# on windows it seems we can't exactly be sure wtf
|
||||||
# will happen..
|
# will happen..
|
||||||
assert subexc.type in (
|
assert subexc.type in (
|
||||||
tractor.RemoteActorError,
|
tractor.RemoteActorError,
|
||||||
trio.Cancelled,
|
trio.Cancelled,
|
||||||
trio.MultiError
|
BaseExceptionGroup,
|
||||||
|
)
|
||||||
|
|
||||||
|
elif isinstance(subexc, BaseExceptionGroup):
|
||||||
|
for subsub in subexc.exceptions:
|
||||||
|
|
||||||
|
if subsub in (tractor.RemoteActorError,):
|
||||||
|
subsub = subsub.type
|
||||||
|
|
||||||
|
assert type(subsub) in (
|
||||||
|
trio.Cancelled,
|
||||||
|
BaseExceptionGroup,
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
assert isinstance(subexc, tractor.RemoteActorError)
|
assert isinstance(subexc, tractor.RemoteActorError)
|
||||||
|
@ -361,14 +428,21 @@ async def test_nested_multierrors(loglevel, start_method):
|
||||||
# XXX not sure what's up with this..
|
# XXX not sure what's up with this..
|
||||||
# on windows sometimes spawning is just too slow and
|
# on windows sometimes spawning is just too slow and
|
||||||
# we get back the (sent) cancel signal instead
|
# we get back the (sent) cancel signal instead
|
||||||
if platform.system() == 'Windows':
|
if is_win():
|
||||||
assert (subexc.type is trio.MultiError) or (
|
if isinstance(subexc, tractor.RemoteActorError):
|
||||||
subexc.type is tractor.RemoteActorError)
|
assert subexc.type in (
|
||||||
|
BaseExceptionGroup,
|
||||||
|
tractor.RemoteActorError
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
assert subexc.type is trio.MultiError
|
assert isinstance(subexc, BaseExceptionGroup)
|
||||||
else:
|
else:
|
||||||
assert (subexc.type is tractor.RemoteActorError) or (
|
assert subexc.type is ExceptionGroup
|
||||||
subexc.type is trio.Cancelled)
|
else:
|
||||||
|
assert subexc.type in (
|
||||||
|
tractor.RemoteActorError,
|
||||||
|
trio.Cancelled
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@no_windows
|
@no_windows
|
||||||
|
@ -386,13 +460,13 @@ def test_cancel_via_SIGINT(
|
||||||
with trio.fail_after(2):
|
with trio.fail_after(2):
|
||||||
async with tractor.open_nursery() as tn:
|
async with tractor.open_nursery() as tn:
|
||||||
await tn.start_actor('sucka')
|
await tn.start_actor('sucka')
|
||||||
if spawn_backend == 'mp':
|
if 'mp' in spawn_backend:
|
||||||
time.sleep(0.1)
|
time.sleep(0.1)
|
||||||
os.kill(pid, signal.SIGINT)
|
os.kill(pid, signal.SIGINT)
|
||||||
await trio.sleep_forever()
|
await trio.sleep_forever()
|
||||||
|
|
||||||
with pytest.raises(KeyboardInterrupt):
|
with pytest.raises(KeyboardInterrupt):
|
||||||
tractor.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
@no_windows
|
@no_windows
|
||||||
|
@ -406,6 +480,9 @@ def test_cancel_via_SIGINT_other_task(
|
||||||
from a seperate ``trio`` child task.
|
from a seperate ``trio`` child task.
|
||||||
"""
|
"""
|
||||||
pid = os.getpid()
|
pid = os.getpid()
|
||||||
|
timeout: float = 2
|
||||||
|
if is_win(): # smh
|
||||||
|
timeout += 1
|
||||||
|
|
||||||
async def spawn_and_sleep_forever(task_status=trio.TASK_STATUS_IGNORED):
|
async def spawn_and_sleep_forever(task_status=trio.TASK_STATUS_IGNORED):
|
||||||
async with tractor.open_nursery() as tn:
|
async with tractor.open_nursery() as tn:
|
||||||
|
@ -419,16 +496,15 @@ 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(2):
|
with trio.fail_after(timeout):
|
||||||
async with trio.open_nursery() as n:
|
async with trio.open_nursery() as n:
|
||||||
await n.start(spawn_and_sleep_forever)
|
await n.start(spawn_and_sleep_forever)
|
||||||
if spawn_backend == 'mp':
|
if 'mp' in spawn_backend:
|
||||||
time.sleep(0.1)
|
time.sleep(0.1)
|
||||||
os.kill(pid, signal.SIGINT)
|
os.kill(pid, signal.SIGINT)
|
||||||
|
|
||||||
with pytest.raises(KeyboardInterrupt):
|
with pytest.raises(KeyboardInterrupt):
|
||||||
tractor.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def spin_for(period=3):
|
async def spin_for(period=3):
|
||||||
|
@ -438,7 +514,7 @@ async def spin_for(period=3):
|
||||||
|
|
||||||
async def spawn():
|
async def spawn():
|
||||||
async with tractor.open_nursery() as tn:
|
async with tractor.open_nursery() as tn:
|
||||||
portal = await tn.run_in_actor(
|
await tn.run_in_actor(
|
||||||
spin_for,
|
spin_for,
|
||||||
name='sleeper',
|
name='sleeper',
|
||||||
)
|
)
|
||||||
|
@ -460,7 +536,7 @@ def test_cancel_while_childs_child_in_sync_sleep(
|
||||||
async def main():
|
async def main():
|
||||||
with trio.fail_after(2):
|
with trio.fail_after(2):
|
||||||
async with tractor.open_nursery() as tn:
|
async with tractor.open_nursery() as tn:
|
||||||
portal = await tn.run_in_actor(
|
await tn.run_in_actor(
|
||||||
spawn,
|
spawn,
|
||||||
name='spawn',
|
name='spawn',
|
||||||
)
|
)
|
||||||
|
@ -468,4 +544,58 @@ def test_cancel_while_childs_child_in_sync_sleep(
|
||||||
assert 0
|
assert 0
|
||||||
|
|
||||||
with pytest.raises(AssertionError):
|
with pytest.raises(AssertionError):
|
||||||
tractor.run(main)
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def test_fast_graceful_cancel_when_spawn_task_in_soft_proc_wait_for_daemon(
|
||||||
|
start_method,
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
This is a very subtle test which demonstrates how cancellation
|
||||||
|
during process collection can result in non-optimal teardown
|
||||||
|
performance on daemon actors. The fix for this test was to handle
|
||||||
|
``trio.Cancelled`` specially in the spawn task waiting in
|
||||||
|
`proc.wait()` such that ``Portal.cancel_actor()`` is called before
|
||||||
|
executing the "hard reap" sequence (which has an up to 3 second
|
||||||
|
delay currently).
|
||||||
|
|
||||||
|
In other words, if we can cancel the actor using a graceful remote
|
||||||
|
cancellation, and it's faster, we might as well do it.
|
||||||
|
|
||||||
|
'''
|
||||||
|
kbi_delay = 0.5
|
||||||
|
timeout: float = 2.9
|
||||||
|
|
||||||
|
if is_win(): # smh
|
||||||
|
timeout += 1
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
start = time.time()
|
||||||
|
try:
|
||||||
|
async with trio.open_nursery() as nurse:
|
||||||
|
async with tractor.open_nursery() as tn:
|
||||||
|
p = await tn.start_actor(
|
||||||
|
'fast_boi',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async def delayed_kbi():
|
||||||
|
await trio.sleep(kbi_delay)
|
||||||
|
print(f'RAISING KBI after {kbi_delay} s')
|
||||||
|
raise KeyboardInterrupt
|
||||||
|
|
||||||
|
# start task which raises a kbi **after**
|
||||||
|
# the actor nursery ``__aexit__()`` has
|
||||||
|
# been run.
|
||||||
|
nurse.start_soon(delayed_kbi)
|
||||||
|
|
||||||
|
await p.run(do_nuthin)
|
||||||
|
finally:
|
||||||
|
duration = time.time() - start
|
||||||
|
if duration > timeout:
|
||||||
|
raise trio.TooSlowError(
|
||||||
|
'daemon cancel was slower then necessary..'
|
||||||
|
)
|
||||||
|
|
||||||
|
with pytest.raises(KeyboardInterrupt):
|
||||||
|
trio.run(main)
|
||||||
|
|
|
@ -0,0 +1,173 @@
|
||||||
|
'''
|
||||||
|
Test a service style daemon that maintains a nursery for spawning
|
||||||
|
"remote async tasks" including both spawning other long living
|
||||||
|
sub-sub-actor daemons.
|
||||||
|
|
||||||
|
'''
|
||||||
|
from typing import Optional
|
||||||
|
import asyncio
|
||||||
|
from contextlib import asynccontextmanager as acm
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
from trio_typing import TaskStatus
|
||||||
|
import tractor
|
||||||
|
from tractor import RemoteActorError
|
||||||
|
from async_generator import aclosing
|
||||||
|
|
||||||
|
|
||||||
|
async def aio_streamer(
|
||||||
|
from_trio: asyncio.Queue,
|
||||||
|
to_trio: trio.abc.SendChannel,
|
||||||
|
) -> trio.abc.ReceiveChannel:
|
||||||
|
|
||||||
|
# required first msg to sync caller
|
||||||
|
to_trio.send_nowait(None)
|
||||||
|
|
||||||
|
from itertools import cycle
|
||||||
|
for i in cycle(range(10)):
|
||||||
|
to_trio.send_nowait(i)
|
||||||
|
await asyncio.sleep(0.01)
|
||||||
|
|
||||||
|
|
||||||
|
async def trio_streamer():
|
||||||
|
from itertools import cycle
|
||||||
|
for i in cycle(range(10)):
|
||||||
|
yield i
|
||||||
|
await trio.sleep(0.01)
|
||||||
|
|
||||||
|
|
||||||
|
async def trio_sleep_and_err(delay: float = 0.5):
|
||||||
|
await trio.sleep(delay)
|
||||||
|
# name error
|
||||||
|
doggy() # noqa
|
||||||
|
|
||||||
|
|
||||||
|
_cached_stream: Optional[
|
||||||
|
trio.abc.ReceiveChannel
|
||||||
|
] = None
|
||||||
|
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def wrapper_mngr(
|
||||||
|
):
|
||||||
|
from tractor.trionics import broadcast_receiver
|
||||||
|
global _cached_stream
|
||||||
|
in_aio = tractor.current_actor().is_infected_aio()
|
||||||
|
|
||||||
|
if in_aio:
|
||||||
|
if _cached_stream:
|
||||||
|
|
||||||
|
from_aio = _cached_stream
|
||||||
|
|
||||||
|
# if we already have a cached feed deliver a rx side clone
|
||||||
|
# to consumer
|
||||||
|
async with broadcast_receiver(from_aio, 6) as from_aio:
|
||||||
|
yield from_aio
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
async with tractor.to_asyncio.open_channel_from(
|
||||||
|
aio_streamer,
|
||||||
|
) as (first, from_aio):
|
||||||
|
assert not first
|
||||||
|
|
||||||
|
# cache it so next task uses broadcast receiver
|
||||||
|
_cached_stream = from_aio
|
||||||
|
|
||||||
|
yield from_aio
|
||||||
|
else:
|
||||||
|
async with aclosing(trio_streamer()) as stream:
|
||||||
|
# cache it so next task uses broadcast receiver
|
||||||
|
_cached_stream = stream
|
||||||
|
yield stream
|
||||||
|
|
||||||
|
|
||||||
|
_nursery: trio.Nursery = None
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def trio_main(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
):
|
||||||
|
# sync
|
||||||
|
await ctx.started()
|
||||||
|
|
||||||
|
# stash a "service nursery" as "actor local" (aka a Python global)
|
||||||
|
global _nursery
|
||||||
|
n = _nursery
|
||||||
|
assert n
|
||||||
|
|
||||||
|
async def consume_stream():
|
||||||
|
async with wrapper_mngr() as stream:
|
||||||
|
async for msg in stream:
|
||||||
|
print(msg)
|
||||||
|
|
||||||
|
# run 2 tasks to ensure broadcaster chan use
|
||||||
|
n.start_soon(consume_stream)
|
||||||
|
n.start_soon(consume_stream)
|
||||||
|
|
||||||
|
n.start_soon(trio_sleep_and_err)
|
||||||
|
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def open_actor_local_nursery(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
):
|
||||||
|
global _nursery
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
_nursery = n
|
||||||
|
await ctx.started()
|
||||||
|
await trio.sleep(10)
|
||||||
|
# await trio.sleep(1)
|
||||||
|
|
||||||
|
# XXX: this causes the hang since
|
||||||
|
# the caller does not unblock from its own
|
||||||
|
# ``trio.sleep_forever()``.
|
||||||
|
|
||||||
|
# TODO: we need to test a simple ctx task starting remote tasks
|
||||||
|
# that error and then blocking on a ``Nursery.start()`` which
|
||||||
|
# never yields back.. aka a scenario where the
|
||||||
|
# ``tractor.context`` task IS NOT in the service n's cancel
|
||||||
|
# scope.
|
||||||
|
n.cancel_scope.cancel()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'asyncio_mode',
|
||||||
|
[True, False],
|
||||||
|
ids='asyncio_mode={}'.format,
|
||||||
|
)
|
||||||
|
def test_actor_managed_trio_nursery_task_error_cancels_aio(
|
||||||
|
asyncio_mode: bool,
|
||||||
|
arb_addr
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Verify that a ``trio`` nursery created managed in a child actor
|
||||||
|
correctly relays errors to the parent actor when one of its spawned
|
||||||
|
tasks errors even when running in infected asyncio mode and using
|
||||||
|
broadcast receivers for multi-task-per-actor subscription.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
# cancel the nursery shortly after boot
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
p = await n.start_actor(
|
||||||
|
'nursery_mngr',
|
||||||
|
infect_asyncio=asyncio_mode,
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
async with (
|
||||||
|
p.open_context(open_actor_local_nursery) as (ctx, first),
|
||||||
|
p.open_context(trio_main) as (ctx, first),
|
||||||
|
):
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
with pytest.raises(RemoteActorError) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
# verify boxed error
|
||||||
|
err = excinfo.value
|
||||||
|
assert isinstance(err.type(), NameError)
|
|
@ -0,0 +1,84 @@
|
||||||
|
import itertools
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
from tractor import open_actor_cluster
|
||||||
|
from tractor.trionics import gather_contexts
|
||||||
|
|
||||||
|
from conftest import tractor_test
|
||||||
|
|
||||||
|
|
||||||
|
MESSAGE = 'tractoring at full speed'
|
||||||
|
|
||||||
|
|
||||||
|
def test_empty_mngrs_input_raises() -> None:
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
with trio.fail_after(1):
|
||||||
|
async with (
|
||||||
|
open_actor_cluster(
|
||||||
|
modules=[__name__],
|
||||||
|
|
||||||
|
# NOTE: ensure we can passthrough runtime opts
|
||||||
|
loglevel='info',
|
||||||
|
# debug_mode=True,
|
||||||
|
|
||||||
|
) as portals,
|
||||||
|
|
||||||
|
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()
|
||||||
|
),
|
||||||
|
),
|
||||||
|
):
|
||||||
|
assert 0
|
||||||
|
|
||||||
|
with pytest.raises(ValueError):
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def worker(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
await ctx.started()
|
||||||
|
|
||||||
|
async with ctx.open_stream(
|
||||||
|
backpressure=True,
|
||||||
|
) as stream:
|
||||||
|
|
||||||
|
# TODO: this with the below assert causes a hang bug?
|
||||||
|
# with trio.move_on_after(1):
|
||||||
|
|
||||||
|
async for msg in stream:
|
||||||
|
# do something with msg
|
||||||
|
print(msg)
|
||||||
|
assert msg == MESSAGE
|
||||||
|
|
||||||
|
# TODO: does this ever cause a hang
|
||||||
|
# assert 0
|
||||||
|
|
||||||
|
|
||||||
|
@tractor_test
|
||||||
|
async def test_streaming_to_actor_cluster() -> None:
|
||||||
|
|
||||||
|
async with (
|
||||||
|
open_actor_cluster(modules=[__name__]) as portals,
|
||||||
|
|
||||||
|
gather_contexts(
|
||||||
|
mngrs=[p.open_context(worker) for p in portals.values()],
|
||||||
|
) as contexts,
|
||||||
|
|
||||||
|
gather_contexts(
|
||||||
|
mngrs=[ctx[0].open_stream() for ctx in contexts],
|
||||||
|
) as streams,
|
||||||
|
|
||||||
|
):
|
||||||
|
with trio.move_on_after(1):
|
||||||
|
for stream in itertools.cycle(streams):
|
||||||
|
await stream.send(MESSAGE)
|
|
@ -0,0 +1,798 @@
|
||||||
|
'''
|
||||||
|
``async with ():`` inlined context-stream cancellation testing.
|
||||||
|
|
||||||
|
Verify the we raise errors when streams are opened prior to sync-opening
|
||||||
|
a ``tractor.Context`` beforehand.
|
||||||
|
|
||||||
|
'''
|
||||||
|
from contextlib import asynccontextmanager as acm
|
||||||
|
from itertools import count
|
||||||
|
import platform
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
from tractor._exceptions import StreamOverrun
|
||||||
|
|
||||||
|
from conftest import tractor_test
|
||||||
|
|
||||||
|
# ``Context`` semantics are as follows,
|
||||||
|
# ------------------------------------
|
||||||
|
|
||||||
|
# - standard setup/teardown:
|
||||||
|
# ``Portal.open_context()`` starts a new
|
||||||
|
# remote task context in another actor. The target actor's task must
|
||||||
|
# call ``Context.started()`` to unblock this entry on the caller side.
|
||||||
|
# the callee task executes until complete and returns a final value
|
||||||
|
# which is delivered to the caller side and retreived via
|
||||||
|
# ``Context.result()``.
|
||||||
|
|
||||||
|
# - cancel termination:
|
||||||
|
# context can be cancelled on either side where either end's task can
|
||||||
|
# call ``Context.cancel()`` which raises a local ``trio.Cancelled``
|
||||||
|
# and sends a task cancel request to the remote task which in turn
|
||||||
|
# raises a ``trio.Cancelled`` in that scope, catches it, and re-raises
|
||||||
|
# as ``ContextCancelled``. This is then caught by
|
||||||
|
# ``Portal.open_context()``'s exit and we get a graceful termination
|
||||||
|
# of the linked tasks.
|
||||||
|
|
||||||
|
# - error termination:
|
||||||
|
# error is caught after all context-cancel-scope tasks are cancelled
|
||||||
|
# via regular ``trio`` cancel scope semantics, error is sent to other
|
||||||
|
# side and unpacked as a `RemoteActorError`.
|
||||||
|
|
||||||
|
|
||||||
|
# ``Context.open_stream() as stream: MsgStream:`` msg semantics are:
|
||||||
|
# -----------------------------------------------------------------
|
||||||
|
|
||||||
|
# - either side can ``.send()`` which emits a 'yield' msgs and delivers
|
||||||
|
# a value to the a ``MsgStream.receive()`` call.
|
||||||
|
|
||||||
|
# - stream closure: one end relays a 'stop' message which terminates an
|
||||||
|
# ongoing ``MsgStream`` iteration.
|
||||||
|
|
||||||
|
# - cancel/error termination: as per the context semantics above but
|
||||||
|
# with implicit stream closure on the cancelling end.
|
||||||
|
|
||||||
|
|
||||||
|
_state: bool = False
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def too_many_starteds(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Call ``Context.started()`` more then once (an error).
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
try:
|
||||||
|
await ctx.started()
|
||||||
|
except RuntimeError:
|
||||||
|
raise
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def not_started_but_stream_opened(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Enter ``Context.open_stream()`` without calling ``.started()``.
|
||||||
|
|
||||||
|
'''
|
||||||
|
try:
|
||||||
|
async with ctx.open_stream():
|
||||||
|
assert 0
|
||||||
|
except RuntimeError:
|
||||||
|
raise
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'target',
|
||||||
|
[too_many_starteds, not_started_but_stream_opened],
|
||||||
|
ids='misuse_type={}'.format,
|
||||||
|
)
|
||||||
|
def test_started_misuse(target):
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.start_actor(
|
||||||
|
target.__name__,
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(target) as (ctx, sent):
|
||||||
|
await trio.sleep(1)
|
||||||
|
|
||||||
|
with pytest.raises(tractor.RemoteActorError):
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def simple_setup_teardown(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
data: int,
|
||||||
|
block_forever: bool = False,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
# startup phase
|
||||||
|
global _state
|
||||||
|
_state = True
|
||||||
|
|
||||||
|
# signal to parent that we're up
|
||||||
|
await ctx.started(data + 1)
|
||||||
|
|
||||||
|
try:
|
||||||
|
if block_forever:
|
||||||
|
# block until cancelled
|
||||||
|
await trio.sleep_forever()
|
||||||
|
else:
|
||||||
|
return 'yo'
|
||||||
|
finally:
|
||||||
|
_state = False
|
||||||
|
|
||||||
|
|
||||||
|
async def assert_state(value: bool):
|
||||||
|
global _state
|
||||||
|
assert _state == value
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'error_parent',
|
||||||
|
[False, ValueError, KeyboardInterrupt],
|
||||||
|
)
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'callee_blocks_forever',
|
||||||
|
[False, True],
|
||||||
|
ids=lambda item: f'callee_blocks_forever={item}'
|
||||||
|
)
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'pointlessly_open_stream',
|
||||||
|
[False, True],
|
||||||
|
ids=lambda item: f'open_stream={item}'
|
||||||
|
)
|
||||||
|
def test_simple_context(
|
||||||
|
error_parent,
|
||||||
|
callee_blocks_forever,
|
||||||
|
pointlessly_open_stream,
|
||||||
|
):
|
||||||
|
|
||||||
|
timeout = 1.5 if not platform.system() == 'Windows' else 4
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
with trio.fail_after(timeout):
|
||||||
|
async with tractor.open_nursery() as nursery:
|
||||||
|
|
||||||
|
portal = await nursery.start_actor(
|
||||||
|
'simple_context',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with portal.open_context(
|
||||||
|
simple_setup_teardown,
|
||||||
|
data=10,
|
||||||
|
block_forever=callee_blocks_forever,
|
||||||
|
) as (ctx, sent):
|
||||||
|
|
||||||
|
assert sent == 11
|
||||||
|
|
||||||
|
if callee_blocks_forever:
|
||||||
|
await portal.run(assert_state, value=True)
|
||||||
|
else:
|
||||||
|
assert await ctx.result() == 'yo'
|
||||||
|
|
||||||
|
if not error_parent:
|
||||||
|
await ctx.cancel()
|
||||||
|
|
||||||
|
if pointlessly_open_stream:
|
||||||
|
async with ctx.open_stream():
|
||||||
|
if error_parent:
|
||||||
|
raise error_parent
|
||||||
|
|
||||||
|
if callee_blocks_forever:
|
||||||
|
await ctx.cancel()
|
||||||
|
else:
|
||||||
|
# in this case the stream will send a
|
||||||
|
# 'stop' msg to the far end which needs
|
||||||
|
# to be ignored
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
if error_parent:
|
||||||
|
raise error_parent
|
||||||
|
|
||||||
|
finally:
|
||||||
|
|
||||||
|
# after cancellation
|
||||||
|
if not error_parent:
|
||||||
|
await portal.run(assert_state, value=False)
|
||||||
|
|
||||||
|
# shut down daemon
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
if error_parent:
|
||||||
|
try:
|
||||||
|
trio.run(main)
|
||||||
|
except error_parent:
|
||||||
|
pass
|
||||||
|
except trio.MultiError as me:
|
||||||
|
# XXX: on windows it seems we may have to expect the group error
|
||||||
|
from tractor._exceptions import is_multi_cancelled
|
||||||
|
assert is_multi_cancelled(me)
|
||||||
|
else:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
# basic stream terminations:
|
||||||
|
# - callee context closes without using stream
|
||||||
|
# - caller context closes without using stream
|
||||||
|
# - caller context calls `Context.cancel()` while streaming
|
||||||
|
# is ongoing resulting in callee being cancelled
|
||||||
|
# - callee calls `Context.cancel()` while streaming and caller
|
||||||
|
# sees stream terminated in `RemoteActorError`
|
||||||
|
|
||||||
|
# TODO: future possible features
|
||||||
|
# - restart request: far end raises `ContextRestart`
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def close_ctx_immediately(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
await ctx.started()
|
||||||
|
global _state
|
||||||
|
|
||||||
|
async with ctx.open_stream():
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@tractor_test
|
||||||
|
async def test_callee_closes_ctx_after_stream_open():
|
||||||
|
'callee context closes without using stream'
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'fast_stream_closer',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
with trio.fail_after(2):
|
||||||
|
async with portal.open_context(
|
||||||
|
close_ctx_immediately,
|
||||||
|
|
||||||
|
# flag to avoid waiting the final result
|
||||||
|
# cancel_on_exit=True,
|
||||||
|
|
||||||
|
) as (ctx, sent):
|
||||||
|
|
||||||
|
assert sent is None
|
||||||
|
|
||||||
|
with trio.fail_after(0.5):
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
# should fall through since ``StopAsyncIteration``
|
||||||
|
# should be raised through translation of
|
||||||
|
# a ``trio.EndOfChannel`` by
|
||||||
|
# ``trio.abc.ReceiveChannel.__anext__()``
|
||||||
|
async for _ in stream:
|
||||||
|
assert 0
|
||||||
|
else:
|
||||||
|
|
||||||
|
# verify stream is now closed
|
||||||
|
try:
|
||||||
|
await stream.receive()
|
||||||
|
except trio.EndOfChannel:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# TODO: should be just raise the closed resource err
|
||||||
|
# directly here to enforce not allowing a re-open
|
||||||
|
# of a stream to the context (at least until a time of
|
||||||
|
# if/when we decide that's a good idea?)
|
||||||
|
try:
|
||||||
|
with trio.fail_after(0.5):
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
pass
|
||||||
|
except trio.ClosedResourceError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def expect_cancelled(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
global _state
|
||||||
|
_state = True
|
||||||
|
|
||||||
|
await ctx.started()
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
async for msg in stream:
|
||||||
|
await stream.send(msg) # echo server
|
||||||
|
|
||||||
|
except trio.Cancelled:
|
||||||
|
# expected case
|
||||||
|
_state = False
|
||||||
|
raise
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert 0, "Wasn't cancelled!?"
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'use_ctx_cancel_method',
|
||||||
|
[False, True],
|
||||||
|
)
|
||||||
|
@tractor_test
|
||||||
|
async def test_caller_closes_ctx_after_callee_opens_stream(
|
||||||
|
use_ctx_cancel_method: bool,
|
||||||
|
):
|
||||||
|
'caller context closes without using stream'
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'ctx_cancelled',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
expect_cancelled,
|
||||||
|
) as (ctx, sent):
|
||||||
|
await portal.run(assert_state, value=True)
|
||||||
|
|
||||||
|
assert sent is None
|
||||||
|
|
||||||
|
# call cancel explicitly
|
||||||
|
if use_ctx_cancel_method:
|
||||||
|
|
||||||
|
await ctx.cancel()
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
async for msg in stream:
|
||||||
|
pass
|
||||||
|
|
||||||
|
except tractor.ContextCancelled:
|
||||||
|
raise # XXX: must be propagated to __aexit__
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert 0, "Should have context cancelled?"
|
||||||
|
|
||||||
|
# channel should still be up
|
||||||
|
assert portal.channel.connected()
|
||||||
|
|
||||||
|
# ctx is closed here
|
||||||
|
await portal.run(assert_state, value=False)
|
||||||
|
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
with trio.fail_after(0.2):
|
||||||
|
await ctx.result()
|
||||||
|
assert 0, "Callee should have blocked!?"
|
||||||
|
except trio.TooSlowError:
|
||||||
|
await ctx.cancel()
|
||||||
|
try:
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
async for msg in stream:
|
||||||
|
pass
|
||||||
|
except tractor.ContextCancelled:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
assert 0, "Should have received closed resource error?"
|
||||||
|
|
||||||
|
# ctx is closed here
|
||||||
|
await portal.run(assert_state, value=False)
|
||||||
|
|
||||||
|
# channel should not have been destroyed yet, only the
|
||||||
|
# inter-actor-task context
|
||||||
|
assert portal.channel.connected()
|
||||||
|
|
||||||
|
# teardown the actor
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
@tractor_test
|
||||||
|
async def test_multitask_caller_cancels_from_nonroot_task():
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'ctx_cancelled',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
expect_cancelled,
|
||||||
|
) as (ctx, sent):
|
||||||
|
|
||||||
|
await portal.run(assert_state, value=True)
|
||||||
|
assert sent is None
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
async def send_msg_then_cancel():
|
||||||
|
await stream.send('yo')
|
||||||
|
await portal.run(assert_state, value=True)
|
||||||
|
await ctx.cancel()
|
||||||
|
await portal.run(assert_state, value=False)
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
n.start_soon(send_msg_then_cancel)
|
||||||
|
|
||||||
|
try:
|
||||||
|
async for msg in stream:
|
||||||
|
assert msg == 'yo'
|
||||||
|
|
||||||
|
except tractor.ContextCancelled:
|
||||||
|
raise # XXX: must be propagated to __aexit__
|
||||||
|
|
||||||
|
# channel should still be up
|
||||||
|
assert portal.channel.connected()
|
||||||
|
|
||||||
|
# ctx is closed here
|
||||||
|
await portal.run(assert_state, value=False)
|
||||||
|
|
||||||
|
# channel should not have been destroyed yet, only the
|
||||||
|
# inter-actor-task context
|
||||||
|
assert portal.channel.connected()
|
||||||
|
|
||||||
|
# teardown the actor
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def cancel_self(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
global _state
|
||||||
|
_state = True
|
||||||
|
|
||||||
|
await ctx.cancel()
|
||||||
|
|
||||||
|
# should inline raise immediately
|
||||||
|
try:
|
||||||
|
async with ctx.open_stream():
|
||||||
|
pass
|
||||||
|
except tractor.ContextCancelled:
|
||||||
|
# suppress for now so we can do checkpoint tests below
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
raise RuntimeError('Context didnt cancel itself?!')
|
||||||
|
|
||||||
|
# check a real ``trio.Cancelled`` is raised on a checkpoint
|
||||||
|
try:
|
||||||
|
with trio.fail_after(0.1):
|
||||||
|
await trio.sleep_forever()
|
||||||
|
except trio.Cancelled:
|
||||||
|
raise
|
||||||
|
|
||||||
|
except trio.TooSlowError:
|
||||||
|
# should never get here
|
||||||
|
assert 0
|
||||||
|
|
||||||
|
|
||||||
|
@tractor_test
|
||||||
|
async def test_callee_cancels_before_started():
|
||||||
|
'''
|
||||||
|
Callee calls `Context.cancel()` while streaming and caller
|
||||||
|
sees stream terminated in `ContextCancelled`.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'cancels_self',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
cancel_self,
|
||||||
|
) as (ctx, sent):
|
||||||
|
async with ctx.open_stream():
|
||||||
|
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
# raises a special cancel signal
|
||||||
|
except tractor.ContextCancelled as ce:
|
||||||
|
ce.type == trio.Cancelled
|
||||||
|
|
||||||
|
# the traceback should be informative
|
||||||
|
assert 'cancelled itself' in ce.msgdata['tb_str']
|
||||||
|
|
||||||
|
# teardown the actor
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def never_open_stream(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Context which never opens a stream and blocks.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def keep_sending_from_callee(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
msg_buffer_size: Optional[int] = None,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Send endlessly on the calleee stream.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
async with ctx.open_stream(
|
||||||
|
msg_buffer_size=msg_buffer_size,
|
||||||
|
) as stream:
|
||||||
|
for msg in count():
|
||||||
|
print(f'callee sending {msg}')
|
||||||
|
await stream.send(msg)
|
||||||
|
await trio.sleep(0.01)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'overrun_by',
|
||||||
|
[
|
||||||
|
('caller', 1, never_open_stream),
|
||||||
|
('cancel_caller_during_overrun', 1, never_open_stream),
|
||||||
|
('callee', 0, keep_sending_from_callee),
|
||||||
|
],
|
||||||
|
ids='overrun_condition={}'.format,
|
||||||
|
)
|
||||||
|
def test_one_end_stream_not_opened(overrun_by):
|
||||||
|
'''
|
||||||
|
This should exemplify the bug from:
|
||||||
|
https://github.com/goodboy/tractor/issues/265
|
||||||
|
|
||||||
|
'''
|
||||||
|
overrunner, buf_size_increase, entrypoint = overrun_by
|
||||||
|
from tractor._runtime import Actor
|
||||||
|
buf_size = buf_size_increase + Actor.msg_buffer_size
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.start_actor(
|
||||||
|
entrypoint.__name__,
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
entrypoint,
|
||||||
|
) as (ctx, sent):
|
||||||
|
assert sent is None
|
||||||
|
|
||||||
|
if 'caller' in overrunner:
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
for i in range(buf_size):
|
||||||
|
print(f'sending {i}')
|
||||||
|
await stream.send(i)
|
||||||
|
|
||||||
|
if 'cancel' in overrunner:
|
||||||
|
# without this we block waiting on the child side
|
||||||
|
await ctx.cancel()
|
||||||
|
|
||||||
|
else:
|
||||||
|
# expect overrun error to be relayed back
|
||||||
|
# and this sleep interrupted
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
else:
|
||||||
|
# callee overruns caller case so we do nothing here
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
# 2 overrun cases and the no overrun case (which pushes right up to
|
||||||
|
# the msg limit)
|
||||||
|
if overrunner == 'caller' or 'cance' in overrunner:
|
||||||
|
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
assert excinfo.value.type == StreamOverrun
|
||||||
|
|
||||||
|
elif overrunner == 'callee':
|
||||||
|
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
# TODO: embedded remote errors so that we can verify the source
|
||||||
|
# error? the callee delivers an error which is an overrun
|
||||||
|
# wrapped in a remote actor error.
|
||||||
|
assert excinfo.value.type == tractor.RemoteActorError
|
||||||
|
|
||||||
|
else:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def echo_back_sequence(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
seq: list[int],
|
||||||
|
msg_buffer_size: Optional[int] = None,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Send endlessly on the calleee stream.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
async with ctx.open_stream(
|
||||||
|
msg_buffer_size=msg_buffer_size,
|
||||||
|
) as stream:
|
||||||
|
|
||||||
|
seq = list(seq) # bleh, `msgpack`...
|
||||||
|
count = 0
|
||||||
|
while count < 3:
|
||||||
|
batch = []
|
||||||
|
async for msg in stream:
|
||||||
|
batch.append(msg)
|
||||||
|
if batch == seq:
|
||||||
|
break
|
||||||
|
|
||||||
|
for msg in batch:
|
||||||
|
print(f'callee sending {msg}')
|
||||||
|
await stream.send(msg)
|
||||||
|
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
return 'yo'
|
||||||
|
|
||||||
|
|
||||||
|
def test_stream_backpressure():
|
||||||
|
'''
|
||||||
|
Demonstrate small overruns of each task back and forth
|
||||||
|
on a stream not raising any errors by default.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'callee_sends_forever',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
seq = list(range(3))
|
||||||
|
async with portal.open_context(
|
||||||
|
echo_back_sequence,
|
||||||
|
seq=seq,
|
||||||
|
msg_buffer_size=1,
|
||||||
|
) as (ctx, sent):
|
||||||
|
assert sent is None
|
||||||
|
|
||||||
|
async with ctx.open_stream(msg_buffer_size=1) as stream:
|
||||||
|
count = 0
|
||||||
|
while count < 3:
|
||||||
|
for msg in seq:
|
||||||
|
print(f'caller sending {msg}')
|
||||||
|
await stream.send(msg)
|
||||||
|
await trio.sleep(0.1)
|
||||||
|
|
||||||
|
batch = []
|
||||||
|
async for msg in stream:
|
||||||
|
batch.append(msg)
|
||||||
|
if batch == seq:
|
||||||
|
break
|
||||||
|
|
||||||
|
count += 1
|
||||||
|
|
||||||
|
# here the context should return
|
||||||
|
assert await ctx.result() == 'yo'
|
||||||
|
|
||||||
|
# cancel the daemon
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def sleep_forever(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
) -> None:
|
||||||
|
await ctx.started()
|
||||||
|
async with ctx.open_stream():
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def attach_to_sleep_forever():
|
||||||
|
'''
|
||||||
|
Cancel a context **before** any underlying error is raised in order
|
||||||
|
to trigger a local reception of a ``ContextCancelled`` which **should not**
|
||||||
|
be re-raised in the local surrounding ``Context`` *iff* the cancel was
|
||||||
|
requested by **this** side of the context.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async with tractor.wait_for_actor('sleeper') as p2:
|
||||||
|
async with (
|
||||||
|
p2.open_context(sleep_forever) as (peer_ctx, first),
|
||||||
|
peer_ctx.open_stream(),
|
||||||
|
):
|
||||||
|
try:
|
||||||
|
yield
|
||||||
|
finally:
|
||||||
|
# XXX: previously this would trigger local
|
||||||
|
# ``ContextCancelled`` to be received and raised in the
|
||||||
|
# local context overriding any local error due to
|
||||||
|
# logic inside ``_invoke()`` which checked for
|
||||||
|
# an error set on ``Context._error`` and raised it in
|
||||||
|
# under a cancellation scenario.
|
||||||
|
|
||||||
|
# The problem is you can have a remote cancellation
|
||||||
|
# that is part of a local error and we shouldn't raise
|
||||||
|
# ``ContextCancelled`` **iff** we weren't the side of
|
||||||
|
# the context to initiate it, i.e.
|
||||||
|
# ``Context._cancel_called`` should **NOT** have been
|
||||||
|
# set. The special logic to handle this case is now
|
||||||
|
# inside ``Context._may_raise_from_remote_msg()`` XD
|
||||||
|
await peer_ctx.cancel()
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def error_before_started(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
This simulates exactly an original bug discovered in:
|
||||||
|
https://github.com/pikers/piker/issues/244
|
||||||
|
|
||||||
|
'''
|
||||||
|
async with attach_to_sleep_forever():
|
||||||
|
# send an unserializable type which should raise a type error
|
||||||
|
# here and **NOT BE SWALLOWED** by the surrounding acm!!?!
|
||||||
|
await ctx.started(object())
|
||||||
|
|
||||||
|
|
||||||
|
def test_do_not_swallow_error_before_started_by_remote_contextcancelled():
|
||||||
|
'''
|
||||||
|
Verify that an error raised in a remote context which itself opens another
|
||||||
|
remote context, which it cancels, does not ovverride the original error that
|
||||||
|
caused the cancellation of the secondardy context.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.start_actor(
|
||||||
|
'errorer',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
await n.start_actor(
|
||||||
|
'sleeper',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with (
|
||||||
|
portal.open_context(
|
||||||
|
error_before_started
|
||||||
|
) as (ctx, sent),
|
||||||
|
):
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
with pytest.raises(tractor.RemoteActorError) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
assert excinfo.value.type == TypeError
|
File diff suppressed because it is too large
Load Diff
|
@ -20,8 +20,11 @@ async def test_reg_then_unreg(arb_addr):
|
||||||
assert actor.is_arbiter
|
assert actor.is_arbiter
|
||||||
assert len(actor._registry) == 1 # only self is registered
|
assert len(actor._registry) == 1 # only self is registered
|
||||||
|
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
portal = await n.start_actor('actor', rpc_module_paths=[__name__])
|
arbiter_addr=arb_addr,
|
||||||
|
) as n:
|
||||||
|
|
||||||
|
portal = await n.start_actor('actor', enable_modules=[__name__])
|
||||||
uid = portal.channel.uid
|
uid = portal.channel.uid
|
||||||
|
|
||||||
async with tractor.get_arbiter(*arb_addr) as aportal:
|
async with tractor.get_arbiter(*arb_addr) as aportal:
|
||||||
|
@ -39,7 +42,7 @@ async def test_reg_then_unreg(arb_addr):
|
||||||
|
|
||||||
await trio.sleep(0.1)
|
await trio.sleep(0.1)
|
||||||
assert uid not in aportal.actor._registry
|
assert uid not in aportal.actor._registry
|
||||||
sockaddrs = actor._registry[uid]
|
sockaddrs = actor._registry.get(uid)
|
||||||
assert not sockaddrs
|
assert not sockaddrs
|
||||||
|
|
||||||
|
|
||||||
|
@ -66,7 +69,7 @@ async def say_hello_use_wait(other_actor):
|
||||||
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
@pytest.mark.parametrize('func', [say_hello, say_hello_use_wait])
|
@pytest.mark.parametrize('func', [say_hello, say_hello_use_wait])
|
||||||
async def test_trynamic_trio(func, start_method):
|
async def test_trynamic_trio(func, start_method, arb_addr):
|
||||||
"""Main tractor entry point, the "master" process (for now
|
"""Main tractor entry point, the "master" process (for now
|
||||||
acts as the "director").
|
acts as the "director").
|
||||||
"""
|
"""
|
||||||
|
@ -108,49 +111,69 @@ async def cancel(use_signal, delay=0):
|
||||||
|
|
||||||
|
|
||||||
async def stream_from(portal):
|
async def stream_from(portal):
|
||||||
async for value in await portal.result():
|
async with portal.open_stream_from(stream_forever) as stream:
|
||||||
|
async for value in stream:
|
||||||
print(value)
|
print(value)
|
||||||
|
|
||||||
|
|
||||||
|
async def unpack_reg(actor_or_portal):
|
||||||
|
'''
|
||||||
|
Get and unpack a "registry" RPC request from the "arbiter" registry
|
||||||
|
system.
|
||||||
|
|
||||||
|
'''
|
||||||
|
if getattr(actor_or_portal, 'get_registry', None):
|
||||||
|
msg = await actor_or_portal.get_registry()
|
||||||
|
else:
|
||||||
|
msg = await actor_or_portal.run_from_ns('self', 'get_registry')
|
||||||
|
|
||||||
|
return {tuple(key.split('.')): val for key, val in msg.items()}
|
||||||
|
|
||||||
|
|
||||||
async def spawn_and_check_registry(
|
async def spawn_and_check_registry(
|
||||||
arb_addr: tuple,
|
arb_addr: tuple,
|
||||||
use_signal: bool,
|
use_signal: bool,
|
||||||
remote_arbiter: bool = False,
|
remote_arbiter: bool = False,
|
||||||
with_streaming: bool = False,
|
with_streaming: bool = False,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
):
|
||||||
|
async with tractor.get_arbiter(*arb_addr) as portal:
|
||||||
|
# runtime needs to be up to call this
|
||||||
actor = tractor.current_actor()
|
actor = tractor.current_actor()
|
||||||
|
|
||||||
if remote_arbiter:
|
if remote_arbiter:
|
||||||
assert not actor.is_arbiter
|
assert not actor.is_arbiter
|
||||||
|
|
||||||
async with tractor.get_arbiter(*arb_addr) as portal:
|
|
||||||
|
|
||||||
if actor.is_arbiter:
|
if actor.is_arbiter:
|
||||||
|
|
||||||
async def get_reg():
|
|
||||||
return actor._registry
|
|
||||||
|
|
||||||
extra = 1 # arbiter is local root actor
|
extra = 1 # arbiter is local root actor
|
||||||
|
get_reg = partial(unpack_reg, actor)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
get_reg = partial(portal.run_from_ns, 'self', 'get_registry')
|
get_reg = partial(unpack_reg, portal)
|
||||||
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 = await get_reg()
|
registry = await get_reg()
|
||||||
assert actor.uid in registry
|
assert actor.uid in registry
|
||||||
|
|
||||||
if with_streaming:
|
|
||||||
to_run = stream_forever
|
|
||||||
else:
|
|
||||||
to_run = trio.sleep_forever
|
|
||||||
|
|
||||||
async with trio.open_nursery() as trion:
|
|
||||||
try:
|
try:
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery() as n:
|
||||||
|
async with 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}'
|
||||||
portals[name] = await n.run_in_actor(to_run, name=name)
|
if with_streaming:
|
||||||
|
portals[name] = await n.start_actor(
|
||||||
|
name=name, enable_modules=[__name__])
|
||||||
|
|
||||||
|
else: # no streaming
|
||||||
|
portals[name] = await n.run_in_actor(
|
||||||
|
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):
|
||||||
|
@ -171,13 +194,12 @@ async def spawn_and_check_registry(
|
||||||
trion.start_soon(cancel, use_signal, 1)
|
trion.start_soon(cancel, use_signal, 1)
|
||||||
|
|
||||||
last_p = pts[-1]
|
last_p = pts[-1]
|
||||||
async for value in await last_p.result():
|
await stream_from(last_p)
|
||||||
print(value)
|
|
||||||
else:
|
else:
|
||||||
await cancel(use_signal)
|
await cancel(use_signal)
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
with trio.CancelScope(shield=True):
|
|
||||||
await trio.sleep(0.5)
|
await trio.sleep(0.5)
|
||||||
|
|
||||||
# all subactors should have de-registered
|
# all subactors should have de-registered
|
||||||
|
@ -198,7 +220,7 @@ def test_subactors_unregister_on_cancel(
|
||||||
deregistering themselves with the arbiter.
|
deregistering themselves with the arbiter.
|
||||||
"""
|
"""
|
||||||
with pytest.raises(KeyboardInterrupt):
|
with pytest.raises(KeyboardInterrupt):
|
||||||
tractor.run(
|
trio.run(
|
||||||
partial(
|
partial(
|
||||||
spawn_and_check_registry,
|
spawn_and_check_registry,
|
||||||
arb_addr,
|
arb_addr,
|
||||||
|
@ -206,7 +228,6 @@ def test_subactors_unregister_on_cancel(
|
||||||
remote_arbiter=False,
|
remote_arbiter=False,
|
||||||
with_streaming=with_streaming,
|
with_streaming=with_streaming,
|
||||||
),
|
),
|
||||||
arbiter_addr=arb_addr
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@ -224,7 +245,7 @@ def test_subactors_unregister_on_cancel_remote_daemon(
|
||||||
tree) arbiter.
|
tree) arbiter.
|
||||||
"""
|
"""
|
||||||
with pytest.raises(KeyboardInterrupt):
|
with pytest.raises(KeyboardInterrupt):
|
||||||
tractor.run(
|
trio.run(
|
||||||
partial(
|
partial(
|
||||||
spawn_and_check_registry,
|
spawn_and_check_registry,
|
||||||
arb_addr,
|
arb_addr,
|
||||||
|
@ -232,8 +253,6 @@ def test_subactors_unregister_on_cancel_remote_daemon(
|
||||||
remote_arbiter=True,
|
remote_arbiter=True,
|
||||||
with_streaming=with_streaming,
|
with_streaming=with_streaming,
|
||||||
),
|
),
|
||||||
# XXX: required to use remote daemon!
|
|
||||||
arbiter_addr=arb_addr
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@ -255,20 +274,27 @@ async def close_chans_before_nursery(
|
||||||
else:
|
else:
|
||||||
entries_at_end = 1
|
entries_at_end = 1
|
||||||
|
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
):
|
||||||
async with tractor.get_arbiter(*arb_addr) as aportal:
|
async with tractor.get_arbiter(*arb_addr) as aportal:
|
||||||
try:
|
try:
|
||||||
get_reg = partial(aportal.run_from_ns, 'self', 'get_registry')
|
get_reg = partial(unpack_reg, aportal)
|
||||||
|
|
||||||
async with tractor.open_nursery() as tn:
|
async with tractor.open_nursery() as tn:
|
||||||
portal1 = await tn.run_in_actor(
|
portal1 = await tn.start_actor(
|
||||||
stream_forever,
|
name='consumer1', enable_modules=[__name__])
|
||||||
name='consumer1',
|
portal2 = await tn.start_actor(
|
||||||
)
|
'consumer2', enable_modules=[__name__])
|
||||||
agen1 = await portal1.result()
|
|
||||||
|
|
||||||
portal2 = await tn.start_actor('consumer2', rpc_module_paths=[__name__])
|
|
||||||
agen2 = await portal2.run(stream_forever)
|
|
||||||
|
|
||||||
|
# TODO: compact this back as was in last commit once
|
||||||
|
# 3.9+, see https://github.com/goodboy/tractor/issues/207
|
||||||
|
async with portal1.open_stream_from(
|
||||||
|
stream_forever
|
||||||
|
) as agen1:
|
||||||
|
async with portal2.open_stream_from(
|
||||||
|
stream_forever
|
||||||
|
) as agen2:
|
||||||
async with trio.open_nursery() as n:
|
async with trio.open_nursery() as n:
|
||||||
n.start_soon(streamer, agen1)
|
n.start_soon(streamer, agen1)
|
||||||
n.start_soon(cancel, use_signal, .5)
|
n.start_soon(cancel, use_signal, .5)
|
||||||
|
@ -281,15 +307,16 @@ async def close_chans_before_nursery(
|
||||||
# reliably triggered by an external SIGINT.
|
# reliably triggered by an external SIGINT.
|
||||||
# tractor.current_actor()._root_nursery.cancel_scope.cancel()
|
# tractor.current_actor()._root_nursery.cancel_scope.cancel()
|
||||||
|
|
||||||
# XXX: THIS IS THE KEY THING that happens
|
# XXX: THIS IS THE KEY THING that
|
||||||
# **before** exiting the actor nursery block
|
# happens **before** exiting the
|
||||||
|
# actor nursery block
|
||||||
|
|
||||||
# also kill off channels cuz why not
|
# also kill off channels cuz why not
|
||||||
await agen1.aclose()
|
await agen1.aclose()
|
||||||
await agen2.aclose()
|
await agen2.aclose()
|
||||||
finally:
|
finally:
|
||||||
with trio.CancelScope(shield=True):
|
with trio.CancelScope(shield=True):
|
||||||
await trio.sleep(.5)
|
await trio.sleep(1)
|
||||||
|
|
||||||
# all subactors should have de-registered
|
# all subactors should have de-registered
|
||||||
registry = await get_reg()
|
registry = await get_reg()
|
||||||
|
@ -309,15 +336,13 @@ def test_close_channel_explicit(
|
||||||
results in subactor(s) deregistering from the arbiter.
|
results in subactor(s) deregistering from the arbiter.
|
||||||
"""
|
"""
|
||||||
with pytest.raises(KeyboardInterrupt):
|
with pytest.raises(KeyboardInterrupt):
|
||||||
tractor.run(
|
trio.run(
|
||||||
partial(
|
partial(
|
||||||
close_chans_before_nursery,
|
close_chans_before_nursery,
|
||||||
arb_addr,
|
arb_addr,
|
||||||
use_signal,
|
use_signal,
|
||||||
remote_arbiter=False,
|
remote_arbiter=False,
|
||||||
),
|
),
|
||||||
# XXX: required to use remote daemon!
|
|
||||||
arbiter_addr=arb_addr
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@ -333,13 +358,11 @@ def test_close_channel_explicit_remote_arbiter(
|
||||||
results in subactor(s) deregistering from the arbiter.
|
results in subactor(s) deregistering from the arbiter.
|
||||||
"""
|
"""
|
||||||
with pytest.raises(KeyboardInterrupt):
|
with pytest.raises(KeyboardInterrupt):
|
||||||
tractor.run(
|
trio.run(
|
||||||
partial(
|
partial(
|
||||||
close_chans_before_nursery,
|
close_chans_before_nursery,
|
||||||
arb_addr,
|
arb_addr,
|
||||||
use_signal,
|
use_signal,
|
||||||
remote_arbiter=True,
|
remote_arbiter=True,
|
||||||
),
|
),
|
||||||
# XXX: required to use remote daemon!
|
|
||||||
arbiter_addr=arb_addr
|
|
||||||
)
|
)
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
"""
|
'''
|
||||||
Let's make sure them docs work yah?
|
Let's make sure them docs work yah?
|
||||||
"""
|
|
||||||
|
'''
|
||||||
from contextlib import contextmanager
|
from contextlib import contextmanager
|
||||||
import itertools
|
import itertools
|
||||||
import os
|
import os
|
||||||
|
@ -11,17 +12,17 @@ import shutil
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from conftest import repodir
|
from conftest import (
|
||||||
|
examples_dir,
|
||||||
|
)
|
||||||
def examples_dir():
|
|
||||||
"""Return the abspath to the examples directory.
|
|
||||||
"""
|
|
||||||
return os.path.join(repodir(), 'examples')
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def run_example_in_subproc(loglevel, testdir, arb_addr):
|
def run_example_in_subproc(
|
||||||
|
loglevel: str,
|
||||||
|
testdir,
|
||||||
|
arb_addr: tuple[str, int],
|
||||||
|
):
|
||||||
|
|
||||||
@contextmanager
|
@contextmanager
|
||||||
def run(script_code):
|
def run(script_code):
|
||||||
|
@ -31,8 +32,8 @@ def run_example_in_subproc(loglevel, testdir, arb_addr):
|
||||||
# on windows we need to create a special __main__.py which will
|
# on windows we need to create a special __main__.py which will
|
||||||
# be executed with ``python -m <modulename>`` on windows..
|
# be executed with ``python -m <modulename>`` on windows..
|
||||||
shutil.copyfile(
|
shutil.copyfile(
|
||||||
os.path.join(examples_dir(), '__main__.py'),
|
examples_dir() / '__main__.py',
|
||||||
os.path.join(str(testdir), '__main__.py')
|
str(testdir / '__main__.py'),
|
||||||
)
|
)
|
||||||
|
|
||||||
# drop the ``if __name__ == '__main__'`` guard onwards from
|
# drop the ``if __name__ == '__main__'`` guard onwards from
|
||||||
|
@ -61,10 +62,11 @@ def run_example_in_subproc(loglevel, testdir, arb_addr):
|
||||||
str(script_file),
|
str(script_file),
|
||||||
]
|
]
|
||||||
|
|
||||||
|
# XXX: BE FOREVER WARNED: if you enable lots of tractor logging
|
||||||
|
# in the subprocess it may cause infinite blocking on the pipes
|
||||||
|
# due to backpressure!!!
|
||||||
proc = testdir.popen(
|
proc = testdir.popen(
|
||||||
cmdargs,
|
cmdargs,
|
||||||
stdout=subprocess.PIPE,
|
|
||||||
stderr=subprocess.PIPE,
|
|
||||||
**kwargs,
|
**kwargs,
|
||||||
)
|
)
|
||||||
assert not proc.returncode
|
assert not proc.returncode
|
||||||
|
@ -77,13 +79,19 @@ def run_example_in_subproc(loglevel, testdir, arb_addr):
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'example_script',
|
'example_script',
|
||||||
|
|
||||||
|
# walk yields: (dirpath, dirnames, filenames)
|
||||||
[
|
[
|
||||||
f for f in os.listdir(examples_dir())
|
(p[0], f) for p in os.walk(examples_dir()) for f in p[2]
|
||||||
if (
|
|
||||||
('__' not in f) and
|
if '__' not in f
|
||||||
('debugging' not in f)
|
and f[0] != '_'
|
||||||
)
|
and 'debugging' not in p[0]
|
||||||
|
and 'integration' not in p[0]
|
||||||
|
and 'advanced_faults' not in p[0]
|
||||||
],
|
],
|
||||||
|
|
||||||
|
ids=lambda t: t[1],
|
||||||
)
|
)
|
||||||
def test_example(run_example_in_subproc, example_script):
|
def test_example(run_example_in_subproc, example_script):
|
||||||
"""Load and run scripts from this repo's ``examples/`` dir as a user
|
"""Load and run scripts from this repo's ``examples/`` dir as a user
|
||||||
|
@ -94,18 +102,34 @@ def test_example(run_example_in_subproc, example_script):
|
||||||
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(examples_dir(), example_script)
|
ex_file = os.path.join(*example_script)
|
||||||
|
|
||||||
|
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")
|
||||||
|
|
||||||
with open(ex_file, 'r') as ex:
|
with open(ex_file, 'r') as ex:
|
||||||
code = ex.read()
|
code = ex.read()
|
||||||
|
|
||||||
with run_example_in_subproc(code) as proc:
|
with run_example_in_subproc(code) as proc:
|
||||||
proc.wait()
|
proc.wait()
|
||||||
err, _ = proc.stderr.read(), proc.stdout.read()
|
err, _ = proc.stderr.read(), proc.stdout.read()
|
||||||
|
# print(f'STDERR: {err}')
|
||||||
|
# print(f'STDOUT: {out}')
|
||||||
|
|
||||||
# if we get some gnarly output let's aggregate and raise
|
# if we get some gnarly output let's aggregate and raise
|
||||||
|
if err:
|
||||||
errmsg = err.decode()
|
errmsg = err.decode()
|
||||||
errlines = errmsg.splitlines()
|
errlines = errmsg.splitlines()
|
||||||
if err and 'Error' in errlines[-1]:
|
last_error = errlines[-1]
|
||||||
|
if (
|
||||||
|
'Error' in last_error
|
||||||
|
|
||||||
|
# XXX: currently we print this to console, but maybe
|
||||||
|
# shouldn't eventually once we figure out what's
|
||||||
|
# a better way to be explicit about aio side
|
||||||
|
# cancels?
|
||||||
|
and 'asyncio.exceptions.CancelledError' not in last_error
|
||||||
|
):
|
||||||
raise Exception(errmsg)
|
raise Exception(errmsg)
|
||||||
|
|
||||||
assert proc.returncode == 0
|
assert proc.returncode == 0
|
||||||
|
|
|
@ -0,0 +1,564 @@
|
||||||
|
'''
|
||||||
|
The hipster way to force SC onto the stdlib's "async": 'infection mode'.
|
||||||
|
|
||||||
|
'''
|
||||||
|
from typing import Optional, Iterable, Union
|
||||||
|
import asyncio
|
||||||
|
import builtins
|
||||||
|
import itertools
|
||||||
|
import importlib
|
||||||
|
|
||||||
|
from exceptiongroup import BaseExceptionGroup
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
from tractor import (
|
||||||
|
to_asyncio,
|
||||||
|
RemoteActorError,
|
||||||
|
)
|
||||||
|
from tractor.trionics import BroadcastReceiver
|
||||||
|
|
||||||
|
|
||||||
|
async def sleep_and_err(
|
||||||
|
sleep_for: float = 0.1,
|
||||||
|
|
||||||
|
# just signature placeholders for compat with
|
||||||
|
# ``to_asyncio.open_channel_from()``
|
||||||
|
to_trio: Optional[trio.MemorySendChannel] = None,
|
||||||
|
from_trio: Optional[asyncio.Queue] = None,
|
||||||
|
|
||||||
|
):
|
||||||
|
if to_trio:
|
||||||
|
to_trio.send_nowait('start')
|
||||||
|
|
||||||
|
await asyncio.sleep(sleep_for)
|
||||||
|
assert 0
|
||||||
|
|
||||||
|
|
||||||
|
async def sleep_forever():
|
||||||
|
await asyncio.sleep(float('inf'))
|
||||||
|
|
||||||
|
|
||||||
|
async def trio_cancels_single_aio_task():
|
||||||
|
|
||||||
|
# spawn an ``asyncio`` task to run a func and return result
|
||||||
|
with trio.move_on_after(.2):
|
||||||
|
await tractor.to_asyncio.run_task(sleep_forever)
|
||||||
|
|
||||||
|
|
||||||
|
def test_trio_cancels_aio_on_actor_side(arb_addr):
|
||||||
|
'''
|
||||||
|
Spawn an infected actor that is cancelled by the ``trio`` side
|
||||||
|
task using std cancel scope apis.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr
|
||||||
|
) as n:
|
||||||
|
await n.run_in_actor(
|
||||||
|
trio_cancels_single_aio_task,
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
async def asyncio_actor(
|
||||||
|
|
||||||
|
target: str,
|
||||||
|
expect_err: Optional[Exception] = None
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
assert tractor.current_actor().is_infected_aio()
|
||||||
|
target = globals()[target]
|
||||||
|
|
||||||
|
if '.' in expect_err:
|
||||||
|
modpath, _, name = expect_err.rpartition('.')
|
||||||
|
mod = importlib.import_module(modpath)
|
||||||
|
error_type = getattr(mod, name)
|
||||||
|
|
||||||
|
else: # toplevel builtin error type
|
||||||
|
error_type = builtins.__dict__.get(expect_err)
|
||||||
|
|
||||||
|
try:
|
||||||
|
# spawn an ``asyncio`` task to run a func and return result
|
||||||
|
await tractor.to_asyncio.run_task(target)
|
||||||
|
|
||||||
|
except BaseException as err:
|
||||||
|
if expect_err:
|
||||||
|
assert isinstance(err, error_type)
|
||||||
|
|
||||||
|
raise
|
||||||
|
|
||||||
|
|
||||||
|
def test_aio_simple_error(arb_addr):
|
||||||
|
'''
|
||||||
|
Verify a simple remote asyncio error propagates back through trio
|
||||||
|
to the parent actor.
|
||||||
|
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr
|
||||||
|
) as n:
|
||||||
|
await n.run_in_actor(
|
||||||
|
asyncio_actor,
|
||||||
|
target='sleep_and_err',
|
||||||
|
expect_err='AssertionError',
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
with pytest.raises(RemoteActorError) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
err = excinfo.value
|
||||||
|
assert isinstance(err, RemoteActorError)
|
||||||
|
assert err.type == AssertionError
|
||||||
|
|
||||||
|
|
||||||
|
def test_tractor_cancels_aio(arb_addr):
|
||||||
|
'''
|
||||||
|
Verify we can cancel a spawned asyncio task gracefully.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.run_in_actor(
|
||||||
|
asyncio_actor,
|
||||||
|
target='sleep_forever',
|
||||||
|
expect_err='trio.Cancelled',
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
# cancel the entire remote runtime
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def test_trio_cancels_aio(arb_addr):
|
||||||
|
'''
|
||||||
|
Much like the above test with ``tractor.Portal.cancel_actor()``
|
||||||
|
except we just use a standard ``trio`` cancellation api.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
with trio.move_on_after(1):
|
||||||
|
# cancel the nursery shortly after boot
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
await n.run_in_actor(
|
||||||
|
asyncio_actor,
|
||||||
|
target='sleep_forever',
|
||||||
|
expect_err='trio.Cancelled',
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def trio_ctx(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
):
|
||||||
|
|
||||||
|
await ctx.started('start')
|
||||||
|
|
||||||
|
# this will block until the ``asyncio`` task sends a "first"
|
||||||
|
# message.
|
||||||
|
with trio.fail_after(2):
|
||||||
|
async with (
|
||||||
|
trio.open_nursery() as n,
|
||||||
|
|
||||||
|
tractor.to_asyncio.open_channel_from(
|
||||||
|
sleep_and_err,
|
||||||
|
) as (first, chan),
|
||||||
|
):
|
||||||
|
|
||||||
|
assert first == 'start'
|
||||||
|
|
||||||
|
# spawn another asyncio task for the cuck of it.
|
||||||
|
n.start_soon(
|
||||||
|
tractor.to_asyncio.run_task,
|
||||||
|
sleep_forever,
|
||||||
|
)
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'parent_cancels', [False, True],
|
||||||
|
ids='parent_actor_cancels_child={}'.format
|
||||||
|
)
|
||||||
|
def test_context_spawns_aio_task_that_errors(
|
||||||
|
arb_addr,
|
||||||
|
parent_cancels: bool,
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Verify that spawning a task via an intertask channel ctx mngr that
|
||||||
|
errors correctly propagates the error back from the `asyncio`-side
|
||||||
|
task.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
with trio.fail_after(2):
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
p = await n.start_actor(
|
||||||
|
'aio_daemon',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
infect_asyncio=True,
|
||||||
|
# debug_mode=True,
|
||||||
|
loglevel='cancel',
|
||||||
|
)
|
||||||
|
async with p.open_context(
|
||||||
|
trio_ctx,
|
||||||
|
) as (ctx, first):
|
||||||
|
|
||||||
|
assert first == 'start'
|
||||||
|
|
||||||
|
if parent_cancels:
|
||||||
|
await p.cancel_actor()
|
||||||
|
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
with pytest.raises(RemoteActorError) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
err = excinfo.value
|
||||||
|
assert isinstance(err, RemoteActorError)
|
||||||
|
if parent_cancels:
|
||||||
|
assert err.type == trio.Cancelled
|
||||||
|
else:
|
||||||
|
assert err.type == AssertionError
|
||||||
|
|
||||||
|
|
||||||
|
async def aio_cancel():
|
||||||
|
''''
|
||||||
|
Cancel urself boi.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await asyncio.sleep(0.5)
|
||||||
|
task = asyncio.current_task()
|
||||||
|
|
||||||
|
# cancel and enter sleep
|
||||||
|
task.cancel()
|
||||||
|
await sleep_forever()
|
||||||
|
|
||||||
|
|
||||||
|
def test_aio_cancelled_from_aio_causes_trio_cancelled(arb_addr):
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
await n.run_in_actor(
|
||||||
|
asyncio_actor,
|
||||||
|
target='aio_cancel',
|
||||||
|
expect_err='tractor.to_asyncio.AsyncioCancelled',
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
with pytest.raises(RemoteActorError) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
# ensure boxed error is correct
|
||||||
|
assert excinfo.value.type == to_asyncio.AsyncioCancelled
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: verify open_channel_from will fail on this..
|
||||||
|
async def no_to_trio_in_args():
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
async def push_from_aio_task(
|
||||||
|
|
||||||
|
sequence: Iterable,
|
||||||
|
to_trio: trio.abc.SendChannel,
|
||||||
|
expect_cancel: False,
|
||||||
|
fail_early: bool,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
try:
|
||||||
|
# sync caller ctx manager
|
||||||
|
to_trio.send_nowait(True)
|
||||||
|
|
||||||
|
for i in sequence:
|
||||||
|
print(f'asyncio sending {i}')
|
||||||
|
to_trio.send_nowait(i)
|
||||||
|
await asyncio.sleep(0.001)
|
||||||
|
|
||||||
|
if i == 50 and fail_early:
|
||||||
|
raise Exception
|
||||||
|
|
||||||
|
print('asyncio streamer complete!')
|
||||||
|
|
||||||
|
except asyncio.CancelledError:
|
||||||
|
if not expect_cancel:
|
||||||
|
pytest.fail("aio task was cancelled unexpectedly")
|
||||||
|
raise
|
||||||
|
else:
|
||||||
|
if expect_cancel:
|
||||||
|
pytest.fail("aio task wasn't cancelled as expected!?")
|
||||||
|
|
||||||
|
|
||||||
|
async def stream_from_aio(
|
||||||
|
|
||||||
|
exit_early: bool = False,
|
||||||
|
raise_err: bool = False,
|
||||||
|
aio_raise_err: bool = False,
|
||||||
|
fan_out: bool = False,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
seq = range(100)
|
||||||
|
expect = list(seq)
|
||||||
|
|
||||||
|
try:
|
||||||
|
pulled = []
|
||||||
|
|
||||||
|
async with to_asyncio.open_channel_from(
|
||||||
|
push_from_aio_task,
|
||||||
|
sequence=seq,
|
||||||
|
expect_cancel=raise_err or exit_early,
|
||||||
|
fail_early=aio_raise_err,
|
||||||
|
) as (first, chan):
|
||||||
|
|
||||||
|
assert first is True
|
||||||
|
|
||||||
|
async def consume(
|
||||||
|
chan: Union[
|
||||||
|
to_asyncio.LinkedTaskChannel,
|
||||||
|
BroadcastReceiver,
|
||||||
|
],
|
||||||
|
):
|
||||||
|
async for value in chan:
|
||||||
|
print(f'trio received {value}')
|
||||||
|
pulled.append(value)
|
||||||
|
|
||||||
|
if value == 50:
|
||||||
|
if raise_err:
|
||||||
|
raise Exception
|
||||||
|
elif exit_early:
|
||||||
|
break
|
||||||
|
|
||||||
|
if fan_out:
|
||||||
|
# start second task that get's the same stream value set.
|
||||||
|
async with (
|
||||||
|
|
||||||
|
# NOTE: this has to come first to avoid
|
||||||
|
# the channel being closed before the nursery
|
||||||
|
# tasks are joined..
|
||||||
|
chan.subscribe() as br,
|
||||||
|
|
||||||
|
trio.open_nursery() as n,
|
||||||
|
):
|
||||||
|
n.start_soon(consume, br)
|
||||||
|
await consume(chan)
|
||||||
|
|
||||||
|
else:
|
||||||
|
await consume(chan)
|
||||||
|
finally:
|
||||||
|
|
||||||
|
if (
|
||||||
|
not raise_err and
|
||||||
|
not exit_early and
|
||||||
|
not aio_raise_err
|
||||||
|
):
|
||||||
|
if fan_out:
|
||||||
|
# we get double the pulled values in the
|
||||||
|
# ``.subscribe()`` fan out case.
|
||||||
|
doubled = list(itertools.chain(*zip(expect, expect)))
|
||||||
|
expect = doubled[:len(pulled)]
|
||||||
|
assert list(sorted(pulled)) == expect
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert pulled == expect
|
||||||
|
else:
|
||||||
|
assert not fan_out
|
||||||
|
assert pulled == expect[:51]
|
||||||
|
|
||||||
|
print('trio guest mode task completed!')
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'fan_out', [False, True],
|
||||||
|
ids='fan_out_w_chan_subscribe={}'.format
|
||||||
|
)
|
||||||
|
def test_basic_interloop_channel_stream(arb_addr, fan_out):
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.run_in_actor(
|
||||||
|
stream_from_aio,
|
||||||
|
infect_asyncio=True,
|
||||||
|
fan_out=fan_out,
|
||||||
|
)
|
||||||
|
await portal.result()
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: parametrize the above test and avoid the duplication here?
|
||||||
|
def test_trio_error_cancels_intertask_chan(arb_addr):
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.run_in_actor(
|
||||||
|
stream_from_aio,
|
||||||
|
raise_err=True,
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
# should trigger remote actor error
|
||||||
|
await portal.result()
|
||||||
|
|
||||||
|
with pytest.raises(BaseExceptionGroup) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
# ensure boxed errors
|
||||||
|
for exc in excinfo.value.exceptions:
|
||||||
|
assert exc.type == Exception
|
||||||
|
|
||||||
|
|
||||||
|
def test_trio_closes_early_and_channel_exits(arb_addr):
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.run_in_actor(
|
||||||
|
stream_from_aio,
|
||||||
|
exit_early=True,
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
# should trigger remote actor error
|
||||||
|
await portal.result()
|
||||||
|
|
||||||
|
# should be a quiet exit on a simple channel exit
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def test_aio_errors_and_channel_propagates_and_closes(arb_addr):
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
portal = await n.run_in_actor(
|
||||||
|
stream_from_aio,
|
||||||
|
aio_raise_err=True,
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
# should trigger remote actor error
|
||||||
|
await portal.result()
|
||||||
|
|
||||||
|
with pytest.raises(BaseExceptionGroup) as excinfo:
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
# ensure boxed errors
|
||||||
|
for exc in excinfo.value.exceptions:
|
||||||
|
assert exc.type == Exception
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def trio_to_aio_echo_server(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
):
|
||||||
|
|
||||||
|
async def aio_echo_server(
|
||||||
|
to_trio: trio.MemorySendChannel,
|
||||||
|
from_trio: asyncio.Queue,
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
to_trio.send_nowait('start')
|
||||||
|
|
||||||
|
while True:
|
||||||
|
msg = await from_trio.get()
|
||||||
|
|
||||||
|
# echo the msg back
|
||||||
|
to_trio.send_nowait(msg)
|
||||||
|
|
||||||
|
# if we get the terminate sentinel
|
||||||
|
# break the echo loop
|
||||||
|
if msg is None:
|
||||||
|
print('breaking aio echo loop')
|
||||||
|
break
|
||||||
|
|
||||||
|
print('exiting asyncio task')
|
||||||
|
|
||||||
|
async with to_asyncio.open_channel_from(
|
||||||
|
aio_echo_server,
|
||||||
|
) as (first, chan):
|
||||||
|
|
||||||
|
assert first == 'start'
|
||||||
|
await ctx.started(first)
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
|
||||||
|
async for msg in stream:
|
||||||
|
print(f'asyncio echoing {msg}')
|
||||||
|
await chan.send(msg)
|
||||||
|
|
||||||
|
out = await chan.receive()
|
||||||
|
# echo back to parent actor-task
|
||||||
|
await stream.send(out)
|
||||||
|
|
||||||
|
if out is None:
|
||||||
|
try:
|
||||||
|
out = await chan.receive()
|
||||||
|
except trio.EndOfChannel:
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
raise RuntimeError('aio channel never stopped?')
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'raise_error_mid_stream',
|
||||||
|
[False, Exception, KeyboardInterrupt],
|
||||||
|
ids='raise_error={}'.format,
|
||||||
|
)
|
||||||
|
def test_echoserver_detailed_mechanics(
|
||||||
|
arb_addr,
|
||||||
|
raise_error_mid_stream,
|
||||||
|
):
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
p = await n.start_actor(
|
||||||
|
'aio_server',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
infect_asyncio=True,
|
||||||
|
)
|
||||||
|
async with p.open_context(
|
||||||
|
trio_to_aio_echo_server,
|
||||||
|
) as (ctx, first):
|
||||||
|
|
||||||
|
assert first == 'start'
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
for i in range(100):
|
||||||
|
await stream.send(i)
|
||||||
|
out = await stream.receive()
|
||||||
|
assert i == out
|
||||||
|
|
||||||
|
if raise_error_mid_stream and i == 50:
|
||||||
|
raise raise_error_mid_stream
|
||||||
|
|
||||||
|
# send terminate msg
|
||||||
|
await stream.send(None)
|
||||||
|
out = await stream.receive()
|
||||||
|
assert out is None
|
||||||
|
|
||||||
|
if out is None:
|
||||||
|
# ensure the stream is stopped
|
||||||
|
# with trio.fail_after(0.1):
|
||||||
|
try:
|
||||||
|
await stream.receive()
|
||||||
|
except trio.EndOfChannel:
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
pytest.fail(
|
||||||
|
"stream wasn't stopped after sentinel?!")
|
||||||
|
|
||||||
|
# TODO: the case where this blocks and
|
||||||
|
# is cancelled by kbi or out of task cancellation
|
||||||
|
await p.cancel_actor()
|
||||||
|
|
||||||
|
if raise_error_mid_stream:
|
||||||
|
with pytest.raises(raise_error_mid_stream):
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
else:
|
||||||
|
trio.run(main)
|
|
@ -7,9 +7,10 @@ import platform
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
from tractor.testing import tractor_test
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
from conftest import tractor_test
|
||||||
|
|
||||||
|
|
||||||
def test_must_define_ctx():
|
def test_must_define_ctx():
|
||||||
|
|
||||||
|
@ -32,13 +33,16 @@ async def async_gen_stream(sequence):
|
||||||
|
|
||||||
# block indefinitely waiting to be cancelled by ``aclose()`` call
|
# block indefinitely waiting to be cancelled by ``aclose()`` call
|
||||||
with trio.CancelScope() as cs:
|
with trio.CancelScope() as cs:
|
||||||
await trio.sleep(float('inf'))
|
await trio.sleep_forever()
|
||||||
assert 0
|
assert 0
|
||||||
assert cs.cancelled_caught
|
assert cs.cancelled_caught
|
||||||
|
|
||||||
|
|
||||||
@tractor.stream
|
@tractor.stream
|
||||||
async def context_stream(ctx, sequence):
|
async def context_stream(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
sequence
|
||||||
|
):
|
||||||
for i in sequence:
|
for i in sequence:
|
||||||
await ctx.send_yield(i)
|
await ctx.send_yield(i)
|
||||||
await trio.sleep(0.1)
|
await trio.sleep(0.1)
|
||||||
|
@ -50,26 +54,38 @@ async def context_stream(ctx, sequence):
|
||||||
assert cs.cancelled_caught
|
assert cs.cancelled_caught
|
||||||
|
|
||||||
|
|
||||||
async def stream_from_single_subactor(stream_func):
|
async def stream_from_single_subactor(
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
|
stream_func,
|
||||||
|
):
|
||||||
"""Verify we can spawn a daemon actor and retrieve streamed data.
|
"""Verify we can spawn a daemon actor and retrieve streamed data.
|
||||||
"""
|
"""
|
||||||
|
# only one per host address, spawns an actor if None
|
||||||
|
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
start_method=start_method,
|
||||||
|
) as nursery:
|
||||||
|
|
||||||
async with tractor.find_actor('streamerd') as portals:
|
async with tractor.find_actor('streamerd') as portals:
|
||||||
|
|
||||||
if not portals:
|
if not portals:
|
||||||
# only one per host address, spawns an actor if None
|
|
||||||
async with tractor.open_nursery() as nursery:
|
|
||||||
# no brokerd actor found
|
# no brokerd actor found
|
||||||
portal = await nursery.start_actor(
|
portal = await nursery.start_actor(
|
||||||
'streamerd',
|
'streamerd',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
seq = range(10)
|
seq = range(10)
|
||||||
|
|
||||||
stream = await portal.run(
|
with trio.fail_after(5):
|
||||||
stream_func, # one of the funcs above
|
async with portal.open_stream_from(
|
||||||
|
stream_func,
|
||||||
sequence=list(seq), # has to be msgpack serializable
|
sequence=list(seq), # has to be msgpack serializable
|
||||||
)
|
) as stream:
|
||||||
|
|
||||||
# it'd sure be nice to have an asyncitertools here...
|
# it'd sure be nice to have an asyncitertools here...
|
||||||
iseq = iter(seq)
|
iseq = iter(seq)
|
||||||
ival = next(iseq)
|
ival = next(iseq)
|
||||||
|
@ -86,12 +102,14 @@ async def stream_from_single_subactor(stream_func):
|
||||||
|
|
||||||
await trio.sleep(0.3)
|
await trio.sleep(0.3)
|
||||||
|
|
||||||
|
# ensure EOC signalled-state translates
|
||||||
|
# XXX: not really sure this is correct,
|
||||||
|
# shouldn't it be a `ClosedResourceError`?
|
||||||
try:
|
try:
|
||||||
await stream.__anext__()
|
await stream.__anext__()
|
||||||
except StopAsyncIteration:
|
except StopAsyncIteration:
|
||||||
# stop all spawned subactors
|
# stop all spawned subactors
|
||||||
await portal.cancel_actor()
|
await portal.cancel_actor()
|
||||||
# await nursery.cancel()
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
|
@ -100,13 +118,13 @@ async def stream_from_single_subactor(stream_func):
|
||||||
def test_stream_from_single_subactor(arb_addr, start_method, stream_func):
|
def test_stream_from_single_subactor(arb_addr, start_method, stream_func):
|
||||||
"""Verify streaming from a spawned async generator.
|
"""Verify streaming from a spawned async generator.
|
||||||
"""
|
"""
|
||||||
tractor.run(
|
trio.run(
|
||||||
partial(
|
partial(
|
||||||
stream_from_single_subactor,
|
stream_from_single_subactor,
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
stream_func=stream_func,
|
stream_func=stream_func,
|
||||||
),
|
),
|
||||||
arbiter_addr=arb_addr,
|
|
||||||
start_method=start_method,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@ -118,7 +136,7 @@ async def stream_data(seed):
|
||||||
yield i
|
yield i
|
||||||
|
|
||||||
# trigger scheduler to simulate practical usage
|
# trigger scheduler to simulate practical usage
|
||||||
await trio.sleep(0)
|
await trio.sleep(0.0001)
|
||||||
|
|
||||||
|
|
||||||
# this is the third actor; the aggregator
|
# this is the third actor; the aggregator
|
||||||
|
@ -132,7 +150,7 @@ async def aggregate(seed):
|
||||||
# fork point
|
# fork point
|
||||||
portal = await nursery.start_actor(
|
portal = await nursery.start_actor(
|
||||||
name=f'streamer_{i}',
|
name=f'streamer_{i}',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
portals.append(portal)
|
portals.append(portal)
|
||||||
|
@ -141,9 +159,12 @@ async def aggregate(seed):
|
||||||
|
|
||||||
async def push_to_chan(portal, send_chan):
|
async def push_to_chan(portal, send_chan):
|
||||||
async with send_chan:
|
async with send_chan:
|
||||||
async for value in await portal.run(
|
|
||||||
__name__, 'stream_data', seed=seed
|
async with portal.open_stream_from(
|
||||||
):
|
stream_data, seed=seed,
|
||||||
|
) as stream:
|
||||||
|
|
||||||
|
async for value in stream:
|
||||||
# leverage trio's built-in backpressure
|
# leverage trio's built-in backpressure
|
||||||
await send_chan.send(value)
|
await send_chan.send(value)
|
||||||
|
|
||||||
|
@ -183,26 +204,29 @@ async def a_quadruple_example():
|
||||||
seed = int(1e3)
|
seed = int(1e3)
|
||||||
pre_start = time.time()
|
pre_start = time.time()
|
||||||
|
|
||||||
portal = await nursery.run_in_actor(
|
portal = await nursery.start_actor(
|
||||||
aggregate,
|
|
||||||
seed=seed,
|
|
||||||
name='aggregator',
|
name='aggregator',
|
||||||
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
start = time.time()
|
start = time.time()
|
||||||
# the portal call returns exactly what you'd expect
|
# the portal call returns exactly what you'd expect
|
||||||
# as if the remote "aggregate" function was called locally
|
# as if the remote "aggregate" function was called locally
|
||||||
result_stream = []
|
result_stream = []
|
||||||
async for value in await portal.result():
|
|
||||||
|
async with portal.open_stream_from(aggregate, seed=seed) as stream:
|
||||||
|
async for value in stream:
|
||||||
result_stream.append(value)
|
result_stream.append(value)
|
||||||
|
|
||||||
print(f"STREAM TIME = {time.time() - start}")
|
print(f"STREAM TIME = {time.time() - start}")
|
||||||
print(f"STREAM + SPAWN TIME = {time.time() - pre_start}")
|
print(f"STREAM + SPAWN TIME = {time.time() - pre_start}")
|
||||||
assert result_stream == list(range(seed))
|
assert result_stream == list(range(seed))
|
||||||
|
await portal.cancel_actor()
|
||||||
return result_stream
|
return result_stream
|
||||||
|
|
||||||
|
|
||||||
async def cancel_after(wait):
|
async def cancel_after(wait, arb_addr):
|
||||||
|
async with tractor.open_root_actor(arbiter_addr=arb_addr):
|
||||||
with trio.move_on_after(wait):
|
with trio.move_on_after(wait):
|
||||||
return await a_quadruple_example()
|
return await a_quadruple_example()
|
||||||
|
|
||||||
|
@ -216,7 +240,7 @@ def time_quad_ex(arb_addr, ci_env, spawn_backend):
|
||||||
|
|
||||||
timeout = 7 if platform.system() in ('Windows', 'Darwin') else 4
|
timeout = 7 if platform.system() in ('Windows', 'Darwin') else 4
|
||||||
start = time.time()
|
start = time.time()
|
||||||
results = tractor.run(cancel_after, timeout, arbiter_addr=arb_addr)
|
results = trio.run(cancel_after, timeout, arb_addr)
|
||||||
diff = time.time() - start
|
diff = time.time() - start
|
||||||
assert results
|
assert results
|
||||||
return results, diff
|
return results, diff
|
||||||
|
@ -227,7 +251,7 @@ def test_a_quadruple_example(time_quad_ex, ci_env, spawn_backend):
|
||||||
|
|
||||||
results, diff = time_quad_ex
|
results, diff = time_quad_ex
|
||||||
assert results
|
assert results
|
||||||
this_fast = 6 if platform.system() in ('Windows', 'Darwin') else 2.5
|
this_fast = 6 if platform.system() in ('Windows', 'Darwin') else 3
|
||||||
assert diff < this_fast
|
assert diff < this_fast
|
||||||
|
|
||||||
|
|
||||||
|
@ -243,7 +267,7 @@ def test_not_fast_enough_quad(
|
||||||
"""
|
"""
|
||||||
results, diff = time_quad_ex
|
results, diff = time_quad_ex
|
||||||
delay = max(diff - cancel_delay, 0)
|
delay = max(diff - cancel_delay, 0)
|
||||||
results = tractor.run(cancel_after, delay, arbiter_addr=arb_addr)
|
results = trio.run(cancel_after, delay, arb_addr)
|
||||||
system = platform.system()
|
system = platform.system()
|
||||||
if system in ('Windows', 'Darwin') and results is not None:
|
if system in ('Windows', 'Darwin') and results is not None:
|
||||||
# In CI envoirments it seems later runs are quicker then the first
|
# In CI envoirments it seems later runs are quicker then the first
|
||||||
|
@ -272,11 +296,14 @@ async def test_respawn_consumer_task(
|
||||||
|
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
stream = await(await n.run_in_actor(
|
portal = await n.start_actor(
|
||||||
|
name='streamer',
|
||||||
|
enable_modules=[__name__]
|
||||||
|
)
|
||||||
|
async with portal.open_stream_from(
|
||||||
stream_data,
|
stream_data,
|
||||||
seed=11,
|
seed=11,
|
||||||
name='streamer',
|
) as stream:
|
||||||
)).result()
|
|
||||||
|
|
||||||
expect = set(range(11))
|
expect = set(range(11))
|
||||||
received = []
|
received = []
|
||||||
|
@ -290,7 +317,7 @@ async def test_respawn_consumer_task(
|
||||||
task_status.started(cs)
|
task_status.started(cs)
|
||||||
|
|
||||||
# shield stream's underlying channel from cancellation
|
# shield stream's underlying channel from cancellation
|
||||||
with stream.shield():
|
# with stream.shield():
|
||||||
|
|
||||||
async for v in stream:
|
async for v in stream:
|
||||||
print(f'from stream: {v}')
|
print(f'from stream: {v}')
|
||||||
|
@ -317,3 +344,9 @@ async def test_respawn_consumer_task(
|
||||||
if not expect:
|
if not expect:
|
||||||
print("all values streamed, BREAKING")
|
print("all values streamed, BREAKING")
|
||||||
break
|
break
|
||||||
|
|
||||||
|
cs.cancel()
|
||||||
|
|
||||||
|
# TODO: this is justification for a
|
||||||
|
# ``ActorNursery.stream_from_actor()`` helper?
|
||||||
|
await portal.cancel_actor()
|
|
@ -11,32 +11,26 @@ from conftest import tractor_test
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.trio
|
@pytest.mark.trio
|
||||||
async def test_no_arbitter():
|
async def test_no_runtime():
|
||||||
"""An arbitter must be established before any nurseries
|
"""An arbitter must be established before any nurseries
|
||||||
can be created.
|
can be created.
|
||||||
|
|
||||||
(In other words ``tractor.run`` must be used instead of ``trio.run`` as is
|
(In other words ``tractor.open_root_actor()`` must be engaged at
|
||||||
done by the ``pytest-trio`` plugin.)
|
some point?)
|
||||||
"""
|
"""
|
||||||
with pytest.raises(RuntimeError) :
|
with pytest.raises(RuntimeError) :
|
||||||
with tractor.open_nursery():
|
async with tractor.find_actor('doggy'):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
def test_no_main():
|
|
||||||
"""An async function **must** be passed to ``tractor.run()``.
|
|
||||||
"""
|
|
||||||
with pytest.raises(TypeError):
|
|
||||||
tractor.run(None)
|
|
||||||
|
|
||||||
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_self_is_registered(arb_addr):
|
async def test_self_is_registered(arb_addr):
|
||||||
"Verify waiting on the arbiter to register itself using the standard api."
|
"Verify waiting on the arbiter to register itself using the standard api."
|
||||||
actor = tractor.current_actor()
|
actor = tractor.current_actor()
|
||||||
assert actor.is_arbiter
|
assert actor.is_arbiter
|
||||||
async with tractor.wait_for_actor('arbiter') as portal:
|
with trio.fail_after(0.2):
|
||||||
assert portal.channel.uid[0] == 'arbiter'
|
async with tractor.wait_for_actor('root') as portal:
|
||||||
|
assert portal.channel.uid[0] == 'root'
|
||||||
|
|
||||||
|
|
||||||
@tractor_test
|
@tractor_test
|
||||||
|
@ -46,7 +40,10 @@ async def test_self_is_registered_localportal(arb_addr):
|
||||||
assert actor.is_arbiter
|
assert actor.is_arbiter
|
||||||
async with tractor.get_arbiter(*arb_addr) as portal:
|
async with tractor.get_arbiter(*arb_addr) as portal:
|
||||||
assert isinstance(portal, tractor._portal.LocalPortal)
|
assert isinstance(portal, tractor._portal.LocalPortal)
|
||||||
sockaddr = await portal.run_from_ns('self', 'wait_for_actor', name='arbiter')
|
|
||||||
|
with trio.fail_after(0.2):
|
||||||
|
sockaddr = await portal.run_from_ns(
|
||||||
|
'self', 'wait_for_actor', name='root')
|
||||||
assert sockaddr[0] == arb_addr
|
assert sockaddr[0] == arb_addr
|
||||||
|
|
||||||
|
|
||||||
|
@ -56,6 +53,10 @@ def test_local_actor_async_func(arb_addr):
|
||||||
nums = []
|
nums = []
|
||||||
|
|
||||||
async def print_loop():
|
async def print_loop():
|
||||||
|
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
):
|
||||||
# arbiter is started in-proc if dne
|
# arbiter is started in-proc if dne
|
||||||
assert tractor.current_actor().is_arbiter
|
assert tractor.current_actor().is_arbiter
|
||||||
|
|
||||||
|
@ -64,7 +65,7 @@ def test_local_actor_async_func(arb_addr):
|
||||||
await trio.sleep(0.1)
|
await trio.sleep(0.1)
|
||||||
|
|
||||||
start = time.time()
|
start = time.time()
|
||||||
tractor.run(print_loop, arbiter_addr=arb_addr)
|
trio.run(print_loop)
|
||||||
|
|
||||||
# ensure the sleeps were actually awaited
|
# ensure the sleeps were actually awaited
|
||||||
assert time.time() - start >= 1
|
assert time.time() - start >= 1
|
||||||
|
|
|
@ -1,10 +1,11 @@
|
||||||
"""
|
"""
|
||||||
Multiple python programs invoking ``tractor.run()``
|
Multiple python programs invoking the runtime.
|
||||||
"""
|
"""
|
||||||
import platform
|
import platform
|
||||||
import time
|
import time
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
from conftest import (
|
from conftest import (
|
||||||
tractor_test,
|
tractor_test,
|
||||||
|
@ -45,8 +46,13 @@ async def test_cancel_remote_arbiter(daemon, arb_addr):
|
||||||
def test_register_duplicate_name(daemon, arb_addr):
|
def test_register_duplicate_name(daemon, arb_addr):
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
) as n:
|
||||||
|
|
||||||
assert not tractor.current_actor().is_arbiter
|
assert not tractor.current_actor().is_arbiter
|
||||||
async with tractor.open_nursery() as n:
|
|
||||||
p1 = await n.start_actor('doggy')
|
p1 = await n.start_actor('doggy')
|
||||||
p2 = await n.start_actor('doggy')
|
p2 = await n.start_actor('doggy')
|
||||||
|
|
||||||
|
@ -57,4 +63,4 @@ def test_register_duplicate_name(daemon, arb_addr):
|
||||||
|
|
||||||
# run it manually since we want to start **after**
|
# run it manually since we want to start **after**
|
||||||
# the other "daemon" program
|
# the other "daemon" program
|
||||||
tractor.run(main, arbiter_addr=arb_addr)
|
trio.run(main)
|
||||||
|
|
|
@ -4,20 +4,22 @@ from itertools import cycle
|
||||||
import pytest
|
import pytest
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
from tractor.testing import tractor_test
|
from tractor.experimental import msgpub
|
||||||
|
|
||||||
|
from conftest import tractor_test
|
||||||
|
|
||||||
|
|
||||||
def test_type_checks():
|
def test_type_checks():
|
||||||
|
|
||||||
with pytest.raises(TypeError) as err:
|
with pytest.raises(TypeError) as err:
|
||||||
@tractor.msg.pub
|
@msgpub
|
||||||
async def no_get_topics(yo):
|
async def no_get_topics(yo):
|
||||||
yield
|
yield
|
||||||
|
|
||||||
assert "must define a `get_topics`" in str(err.value)
|
assert "must define a `get_topics`" in str(err.value)
|
||||||
|
|
||||||
with pytest.raises(TypeError) as err:
|
with pytest.raises(TypeError) as err:
|
||||||
@tractor.msg.pub
|
@msgpub
|
||||||
def not_async_gen(yo):
|
def not_async_gen(yo):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
@ -32,7 +34,7 @@ def is_even(i):
|
||||||
_get_topics = None
|
_get_topics = None
|
||||||
|
|
||||||
|
|
||||||
@tractor.msg.pub
|
@msgpub
|
||||||
async def pubber(get_topics, seed=10):
|
async def pubber(get_topics, seed=10):
|
||||||
|
|
||||||
# ensure topic subscriptions are as expected
|
# ensure topic subscriptions are as expected
|
||||||
|
@ -46,8 +48,9 @@ async def pubber(get_topics, seed=10):
|
||||||
|
|
||||||
|
|
||||||
async def subs(
|
async def subs(
|
||||||
which, pub_actor_name, seed=10,
|
which,
|
||||||
portal=None,
|
pub_actor_name,
|
||||||
|
seed=10,
|
||||||
task_status=trio.TASK_STATUS_IGNORED,
|
task_status=trio.TASK_STATUS_IGNORED,
|
||||||
):
|
):
|
||||||
if len(which) == 1:
|
if len(which) == 1:
|
||||||
|
@ -60,12 +63,15 @@ async def subs(
|
||||||
def pred(i):
|
def pred(i):
|
||||||
return isinstance(i, int)
|
return isinstance(i, int)
|
||||||
|
|
||||||
async with tractor.find_actor(pub_actor_name) as portal:
|
# TODO: https://github.com/goodboy/tractor/issues/207
|
||||||
stream = await portal.run(
|
async with tractor.wait_for_actor(pub_actor_name) as portal:
|
||||||
|
assert portal
|
||||||
|
|
||||||
|
async with portal.open_stream_from(
|
||||||
pubber,
|
pubber,
|
||||||
topics=which,
|
topics=which,
|
||||||
seed=seed,
|
seed=seed,
|
||||||
)
|
) as stream:
|
||||||
task_status.started(stream)
|
task_status.started(stream)
|
||||||
times = 10
|
times = 10
|
||||||
count = 0
|
count = 0
|
||||||
|
@ -79,12 +85,11 @@ async def subs(
|
||||||
|
|
||||||
await stream.aclose()
|
await stream.aclose()
|
||||||
|
|
||||||
stream = await portal.run(
|
async with portal.open_stream_from(
|
||||||
pubber,
|
pubber,
|
||||||
topics=['odd'],
|
topics=['odd'],
|
||||||
seed=seed,
|
seed=seed,
|
||||||
)
|
) as stream:
|
||||||
|
|
||||||
await stream.__anext__()
|
await stream.__anext__()
|
||||||
count = 0
|
count = 0
|
||||||
# async with aclosing(stream) as stream:
|
# async with aclosing(stream) as stream:
|
||||||
|
@ -100,7 +105,7 @@ async def subs(
|
||||||
await stream.aclose()
|
await stream.aclose()
|
||||||
|
|
||||||
|
|
||||||
@tractor.msg.pub(tasks=['one', 'two'])
|
@msgpub(tasks=['one', 'two'])
|
||||||
async def multilock_pubber(get_topics):
|
async def multilock_pubber(get_topics):
|
||||||
yield {'doggy': 10}
|
yield {'doggy': 10}
|
||||||
|
|
||||||
|
@ -128,11 +133,10 @@ async def test_required_args(callwith_expecterror):
|
||||||
await func(**kwargs)
|
await func(**kwargs)
|
||||||
else:
|
else:
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery() as n:
|
||||||
# await func(**kwargs)
|
|
||||||
portal = await n.run_in_actor(
|
portal = await n.start_actor(
|
||||||
multilock_pubber,
|
|
||||||
name='pubber',
|
name='pubber',
|
||||||
**kwargs
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
async with tractor.wait_for_actor('pubber'):
|
async with tractor.wait_for_actor('pubber'):
|
||||||
|
@ -140,9 +144,15 @@ async def test_required_args(callwith_expecterror):
|
||||||
|
|
||||||
await trio.sleep(0.5)
|
await trio.sleep(0.5)
|
||||||
|
|
||||||
async for val in await portal.result():
|
async with portal.open_stream_from(
|
||||||
|
multilock_pubber,
|
||||||
|
**kwargs
|
||||||
|
) as stream:
|
||||||
|
async for val in stream:
|
||||||
assert val == {'doggy': 10}
|
assert val == {'doggy': 10}
|
||||||
|
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
'pub_actor',
|
'pub_actor',
|
||||||
|
@ -159,16 +169,20 @@ def test_multi_actor_subs_arbiter_pub(
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
enable_modules=[__name__],
|
||||||
|
) as n:
|
||||||
|
|
||||||
name = 'arbiter'
|
name = 'root'
|
||||||
|
|
||||||
if pub_actor == 'streamer':
|
if pub_actor == 'streamer':
|
||||||
# start the publisher as a daemon
|
# start the publisher as a daemon
|
||||||
master_portal = await n.start_actor(
|
master_portal = await n.start_actor(
|
||||||
'streamer',
|
'streamer',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
name = 'streamer'
|
||||||
|
|
||||||
even_portal = await n.run_in_actor(
|
even_portal = await n.run_in_actor(
|
||||||
subs,
|
subs,
|
||||||
|
@ -227,7 +241,6 @@ def test_multi_actor_subs_arbiter_pub(
|
||||||
assert 'even' not in get_topics()
|
assert 'even' not in get_topics()
|
||||||
|
|
||||||
await odd_portal.cancel_actor()
|
await odd_portal.cancel_actor()
|
||||||
await trio.sleep(2)
|
|
||||||
|
|
||||||
if pub_actor == 'arbiter':
|
if pub_actor == 'arbiter':
|
||||||
while get_topics():
|
while get_topics():
|
||||||
|
@ -237,11 +250,7 @@ def test_multi_actor_subs_arbiter_pub(
|
||||||
else:
|
else:
|
||||||
await master_portal.cancel_actor()
|
await master_portal.cancel_actor()
|
||||||
|
|
||||||
tractor.run(
|
trio.run(main)
|
||||||
main,
|
|
||||||
arbiter_addr=arb_addr,
|
|
||||||
rpc_module_paths=[__name__],
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def test_single_subactor_pub_multitask_subs(
|
def test_single_subactor_pub_multitask_subs(
|
||||||
|
@ -250,11 +259,14 @@ def test_single_subactor_pub_multitask_subs(
|
||||||
):
|
):
|
||||||
async def main():
|
async def main():
|
||||||
|
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
enable_modules=[__name__],
|
||||||
|
) as n:
|
||||||
|
|
||||||
portal = await n.start_actor(
|
portal = await n.start_actor(
|
||||||
'streamer',
|
'streamer',
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
async with tractor.wait_for_actor('streamer'):
|
async with tractor.wait_for_actor('streamer'):
|
||||||
# block until 2nd actor is initialized
|
# block until 2nd actor is initialized
|
||||||
|
@ -278,8 +290,4 @@ def test_single_subactor_pub_multitask_subs(
|
||||||
|
|
||||||
await portal.cancel_actor()
|
await portal.cancel_actor()
|
||||||
|
|
||||||
tractor.run(
|
trio.run(main)
|
||||||
main,
|
|
||||||
arbiter_addr=arb_addr,
|
|
||||||
rpc_module_paths=[__name__],
|
|
||||||
)
|
|
||||||
|
|
|
@ -0,0 +1,182 @@
|
||||||
|
'''
|
||||||
|
Async context manager cache api testing: ``trionics.maybe_open_context():``
|
||||||
|
|
||||||
|
'''
|
||||||
|
from contextlib import asynccontextmanager as acm
|
||||||
|
import platform
|
||||||
|
from typing import Awaitable
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
_resource: int = 0
|
||||||
|
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def maybe_increment_counter(task_name: str):
|
||||||
|
global _resource
|
||||||
|
|
||||||
|
_resource += 1
|
||||||
|
await trio.lowlevel.checkpoint()
|
||||||
|
yield _resource
|
||||||
|
await trio.lowlevel.checkpoint()
|
||||||
|
_resource -= 1
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'key_on',
|
||||||
|
['key_value', 'kwargs'],
|
||||||
|
ids="key_on={}".format,
|
||||||
|
)
|
||||||
|
def test_resource_only_entered_once(key_on):
|
||||||
|
global _resource
|
||||||
|
_resource = 0
|
||||||
|
|
||||||
|
kwargs = {}
|
||||||
|
key = None
|
||||||
|
if key_on == 'key_value':
|
||||||
|
key = 'some_common_key'
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
cache_active: bool = False
|
||||||
|
|
||||||
|
async def enter_cached_mngr(name: str):
|
||||||
|
nonlocal cache_active
|
||||||
|
|
||||||
|
if key_on == 'kwargs':
|
||||||
|
# make a common kwargs input to key on it
|
||||||
|
kwargs = {'task_name': 'same_task_name'}
|
||||||
|
assert key is None
|
||||||
|
else:
|
||||||
|
# different task names per task will be used
|
||||||
|
kwargs = {'task_name': name}
|
||||||
|
|
||||||
|
async with tractor.trionics.maybe_open_context(
|
||||||
|
maybe_increment_counter,
|
||||||
|
kwargs=kwargs,
|
||||||
|
key=key,
|
||||||
|
|
||||||
|
) as (cache_hit, resource):
|
||||||
|
if cache_hit:
|
||||||
|
try:
|
||||||
|
cache_active = True
|
||||||
|
assert resource == 1
|
||||||
|
await trio.sleep_forever()
|
||||||
|
finally:
|
||||||
|
cache_active = False
|
||||||
|
else:
|
||||||
|
assert resource == 1
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
with trio.move_on_after(0.5):
|
||||||
|
async with (
|
||||||
|
tractor.open_root_actor(),
|
||||||
|
trio.open_nursery() as n,
|
||||||
|
):
|
||||||
|
|
||||||
|
for i in range(10):
|
||||||
|
n.start_soon(enter_cached_mngr, f'task_{i}')
|
||||||
|
await trio.sleep(0.001)
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def streamer(
|
||||||
|
ctx: tractor.Context,
|
||||||
|
seq: list[int] = list(range(1000)),
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
await ctx.started()
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
for val in seq:
|
||||||
|
await stream.send(val)
|
||||||
|
await trio.sleep(0.001)
|
||||||
|
|
||||||
|
print('producer finished')
|
||||||
|
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def open_stream() -> Awaitable[tractor.MsgStream]:
|
||||||
|
|
||||||
|
async with tractor.open_nursery() as tn:
|
||||||
|
portal = await tn.start_actor('streamer', enable_modules=[__name__])
|
||||||
|
async with (
|
||||||
|
portal.open_context(streamer) as (ctx, first),
|
||||||
|
ctx.open_stream() as stream,
|
||||||
|
):
|
||||||
|
yield stream
|
||||||
|
|
||||||
|
await portal.cancel_actor()
|
||||||
|
print('CANCELLED STREAMER')
|
||||||
|
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def maybe_open_stream(taskname: str):
|
||||||
|
async with tractor.trionics.maybe_open_context(
|
||||||
|
# NOTE: all secondary tasks should cache hit on the same key
|
||||||
|
acm_func=open_stream,
|
||||||
|
) as (cache_hit, stream):
|
||||||
|
|
||||||
|
if cache_hit:
|
||||||
|
print(f'{taskname} loaded from cache')
|
||||||
|
|
||||||
|
# add a new broadcast subscription for the quote stream
|
||||||
|
# if this feed is already allocated by the first
|
||||||
|
# task that entereed
|
||||||
|
async with stream.subscribe() as bstream:
|
||||||
|
yield bstream
|
||||||
|
else:
|
||||||
|
# yield the actual stream
|
||||||
|
yield stream
|
||||||
|
|
||||||
|
|
||||||
|
def test_open_local_sub_to_stream():
|
||||||
|
'''
|
||||||
|
Verify a single inter-actor stream can can be fanned-out shared to
|
||||||
|
N local tasks using ``trionics.maybe_open_context():``.
|
||||||
|
|
||||||
|
'''
|
||||||
|
timeout = 3 if platform.system() != "Windows" else 10
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
full = list(range(1000))
|
||||||
|
|
||||||
|
async def get_sub_and_pull(taskname: str):
|
||||||
|
async with (
|
||||||
|
maybe_open_stream(taskname) as stream,
|
||||||
|
):
|
||||||
|
if '0' in taskname:
|
||||||
|
assert isinstance(stream, tractor.MsgStream)
|
||||||
|
else:
|
||||||
|
assert isinstance(
|
||||||
|
stream,
|
||||||
|
tractor.trionics.BroadcastReceiver
|
||||||
|
)
|
||||||
|
|
||||||
|
first = await stream.receive()
|
||||||
|
print(f'{taskname} started with value {first}')
|
||||||
|
seq = []
|
||||||
|
async for msg in stream:
|
||||||
|
seq.append(msg)
|
||||||
|
|
||||||
|
assert set(seq).issubset(set(full))
|
||||||
|
print(f'{taskname} finished')
|
||||||
|
|
||||||
|
with trio.fail_after(timeout):
|
||||||
|
# TODO: turns out this isn't multi-task entrant XD
|
||||||
|
# We probably need an indepotent entry semantic?
|
||||||
|
async with tractor.open_root_actor():
|
||||||
|
async with (
|
||||||
|
trio.open_nursery() as nurse,
|
||||||
|
):
|
||||||
|
for i in range(10):
|
||||||
|
nurse.start_soon(get_sub_and_pull, f'task_{i}')
|
||||||
|
await trio.sleep(0.001)
|
||||||
|
|
||||||
|
print('all consumer tasks finished')
|
||||||
|
|
||||||
|
trio.run(main)
|
|
@ -53,7 +53,7 @@ def test_rpc_errors(arb_addr, to_call, testdir):
|
||||||
exposed_mods, funcname, inside_err = to_call
|
exposed_mods, funcname, inside_err = to_call
|
||||||
subactor_exposed_mods = []
|
subactor_exposed_mods = []
|
||||||
func_defined = globals().get(funcname, False)
|
func_defined = globals().get(funcname, False)
|
||||||
subactor_requests_to = 'arbiter'
|
subactor_requests_to = 'root'
|
||||||
remote_err = tractor.RemoteActorError
|
remote_err = tractor.RemoteActorError
|
||||||
|
|
||||||
# remote module that fails at import time
|
# remote module that fails at import time
|
||||||
|
@ -74,11 +74,15 @@ def test_rpc_errors(arb_addr, to_call, testdir):
|
||||||
remote_err = inside_err
|
remote_err = inside_err
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
actor = tractor.current_actor()
|
|
||||||
assert actor.is_arbiter
|
|
||||||
|
|
||||||
# spawn a subactor which calls us back
|
# spawn a subactor which calls us back
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
enable_modules=exposed_mods.copy(),
|
||||||
|
) as n:
|
||||||
|
|
||||||
|
actor = tractor.current_actor()
|
||||||
|
assert actor.is_arbiter
|
||||||
await n.run_in_actor(
|
await n.run_in_actor(
|
||||||
sleep_back_actor,
|
sleep_back_actor,
|
||||||
actor_name=subactor_requests_to,
|
actor_name=subactor_requests_to,
|
||||||
|
@ -90,15 +94,11 @@ def test_rpc_errors(arb_addr, to_call, testdir):
|
||||||
func_name=funcname,
|
func_name=funcname,
|
||||||
exposed_mods=exposed_mods,
|
exposed_mods=exposed_mods,
|
||||||
func_defined=True if func_defined else False,
|
func_defined=True if func_defined else False,
|
||||||
rpc_module_paths=subactor_exposed_mods,
|
enable_modules=subactor_exposed_mods,
|
||||||
)
|
)
|
||||||
|
|
||||||
def run():
|
def run():
|
||||||
tractor.run(
|
trio.run(main)
|
||||||
main,
|
|
||||||
arbiter_addr=arb_addr,
|
|
||||||
rpc_module_paths=exposed_mods.copy(),
|
|
||||||
)
|
|
||||||
|
|
||||||
# handle both parameterized cases
|
# handle both parameterized cases
|
||||||
if exposed_mods and func_defined:
|
if exposed_mods and func_defined:
|
||||||
|
|
|
@ -0,0 +1,73 @@
|
||||||
|
"""
|
||||||
|
Verifying internal runtime state and undocumented extras.
|
||||||
|
|
||||||
|
"""
|
||||||
|
import os
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
from conftest import tractor_test
|
||||||
|
|
||||||
|
|
||||||
|
_file_path: str = ''
|
||||||
|
|
||||||
|
|
||||||
|
def unlink_file():
|
||||||
|
print('Removing tmp file!')
|
||||||
|
os.remove(_file_path)
|
||||||
|
|
||||||
|
|
||||||
|
async def crash_and_clean_tmpdir(
|
||||||
|
tmp_file_path: str,
|
||||||
|
error: bool = True,
|
||||||
|
):
|
||||||
|
global _file_path
|
||||||
|
_file_path = tmp_file_path
|
||||||
|
|
||||||
|
actor = tractor.current_actor()
|
||||||
|
actor.lifetime_stack.callback(unlink_file)
|
||||||
|
|
||||||
|
assert os.path.isfile(tmp_file_path)
|
||||||
|
await trio.sleep(0.1)
|
||||||
|
if error:
|
||||||
|
assert 0
|
||||||
|
else:
|
||||||
|
actor.cancel_soon()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'error_in_child',
|
||||||
|
[True, False],
|
||||||
|
)
|
||||||
|
@tractor_test
|
||||||
|
async def test_lifetime_stack_wipes_tmpfile(
|
||||||
|
tmp_path,
|
||||||
|
error_in_child: bool,
|
||||||
|
):
|
||||||
|
child_tmp_file = tmp_path / "child.txt"
|
||||||
|
child_tmp_file.touch()
|
||||||
|
assert child_tmp_file.exists()
|
||||||
|
path = str(child_tmp_file)
|
||||||
|
|
||||||
|
try:
|
||||||
|
with trio.move_on_after(0.5):
|
||||||
|
async with tractor.open_nursery() as n:
|
||||||
|
await ( # inlined portal
|
||||||
|
await n.run_in_actor(
|
||||||
|
crash_and_clean_tmpdir,
|
||||||
|
tmp_file_path=path,
|
||||||
|
error=error_in_child,
|
||||||
|
)
|
||||||
|
).result()
|
||||||
|
|
||||||
|
except (
|
||||||
|
tractor.RemoteActorError,
|
||||||
|
tractor.BaseExceptionGroup,
|
||||||
|
):
|
||||||
|
pass
|
||||||
|
|
||||||
|
# tmp file should have been wiped by
|
||||||
|
# teardown stack.
|
||||||
|
assert not child_tmp_file.exists()
|
|
@ -1,7 +1,8 @@
|
||||||
"""
|
"""
|
||||||
Spawning basics
|
Spawning basics
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from functools import partial
|
from typing import Optional
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
import trio
|
import trio
|
||||||
|
@ -12,23 +13,36 @@ from conftest import tractor_test
|
||||||
data_to_pass_down = {'doggy': 10, 'kitty': 4}
|
data_to_pass_down = {'doggy': 10, 'kitty': 4}
|
||||||
|
|
||||||
|
|
||||||
async def spawn(is_arbiter, data):
|
async def spawn(
|
||||||
|
is_arbiter: bool,
|
||||||
|
data: dict,
|
||||||
|
arb_addr: tuple[str, int],
|
||||||
|
):
|
||||||
namespaces = [__name__]
|
namespaces = [__name__]
|
||||||
|
|
||||||
await trio.sleep(0.1)
|
await trio.sleep(0.1)
|
||||||
|
|
||||||
|
async with tractor.open_root_actor(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
):
|
||||||
|
|
||||||
actor = tractor.current_actor()
|
actor = tractor.current_actor()
|
||||||
assert actor.is_arbiter == is_arbiter
|
assert actor.is_arbiter == is_arbiter
|
||||||
data == data_to_pass_down
|
data = data_to_pass_down
|
||||||
|
|
||||||
if actor.is_arbiter:
|
if actor.is_arbiter:
|
||||||
async with tractor.open_nursery() as nursery:
|
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
) as nursery:
|
||||||
|
|
||||||
# forks here
|
# forks here
|
||||||
portal = await nursery.run_in_actor(
|
portal = await nursery.run_in_actor(
|
||||||
spawn,
|
spawn,
|
||||||
is_arbiter=False,
|
is_arbiter=False,
|
||||||
name='sub-actor',
|
name='sub-actor',
|
||||||
data=data,
|
data=data,
|
||||||
rpc_module_paths=namespaces,
|
arb_addr=arb_addr,
|
||||||
|
enable_modules=namespaces,
|
||||||
)
|
)
|
||||||
|
|
||||||
assert len(nursery._children) == 1
|
assert len(nursery._children) == 1
|
||||||
|
@ -42,16 +56,16 @@ async def spawn(is_arbiter, data):
|
||||||
|
|
||||||
|
|
||||||
def test_local_arbiter_subactor_global_state(arb_addr):
|
def test_local_arbiter_subactor_global_state(arb_addr):
|
||||||
result = tractor.run(
|
result = trio.run(
|
||||||
partial(spawn, data=data_to_pass_down),
|
spawn,
|
||||||
True,
|
True,
|
||||||
name='arbiter',
|
data_to_pass_down,
|
||||||
arbiter_addr=arb_addr,
|
arb_addr,
|
||||||
)
|
)
|
||||||
assert result == 10
|
assert result == 10
|
||||||
|
|
||||||
|
|
||||||
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.
|
||||||
"""
|
"""
|
||||||
|
@ -67,7 +81,7 @@ async def test_movie_theatre_convo(start_method):
|
||||||
portal = await n.start_actor(
|
portal = await n.start_actor(
|
||||||
'frank',
|
'frank',
|
||||||
# enable the actor to run funcs from this current module
|
# enable the actor to run funcs from this current module
|
||||||
rpc_module_paths=[__name__],
|
enable_modules=[__name__],
|
||||||
)
|
)
|
||||||
|
|
||||||
print(await portal.run(movie_theatre_question))
|
print(await portal.run(movie_theatre_question))
|
||||||
|
@ -80,24 +94,38 @@ async def test_movie_theatre_convo(start_method):
|
||||||
await portal.cancel_actor()
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
def cellar_door():
|
async def cellar_door(return_value: Optional[str]):
|
||||||
return "Dang that's beautiful"
|
return return_value
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'return_value', ["Dang that's beautiful", None],
|
||||||
|
ids=['return_str', 'return_None'],
|
||||||
|
)
|
||||||
@tractor_test
|
@tractor_test
|
||||||
async def test_most_beautiful_word(start_method):
|
async def test_most_beautiful_word(
|
||||||
"""The main ``tractor`` routine.
|
start_method,
|
||||||
"""
|
return_value
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
The main ``tractor`` routine.
|
||||||
|
|
||||||
|
'''
|
||||||
|
with trio.fail_after(1):
|
||||||
async with tractor.open_nursery() as n:
|
async with tractor.open_nursery() as n:
|
||||||
|
|
||||||
portal = await n.run_in_actor(
|
portal = await n.run_in_actor(
|
||||||
cellar_door,
|
cellar_door,
|
||||||
|
return_value=return_value,
|
||||||
name='some_linguist',
|
name='some_linguist',
|
||||||
)
|
)
|
||||||
|
|
||||||
|
print(await portal.result())
|
||||||
# The ``async with`` will unblock here since the 'some_linguist'
|
# The ``async with`` will unblock here since the 'some_linguist'
|
||||||
# actor has completed its main task ``cellar_door``.
|
# actor has completed its main task ``cellar_door``.
|
||||||
|
|
||||||
|
# this should pull the cached final result already captured during
|
||||||
|
# the nursery block exit.
|
||||||
print(await portal.result())
|
print(await portal.result())
|
||||||
|
|
||||||
|
|
||||||
|
@ -114,26 +142,27 @@ def test_loglevel_propagated_to_subactor(
|
||||||
capfd,
|
capfd,
|
||||||
arb_addr,
|
arb_addr,
|
||||||
):
|
):
|
||||||
if start_method == 'forkserver':
|
if start_method == 'mp_forkserver':
|
||||||
pytest.skip(
|
pytest.skip(
|
||||||
"a bug with `capfd` seems to make forkserver capture not work?")
|
"a bug with `capfd` seems to make forkserver capture not work?")
|
||||||
|
|
||||||
level = 'critical'
|
level = 'critical'
|
||||||
|
|
||||||
async def main():
|
async def main():
|
||||||
async with tractor.open_nursery() as tn:
|
async with tractor.open_nursery(
|
||||||
|
name='arbiter',
|
||||||
|
start_method=start_method,
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
|
||||||
|
) as tn:
|
||||||
await tn.run_in_actor(
|
await tn.run_in_actor(
|
||||||
check_loglevel,
|
check_loglevel,
|
||||||
|
loglevel=level,
|
||||||
level=level,
|
level=level,
|
||||||
)
|
)
|
||||||
|
|
||||||
tractor.run(
|
trio.run(main)
|
||||||
main,
|
|
||||||
name='arbiter',
|
|
||||||
loglevel=level,
|
|
||||||
start_method=start_method,
|
|
||||||
arbiter_addr=arb_addr,
|
|
||||||
)
|
|
||||||
# ensure subactor spits log message on stderr
|
# ensure subactor spits log message on stderr
|
||||||
captured = capfd.readouterr()
|
captured = capfd.readouterr()
|
||||||
assert 'yoyoyo' in captured.err
|
assert 'yoyoyo' in captured.err
|
||||||
|
|
|
@ -0,0 +1,514 @@
|
||||||
|
"""
|
||||||
|
Broadcast channels for fan-out to local tasks.
|
||||||
|
|
||||||
|
"""
|
||||||
|
from contextlib import asynccontextmanager
|
||||||
|
from functools import partial
|
||||||
|
from itertools import cycle
|
||||||
|
import time
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
from trio.lowlevel import current_task
|
||||||
|
import tractor
|
||||||
|
from tractor.trionics import (
|
||||||
|
broadcast_receiver,
|
||||||
|
Lagged,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@tractor.context
|
||||||
|
async def echo_sequences(
|
||||||
|
|
||||||
|
ctx: tractor.Context,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''Bidir streaming endpoint which will stream
|
||||||
|
back any sequence it is sent item-wise.
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ctx.started()
|
||||||
|
|
||||||
|
async with ctx.open_stream() as stream:
|
||||||
|
async for sequence in stream:
|
||||||
|
seq = list(sequence)
|
||||||
|
for value in seq:
|
||||||
|
await stream.send(value)
|
||||||
|
print(f'producer sent {value}')
|
||||||
|
|
||||||
|
|
||||||
|
async def ensure_sequence(
|
||||||
|
|
||||||
|
stream: tractor.MsgStream,
|
||||||
|
sequence: list,
|
||||||
|
delay: Optional[float] = None,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
name = current_task().name
|
||||||
|
async with stream.subscribe() as bcaster:
|
||||||
|
assert not isinstance(bcaster, type(stream))
|
||||||
|
async for value in bcaster:
|
||||||
|
print(f'{name} rx: {value}')
|
||||||
|
assert value == sequence[0]
|
||||||
|
sequence.remove(value)
|
||||||
|
|
||||||
|
if delay:
|
||||||
|
await trio.sleep(delay)
|
||||||
|
|
||||||
|
if not sequence:
|
||||||
|
# fully consumed
|
||||||
|
break
|
||||||
|
|
||||||
|
|
||||||
|
@asynccontextmanager
|
||||||
|
async def open_sequence_streamer(
|
||||||
|
|
||||||
|
sequence: list[int],
|
||||||
|
arb_addr: tuple[str, int],
|
||||||
|
start_method: str,
|
||||||
|
|
||||||
|
) -> tractor.MsgStream:
|
||||||
|
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
arbiter_addr=arb_addr,
|
||||||
|
start_method=start_method,
|
||||||
|
) as tn:
|
||||||
|
|
||||||
|
portal = await tn.start_actor(
|
||||||
|
'sequence_echoer',
|
||||||
|
enable_modules=[__name__],
|
||||||
|
)
|
||||||
|
|
||||||
|
async with portal.open_context(
|
||||||
|
echo_sequences,
|
||||||
|
) as (ctx, first):
|
||||||
|
|
||||||
|
assert first is None
|
||||||
|
async with ctx.open_stream(backpressure=True) as stream:
|
||||||
|
yield stream
|
||||||
|
|
||||||
|
await portal.cancel_actor()
|
||||||
|
|
||||||
|
|
||||||
|
def test_stream_fan_out_to_local_subscriptions(
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
|
):
|
||||||
|
|
||||||
|
sequence = list(range(1000))
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with open_sequence_streamer(
|
||||||
|
sequence,
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
|
) as stream:
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
for i in range(10):
|
||||||
|
n.start_soon(
|
||||||
|
ensure_sequence,
|
||||||
|
stream,
|
||||||
|
sequence.copy(),
|
||||||
|
name=f'consumer_{i}',
|
||||||
|
)
|
||||||
|
|
||||||
|
await stream.send(tuple(sequence))
|
||||||
|
|
||||||
|
async for value in stream:
|
||||||
|
print(f'source stream rx: {value}')
|
||||||
|
assert value == sequence[0]
|
||||||
|
sequence.remove(value)
|
||||||
|
|
||||||
|
if not sequence:
|
||||||
|
# fully consumed
|
||||||
|
break
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'task_delays',
|
||||||
|
[
|
||||||
|
(0.01, 0.001),
|
||||||
|
(0.001, 0.01),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
def test_consumer_and_parent_maybe_lag(
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
|
task_delays,
|
||||||
|
):
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
sequence = list(range(300))
|
||||||
|
parent_delay, sub_delay = task_delays
|
||||||
|
|
||||||
|
async with open_sequence_streamer(
|
||||||
|
sequence,
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
|
) as stream:
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
|
||||||
|
n.start_soon(
|
||||||
|
ensure_sequence,
|
||||||
|
stream,
|
||||||
|
sequence.copy(),
|
||||||
|
sub_delay,
|
||||||
|
name='consumer_task',
|
||||||
|
)
|
||||||
|
|
||||||
|
await stream.send(tuple(sequence))
|
||||||
|
|
||||||
|
# async for value in stream:
|
||||||
|
lagged = False
|
||||||
|
lag_count = 0
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
value = await stream.receive()
|
||||||
|
print(f'source stream rx: {value}')
|
||||||
|
|
||||||
|
if lagged:
|
||||||
|
# re set the sequence starting at our last
|
||||||
|
# value
|
||||||
|
sequence = sequence[sequence.index(value) + 1:]
|
||||||
|
else:
|
||||||
|
assert value == sequence[0]
|
||||||
|
sequence.remove(value)
|
||||||
|
|
||||||
|
lagged = False
|
||||||
|
|
||||||
|
except Lagged:
|
||||||
|
lagged = True
|
||||||
|
print(f'source stream lagged after {value}')
|
||||||
|
lag_count += 1
|
||||||
|
continue
|
||||||
|
|
||||||
|
# lag the parent
|
||||||
|
await trio.sleep(parent_delay)
|
||||||
|
|
||||||
|
if not sequence:
|
||||||
|
# fully consumed
|
||||||
|
break
|
||||||
|
print(f'parent + source stream lagged: {lag_count}')
|
||||||
|
|
||||||
|
if parent_delay > sub_delay:
|
||||||
|
assert lag_count > 0
|
||||||
|
|
||||||
|
except Lagged:
|
||||||
|
# child was lagged
|
||||||
|
assert parent_delay < sub_delay
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def test_faster_task_to_recv_is_cancelled_by_slower(
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Ensure that if a faster task consuming from a stream is cancelled
|
||||||
|
the slower task can continue to receive all expected values.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
sequence = list(range(1000))
|
||||||
|
|
||||||
|
async with open_sequence_streamer(
|
||||||
|
sequence,
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
|
|
||||||
|
) as stream:
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
n.start_soon(
|
||||||
|
ensure_sequence,
|
||||||
|
stream,
|
||||||
|
sequence.copy(),
|
||||||
|
0,
|
||||||
|
name='consumer_task',
|
||||||
|
)
|
||||||
|
|
||||||
|
await stream.send(tuple(sequence))
|
||||||
|
|
||||||
|
# pull 3 values, cancel the subtask, then
|
||||||
|
# expect to be able to pull all values still
|
||||||
|
for i in range(20):
|
||||||
|
try:
|
||||||
|
value = await stream.receive()
|
||||||
|
print(f'source stream rx: {value}')
|
||||||
|
await trio.sleep(0.01)
|
||||||
|
except Lagged:
|
||||||
|
print(f'parent overrun after {value}')
|
||||||
|
continue
|
||||||
|
|
||||||
|
print('cancelling faster subtask')
|
||||||
|
n.cancel_scope.cancel()
|
||||||
|
|
||||||
|
try:
|
||||||
|
value = await stream.receive()
|
||||||
|
print(f'source stream after cancel: {value}')
|
||||||
|
except Lagged:
|
||||||
|
print(f'parent overrun after {value}')
|
||||||
|
|
||||||
|
# expect to see all remaining values
|
||||||
|
with trio.fail_after(0.5):
|
||||||
|
async for value in stream:
|
||||||
|
assert stream._broadcaster._state.recv_ready is None
|
||||||
|
print(f'source stream rx: {value}')
|
||||||
|
if value == 999:
|
||||||
|
# fully consumed and we missed no values once
|
||||||
|
# the faster subtask was cancelled
|
||||||
|
break
|
||||||
|
|
||||||
|
# await tractor.breakpoint()
|
||||||
|
# await stream.receive()
|
||||||
|
print(f'final value: {value}')
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def test_subscribe_errors_after_close():
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
size = 1
|
||||||
|
tx, rx = trio.open_memory_channel(size)
|
||||||
|
async with broadcast_receiver(rx, size) as brx:
|
||||||
|
pass
|
||||||
|
|
||||||
|
try:
|
||||||
|
# open and close
|
||||||
|
async with brx.subscribe():
|
||||||
|
pass
|
||||||
|
|
||||||
|
except trio.ClosedResourceError:
|
||||||
|
assert brx.key not in brx._state.subs
|
||||||
|
|
||||||
|
else:
|
||||||
|
assert 0
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def test_ensure_slow_consumers_lag_out(
|
||||||
|
arb_addr,
|
||||||
|
start_method,
|
||||||
|
):
|
||||||
|
'''This is a pure local task test; no tractor
|
||||||
|
machinery is really required.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
# make sure it all works within the runtime
|
||||||
|
async with tractor.open_root_actor():
|
||||||
|
|
||||||
|
num_laggers = 4
|
||||||
|
laggers: dict[str, int] = {}
|
||||||
|
retries = 3
|
||||||
|
size = 100
|
||||||
|
tx, rx = trio.open_memory_channel(size)
|
||||||
|
brx = broadcast_receiver(rx, size)
|
||||||
|
|
||||||
|
async def sub_and_print(
|
||||||
|
delay: float,
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
task = current_task()
|
||||||
|
start = time.time()
|
||||||
|
|
||||||
|
async with brx.subscribe() as lbrx:
|
||||||
|
while True:
|
||||||
|
print(f'{task.name}: starting consume loop')
|
||||||
|
try:
|
||||||
|
async for value in lbrx:
|
||||||
|
print(f'{task.name}: {value}')
|
||||||
|
await trio.sleep(delay)
|
||||||
|
|
||||||
|
if task.name == 'sub_1':
|
||||||
|
# trigger checkpoint to clean out other subs
|
||||||
|
await trio.sleep(0.01)
|
||||||
|
|
||||||
|
# the non-lagger got
|
||||||
|
# a ``trio.EndOfChannel``
|
||||||
|
# because the ``tx`` below was closed
|
||||||
|
assert len(lbrx._state.subs) == 1
|
||||||
|
|
||||||
|
await lbrx.aclose()
|
||||||
|
|
||||||
|
assert len(lbrx._state.subs) == 0
|
||||||
|
|
||||||
|
except trio.ClosedResourceError:
|
||||||
|
# only the fast sub will try to re-enter
|
||||||
|
# iteration on the now closed bcaster
|
||||||
|
assert task.name == 'sub_1'
|
||||||
|
return
|
||||||
|
|
||||||
|
except Lagged:
|
||||||
|
lag_time = time.time() - start
|
||||||
|
lags = laggers[task.name]
|
||||||
|
print(
|
||||||
|
f'restarting slow task {task.name} '
|
||||||
|
f'that bailed out on {lags}:{value} '
|
||||||
|
f'after {lag_time:.3f}')
|
||||||
|
if lags <= retries:
|
||||||
|
laggers[task.name] += 1
|
||||||
|
continue
|
||||||
|
else:
|
||||||
|
print(
|
||||||
|
f'{task.name} was too slow and terminated '
|
||||||
|
f'on {lags}:{value}')
|
||||||
|
return
|
||||||
|
|
||||||
|
async with trio.open_nursery() as nursery:
|
||||||
|
|
||||||
|
for i in range(1, num_laggers):
|
||||||
|
|
||||||
|
task_name = f'sub_{i}'
|
||||||
|
laggers[task_name] = 0
|
||||||
|
nursery.start_soon(
|
||||||
|
partial(
|
||||||
|
sub_and_print,
|
||||||
|
delay=i*0.001,
|
||||||
|
),
|
||||||
|
name=task_name,
|
||||||
|
)
|
||||||
|
|
||||||
|
# allow subs to sched
|
||||||
|
await trio.sleep(0.1)
|
||||||
|
|
||||||
|
async with tx:
|
||||||
|
for i in cycle(range(size)):
|
||||||
|
await tx.send(i)
|
||||||
|
if len(brx._state.subs) == 2:
|
||||||
|
# only one, the non lagger, sub is left
|
||||||
|
break
|
||||||
|
|
||||||
|
# the non-lagger
|
||||||
|
assert laggers.pop('sub_1') == 0
|
||||||
|
|
||||||
|
for n, v in laggers.items():
|
||||||
|
assert v == 4
|
||||||
|
|
||||||
|
assert tx._closed
|
||||||
|
assert not tx._state.open_send_channels
|
||||||
|
|
||||||
|
# check that "first" bcaster that we created
|
||||||
|
# above, never was iterated and is thus overrun
|
||||||
|
try:
|
||||||
|
await brx.receive()
|
||||||
|
except Lagged:
|
||||||
|
# expect tokio style index truncation
|
||||||
|
seq = brx._state.subs[brx.key]
|
||||||
|
assert seq == len(brx._state.queue) - 1
|
||||||
|
|
||||||
|
# all backpressured entries in the underlying
|
||||||
|
# channel should have been copied into the caster
|
||||||
|
# queue trailing-window
|
||||||
|
async for i in rx:
|
||||||
|
print(f'bped: {i}')
|
||||||
|
assert i in brx._state.queue
|
||||||
|
|
||||||
|
# should be noop
|
||||||
|
await brx.aclose()
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def test_first_recver_is_cancelled():
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
# make sure it all works within the runtime
|
||||||
|
async with tractor.open_root_actor():
|
||||||
|
|
||||||
|
tx, rx = trio.open_memory_channel(1)
|
||||||
|
brx = broadcast_receiver(rx, 1)
|
||||||
|
cs = trio.CancelScope()
|
||||||
|
|
||||||
|
async def sub_and_recv():
|
||||||
|
with cs:
|
||||||
|
async with brx.subscribe() as bc:
|
||||||
|
async for value in bc:
|
||||||
|
print(value)
|
||||||
|
|
||||||
|
async def cancel_and_send():
|
||||||
|
await trio.sleep(0.2)
|
||||||
|
cs.cancel()
|
||||||
|
await tx.send(1)
|
||||||
|
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
|
||||||
|
n.start_soon(sub_and_recv)
|
||||||
|
await trio.sleep(0.1)
|
||||||
|
assert brx._state.recv_ready
|
||||||
|
|
||||||
|
n.start_soon(cancel_and_send)
|
||||||
|
|
||||||
|
# ensure that we don't hang because no-task is now
|
||||||
|
# waiting on the underlying receive..
|
||||||
|
with trio.fail_after(0.5):
|
||||||
|
value = await brx.receive()
|
||||||
|
print(f'parent: {value}')
|
||||||
|
assert value == 1
|
||||||
|
|
||||||
|
trio.run(main)
|
||||||
|
|
||||||
|
|
||||||
|
def test_no_raise_on_lag():
|
||||||
|
'''
|
||||||
|
Run a simple 2-task broadcast where one task is slow but configured
|
||||||
|
so that it does not raise `Lagged` on overruns using
|
||||||
|
`raise_on_lasg=False` and verify that the task does not raise.
|
||||||
|
|
||||||
|
'''
|
||||||
|
size = 100
|
||||||
|
tx, rx = trio.open_memory_channel(size)
|
||||||
|
brx = broadcast_receiver(rx, size)
|
||||||
|
|
||||||
|
async def slow():
|
||||||
|
async with brx.subscribe(
|
||||||
|
raise_on_lag=False,
|
||||||
|
) as br:
|
||||||
|
async for msg in br:
|
||||||
|
print(f'slow task got: {msg}')
|
||||||
|
await trio.sleep(0.1)
|
||||||
|
|
||||||
|
async def fast():
|
||||||
|
async with brx.subscribe() as br:
|
||||||
|
async for msg in br:
|
||||||
|
print(f'fast task got: {msg}')
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
async with (
|
||||||
|
tractor.open_root_actor(
|
||||||
|
# NOTE: so we see the warning msg emitted by the bcaster
|
||||||
|
# internals when the no raise flag is set.
|
||||||
|
loglevel='warning',
|
||||||
|
),
|
||||||
|
trio.open_nursery() as n,
|
||||||
|
):
|
||||||
|
n.start_soon(slow)
|
||||||
|
n.start_soon(fast)
|
||||||
|
|
||||||
|
for i in range(1000):
|
||||||
|
await tx.send(i)
|
||||||
|
|
||||||
|
# simulate user nailing ctl-c after realizing
|
||||||
|
# there's a lag in the slow task.
|
||||||
|
await trio.sleep(1)
|
||||||
|
raise KeyboardInterrupt
|
||||||
|
|
||||||
|
with pytest.raises(KeyboardInterrupt):
|
||||||
|
trio.run(main)
|
|
@ -0,0 +1,82 @@
|
||||||
|
'''
|
||||||
|
Reminders for oddities in `trio` that we need to stay aware of and/or
|
||||||
|
want to see changed.
|
||||||
|
|
||||||
|
'''
|
||||||
|
import pytest
|
||||||
|
import trio
|
||||||
|
from trio_typing import TaskStatus
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
'use_start_soon', [
|
||||||
|
pytest.param(
|
||||||
|
True,
|
||||||
|
marks=pytest.mark.xfail(reason="see python-trio/trio#2258")
|
||||||
|
),
|
||||||
|
False,
|
||||||
|
]
|
||||||
|
)
|
||||||
|
def test_stashed_child_nursery(use_start_soon):
|
||||||
|
|
||||||
|
_child_nursery = None
|
||||||
|
|
||||||
|
async def waits_on_signal(
|
||||||
|
ev: trio.Event(),
|
||||||
|
task_status: TaskStatus[trio.Nursery] = trio.TASK_STATUS_IGNORED,
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Do some stuf, then signal other tasks, then yield back to "starter".
|
||||||
|
|
||||||
|
'''
|
||||||
|
await ev.wait()
|
||||||
|
task_status.started()
|
||||||
|
|
||||||
|
async def mk_child_nursery(
|
||||||
|
task_status: TaskStatus = trio.TASK_STATUS_IGNORED,
|
||||||
|
):
|
||||||
|
'''
|
||||||
|
Allocate a child sub-nursery and stash it as a global.
|
||||||
|
|
||||||
|
'''
|
||||||
|
nonlocal _child_nursery
|
||||||
|
|
||||||
|
async with trio.open_nursery() as cn:
|
||||||
|
_child_nursery = cn
|
||||||
|
task_status.started(cn)
|
||||||
|
|
||||||
|
# block until cancelled by parent.
|
||||||
|
await trio.sleep_forever()
|
||||||
|
|
||||||
|
async def sleep_and_err(
|
||||||
|
ev: trio.Event,
|
||||||
|
task_status: TaskStatus = trio.TASK_STATUS_IGNORED,
|
||||||
|
):
|
||||||
|
await trio.sleep(0.5)
|
||||||
|
doggy() # noqa
|
||||||
|
ev.set()
|
||||||
|
task_status.started()
|
||||||
|
|
||||||
|
async def main():
|
||||||
|
|
||||||
|
async with (
|
||||||
|
trio.open_nursery() as pn,
|
||||||
|
):
|
||||||
|
cn = await pn.start(mk_child_nursery)
|
||||||
|
assert cn
|
||||||
|
|
||||||
|
ev = trio.Event()
|
||||||
|
|
||||||
|
if use_start_soon:
|
||||||
|
# this causes inf hang
|
||||||
|
cn.start_soon(sleep_and_err, ev)
|
||||||
|
|
||||||
|
else:
|
||||||
|
# this does not.
|
||||||
|
await cn.start(sleep_and_err, ev)
|
||||||
|
|
||||||
|
with trio.fail_after(1):
|
||||||
|
await cn.start(waits_on_signal, ev)
|
||||||
|
|
||||||
|
with pytest.raises(NameError):
|
||||||
|
trio.run(main)
|
|
@ -1,182 +1,86 @@
|
||||||
"""
|
# tractor: structured concurrent "actors".
|
||||||
tractor: An actor model micro-framework built on
|
# Copyright 2018-eternity Tyler Goodlet.
|
||||||
``trio`` and ``multiprocessing``.
|
|
||||||
"""
|
|
||||||
import importlib
|
|
||||||
from functools import partial
|
|
||||||
from typing import Tuple, Any, Optional, List
|
|
||||||
import typing
|
|
||||||
|
|
||||||
import trio # type: ignore
|
# This program is free software: you can redistribute it and/or modify
|
||||||
from trio import MultiError
|
# 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.
|
||||||
|
|
||||||
from . import log
|
# This program is distributed in the hope that it will be useful,
|
||||||
from ._ipc import _connect_chan, Channel
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
from ._streaming import Context, stream
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
from ._discovery import get_arbiter, find_actor, wait_for_actor
|
# GNU Affero General Public License for more details.
|
||||||
from ._actor import Actor, _start_actor, Arbiter
|
|
||||||
from ._trionics import open_nursery
|
# You should have received a copy of the GNU Affero General Public License
|
||||||
from ._state import current_actor, is_root_process
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
from . import _state
|
|
||||||
from ._exceptions import RemoteActorError, ModuleNotExposed
|
"""
|
||||||
from ._debug import breakpoint, post_mortem
|
tractor: structured concurrent "actors".
|
||||||
from . import _spawn
|
|
||||||
|
"""
|
||||||
|
from exceptiongroup import BaseExceptionGroup
|
||||||
|
|
||||||
|
from ._clustering import open_actor_cluster
|
||||||
|
from ._ipc import Channel
|
||||||
|
from ._streaming import (
|
||||||
|
Context,
|
||||||
|
MsgStream,
|
||||||
|
stream,
|
||||||
|
context,
|
||||||
|
)
|
||||||
|
from ._discovery import (
|
||||||
|
get_arbiter,
|
||||||
|
find_actor,
|
||||||
|
wait_for_actor,
|
||||||
|
query_actor,
|
||||||
|
)
|
||||||
|
from ._supervise import open_nursery
|
||||||
|
from ._state import (
|
||||||
|
current_actor,
|
||||||
|
is_root_process,
|
||||||
|
)
|
||||||
|
from ._exceptions import (
|
||||||
|
RemoteActorError,
|
||||||
|
ModuleNotExposed,
|
||||||
|
ContextCancelled,
|
||||||
|
)
|
||||||
|
from ._debug import (
|
||||||
|
breakpoint,
|
||||||
|
post_mortem,
|
||||||
|
)
|
||||||
from . import msg
|
from . import msg
|
||||||
|
from ._root import (
|
||||||
|
run_daemon,
|
||||||
|
open_root_actor,
|
||||||
|
)
|
||||||
|
from ._portal import Portal
|
||||||
|
from ._runtime import Actor
|
||||||
|
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
|
'Actor',
|
||||||
|
'Channel',
|
||||||
|
'Context',
|
||||||
|
'ContextCancelled',
|
||||||
|
'ModuleNotExposed',
|
||||||
|
'MsgStream',
|
||||||
|
'BaseExceptionGroup',
|
||||||
|
'Portal',
|
||||||
|
'RemoteActorError',
|
||||||
'breakpoint',
|
'breakpoint',
|
||||||
'post_mortem',
|
'context',
|
||||||
'current_actor',
|
'current_actor',
|
||||||
'find_actor',
|
'find_actor',
|
||||||
'get_arbiter',
|
'get_arbiter',
|
||||||
|
'is_root_process',
|
||||||
|
'msg',
|
||||||
|
'open_actor_cluster',
|
||||||
'open_nursery',
|
'open_nursery',
|
||||||
'wait_for_actor',
|
'open_root_actor',
|
||||||
'Channel',
|
'post_mortem',
|
||||||
'Context',
|
'query_actor',
|
||||||
|
'run_daemon',
|
||||||
'stream',
|
'stream',
|
||||||
'MultiError',
|
'to_asyncio',
|
||||||
'RemoteActorError',
|
'wait_for_actor',
|
||||||
'ModuleNotExposed',
|
|
||||||
'msg'
|
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
# set at startup and after forks
|
|
||||||
_default_arbiter_host = '127.0.0.1'
|
|
||||||
_default_arbiter_port = 1616
|
|
||||||
|
|
||||||
|
|
||||||
async def _main(
|
|
||||||
async_fn: typing.Callable[..., typing.Awaitable],
|
|
||||||
args: Tuple,
|
|
||||||
arbiter_addr: Tuple[str, int],
|
|
||||||
name: Optional[str] = None,
|
|
||||||
start_method: Optional[str] = None,
|
|
||||||
debug_mode: bool = False,
|
|
||||||
**kwargs,
|
|
||||||
) -> typing.Any:
|
|
||||||
"""Async entry point for ``tractor``.
|
|
||||||
"""
|
|
||||||
logger = log.get_logger('tractor')
|
|
||||||
|
|
||||||
# mark top most level process as root actor
|
|
||||||
_state._runtime_vars['_is_root'] = True
|
|
||||||
|
|
||||||
if start_method is not None:
|
|
||||||
_spawn.try_set_start_method(start_method)
|
|
||||||
|
|
||||||
if debug_mode and _spawn._spawn_method == 'trio':
|
|
||||||
_state._runtime_vars['_debug_mode'] = True
|
|
||||||
|
|
||||||
# expose internal debug module to every actor allowing
|
|
||||||
# for use of ``await tractor.breakpoint()``
|
|
||||||
kwargs.setdefault('rpc_module_paths', []).append('tractor._debug')
|
|
||||||
|
|
||||||
elif debug_mode:
|
|
||||||
raise RuntimeError(
|
|
||||||
"Debug mode is only supported for the `trio` backend!"
|
|
||||||
)
|
|
||||||
|
|
||||||
main = partial(async_fn, *args)
|
|
||||||
|
|
||||||
arbiter_addr = (host, port) = arbiter_addr or (
|
|
||||||
_default_arbiter_host,
|
|
||||||
_default_arbiter_port
|
|
||||||
)
|
|
||||||
|
|
||||||
loglevel = kwargs.get('loglevel', log.get_loglevel())
|
|
||||||
if loglevel is not None:
|
|
||||||
log._default_loglevel = loglevel
|
|
||||||
log.get_console_log(loglevel)
|
|
||||||
|
|
||||||
# make a temporary connection to see if an arbiter exists
|
|
||||||
arbiter_found = False
|
|
||||||
try:
|
|
||||||
async with _connect_chan(host, port):
|
|
||||||
arbiter_found = True
|
|
||||||
except OSError:
|
|
||||||
logger.warning(f"No actor could be found @ {host}:{port}")
|
|
||||||
|
|
||||||
# create a local actor and start up its main routine/task
|
|
||||||
if arbiter_found: # we were able to connect to an arbiter
|
|
||||||
logger.info(f"Arbiter seems to exist @ {host}:{port}")
|
|
||||||
actor = Actor(
|
|
||||||
name or 'anonymous',
|
|
||||||
arbiter_addr=arbiter_addr,
|
|
||||||
**kwargs
|
|
||||||
)
|
|
||||||
host, port = (host, 0)
|
|
||||||
else:
|
|
||||||
# start this local actor as the arbiter
|
|
||||||
actor = Arbiter(
|
|
||||||
name or 'arbiter', arbiter_addr=arbiter_addr, **kwargs)
|
|
||||||
|
|
||||||
# ``Actor._async_main()`` creates an internal nursery if one is not
|
|
||||||
# provided and thus blocks here until it's main task completes.
|
|
||||||
# Note that if the current actor is the arbiter it is desirable
|
|
||||||
# for it to stay up indefinitely until a re-election process has
|
|
||||||
# taken place - which is not implemented yet FYI).
|
|
||||||
|
|
||||||
try:
|
|
||||||
return await _start_actor(
|
|
||||||
actor, main, host, port, arbiter_addr=arbiter_addr
|
|
||||||
)
|
|
||||||
finally:
|
|
||||||
logger.info("Root actor terminated")
|
|
||||||
|
|
||||||
|
|
||||||
def run(
|
|
||||||
async_fn: typing.Callable[..., typing.Awaitable],
|
|
||||||
*args,
|
|
||||||
name: Optional[str] = None,
|
|
||||||
arbiter_addr: Tuple[str, int] = (
|
|
||||||
_default_arbiter_host,
|
|
||||||
_default_arbiter_port,
|
|
||||||
),
|
|
||||||
# either the `multiprocessing` start method:
|
|
||||||
# https://docs.python.org/3/library/multiprocessing.html#contexts-and-start-methods
|
|
||||||
# OR `trio` (the new default).
|
|
||||||
start_method: Optional[str] = None,
|
|
||||||
debug_mode: bool = False,
|
|
||||||
**kwargs,
|
|
||||||
) -> Any:
|
|
||||||
"""Run a trio-actor async function in process.
|
|
||||||
|
|
||||||
This is tractor's main entry and the start point for any async actor.
|
|
||||||
"""
|
|
||||||
return trio.run(
|
|
||||||
partial(
|
|
||||||
# our entry
|
|
||||||
_main,
|
|
||||||
|
|
||||||
# user entry point
|
|
||||||
async_fn,
|
|
||||||
args,
|
|
||||||
|
|
||||||
# global kwargs
|
|
||||||
arbiter_addr=arbiter_addr,
|
|
||||||
name=name,
|
|
||||||
start_method=start_method,
|
|
||||||
debug_mode=debug_mode,
|
|
||||||
**kwargs,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def run_daemon(
|
|
||||||
rpc_module_paths: List[str],
|
|
||||||
**kwargs
|
|
||||||
) -> None:
|
|
||||||
"""Spawn daemon actor which will respond to RPC.
|
|
||||||
|
|
||||||
This is a convenience wrapper around
|
|
||||||
``tractor.run(trio.sleep(float('inf')))`` such that the first actor spawned
|
|
||||||
is meant to run forever responding to RPC requests.
|
|
||||||
"""
|
|
||||||
kwargs['rpc_module_paths'] = list(rpc_module_paths)
|
|
||||||
|
|
||||||
for path in rpc_module_paths:
|
|
||||||
importlib.import_module(path)
|
|
||||||
|
|
||||||
return run(partial(trio.sleep, float('inf')), **kwargs)
|
|
||||||
|
|
1118
tractor/_actor.py
1118
tractor/_actor.py
File diff suppressed because it is too large
Load Diff
|
@ -1,4 +1,22 @@
|
||||||
"""This is the "bootloader" for actors started using the native trio backend.
|
# 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/>.
|
||||||
|
|
||||||
|
"""
|
||||||
|
This is the "bootloader" for actors started using the native trio backend.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
import sys
|
import sys
|
||||||
import trio
|
import trio
|
||||||
|
@ -6,7 +24,7 @@ import argparse
|
||||||
|
|
||||||
from ast import literal_eval
|
from ast import literal_eval
|
||||||
|
|
||||||
from ._actor import Actor
|
from ._runtime import Actor
|
||||||
from ._entry import _trio_main
|
from ._entry import _trio_main
|
||||||
|
|
||||||
|
|
||||||
|
@ -19,12 +37,15 @@ def parse_ipaddr(arg):
|
||||||
return (str(host), int(port))
|
return (str(host), int(port))
|
||||||
|
|
||||||
|
|
||||||
|
from ._entry import _trio_main
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument("--uid", type=parse_uid)
|
parser.add_argument("--uid", type=parse_uid)
|
||||||
parser.add_argument("--loglevel", type=str)
|
parser.add_argument("--loglevel", type=str)
|
||||||
parser.add_argument("--parent_addr", type=parse_ipaddr)
|
parser.add_argument("--parent_addr", type=parse_ipaddr)
|
||||||
|
parser.add_argument("--asyncio", action='store_true')
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
subactor = Actor(
|
subactor = Actor(
|
||||||
|
@ -36,5 +57,6 @@ if __name__ == "__main__":
|
||||||
|
|
||||||
_trio_main(
|
_trio_main(
|
||||||
subactor,
|
subactor,
|
||||||
parent_addr=args.parent_addr
|
parent_addr=args.parent_addr,
|
||||||
|
infect_asyncio=args.asyncio,
|
||||||
)
|
)
|
|
@ -0,0 +1,74 @@
|
||||||
|
# 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/>.
|
||||||
|
|
||||||
|
'''
|
||||||
|
Actor cluster helpers.
|
||||||
|
|
||||||
|
'''
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from contextlib import asynccontextmanager as acm
|
||||||
|
from multiprocessing import cpu_count
|
||||||
|
from typing import AsyncGenerator, Optional
|
||||||
|
|
||||||
|
import trio
|
||||||
|
import tractor
|
||||||
|
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def open_actor_cluster(
|
||||||
|
modules: list[str],
|
||||||
|
count: int = cpu_count(),
|
||||||
|
names: list[str] | None = None,
|
||||||
|
hard_kill: bool = False,
|
||||||
|
|
||||||
|
# passed through verbatim to ``open_root_actor()``
|
||||||
|
**runtime_kwargs,
|
||||||
|
|
||||||
|
) -> AsyncGenerator[
|
||||||
|
dict[str, tractor.Portal],
|
||||||
|
None,
|
||||||
|
]:
|
||||||
|
|
||||||
|
portals: dict[str, tractor.Portal] = {}
|
||||||
|
|
||||||
|
if not names:
|
||||||
|
names = [f'worker_{i}' for i in range(count)]
|
||||||
|
|
||||||
|
if not len(names) == count:
|
||||||
|
raise ValueError(
|
||||||
|
'Number of names is {len(names)} but count it {count}')
|
||||||
|
|
||||||
|
async with tractor.open_nursery(
|
||||||
|
**runtime_kwargs,
|
||||||
|
) as an:
|
||||||
|
async with trio.open_nursery() as n:
|
||||||
|
uid = tractor.current_actor().uid
|
||||||
|
|
||||||
|
async def _start(name: str) -> None:
|
||||||
|
name = f'{uid[0]}.{name}'
|
||||||
|
portals[name] = await an.start_actor(
|
||||||
|
enable_modules=modules,
|
||||||
|
name=name,
|
||||||
|
)
|
||||||
|
|
||||||
|
for name in names:
|
||||||
|
n.start_soon(_start, name)
|
||||||
|
|
||||||
|
assert len(portals) == count
|
||||||
|
yield portals
|
||||||
|
|
||||||
|
await an.cancel(hard_kill=hard_kill)
|
1021
tractor/_debug.py
1021
tractor/_debug.py
File diff suppressed because it is too large
Load Diff
|
@ -1,9 +1,29 @@
|
||||||
|
# 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/>.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Actor discovery API.
|
Actor discovery API.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
import typing
|
from typing import (
|
||||||
from typing import Tuple, Optional, Union
|
Optional,
|
||||||
from async_generator import asynccontextmanager
|
Union,
|
||||||
|
AsyncGenerator,
|
||||||
|
)
|
||||||
|
from contextlib import asynccontextmanager as acm
|
||||||
|
|
||||||
from ._ipc import _connect_chan, Channel
|
from ._ipc import _connect_chan, Channel
|
||||||
from ._portal import (
|
from ._portal import (
|
||||||
|
@ -14,14 +34,16 @@ from ._portal import (
|
||||||
from ._state import current_actor, _runtime_vars
|
from ._state import current_actor, _runtime_vars
|
||||||
|
|
||||||
|
|
||||||
@asynccontextmanager
|
@acm
|
||||||
async def get_arbiter(
|
async def get_arbiter(
|
||||||
|
|
||||||
host: str,
|
host: str,
|
||||||
port: int,
|
port: int,
|
||||||
) -> typing.AsyncGenerator[Union[Portal, LocalPortal], None]:
|
|
||||||
"""Return a portal instance connected to a local or remote
|
) -> AsyncGenerator[Union[Portal, LocalPortal], None]:
|
||||||
|
'''Return a portal instance connected to a local or remote
|
||||||
arbiter.
|
arbiter.
|
||||||
"""
|
'''
|
||||||
actor = current_actor()
|
actor = current_actor()
|
||||||
|
|
||||||
if not actor:
|
if not actor:
|
||||||
|
@ -33,39 +55,75 @@ async def get_arbiter(
|
||||||
yield LocalPortal(actor, Channel((host, port)))
|
yield LocalPortal(actor, Channel((host, port)))
|
||||||
else:
|
else:
|
||||||
async with _connect_chan(host, port) as chan:
|
async with _connect_chan(host, port) as chan:
|
||||||
|
|
||||||
async with open_portal(chan) as arb_portal:
|
async with open_portal(chan) as arb_portal:
|
||||||
|
|
||||||
yield arb_portal
|
yield arb_portal
|
||||||
|
|
||||||
|
|
||||||
@asynccontextmanager
|
@acm
|
||||||
async def get_root(
|
async def get_root(
|
||||||
**kwargs,
|
**kwargs,
|
||||||
) -> typing.AsyncGenerator[Union[Portal, LocalPortal], None]:
|
) -> AsyncGenerator[Portal, None]:
|
||||||
|
|
||||||
host, port = _runtime_vars['_root_mailbox']
|
host, port = _runtime_vars['_root_mailbox']
|
||||||
assert host is not None
|
assert host is not None
|
||||||
|
|
||||||
async with _connect_chan(host, port) as chan:
|
async with _connect_chan(host, port) as chan:
|
||||||
async with open_portal(chan, **kwargs) as portal:
|
async with open_portal(chan, **kwargs) as portal:
|
||||||
yield portal
|
yield portal
|
||||||
|
|
||||||
|
|
||||||
@asynccontextmanager
|
@acm
|
||||||
async def find_actor(
|
async def query_actor(
|
||||||
name: str,
|
name: str,
|
||||||
arbiter_sockaddr: Tuple[str, int] = None
|
arbiter_sockaddr: Optional[tuple[str, int]] = None,
|
||||||
) -> typing.AsyncGenerator[Optional[Portal], None]:
|
|
||||||
"""Ask the arbiter to find actor(s) by name.
|
|
||||||
|
|
||||||
Returns a connected portal to the last registered matching actor
|
) -> AsyncGenerator[tuple[str, int], None]:
|
||||||
known to the arbiter.
|
'''
|
||||||
"""
|
Simple address lookup for a given actor name.
|
||||||
|
|
||||||
|
Returns the (socket) address or ``None``.
|
||||||
|
|
||||||
|
'''
|
||||||
actor = current_actor()
|
actor = current_actor()
|
||||||
async with get_arbiter(*arbiter_sockaddr or actor._arb_addr) as arb_portal:
|
async with get_arbiter(
|
||||||
sockaddr = await arb_portal.run_from_ns('self', 'find_actor', name=name)
|
*arbiter_sockaddr or actor._arb_addr
|
||||||
|
) as arb_portal:
|
||||||
|
|
||||||
|
sockaddr = await arb_portal.run_from_ns(
|
||||||
|
'self',
|
||||||
|
'find_actor',
|
||||||
|
name=name,
|
||||||
|
)
|
||||||
|
|
||||||
# TODO: return portals to all available actors - for now just
|
# TODO: return portals to all available actors - for now just
|
||||||
# the last one that registered
|
# the last one that registered
|
||||||
if name == 'arbiter' and actor.is_arbiter:
|
if name == 'arbiter' and actor.is_arbiter:
|
||||||
raise RuntimeError("The current actor is the arbiter")
|
raise RuntimeError("The current actor is the arbiter")
|
||||||
elif sockaddr:
|
|
||||||
|
yield sockaddr if sockaddr else None
|
||||||
|
|
||||||
|
|
||||||
|
@acm
|
||||||
|
async def find_actor(
|
||||||
|
name: str,
|
||||||
|
arbiter_sockaddr: tuple[str, int] | None = None
|
||||||
|
|
||||||
|
) -> AsyncGenerator[Optional[Portal], None]:
|
||||||
|
'''
|
||||||
|
Ask the arbiter to find actor(s) by name.
|
||||||
|
|
||||||
|
Returns a connected portal to the last registered matching actor
|
||||||
|
known to the arbiter.
|
||||||
|
|
||||||
|
'''
|
||||||
|
async with query_actor(
|
||||||
|
name=name,
|
||||||
|
arbiter_sockaddr=arbiter_sockaddr,
|
||||||
|
) as sockaddr:
|
||||||
|
|
||||||
|
if sockaddr:
|
||||||
async with _connect_chan(*sockaddr) 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
|
||||||
|
@ -73,19 +131,27 @@ async def find_actor(
|
||||||
yield None
|
yield None
|
||||||
|
|
||||||
|
|
||||||
@asynccontextmanager
|
@acm
|
||||||
async def wait_for_actor(
|
async def wait_for_actor(
|
||||||
name: str,
|
name: str,
|
||||||
arbiter_sockaddr: Tuple[str, int] = None
|
arbiter_sockaddr: tuple[str, int] | None = None
|
||||||
) -> typing.AsyncGenerator[Portal, None]:
|
) -> AsyncGenerator[Portal, None]:
|
||||||
"""Wait on an actor to register with the arbiter.
|
"""Wait on an actor to register with the arbiter.
|
||||||
|
|
||||||
A portal to the first registered actor is returned.
|
A portal to the first registered actor is returned.
|
||||||
"""
|
"""
|
||||||
actor = current_actor()
|
actor = current_actor()
|
||||||
async with get_arbiter(*arbiter_sockaddr or actor._arb_addr) as arb_portal:
|
|
||||||
sockaddrs = await arb_portal.run_from_ns('self', 'wait_for_actor', name=name)
|
async with get_arbiter(
|
||||||
|
*arbiter_sockaddr or actor._arb_addr,
|
||||||
|
) as arb_portal:
|
||||||
|
sockaddrs = await arb_portal.run_from_ns(
|
||||||
|
'self',
|
||||||
|
'wait_for_actor',
|
||||||
|
name=name,
|
||||||
|
)
|
||||||
sockaddr = sockaddrs[-1]
|
sockaddr = sockaddrs[-1]
|
||||||
|
|
||||||
async with _connect_chan(*sockaddr) 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
|
||||||
|
|
|
@ -1,28 +1,64 @@
|
||||||
|
# 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/>.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Process entry points.
|
Sub-process entry points.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
from __future__ import annotations
|
||||||
from functools import partial
|
from functools import partial
|
||||||
from typing import Tuple, Any
|
from typing import (
|
||||||
import signal
|
Any,
|
||||||
|
TYPE_CHECKING,
|
||||||
|
)
|
||||||
|
|
||||||
import trio # type: ignore
|
import trio # type: ignore
|
||||||
|
|
||||||
from .log import get_console_log, get_logger
|
from .log import (
|
||||||
|
get_console_log,
|
||||||
|
get_logger,
|
||||||
|
)
|
||||||
from . import _state
|
from . import _state
|
||||||
|
from .to_asyncio import run_as_asyncio_guest
|
||||||
|
from ._runtime import (
|
||||||
|
async_main,
|
||||||
|
Actor,
|
||||||
|
)
|
||||||
|
|
||||||
|
if TYPE_CHECKING:
|
||||||
|
from ._spawn import SpawnMethodKey
|
||||||
|
|
||||||
|
|
||||||
log = get_logger(__name__)
|
log = get_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def _mp_main(
|
def _mp_main(
|
||||||
actor: 'Actor', # type: ignore
|
|
||||||
accept_addr: Tuple[str, int],
|
actor: Actor, # type: ignore
|
||||||
forkserver_info: Tuple[Any, Any, Any, Any, Any],
|
accept_addr: tuple[str, int],
|
||||||
start_method: str,
|
forkserver_info: tuple[Any, Any, Any, Any, Any],
|
||||||
parent_addr: Tuple[str, int] = None,
|
start_method: SpawnMethodKey,
|
||||||
|
parent_addr: tuple[str, int] | None = None,
|
||||||
|
infect_asyncio: bool = False,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
"""The routine called *after fork* which invokes a fresh ``trio.run``
|
'''
|
||||||
"""
|
The routine called *after fork* which invokes a fresh ``trio.run``
|
||||||
|
|
||||||
|
'''
|
||||||
actor._forkserver_info = forkserver_info
|
actor._forkserver_info = forkserver_info
|
||||||
from ._spawn import try_set_start_method
|
from ._spawn import try_set_start_method
|
||||||
spawn_ctx = try_set_start_method(start_method)
|
spawn_ctx = try_set_start_method(start_method)
|
||||||
|
@ -40,11 +76,16 @@ def _mp_main(
|
||||||
|
|
||||||
log.debug(f"parent_addr is {parent_addr}")
|
log.debug(f"parent_addr is {parent_addr}")
|
||||||
trio_main = partial(
|
trio_main = partial(
|
||||||
actor._async_main,
|
async_main,
|
||||||
|
actor,
|
||||||
accept_addr,
|
accept_addr,
|
||||||
parent_addr=parent_addr
|
parent_addr=parent_addr
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
|
if infect_asyncio:
|
||||||
|
actor._infected_aio = True
|
||||||
|
run_as_asyncio_guest(trio_main)
|
||||||
|
else:
|
||||||
trio.run(trio_main)
|
trio.run(trio_main)
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
pass # handle it the same way trio does?
|
pass # handle it the same way trio does?
|
||||||
|
@ -54,18 +95,18 @@ def _mp_main(
|
||||||
|
|
||||||
|
|
||||||
def _trio_main(
|
def _trio_main(
|
||||||
actor: 'Actor', # type: ignore
|
|
||||||
*,
|
|
||||||
parent_addr: Tuple[str, int] = None,
|
|
||||||
) -> None:
|
|
||||||
"""Entry point for a `trio_run_in_process` subactor.
|
|
||||||
"""
|
|
||||||
# Disable sigint handling in children;
|
|
||||||
# we don't need it thanks to our cancellation machinery.
|
|
||||||
signal.signal(signal.SIGINT, signal.SIG_IGN)
|
|
||||||
|
|
||||||
# TODO: make a global func to set this or is it too hacky?
|
actor: Actor, # type: ignore
|
||||||
# os.environ['PYTHONBREAKPOINT'] = 'tractor._debug.breakpoint'
|
*,
|
||||||
|
parent_addr: tuple[str, int] | None = None,
|
||||||
|
infect_asyncio: bool = False,
|
||||||
|
|
||||||
|
) -> None:
|
||||||
|
'''
|
||||||
|
Entry point for a `trio_run_in_process` subactor.
|
||||||
|
|
||||||
|
'''
|
||||||
|
log.info(f"Started new trio process for {actor.uid}")
|
||||||
|
|
||||||
if actor.loglevel is not None:
|
if actor.loglevel is not None:
|
||||||
log.info(
|
log.info(
|
||||||
|
@ -79,11 +120,16 @@ def _trio_main(
|
||||||
|
|
||||||
log.debug(f"parent_addr is {parent_addr}")
|
log.debug(f"parent_addr is {parent_addr}")
|
||||||
trio_main = partial(
|
trio_main = partial(
|
||||||
actor._async_main,
|
async_main,
|
||||||
|
actor,
|
||||||
parent_addr=parent_addr
|
parent_addr=parent_addr
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
if infect_asyncio:
|
||||||
|
actor._infected_aio = True
|
||||||
|
run_as_asyncio_guest(trio_main)
|
||||||
|
else:
|
||||||
trio.run(trio_main)
|
trio.run(trio_main)
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
log.warning(f"Actor {actor.uid} received KBI")
|
log.warning(f"Actor {actor.uid} received KBI")
|
||||||
|
|
|
@ -1,35 +1,58 @@
|
||||||
|
# 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/>.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Our classy exception set.
|
Our classy exception set.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
from typing import (
|
||||||
|
Any,
|
||||||
|
Optional,
|
||||||
|
Type,
|
||||||
|
)
|
||||||
import importlib
|
import importlib
|
||||||
import builtins
|
import builtins
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
|
import exceptiongroup as eg
|
||||||
import trio
|
import trio
|
||||||
|
|
||||||
|
|
||||||
_this_mod = importlib.import_module(__name__)
|
_this_mod = importlib.import_module(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class ActorFailure(Exception):
|
||||||
|
"General actor failure"
|
||||||
|
|
||||||
|
|
||||||
class RemoteActorError(Exception):
|
class RemoteActorError(Exception):
|
||||||
# TODO: local recontruction of remote exception deats
|
# TODO: local recontruction of remote exception deats
|
||||||
"Remote actor exception bundled locally"
|
"Remote actor exception bundled locally"
|
||||||
def __init__(self, message, type_str, **msgdata):
|
def __init__(
|
||||||
|
self,
|
||||||
|
message: str,
|
||||||
|
suberror_type: Optional[Type[BaseException]] = None,
|
||||||
|
**msgdata
|
||||||
|
|
||||||
|
) -> None:
|
||||||
super().__init__(message)
|
super().__init__(message)
|
||||||
for ns in [builtins, _this_mod, trio]:
|
|
||||||
try:
|
|
||||||
self.type = getattr(ns, type_str)
|
|
||||||
break
|
|
||||||
except AttributeError:
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
self.type = Exception
|
|
||||||
|
|
||||||
|
self.type = suberror_type
|
||||||
self.msgdata = msgdata
|
self.msgdata = msgdata
|
||||||
|
|
||||||
# TODO: a trio.MultiError.catch like context manager
|
|
||||||
# for catching underlying remote errors of a particular type
|
|
||||||
|
|
||||||
|
|
||||||
class InternalActorError(RemoteActorError):
|
class InternalActorError(RemoteActorError):
|
||||||
"""Remote internal ``tractor`` error indicating
|
"""Remote internal ``tractor`` error indicating
|
||||||
|
@ -37,6 +60,14 @@ class InternalActorError(RemoteActorError):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
class TransportClosed(trio.ClosedResourceError):
|
||||||
|
"Underlying channel transport was closed prior to use"
|
||||||
|
|
||||||
|
|
||||||
|
class ContextCancelled(RemoteActorError):
|
||||||
|
"Inter-actor task context cancelled itself on the callee side."
|
||||||
|
|
||||||
|
|
||||||
class NoResult(RuntimeError):
|
class NoResult(RuntimeError):
|
||||||
"No final result is expected for this actor"
|
"No final result is expected for this actor"
|
||||||
|
|
||||||
|
@ -45,24 +76,102 @@ class ModuleNotExposed(ModuleNotFoundError):
|
||||||
"The requested module is not exposed for RPC"
|
"The requested module is not exposed for RPC"
|
||||||
|
|
||||||
|
|
||||||
def pack_error(exc):
|
class NoRuntime(RuntimeError):
|
||||||
|
"The root actor has not been initialized yet"
|
||||||
|
|
||||||
|
|
||||||
|
class StreamOverrun(trio.TooSlowError):
|
||||||
|
"This stream was overrun by sender"
|
||||||
|
|
||||||
|
|
||||||
|
class AsyncioCancelled(Exception):
|
||||||
|
'''
|
||||||
|
Asyncio cancelled translation (non-base) error
|
||||||
|
for use with the ``to_asyncio`` module
|
||||||
|
to be raised in the ``trio`` side task
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
|
||||||
|
def pack_error(
|
||||||
|
exc: BaseException,
|
||||||
|
tb=None,
|
||||||
|
|
||||||
|
) -> dict[str, Any]:
|
||||||
"""Create an "error message" for tranmission over
|
"""Create an "error message" for tranmission over
|
||||||
a channel (aka the wire).
|
a channel (aka the wire).
|
||||||
"""
|
"""
|
||||||
|
if tb:
|
||||||
|
tb_str = ''.join(traceback.format_tb(tb))
|
||||||
|
else:
|
||||||
|
tb_str = traceback.format_exc()
|
||||||
|
|
||||||
return {
|
return {
|
||||||
'error': {
|
'error': {
|
||||||
'tb_str': traceback.format_exc(),
|
'tb_str': tb_str,
|
||||||
'type_str': type(exc).__name__,
|
'type_str': type(exc).__name__,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
def unpack_error(msg, chan=None, err_type=RemoteActorError):
|
def unpack_error(
|
||||||
"""Unpack an 'error' message from the wire
|
|
||||||
|
msg: dict[str, Any],
|
||||||
|
chan=None,
|
||||||
|
err_type=RemoteActorError
|
||||||
|
|
||||||
|
) -> Exception:
|
||||||
|
'''
|
||||||
|
Unpack an 'error' message from the wire
|
||||||
into a local ``RemoteActorError``.
|
into a local ``RemoteActorError``.
|
||||||
"""
|
|
||||||
tb_str = msg['error'].get('tb_str', '')
|
'''
|
||||||
return err_type(
|
__tracebackhide__ = True
|
||||||
f"{chan.uid}\n" + tb_str,
|
error = msg['error']
|
||||||
|
|
||||||
|
tb_str = error.get('tb_str', '')
|
||||||
|
message = f"{chan.uid}\n" + tb_str
|
||||||
|
type_name = error['type_str']
|
||||||
|
suberror_type: Type[BaseException] = Exception
|
||||||
|
|
||||||
|
if type_name == 'ContextCancelled':
|
||||||
|
err_type = ContextCancelled
|
||||||
|
suberror_type = trio.Cancelled
|
||||||
|
|
||||||
|
else: # try to lookup a suitable local error type
|
||||||
|
for ns in [
|
||||||
|
builtins,
|
||||||
|
_this_mod,
|
||||||
|
eg,
|
||||||
|
trio,
|
||||||
|
]:
|
||||||
|
try:
|
||||||
|
suberror_type = getattr(ns, type_name)
|
||||||
|
break
|
||||||
|
except AttributeError:
|
||||||
|
continue
|
||||||
|
|
||||||
|
exc = err_type(
|
||||||
|
message,
|
||||||
|
suberror_type=suberror_type,
|
||||||
|
|
||||||
|
# unpack other fields into error type init
|
||||||
**msg['error'],
|
**msg['error'],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
return exc
|
||||||
|
|
||||||
|
|
||||||
|
def is_multi_cancelled(exc: BaseException) -> bool:
|
||||||
|
'''
|
||||||
|
Predicate to determine if a possible ``eg.BaseExceptionGroup`` contains
|
||||||
|
only ``trio.Cancelled`` sub-exceptions (and is likely the result of
|
||||||
|
cancelling a collection of subtasks.
|
||||||
|
|
||||||
|
'''
|
||||||
|
if isinstance(exc, eg.BaseExceptionGroup):
|
||||||
|
return exc.subgroup(
|
||||||
|
lambda exc: isinstance(exc, trio.Cancelled)
|
||||||
|
) is not None
|
||||||
|
|
||||||
|
return False
|
||||||
|
|
|
@ -1,3 +1,19 @@
|
||||||
|
# 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/>.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
This is near-copy of the 3.8 stdlib's ``multiprocessing.forkserver.py``
|
This is near-copy of the 3.8 stdlib's ``multiprocessing.forkserver.py``
|
||||||
with some hackery to prevent any more then a single forkserver and
|
with some hackery to prevent any more then a single forkserver and
|
||||||
|
@ -6,6 +22,8 @@ semaphore tracker per ``MainProcess``.
|
||||||
.. note:: There is no type hinting in this code base (yet) to remain as
|
.. note:: There is no type hinting in this code base (yet) to remain as
|
||||||
a close as possible to upstream.
|
a close as possible to upstream.
|
||||||
"""
|
"""
|
||||||
|
# type: ignore
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import socket
|
import socket
|
||||||
import signal
|
import signal
|
||||||
|
@ -120,6 +138,7 @@ class PatchedForkServer(ForkServer):
|
||||||
with socket.socket(socket.AF_UNIX) as listener:
|
with socket.socket(socket.AF_UNIX) as listener:
|
||||||
address = connection.arbitrary_address('AF_UNIX')
|
address = connection.arbitrary_address('AF_UNIX')
|
||||||
listener.bind(address)
|
listener.bind(address)
|
||||||
|
if not util.is_abstract_socket_namespace(address):
|
||||||
os.chmod(address, 0o600)
|
os.chmod(address, 0o600)
|
||||||
listener.listen()
|
listener.listen()
|
||||||
|
|
||||||
|
|
485
tractor/_ipc.py
485
tractor/_ipc.py
|
@ -1,84 +1,240 @@
|
||||||
|
# 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
|
Inter-process comms abstractions
|
||||||
"""
|
|
||||||
import typing
|
|
||||||
from typing import Any, Tuple, Optional
|
|
||||||
from functools import partial
|
|
||||||
import inspect
|
|
||||||
|
|
||||||
import msgpack
|
"""
|
||||||
|
from __future__ import annotations
|
||||||
|
import platform
|
||||||
|
import struct
|
||||||
|
import typing
|
||||||
|
from collections.abc import (
|
||||||
|
AsyncGenerator,
|
||||||
|
AsyncIterator,
|
||||||
|
)
|
||||||
|
from typing import (
|
||||||
|
Any,
|
||||||
|
runtime_checkable,
|
||||||
|
Optional,
|
||||||
|
Protocol,
|
||||||
|
Type,
|
||||||
|
TypeVar,
|
||||||
|
)
|
||||||
|
|
||||||
|
from tricycle import BufferedReceiveStream
|
||||||
|
import msgspec
|
||||||
import trio
|
import trio
|
||||||
from async_generator import asynccontextmanager
|
from async_generator import asynccontextmanager
|
||||||
|
|
||||||
from .log import get_logger
|
from .log import get_logger
|
||||||
log = get_logger('ipc')
|
from ._exceptions import TransportClosed
|
||||||
|
log = get_logger(__name__)
|
||||||
# :eyeroll:
|
|
||||||
try:
|
|
||||||
import msgpack_numpy
|
|
||||||
Unpacker = msgpack_numpy.Unpacker
|
|
||||||
except ImportError:
|
|
||||||
# just plain ``msgpack`` requires tweaking key settings
|
|
||||||
Unpacker = partial(msgpack.Unpacker, strict_map_key=False)
|
|
||||||
|
|
||||||
|
|
||||||
class MsgpackStream:
|
_is_windows = platform.system() == 'Windows'
|
||||||
"""A ``trio.SocketStream`` delivering ``msgpack`` formatted data.
|
log = get_logger(__name__)
|
||||||
"""
|
|
||||||
def __init__(self, stream: trio.SocketStream) -> None:
|
|
||||||
self.stream = stream
|
def get_stream_addrs(stream: trio.SocketStream) -> tuple:
|
||||||
assert self.stream.socket
|
|
||||||
# should both be IP sockets
|
# should both be IP sockets
|
||||||
lsockname = stream.socket.getsockname()
|
lsockname = stream.socket.getsockname()
|
||||||
assert isinstance(lsockname, tuple)
|
|
||||||
self._laddr = lsockname[:2]
|
|
||||||
rsockname = stream.socket.getpeername()
|
rsockname = stream.socket.getpeername()
|
||||||
assert isinstance(rsockname, tuple)
|
return (
|
||||||
self._raddr = rsockname[:2]
|
tuple(lsockname[:2]),
|
||||||
|
tuple(rsockname[:2]),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
MsgType = TypeVar("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
|
||||||
|
# - https://jcristharif.com/msgspec/usage.html#structs
|
||||||
|
|
||||||
|
|
||||||
|
@runtime_checkable
|
||||||
|
class MsgTransport(Protocol[MsgType]):
|
||||||
|
|
||||||
|
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: 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.
|
||||||
|
|
||||||
|
'''
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
stream: trio.SocketStream,
|
||||||
|
prefix_size: int = 4,
|
||||||
|
|
||||||
|
) -> 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._agen = self._iter_packets()
|
self._agen = self._iter_packets()
|
||||||
self._send_lock = trio.StrictFIFOLock()
|
self._send_lock = trio.StrictFIFOLock()
|
||||||
|
|
||||||
async def _iter_packets(self) -> typing.AsyncGenerator[dict, None]:
|
# public i guess?
|
||||||
"""Yield packets from the underlying stream.
|
self.drained: list[dict] = []
|
||||||
"""
|
|
||||||
unpacker = Unpacker(
|
self.recv_stream = BufferedReceiveStream(transport_stream=stream)
|
||||||
raw=False,
|
self.prefix_size = prefix_size
|
||||||
use_list=False,
|
|
||||||
)
|
# TODO: struct aware messaging coders
|
||||||
|
self.encode = msgspec.msgpack.Encoder().encode
|
||||||
|
self.decode = msgspec.msgpack.Decoder().decode # dict[str, Any])
|
||||||
|
|
||||||
|
async def _iter_packets(self) -> AsyncGenerator[dict, None]:
|
||||||
|
'''Yield packets from the underlying stream.
|
||||||
|
|
||||||
|
'''
|
||||||
|
import msgspec # noqa
|
||||||
|
decodes_failed: int = 0
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
data = await self.stream.receive_some(2**10)
|
header = await self.recv_stream.receive_exactly(4)
|
||||||
log.trace(f"received {data}") # type: ignore
|
|
||||||
except trio.BrokenResourceError:
|
|
||||||
log.error(f"Stream connection {self.raddr} broke")
|
|
||||||
return
|
|
||||||
|
|
||||||
if data == b'':
|
except (
|
||||||
log.debug(f"Stream connection {self.raddr} was closed")
|
ValueError,
|
||||||
return
|
ConnectionResetError,
|
||||||
|
|
||||||
unpacker.feed(data)
|
# not sure entirely why we need this but without it we
|
||||||
for packet in unpacker:
|
# seem to be getting racy failures here on
|
||||||
yield packet
|
# arbiter/registry name subs..
|
||||||
|
trio.BrokenResourceError,
|
||||||
|
):
|
||||||
|
raise TransportClosed(
|
||||||
|
f'transport {self} was already closed prior ro read'
|
||||||
|
)
|
||||||
|
|
||||||
|
if header == b'':
|
||||||
|
raise TransportClosed(
|
||||||
|
f'transport {self} was already closed prior ro read'
|
||||||
|
)
|
||||||
|
|
||||||
|
size, = struct.unpack("<I", header)
|
||||||
|
|
||||||
|
log.transport(f'received header {size}') # type: ignore
|
||||||
|
|
||||||
|
msg_bytes = await self.recv_stream.receive_exactly(size)
|
||||||
|
|
||||||
|
log.transport(f"received {msg_bytes}") # type: ignore
|
||||||
|
try:
|
||||||
|
yield self.decode(msg_bytes)
|
||||||
|
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.error(
|
||||||
|
'`msgspec` failed to decode!?\n'
|
||||||
|
'dumping bytes:\n'
|
||||||
|
f'{msg_str!r}'
|
||||||
|
)
|
||||||
|
decodes_failed += 1
|
||||||
|
else:
|
||||||
|
raise
|
||||||
|
|
||||||
|
async def send(self, msg: Any) -> None:
|
||||||
|
async with self._send_lock:
|
||||||
|
|
||||||
|
bytes_data: bytes = self.encode(msg)
|
||||||
|
|
||||||
|
# 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)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def laddr(self) -> Tuple[Any, ...]:
|
def laddr(self) -> tuple[str, int]:
|
||||||
return self._laddr
|
return self._laddr
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def raddr(self) -> Tuple[Any, ...]:
|
def raddr(self) -> tuple[str, int]:
|
||||||
return self._raddr
|
return self._raddr
|
||||||
|
|
||||||
# XXX: should this instead be called `.sendall()`?
|
|
||||||
async def send(self, data: Any) -> None:
|
|
||||||
async with self._send_lock:
|
|
||||||
return await self.stream.send_all(
|
|
||||||
msgpack.dumps(data, use_bin_type=True))
|
|
||||||
|
|
||||||
async def recv(self) -> Any:
|
async def recv(self) -> Any:
|
||||||
return await self._agen.asend(None)
|
return await self._agen.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):
|
def __aiter__(self):
|
||||||
return self._agen
|
return self._agen
|
||||||
|
|
||||||
|
@ -86,32 +242,87 @@ class MsgpackStream:
|
||||||
return self.stream.socket.fileno() != -1
|
return self.stream.socket.fileno() != -1
|
||||||
|
|
||||||
|
|
||||||
class Channel:
|
def get_msg_transport(
|
||||||
"""An inter-process channel for communication between (remote) actors.
|
|
||||||
|
|
||||||
Currently the only supported transport is a ``trio.SocketStream``.
|
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__(
|
def __init__(
|
||||||
|
|
||||||
self,
|
self,
|
||||||
destaddr: Optional[Tuple[str, int]] = None,
|
destaddr: Optional[tuple[str, int]],
|
||||||
on_reconnect: typing.Callable[..., typing.Awaitable] = None,
|
|
||||||
auto_reconnect: bool = False,
|
msg_transport_type_key: tuple[str, str] = ('msgpack', 'tcp'),
|
||||||
stream: trio.SocketStream = None, # expected to be active
|
|
||||||
|
# TODO: optional reconnection support?
|
||||||
|
# auto_reconnect: bool = False,
|
||||||
|
# on_reconnect: typing.Callable[..., typing.Awaitable] = None,
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
self._recon_seq = on_reconnect
|
|
||||||
self._autorecon = auto_reconnect
|
# self._recon_seq = on_reconnect
|
||||||
self.msgstream: Optional[MsgpackStream] = MsgpackStream(
|
# self._autorecon = auto_reconnect
|
||||||
stream) if stream else None
|
|
||||||
if self.msgstream and destaddr:
|
self._destaddr = destaddr
|
||||||
raise ValueError(
|
self._transport_key = msg_transport_type_key
|
||||||
f"A stream was provided with local addr {self.laddr}"
|
|
||||||
)
|
# Either created in ``.connect()`` or passed in by
|
||||||
self._destaddr = self.msgstream.raddr if self.msgstream else destaddr
|
# user in ``.from_stream()``.
|
||||||
|
self._stream: Optional[trio.SocketStream] = None
|
||||||
|
self.msgstream: Optional[MsgTransport] = None
|
||||||
|
|
||||||
# set after handshake - always uid of far end
|
# set after handshake - always uid of far end
|
||||||
self.uid: Optional[Tuple[str, str]] = None
|
self.uid: Optional[tuple[str, str]] = None
|
||||||
# set if far end actor errors internally
|
|
||||||
self._exc: Optional[Exception] = None
|
|
||||||
self._agen = self._aiter_recv()
|
self._agen = self._aiter_recv()
|
||||||
|
self._exc: Optional[Exception] = None # set if far end actor errors
|
||||||
|
self._closed: bool = False
|
||||||
|
# flag set on ``Portal.cancel_actor()`` indicating
|
||||||
|
# remote (peer) cancellation of the far end actor runtime.
|
||||||
|
self._cancel_called: bool = False # set on ``Portal.cancel_actor()``
|
||||||
|
|
||||||
|
@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 = stream
|
||||||
|
chan.set_msg_transport(stream)
|
||||||
|
return chan
|
||||||
|
|
||||||
|
def set_msg_transport(
|
||||||
|
self,
|
||||||
|
stream: trio.SocketStream,
|
||||||
|
type_key: Optional[tuple[str, str]] = None,
|
||||||
|
|
||||||
|
) -> MsgTransport:
|
||||||
|
type_key = type_key or self._transport_key
|
||||||
|
self.msgstream = get_msg_transport(type_key)(stream)
|
||||||
|
return self.msgstream
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
if self.msgstream:
|
if self.msgstream:
|
||||||
|
@ -121,43 +332,65 @@ class Channel:
|
||||||
return object.__repr__(self)
|
return object.__repr__(self)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def laddr(self) -> Optional[Tuple[Any, ...]]:
|
def laddr(self) -> Optional[tuple[str, int]]:
|
||||||
return self.msgstream.laddr if self.msgstream else None
|
return self.msgstream.laddr if self.msgstream else None
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def raddr(self) -> Optional[Tuple[Any, ...]]:
|
def raddr(self) -> Optional[tuple[str, int]]:
|
||||||
return self.msgstream.raddr if self.msgstream else None
|
return self.msgstream.raddr if self.msgstream else None
|
||||||
|
|
||||||
async def connect(
|
async def connect(
|
||||||
self, destaddr: Tuple[Any, ...] = None,
|
self,
|
||||||
|
destaddr: tuple[Any, ...] | None = None,
|
||||||
**kwargs
|
**kwargs
|
||||||
) -> trio.SocketStream:
|
|
||||||
|
) -> MsgTransport:
|
||||||
|
|
||||||
if self.connected():
|
if self.connected():
|
||||||
raise RuntimeError("channel is already connected?")
|
raise RuntimeError("channel is already connected?")
|
||||||
|
|
||||||
destaddr = destaddr or self._destaddr
|
destaddr = destaddr or self._destaddr
|
||||||
assert isinstance(destaddr, tuple)
|
assert isinstance(destaddr, tuple)
|
||||||
stream = await trio.open_tcp_stream(*destaddr, **kwargs)
|
|
||||||
self.msgstream = MsgpackStream(stream)
|
stream = await trio.open_tcp_stream(
|
||||||
return stream
|
*destaddr,
|
||||||
|
**kwargs
|
||||||
|
)
|
||||||
|
msgstream = self.set_msg_transport(stream)
|
||||||
|
|
||||||
|
log.transport(
|
||||||
|
f'Opened channel[{type(msgstream)}]: {self.laddr} -> {self.raddr}'
|
||||||
|
)
|
||||||
|
return msgstream
|
||||||
|
|
||||||
async def send(self, item: Any) -> None:
|
async def send(self, item: Any) -> None:
|
||||||
log.trace(f"send `{item}`") # type: ignore
|
|
||||||
|
log.transport(f"send `{item}`") # type: ignore
|
||||||
assert self.msgstream
|
assert self.msgstream
|
||||||
|
|
||||||
await self.msgstream.send(item)
|
await self.msgstream.send(item)
|
||||||
|
|
||||||
async def recv(self) -> Any:
|
async def recv(self) -> Any:
|
||||||
assert self.msgstream
|
assert self.msgstream
|
||||||
try:
|
|
||||||
return await self.msgstream.recv()
|
return await self.msgstream.recv()
|
||||||
except trio.BrokenResourceError:
|
|
||||||
if self._autorecon:
|
# try:
|
||||||
await self._reconnect()
|
# return await self.msgstream.recv()
|
||||||
return await self.recv()
|
# except trio.BrokenResourceError:
|
||||||
|
# if self._autorecon:
|
||||||
|
# await self._reconnect()
|
||||||
|
# return await self.recv()
|
||||||
|
# raise
|
||||||
|
|
||||||
async def aclose(self) -> None:
|
async def aclose(self) -> None:
|
||||||
log.debug(f"Closing {self}")
|
|
||||||
|
log.transport(
|
||||||
|
f'Closing channel to {self.uid} '
|
||||||
|
f'{self.laddr} -> {self.raddr}'
|
||||||
|
)
|
||||||
assert self.msgstream
|
assert self.msgstream
|
||||||
await self.msgstream.stream.aclose()
|
await self.msgstream.stream.aclose()
|
||||||
|
self._closed = True
|
||||||
|
|
||||||
async def __aenter__(self):
|
async def __aenter__(self):
|
||||||
await self.connect()
|
await self.connect()
|
||||||
|
@ -169,40 +402,44 @@ class Channel:
|
||||||
def __aiter__(self):
|
def __aiter__(self):
|
||||||
return self._agen
|
return self._agen
|
||||||
|
|
||||||
async def _reconnect(self) -> None:
|
# async def _reconnect(self) -> None:
|
||||||
"""Handle connection failures by polling until a reconnect can be
|
# """Handle connection failures by polling until a reconnect can be
|
||||||
established.
|
# established.
|
||||||
"""
|
# """
|
||||||
down = False
|
# down = False
|
||||||
while True:
|
# while True:
|
||||||
try:
|
# try:
|
||||||
with trio.move_on_after(3) as cancel_scope:
|
# with trio.move_on_after(3) as cancel_scope:
|
||||||
await self.connect()
|
# await self.connect()
|
||||||
cancelled = cancel_scope.cancelled_caught
|
# cancelled = cancel_scope.cancelled_caught
|
||||||
if cancelled:
|
# if cancelled:
|
||||||
log.warning(
|
# log.transport(
|
||||||
"Reconnect timed out after 3 seconds, retrying...")
|
# "Reconnect timed out after 3 seconds, retrying...")
|
||||||
continue
|
# continue
|
||||||
else:
|
# else:
|
||||||
log.warning("Stream connection re-established!")
|
# log.transport("Stream connection re-established!")
|
||||||
# run any reconnection sequence
|
|
||||||
on_recon = self._recon_seq
|
# # TODO: run any reconnection sequence
|
||||||
if on_recon:
|
# # on_recon = self._recon_seq
|
||||||
await on_recon(self)
|
# # if on_recon:
|
||||||
break
|
# # await on_recon(self)
|
||||||
except (OSError, ConnectionRefusedError):
|
|
||||||
if not down:
|
# break
|
||||||
down = True
|
# except (OSError, ConnectionRefusedError):
|
||||||
log.warning(
|
# if not down:
|
||||||
f"Connection to {self.raddr} went down, waiting"
|
# down = True
|
||||||
" for re-establishment")
|
# log.transport(
|
||||||
await trio.sleep(1)
|
# f"Connection to {self.raddr} went down, waiting"
|
||||||
|
# " for re-establishment")
|
||||||
|
# await trio.sleep(1)
|
||||||
|
|
||||||
async def _aiter_recv(
|
async def _aiter_recv(
|
||||||
self
|
self
|
||||||
) -> typing.AsyncGenerator[Any, None]:
|
) -> AsyncGenerator[Any, None]:
|
||||||
"""Async iterate items from underlying stream.
|
'''
|
||||||
"""
|
Async iterate items from underlying stream.
|
||||||
|
|
||||||
|
'''
|
||||||
assert self.msgstream
|
assert self.msgstream
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
|
@ -215,16 +452,14 @@ class Channel:
|
||||||
# await self.msgstream.send(sent)
|
# await self.msgstream.send(sent)
|
||||||
except trio.BrokenResourceError:
|
except trio.BrokenResourceError:
|
||||||
|
|
||||||
if not self._autorecon:
|
# if not self._autorecon:
|
||||||
raise
|
raise
|
||||||
|
|
||||||
await self.aclose()
|
await self.aclose()
|
||||||
|
|
||||||
if self._autorecon: # attempt reconnect
|
# if self._autorecon: # attempt reconnect
|
||||||
await self._reconnect()
|
# await self._reconnect()
|
||||||
continue
|
# continue
|
||||||
else:
|
|
||||||
return
|
|
||||||
|
|
||||||
def connected(self) -> bool:
|
def connected(self) -> bool:
|
||||||
return self.msgstream.connected() if self.msgstream else False
|
return self.msgstream.connected() if self.msgstream else False
|
||||||
|
@ -234,9 +469,11 @@ class Channel:
|
||||||
async def _connect_chan(
|
async def _connect_chan(
|
||||||
host: str, port: int
|
host: str, port: int
|
||||||
) -> typing.AsyncGenerator[Channel, None]:
|
) -> typing.AsyncGenerator[Channel, None]:
|
||||||
"""Create and connect a channel with disconnect on context manager
|
'''
|
||||||
|
Create and connect a channel with disconnect on context manager
|
||||||
teardown.
|
teardown.
|
||||||
"""
|
|
||||||
|
'''
|
||||||
chan = Channel((host, port))
|
chan = Channel((host, port))
|
||||||
await chan.connect()
|
await chan.connect()
|
||||||
yield chan
|
yield chan
|
||||||
|
|
|
@ -1,23 +1,39 @@
|
||||||
|
# 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/>.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Helpers pulled mostly verbatim from ``multiprocessing.spawn``
|
Helpers pulled mostly verbatim from ``multiprocessing.spawn``
|
||||||
to aid with "fixing up" the ``__main__`` module in subprocesses.
|
to aid with "fixing up" the ``__main__`` module in subprocesses.
|
||||||
|
|
||||||
These helpers are needed for any spawing backend that doesn't already handle this.
|
These helpers are needed for any spawing backend that doesn't already
|
||||||
For example when using ``trio_run_in_process`` it is needed but obviously not when
|
handle this. For example when using ``trio_run_in_process`` it is needed
|
||||||
we're already using ``multiprocessing``.
|
but obviously not when we're already using ``multiprocessing``.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import platform
|
import platform
|
||||||
import types
|
import types
|
||||||
import runpy
|
import runpy
|
||||||
from typing import Dict
|
|
||||||
|
|
||||||
|
|
||||||
ORIGINAL_DIR = os.path.abspath(os.getcwd())
|
ORIGINAL_DIR = os.path.abspath(os.getcwd())
|
||||||
|
|
||||||
|
|
||||||
def _mp_figure_out_main() -> Dict[str, str]:
|
def _mp_figure_out_main() -> dict[str, str]:
|
||||||
"""Taken from ``multiprocessing.spawn.get_preparation_data()``.
|
"""Taken from ``multiprocessing.spawn.get_preparation_data()``.
|
||||||
|
|
||||||
Retrieve parent actor `__main__` module data.
|
Retrieve parent actor `__main__` module data.
|
||||||
|
@ -67,7 +83,7 @@ def _fixup_main_from_name(mod_name: str) -> None:
|
||||||
main_module = types.ModuleType("__mp_main__")
|
main_module = types.ModuleType("__mp_main__")
|
||||||
main_content = runpy.run_module(mod_name,
|
main_content = runpy.run_module(mod_name,
|
||||||
run_name="__mp_main__",
|
run_name="__mp_main__",
|
||||||
alter_sys=True)
|
alter_sys=True) # type: ignore
|
||||||
main_module.__dict__.update(main_content)
|
main_module.__dict__.update(main_content)
|
||||||
sys.modules['__main__'] = sys.modules['__mp_main__'] = main_module
|
sys.modules['__main__'] = sys.modules['__mp_main__'] = main_module
|
||||||
|
|
||||||
|
@ -95,6 +111,6 @@ def _fixup_main_from_path(main_path: str) -> None:
|
||||||
# old_main_modules.append(current_main)
|
# old_main_modules.append(current_main)
|
||||||
main_module = types.ModuleType("__mp_main__")
|
main_module = types.ModuleType("__mp_main__")
|
||||||
main_content = runpy.run_path(main_path,
|
main_content = runpy.run_path(main_path,
|
||||||
run_name="__mp_main__")
|
run_name="__mp_main__") # type: ignore
|
||||||
main_module.__dict__.update(main_content)
|
main_module.__dict__.update(main_content)
|
||||||
sys.modules['__main__'] = sys.modules['__mp_main__'] = main_module
|
sys.modules['__main__'] = sys.modules['__mp_main__'] = main_module
|
||||||
|
|
|
@ -1,287 +1,145 @@
|
||||||
"""
|
# tractor: structured concurrent "actors".
|
||||||
Portal api
|
# 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/>.
|
||||||
|
|
||||||
|
'''
|
||||||
|
Memory boundary "Portals": an API for structured
|
||||||
|
concurrency linked tasks running in disparate memory domains.
|
||||||
|
|
||||||
|
'''
|
||||||
|
from __future__ import annotations
|
||||||
import importlib
|
import importlib
|
||||||
import inspect
|
import inspect
|
||||||
import typing
|
from typing import (
|
||||||
from typing import Tuple, Any, Dict, Optional, Set, Iterator
|
Any, Optional,
|
||||||
|
Callable, AsyncGenerator,
|
||||||
|
Type,
|
||||||
|
)
|
||||||
from functools import partial
|
from functools import partial
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from contextlib import contextmanager
|
from pprint import pformat
|
||||||
import warnings
|
import warnings
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
from async_generator import asynccontextmanager
|
from async_generator import asynccontextmanager
|
||||||
|
|
||||||
|
from .trionics import maybe_open_nursery
|
||||||
from ._state import current_actor
|
from ._state import current_actor
|
||||||
from ._ipc import Channel
|
from ._ipc import Channel
|
||||||
from .log import get_logger
|
from .log import get_logger
|
||||||
from ._exceptions import unpack_error, NoResult, RemoteActorError
|
from .msg import NamespacePath
|
||||||
|
from ._exceptions import (
|
||||||
|
unpack_error,
|
||||||
log = get_logger('tractor')
|
NoResult,
|
||||||
|
ContextCancelled,
|
||||||
|
|
||||||
@asynccontextmanager
|
|
||||||
async def maybe_open_nursery(
|
|
||||||
nursery: trio.Nursery = None,
|
|
||||||
shield: bool = False,
|
|
||||||
) -> typing.AsyncGenerator[trio.Nursery, Any]:
|
|
||||||
"""Create a new nursery if None provided.
|
|
||||||
|
|
||||||
Blocks on exit as expected if no input nursery is provided.
|
|
||||||
"""
|
|
||||||
if nursery is not None:
|
|
||||||
yield nursery
|
|
||||||
else:
|
|
||||||
async with trio.open_nursery() as nursery:
|
|
||||||
nursery.cancel_scope.shield = shield
|
|
||||||
yield nursery
|
|
||||||
|
|
||||||
|
|
||||||
class ReceiveStream(trio.abc.ReceiveChannel):
|
|
||||||
"""A wrapper around a ``trio._channel.MemoryReceiveChannel`` with
|
|
||||||
special behaviour for signalling stream termination across an
|
|
||||||
inter-actor ``Channel``. This is the type returned to a local task
|
|
||||||
which invoked a remote streaming function using `Portal.run()`.
|
|
||||||
|
|
||||||
Termination rules:
|
|
||||||
- if the local task signals stop iteration a cancel signal is
|
|
||||||
relayed to the remote task indicating to stop streaming
|
|
||||||
- if the remote task signals the end of a stream, raise a
|
|
||||||
``StopAsyncIteration`` to terminate the local ``async for``
|
|
||||||
|
|
||||||
"""
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
cid: str,
|
|
||||||
rx_chan: trio.abc.ReceiveChannel,
|
|
||||||
portal: 'Portal',
|
|
||||||
) -> None:
|
|
||||||
self._cid = cid
|
|
||||||
self._rx_chan = rx_chan
|
|
||||||
self._portal = portal
|
|
||||||
self._shielded = False
|
|
||||||
|
|
||||||
# delegate directly to underlying mem channel
|
|
||||||
def receive_nowait(self):
|
|
||||||
return self._rx_chan.receive_nowait()
|
|
||||||
|
|
||||||
async def receive(self):
|
|
||||||
try:
|
|
||||||
msg = await self._rx_chan.receive()
|
|
||||||
return msg['yield']
|
|
||||||
except trio.ClosedResourceError:
|
|
||||||
# when the send is closed we assume the stream has
|
|
||||||
# terminated and signal this local iterator to stop
|
|
||||||
await self.aclose()
|
|
||||||
raise StopAsyncIteration
|
|
||||||
except trio.Cancelled:
|
|
||||||
# relay cancels to the remote task
|
|
||||||
await self.aclose()
|
|
||||||
raise
|
|
||||||
except KeyError:
|
|
||||||
# internal error should never get here
|
|
||||||
assert msg.get('cid'), (
|
|
||||||
"Received internal error at portal?")
|
|
||||||
raise unpack_error(msg, self._portal.channel)
|
|
||||||
|
|
||||||
@contextmanager
|
|
||||||
def shield(
|
|
||||||
self
|
|
||||||
) -> Iterator['ReceiveStream']: # noqa
|
|
||||||
"""Shield this stream's underlying channel such that a local consumer task
|
|
||||||
can be cancelled (and possibly restarted) using ``trio.Cancelled``.
|
|
||||||
|
|
||||||
"""
|
|
||||||
self._shielded = True
|
|
||||||
yield self
|
|
||||||
self._shielded = False
|
|
||||||
|
|
||||||
async def aclose(self):
|
|
||||||
"""Cancel associated remote actor task and local memory channel
|
|
||||||
on close.
|
|
||||||
"""
|
|
||||||
if self._rx_chan._closed:
|
|
||||||
log.warning(f"{self} is already closed")
|
|
||||||
return
|
|
||||||
|
|
||||||
if self._shielded:
|
|
||||||
log.warning(f"{self} is shielded, portal channel being kept alive")
|
|
||||||
return
|
|
||||||
|
|
||||||
cid = self._cid
|
|
||||||
with trio.move_on_after(0.5) as cs:
|
|
||||||
cs.shield = True
|
|
||||||
log.warning(
|
|
||||||
f"Cancelling stream {cid} to "
|
|
||||||
f"{self._portal.channel.uid}")
|
|
||||||
|
|
||||||
# NOTE: we're telling the far end actor to cancel a task
|
|
||||||
# corresponding to *this actor*. The far end local channel
|
|
||||||
# instance is passed to `Actor._cancel_task()` implicitly.
|
|
||||||
await self._portal.run_from_ns('self', '_cancel_task', cid=cid)
|
|
||||||
|
|
||||||
if cs.cancelled_caught:
|
|
||||||
# XXX: there's no way to know if the remote task was indeed
|
|
||||||
# cancelled in the case where the connection is broken or
|
|
||||||
# some other network error occurred.
|
|
||||||
if not self._portal.channel.connected():
|
|
||||||
log.warning(
|
|
||||||
"May have failed to cancel remote task "
|
|
||||||
f"{cid} for {self._portal.channel.uid}")
|
|
||||||
|
|
||||||
with trio.CancelScope(shield=True):
|
|
||||||
await self._rx_chan.aclose()
|
|
||||||
|
|
||||||
def clone(self):
|
|
||||||
return self
|
|
||||||
|
|
||||||
|
|
||||||
class Portal:
|
|
||||||
"""A 'portal' to a(n) (remote) ``Actor``.
|
|
||||||
|
|
||||||
Allows for invoking remote routines and receiving results through an
|
|
||||||
underlying ``tractor.Channel`` as though the remote (async)
|
|
||||||
function / generator was invoked locally.
|
|
||||||
|
|
||||||
Think of this like a native async IPC API.
|
|
||||||
"""
|
|
||||||
def __init__(self, channel: Channel) -> None:
|
|
||||||
self.channel = channel
|
|
||||||
# when this is set to a tuple returned from ``_submit()`` then
|
|
||||||
# it is expected that ``result()`` will be awaited at some point
|
|
||||||
# during the portal's lifetime
|
|
||||||
self._result: Optional[Any] = None
|
|
||||||
# set when _submit_for_result is called
|
|
||||||
self._expect_result: Optional[
|
|
||||||
Tuple[str, Any, str, Dict[str, Any]]
|
|
||||||
] = None
|
|
||||||
self._streams: Set[ReceiveStream] = set()
|
|
||||||
self.actor = current_actor()
|
|
||||||
|
|
||||||
async def _submit(
|
|
||||||
self,
|
|
||||||
ns: str,
|
|
||||||
func: str,
|
|
||||||
kwargs,
|
|
||||||
) -> Tuple[str, trio.abc.ReceiveChannel, str, Dict[str, Any]]:
|
|
||||||
"""Submit a function to be scheduled and run by actor, return the
|
|
||||||
associated caller id, response queue, response type str,
|
|
||||||
first message packet as a tuple.
|
|
||||||
|
|
||||||
This is an async call.
|
|
||||||
"""
|
|
||||||
# ship a function call request to the remote actor
|
|
||||||
cid, recv_chan = await self.actor.send_cmd(
|
|
||||||
self.channel, ns, func, kwargs)
|
|
||||||
|
|
||||||
# wait on first response msg and handle (this should be
|
|
||||||
# in an immediate response)
|
|
||||||
|
|
||||||
first_msg = await recv_chan.receive()
|
|
||||||
functype = first_msg.get('functype')
|
|
||||||
|
|
||||||
if functype == 'function' or functype == 'asyncfunction':
|
|
||||||
resp_type = 'return'
|
|
||||||
elif functype == 'asyncgen':
|
|
||||||
resp_type = 'yield'
|
|
||||||
elif 'error' in first_msg:
|
|
||||||
raise unpack_error(first_msg, self.channel)
|
|
||||||
else:
|
|
||||||
raise ValueError(f"{first_msg} is an invalid response packet?")
|
|
||||||
|
|
||||||
return cid, recv_chan, resp_type, first_msg
|
|
||||||
|
|
||||||
async def _submit_for_result(self, ns: str, func: str, **kwargs) -> None:
|
|
||||||
assert self._expect_result is None, \
|
|
||||||
"A pending main result has already been submitted"
|
|
||||||
self._expect_result = await self._submit(ns, func, kwargs)
|
|
||||||
|
|
||||||
async def run(
|
|
||||||
self,
|
|
||||||
func_or_ns: str,
|
|
||||||
fn_name: Optional[str] = None,
|
|
||||||
**kwargs
|
|
||||||
) -> Any:
|
|
||||||
"""Submit a remote function to be scheduled and run by actor,
|
|
||||||
wrap and return its (stream of) result(s).
|
|
||||||
|
|
||||||
This is a blocking call and returns either a value from the
|
|
||||||
remote rpc task or a local async generator instance.
|
|
||||||
"""
|
|
||||||
if isinstance(func_or_ns, str):
|
|
||||||
warnings.warn(
|
|
||||||
"`Portal.run(namespace: str, funcname: str)` is now"
|
|
||||||
"deprecated, pass a function reference directly instead\n"
|
|
||||||
"If you still want to run a remote function by name use"
|
|
||||||
"`Portal.run_from_ns()`",
|
|
||||||
DeprecationWarning,
|
|
||||||
stacklevel=2,
|
|
||||||
)
|
)
|
||||||
fn_mod_path = func_or_ns
|
from ._streaming import (
|
||||||
assert isinstance(fn_name, str)
|
Context,
|
||||||
|
MsgStream,
|
||||||
else: # function reference was passed directly
|
|
||||||
fn = func_or_ns
|
|
||||||
fn_mod_path = fn.__module__
|
|
||||||
fn_name = fn.__name__
|
|
||||||
|
|
||||||
return await self._return_from_resptype(
|
|
||||||
*(await self._submit(fn_mod_path, fn_name, kwargs))
|
|
||||||
)
|
)
|
||||||
|
|
||||||
async def run_from_ns(
|
|
||||||
self,
|
log = get_logger(__name__)
|
||||||
namespace_path: str,
|
|
||||||
function_name: str,
|
|
||||||
**kwargs,
|
def _unwrap_msg(
|
||||||
|
msg: dict[str, Any],
|
||||||
|
channel: Channel
|
||||||
|
|
||||||
) -> Any:
|
) -> Any:
|
||||||
"""Run a function from a (remote) namespace in a new task on the far-end actor.
|
__tracebackhide__ = True
|
||||||
|
|
||||||
This is a more explitcit way to run tasks in a remote-process
|
|
||||||
actor using explicit object-path syntax. Hint: this is how
|
|
||||||
`.run()` works underneath.
|
|
||||||
|
|
||||||
Note::
|
|
||||||
|
|
||||||
A special namespace `self` can be used to invoke `Actor`
|
|
||||||
instance methods in the remote runtime. Currently this should only
|
|
||||||
be used for `tractor` internals.
|
|
||||||
"""
|
|
||||||
return await self._return_from_resptype(
|
|
||||||
*(await self._submit(namespace_path, function_name, kwargs))
|
|
||||||
)
|
|
||||||
|
|
||||||
async def _return_from_resptype(
|
|
||||||
self,
|
|
||||||
cid: str,
|
|
||||||
recv_chan: trio.abc.ReceiveChannel,
|
|
||||||
resptype: str,
|
|
||||||
first_msg: dict
|
|
||||||
) -> Any:
|
|
||||||
# TODO: not this needs some serious work and thinking about how
|
|
||||||
# to make async-generators the fundamental IPC API over channels!
|
|
||||||
# (think `yield from`, `gen.send()`, and functional reactive stuff)
|
|
||||||
if resptype == 'yield': # stream response
|
|
||||||
rchan = ReceiveStream(cid, recv_chan, self)
|
|
||||||
self._streams.add(rchan)
|
|
||||||
return rchan
|
|
||||||
|
|
||||||
elif resptype == 'return': # single response
|
|
||||||
msg = await recv_chan.receive()
|
|
||||||
try:
|
try:
|
||||||
return msg['return']
|
return msg['return']
|
||||||
except KeyError:
|
except KeyError:
|
||||||
# internal error should never get here
|
# internal error should never get here
|
||||||
assert msg.get('cid'), "Received internal error at portal?"
|
assert msg.get('cid'), "Received internal error at portal?"
|
||||||
raise unpack_error(msg, self.channel)
|
raise unpack_error(msg, channel) from None
|
||||||
else:
|
|
||||||
raise ValueError(f"Unknown msg response type: {first_msg}")
|
|
||||||
|
class MessagingError(Exception):
|
||||||
|
'Some kind of unexpected SC messaging dialog issue'
|
||||||
|
|
||||||
|
|
||||||
|
class Portal:
|
||||||
|
'''
|
||||||
|
A 'portal' to a(n) (remote) ``Actor``.
|
||||||
|
|
||||||
|
A portal is "opened" (and eventually closed) by one side of an
|
||||||
|
inter-actor communication context. The side which opens the portal
|
||||||
|
is equivalent to a "caller" in function parlance and usually is
|
||||||
|
either the called actor's parent (in process tree hierarchy terms)
|
||||||
|
or a client interested in scheduling work to be done remotely in a
|
||||||
|
far process.
|
||||||
|
|
||||||
|
The portal api allows the "caller" actor to invoke remote routines
|
||||||
|
and receive results through an underlying ``tractor.Channel`` as
|
||||||
|
though the remote (async) function / generator was called locally.
|
||||||
|
It may be thought of loosely as an RPC api where native Python
|
||||||
|
function calling semantics are supported transparently; hence it is
|
||||||
|
like having a "portal" between the seperate actor memory spaces.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# the timeout for a remote cancel request sent to
|
||||||
|
# a(n) (peer) actor.
|
||||||
|
cancel_timeout = 0.5
|
||||||
|
|
||||||
|
def __init__(self, channel: Channel) -> None:
|
||||||
|
self.channel = channel
|
||||||
|
# during the portal's lifetime
|
||||||
|
self._result_msg: Optional[dict] = None
|
||||||
|
|
||||||
|
# When set to a ``Context`` (when _submit_for_result is called)
|
||||||
|
# it is expected that ``result()`` will be awaited at some
|
||||||
|
# point.
|
||||||
|
self._expect_result: Optional[Context] = None
|
||||||
|
self._streams: set[MsgStream] = set()
|
||||||
|
self.actor = current_actor()
|
||||||
|
|
||||||
|
async def _submit_for_result(
|
||||||
|
self,
|
||||||
|
ns: str,
|
||||||
|
func: str,
|
||||||
|
**kwargs
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
assert self._expect_result is None, \
|
||||||
|
"A pending main result has already been submitted"
|
||||||
|
|
||||||
|
self._expect_result = await self.actor.start_remote_task(
|
||||||
|
self.channel,
|
||||||
|
ns,
|
||||||
|
func,
|
||||||
|
kwargs
|
||||||
|
)
|
||||||
|
|
||||||
|
async def _return_once(
|
||||||
|
self,
|
||||||
|
ctx: Context,
|
||||||
|
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
|
||||||
|
assert ctx._remote_func_type == 'asyncfunc' # single response
|
||||||
|
msg = await ctx._recv_chan.receive()
|
||||||
|
return msg
|
||||||
|
|
||||||
async def result(self) -> Any:
|
async def result(self) -> Any:
|
||||||
"""Return the result(s) from the remote actor's "main" task.
|
'''
|
||||||
"""
|
Return the result(s) from the remote actor's "main" task.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# __tracebackhide__ = True
|
||||||
# Check for non-rpc errors slapped on the
|
# Check for non-rpc errors slapped on the
|
||||||
# channel for which we always raise
|
# channel for which we always raise
|
||||||
exc = self.channel._exc
|
exc = self.channel._exc
|
||||||
|
@ -298,25 +156,19 @@ class Portal:
|
||||||
|
|
||||||
# expecting a "main" result
|
# expecting a "main" result
|
||||||
assert self._expect_result
|
assert self._expect_result
|
||||||
if self._result is None:
|
|
||||||
try:
|
if self._result_msg is None:
|
||||||
self._result = await self._return_from_resptype(
|
self._result_msg = await self._return_once(
|
||||||
*self._expect_result
|
self._expect_result
|
||||||
)
|
)
|
||||||
except RemoteActorError as err:
|
|
||||||
self._result = err
|
|
||||||
|
|
||||||
# re-raise error on every call
|
return _unwrap_msg(self._result_msg, self.channel)
|
||||||
if isinstance(self._result, RemoteActorError):
|
|
||||||
raise self._result
|
|
||||||
|
|
||||||
return self._result
|
|
||||||
|
|
||||||
async def _cancel_streams(self):
|
async def _cancel_streams(self):
|
||||||
# terminate all locally running async generator
|
# terminate all locally running async generator
|
||||||
# IPC calls
|
# IPC calls
|
||||||
if self._streams:
|
if self._streams:
|
||||||
log.warning(
|
log.cancel(
|
||||||
f"Cancelling all streams with {self.channel.uid}")
|
f"Cancelling all streams with {self.channel.uid}")
|
||||||
for stream in self._streams.copy():
|
for stream in self._streams.copy():
|
||||||
try:
|
try:
|
||||||
|
@ -335,80 +187,407 @@ class Portal:
|
||||||
# we'll need to .aclose all those channels here
|
# we'll need to .aclose all those channels here
|
||||||
await self._cancel_streams()
|
await self._cancel_streams()
|
||||||
|
|
||||||
async def cancel_actor(self):
|
async def cancel_actor(
|
||||||
"""Cancel the actor on the other end of this portal.
|
self,
|
||||||
"""
|
timeout: float | None = None,
|
||||||
|
|
||||||
|
) -> bool:
|
||||||
|
'''
|
||||||
|
Cancel the actor on the other end of this portal.
|
||||||
|
|
||||||
|
'''
|
||||||
if not self.channel.connected():
|
if not self.channel.connected():
|
||||||
log.warning("This portal is already closed can't cancel")
|
log.cancel("This channel is already closed can't cancel")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
await self._cancel_streams()
|
log.cancel(
|
||||||
|
|
||||||
log.warning(
|
|
||||||
f"Sending actor cancel request to {self.channel.uid} on "
|
f"Sending actor cancel request to {self.channel.uid} on "
|
||||||
f"{self.channel}")
|
f"{self.channel}")
|
||||||
|
|
||||||
|
self.channel._cancel_called = 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 a proper shield
|
# XXX: sure would be nice to make this work with a proper shield
|
||||||
# with trio.CancelScope() as cancel_scope:
|
with trio.move_on_after(timeout or self.cancel_timeout) as cs:
|
||||||
# with trio.CancelScope(shield=True) as cancel_scope:
|
cs.shield = True
|
||||||
with trio.move_on_after(0.5) as cancel_scope:
|
|
||||||
cancel_scope.shield = True
|
|
||||||
|
|
||||||
await self.run_from_ns('self', 'cancel')
|
await self.run_from_ns('self', 'cancel')
|
||||||
return True
|
return True
|
||||||
|
|
||||||
if cancel_scope.cancelled_caught:
|
if cs.cancelled_caught:
|
||||||
log.warning(f"May have failed to cancel {self.channel.uid}")
|
log.cancel(f"May have failed to cancel {self.channel.uid}")
|
||||||
|
|
||||||
# if we get here some weird cancellation case happened
|
# if we get here some weird cancellation case happened
|
||||||
return False
|
return False
|
||||||
|
|
||||||
except trio.ClosedResourceError:
|
except (
|
||||||
log.warning(
|
trio.ClosedResourceError,
|
||||||
f"{self.channel} for {self.channel.uid} was already closed?")
|
trio.BrokenResourceError,
|
||||||
|
):
|
||||||
|
log.cancel(
|
||||||
|
f"{self.channel} for {self.channel.uid} was already "
|
||||||
|
"closed or broken?")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
async def run_from_ns(
|
||||||
|
self,
|
||||||
|
namespace_path: str,
|
||||||
|
function_name: str,
|
||||||
|
**kwargs,
|
||||||
|
) -> Any:
|
||||||
|
'''
|
||||||
|
Run a function from a (remote) namespace in a new task on the
|
||||||
|
far-end actor.
|
||||||
|
|
||||||
|
This is a more explitcit way to run tasks in a remote-process
|
||||||
|
actor using explicit object-path syntax. Hint: this is how
|
||||||
|
`.run()` works underneath.
|
||||||
|
|
||||||
|
Note::
|
||||||
|
|
||||||
|
A special namespace `self` can be used to invoke `Actor`
|
||||||
|
instance methods in the remote runtime. Currently this
|
||||||
|
should only be used solely for ``tractor`` runtime
|
||||||
|
internals.
|
||||||
|
|
||||||
|
'''
|
||||||
|
ctx = await self.actor.start_remote_task(
|
||||||
|
self.channel,
|
||||||
|
namespace_path,
|
||||||
|
function_name,
|
||||||
|
kwargs,
|
||||||
|
)
|
||||||
|
ctx._portal = self
|
||||||
|
msg = await self._return_once(ctx)
|
||||||
|
return _unwrap_msg(msg, self.channel)
|
||||||
|
|
||||||
|
async def run(
|
||||||
|
self,
|
||||||
|
func: str,
|
||||||
|
fn_name: Optional[str] = None,
|
||||||
|
**kwargs
|
||||||
|
) -> Any:
|
||||||
|
'''
|
||||||
|
Submit a remote function to be scheduled and run by actor, in
|
||||||
|
a new task, wrap and return its (stream of) result(s).
|
||||||
|
|
||||||
|
This is a blocking call and returns either a value from the
|
||||||
|
remote rpc task or a local async generator instance.
|
||||||
|
|
||||||
|
'''
|
||||||
|
if isinstance(func, str):
|
||||||
|
warnings.warn(
|
||||||
|
"`Portal.run(namespace: str, funcname: str)` is now"
|
||||||
|
"deprecated, pass a function reference directly instead\n"
|
||||||
|
"If you still want to run a remote function by name use"
|
||||||
|
"`Portal.run_from_ns()`",
|
||||||
|
DeprecationWarning,
|
||||||
|
stacklevel=2,
|
||||||
|
)
|
||||||
|
fn_mod_path = func
|
||||||
|
assert isinstance(fn_name, str)
|
||||||
|
|
||||||
|
else: # function reference was passed directly
|
||||||
|
if (
|
||||||
|
not inspect.iscoroutinefunction(func) or
|
||||||
|
(
|
||||||
|
inspect.iscoroutinefunction(func) and
|
||||||
|
getattr(func, '_tractor_stream_function', False)
|
||||||
|
)
|
||||||
|
):
|
||||||
|
raise TypeError(
|
||||||
|
f'{func} must be a non-streaming async function!')
|
||||||
|
|
||||||
|
fn_mod_path, fn_name = NamespacePath.from_ref(func).to_tuple()
|
||||||
|
|
||||||
|
ctx = await self.actor.start_remote_task(
|
||||||
|
self.channel,
|
||||||
|
fn_mod_path,
|
||||||
|
fn_name,
|
||||||
|
kwargs,
|
||||||
|
)
|
||||||
|
ctx._portal = self
|
||||||
|
return _unwrap_msg(
|
||||||
|
await self._return_once(ctx),
|
||||||
|
self.channel,
|
||||||
|
)
|
||||||
|
|
||||||
|
@asynccontextmanager
|
||||||
|
async def open_stream_from(
|
||||||
|
self,
|
||||||
|
async_gen_func: Callable, # typing: ignore
|
||||||
|
**kwargs,
|
||||||
|
|
||||||
|
) -> AsyncGenerator[MsgStream, None]:
|
||||||
|
|
||||||
|
if not inspect.isasyncgenfunction(async_gen_func):
|
||||||
|
if not (
|
||||||
|
inspect.iscoroutinefunction(async_gen_func) and
|
||||||
|
getattr(async_gen_func, '_tractor_stream_function', False)
|
||||||
|
):
|
||||||
|
raise TypeError(
|
||||||
|
f'{async_gen_func} must be an async generator function!')
|
||||||
|
|
||||||
|
fn_mod_path, fn_name = NamespacePath.from_ref(
|
||||||
|
async_gen_func).to_tuple()
|
||||||
|
ctx = await self.actor.start_remote_task(
|
||||||
|
self.channel,
|
||||||
|
fn_mod_path,
|
||||||
|
fn_name,
|
||||||
|
kwargs
|
||||||
|
)
|
||||||
|
ctx._portal = self
|
||||||
|
|
||||||
|
# ensure receive-only stream entrypoint
|
||||||
|
assert ctx._remote_func_type == 'asyncgen'
|
||||||
|
|
||||||
|
try:
|
||||||
|
# deliver receive only stream
|
||||||
|
async with MsgStream(
|
||||||
|
ctx, ctx._recv_chan,
|
||||||
|
) as rchan:
|
||||||
|
self._streams.add(rchan)
|
||||||
|
yield rchan
|
||||||
|
|
||||||
|
finally:
|
||||||
|
|
||||||
|
# cancel the far end task on consumer close
|
||||||
|
# NOTE: this is a special case since we assume that if using
|
||||||
|
# this ``.open_fream_from()`` api, the stream is one a one
|
||||||
|
# time use and we couple the far end tasks's lifetime to
|
||||||
|
# the consumer's scope; we don't ever send a `'stop'`
|
||||||
|
# message right now since there shouldn't be a reason to
|
||||||
|
# stop and restart the stream, right?
|
||||||
|
try:
|
||||||
|
with trio.CancelScope(shield=True):
|
||||||
|
await ctx.cancel()
|
||||||
|
|
||||||
|
except trio.ClosedResourceError:
|
||||||
|
# if the far end terminates before we send a cancel the
|
||||||
|
# underlying transport-channel may already be closed.
|
||||||
|
log.cancel(f'Context {ctx} was already closed?')
|
||||||
|
|
||||||
|
# XXX: should this always be done?
|
||||||
|
# await recv_chan.aclose()
|
||||||
|
self._streams.remove(rchan)
|
||||||
|
|
||||||
|
@asynccontextmanager
|
||||||
|
async def open_context(
|
||||||
|
|
||||||
|
self,
|
||||||
|
func: Callable,
|
||||||
|
**kwargs,
|
||||||
|
|
||||||
|
) -> AsyncGenerator[tuple[Context, Any], None]:
|
||||||
|
'''
|
||||||
|
Open an inter-actor task context.
|
||||||
|
|
||||||
|
This is a synchronous API which allows for deterministic
|
||||||
|
setup/teardown of a remote task. The yielded ``Context`` further
|
||||||
|
allows for opening bidirectional streams, explicit cancellation
|
||||||
|
and synchronized final result collection. See ``tractor.Context``.
|
||||||
|
|
||||||
|
'''
|
||||||
|
# conduct target func method structural checks
|
||||||
|
if not inspect.iscoroutinefunction(func) and (
|
||||||
|
getattr(func, '_tractor_contex_function', False)
|
||||||
|
):
|
||||||
|
raise TypeError(
|
||||||
|
f'{func} must be an async generator function!')
|
||||||
|
|
||||||
|
fn_mod_path, fn_name = NamespacePath.from_ref(func).to_tuple()
|
||||||
|
|
||||||
|
ctx = await self.actor.start_remote_task(
|
||||||
|
self.channel,
|
||||||
|
fn_mod_path,
|
||||||
|
fn_name,
|
||||||
|
kwargs
|
||||||
|
)
|
||||||
|
|
||||||
|
assert ctx._remote_func_type == 'context'
|
||||||
|
msg = await ctx._recv_chan.receive()
|
||||||
|
|
||||||
|
try:
|
||||||
|
# the "first" value here is delivered by the callee's
|
||||||
|
# ``Context.started()`` call.
|
||||||
|
first = msg['started']
|
||||||
|
ctx._started_called = True
|
||||||
|
|
||||||
|
except KeyError:
|
||||||
|
assert msg.get('cid'), ("Received internal error at context?")
|
||||||
|
|
||||||
|
if msg.get('error'):
|
||||||
|
# raise kerr from unpack_error(msg, self.channel)
|
||||||
|
raise unpack_error(msg, self.channel) from None
|
||||||
|
else:
|
||||||
|
raise MessagingError(
|
||||||
|
f'Context for {ctx.cid} was expecting a `started` message'
|
||||||
|
f' but received a non-error msg:\n{pformat(msg)}'
|
||||||
|
)
|
||||||
|
|
||||||
|
_err: Optional[BaseException] = None
|
||||||
|
ctx._portal = self
|
||||||
|
|
||||||
|
uid = self.channel.uid
|
||||||
|
cid = ctx.cid
|
||||||
|
etype: Optional[Type[BaseException]] = None
|
||||||
|
|
||||||
|
# deliver context instance and .started() msg value in open tuple.
|
||||||
|
try:
|
||||||
|
async with trio.open_nursery() as scope_nursery:
|
||||||
|
ctx._scope_nursery = scope_nursery
|
||||||
|
|
||||||
|
# do we need this?
|
||||||
|
# await trio.lowlevel.checkpoint()
|
||||||
|
|
||||||
|
yield ctx, first
|
||||||
|
|
||||||
|
except ContextCancelled as err:
|
||||||
|
_err = err
|
||||||
|
if not ctx._cancel_called:
|
||||||
|
# context was cancelled at the far end but was
|
||||||
|
# not part of this end requesting that cancel
|
||||||
|
# so raise for the local task to respond and handle.
|
||||||
|
raise
|
||||||
|
|
||||||
|
# if the context was cancelled by client code
|
||||||
|
# then we don't need to raise since user code
|
||||||
|
# is expecting this and the block should exit.
|
||||||
|
else:
|
||||||
|
log.debug(f'Context {ctx} cancelled gracefully')
|
||||||
|
|
||||||
|
except (
|
||||||
|
BaseException,
|
||||||
|
|
||||||
|
# more specifically, we need to handle these but not
|
||||||
|
# sure it's worth being pedantic:
|
||||||
|
# Exception,
|
||||||
|
# trio.Cancelled,
|
||||||
|
# KeyboardInterrupt,
|
||||||
|
|
||||||
|
) as err:
|
||||||
|
etype = type(err)
|
||||||
|
# the context cancels itself on any cancel
|
||||||
|
# causing error.
|
||||||
|
|
||||||
|
if ctx.chan.connected():
|
||||||
|
log.cancel(
|
||||||
|
'Context cancelled for task, sending cancel request..\n'
|
||||||
|
f'task:{cid}\n'
|
||||||
|
f'actor:{uid}'
|
||||||
|
)
|
||||||
|
await ctx.cancel()
|
||||||
|
else:
|
||||||
|
log.warning(
|
||||||
|
'IPC connection for context is broken?\n'
|
||||||
|
f'task:{cid}\n'
|
||||||
|
f'actor:{uid}'
|
||||||
|
)
|
||||||
|
|
||||||
|
raise
|
||||||
|
|
||||||
|
finally:
|
||||||
|
# in the case where a runtime nursery (due to internal bug)
|
||||||
|
# or a remote actor transmits an error we want to be
|
||||||
|
# sure we get the error the underlying feeder mem chan.
|
||||||
|
# if it's not raised here it *should* be raised from the
|
||||||
|
# msg loop nursery right?
|
||||||
|
if ctx.chan.connected():
|
||||||
|
log.info(
|
||||||
|
'Waiting on final context-task result for\n'
|
||||||
|
f'task: {cid}\n'
|
||||||
|
f'actor: {uid}'
|
||||||
|
)
|
||||||
|
result = await ctx.result()
|
||||||
|
log.runtime(
|
||||||
|
f'Context {fn_name} returned '
|
||||||
|
f'value from callee `{result}`'
|
||||||
|
)
|
||||||
|
|
||||||
|
# though it should be impossible for any tasks
|
||||||
|
# operating *in* this scope to have survived
|
||||||
|
# we tear down the runtime feeder chan last
|
||||||
|
# to avoid premature stream clobbers.
|
||||||
|
if ctx._recv_chan is not None:
|
||||||
|
# should we encapsulate this in the context api?
|
||||||
|
await ctx._recv_chan.aclose()
|
||||||
|
|
||||||
|
if etype:
|
||||||
|
if ctx._cancel_called:
|
||||||
|
log.cancel(
|
||||||
|
f'Context {fn_name} cancelled by caller with\n{etype}'
|
||||||
|
)
|
||||||
|
elif _err is not None:
|
||||||
|
log.cancel(
|
||||||
|
f'Context for task cancelled by callee with {etype}\n'
|
||||||
|
f'target: `{fn_name}`\n'
|
||||||
|
f'task:{cid}\n'
|
||||||
|
f'actor:{uid}'
|
||||||
|
)
|
||||||
|
# XXX: (MEGA IMPORTANT) if this is a root opened process we
|
||||||
|
# wait for any immediate child in debug before popping the
|
||||||
|
# context from the runtime msg loop otherwise inside
|
||||||
|
# ``Actor._push_result()`` the msg will be discarded and in
|
||||||
|
# the case where that msg is global debugger unlock (via
|
||||||
|
# a "stop" msg for a stream), this can result in a deadlock
|
||||||
|
# where the root is waiting on the lock to clear but the
|
||||||
|
# child has already cleared it and clobbered IPC.
|
||||||
|
from ._debug import maybe_wait_for_debugger
|
||||||
|
await maybe_wait_for_debugger()
|
||||||
|
|
||||||
|
# remove the context from runtime tracking
|
||||||
|
self.actor._contexts.pop(
|
||||||
|
(self.channel.uid, ctx.cid),
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class LocalPortal:
|
class LocalPortal:
|
||||||
"""A 'portal' to a local ``Actor``.
|
'''
|
||||||
|
A 'portal' to a local ``Actor``.
|
||||||
|
|
||||||
A compatibility shim for normal portals but for invoking functions
|
A compatibility shim for normal portals but for invoking functions
|
||||||
using an in process actor instance.
|
using an in process actor instance.
|
||||||
"""
|
|
||||||
|
'''
|
||||||
actor: 'Actor' # type: ignore # noqa
|
actor: 'Actor' # type: ignore # noqa
|
||||||
channel: Channel
|
channel: Channel
|
||||||
|
|
||||||
async def run_from_ns(self, ns: str, func_name: str, **kwargs) -> Any:
|
async def run_from_ns(self, ns: str, func_name: str, **kwargs) -> Any:
|
||||||
"""Run a requested local function from a namespace path and
|
'''
|
||||||
|
Run a requested local function from a namespace path and
|
||||||
return it's result.
|
return it's result.
|
||||||
|
|
||||||
"""
|
'''
|
||||||
obj = self.actor if ns == 'self' else importlib.import_module(ns)
|
obj = self.actor if ns == 'self' else importlib.import_module(ns)
|
||||||
func = getattr(obj, func_name)
|
func = getattr(obj, func_name)
|
||||||
if inspect.iscoroutinefunction(func):
|
|
||||||
return await func(**kwargs)
|
return await func(**kwargs)
|
||||||
else:
|
|
||||||
return func(**kwargs)
|
|
||||||
|
|
||||||
|
|
||||||
@asynccontextmanager
|
@asynccontextmanager
|
||||||
async def open_portal(
|
async def open_portal(
|
||||||
|
|
||||||
channel: Channel,
|
channel: Channel,
|
||||||
nursery: Optional[trio.Nursery] = None,
|
nursery: Optional[trio.Nursery] = None,
|
||||||
start_msg_loop: bool = True,
|
start_msg_loop: bool = True,
|
||||||
shield: bool = False,
|
shield: bool = False,
|
||||||
) -> typing.AsyncGenerator[Portal, None]:
|
|
||||||
"""Open a ``Portal`` through the provided ``channel``.
|
|
||||||
|
|
||||||
Spawns a background task to handle message processing.
|
) -> AsyncGenerator[Portal, None]:
|
||||||
"""
|
'''
|
||||||
|
Open a ``Portal`` through the provided ``channel``.
|
||||||
|
|
||||||
|
Spawns a background task to handle message processing (normally
|
||||||
|
done by the actor-runtime implicitly).
|
||||||
|
|
||||||
|
'''
|
||||||
actor = current_actor()
|
actor = current_actor()
|
||||||
assert actor
|
assert actor
|
||||||
was_connected = False
|
was_connected = False
|
||||||
|
|
||||||
async with maybe_open_nursery(nursery, shield=shield) as nursery:
|
async with maybe_open_nursery(nursery, shield=shield) as nursery:
|
||||||
|
|
||||||
if not channel.connected():
|
if not channel.connected():
|
||||||
await channel.connect()
|
await channel.connect()
|
||||||
was_connected = True
|
was_connected = True
|
||||||
|
@ -418,9 +597,11 @@ async def open_portal(
|
||||||
|
|
||||||
msg_loop_cs: Optional[trio.CancelScope] = None
|
msg_loop_cs: Optional[trio.CancelScope] = None
|
||||||
if start_msg_loop:
|
if start_msg_loop:
|
||||||
|
from ._runtime import process_messages
|
||||||
msg_loop_cs = await nursery.start(
|
msg_loop_cs = await nursery.start(
|
||||||
partial(
|
partial(
|
||||||
actor._process_messages,
|
process_messages,
|
||||||
|
actor,
|
||||||
channel,
|
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
|
||||||
|
@ -434,8 +615,9 @@ async def open_portal(
|
||||||
await portal.aclose()
|
await portal.aclose()
|
||||||
|
|
||||||
if was_connected:
|
if was_connected:
|
||||||
# cancel remote channel-msg loop
|
# gracefully signal remote channel-msg loop
|
||||||
await channel.send(None)
|
await channel.send(None)
|
||||||
|
# await channel.aclose()
|
||||||
|
|
||||||
# cancel background msg loop task
|
# cancel background msg loop task
|
||||||
if msg_loop_cs:
|
if msg_loop_cs:
|
||||||
|
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue