forked from goodboy/tractor
Use a global dataclass instead, cuz we like "objects"?
parent
7f8c5cdfe6
commit
47d7b603db
|
@ -1,24 +1,37 @@
|
||||||
from itertools import cycle
|
from itertools import cycle
|
||||||
from pprint import pformat
|
from pprint import pformat
|
||||||
|
from dataclasses import dataclass, field
|
||||||
|
|
||||||
import trio
|
import trio
|
||||||
import tractor
|
import tractor
|
||||||
|
|
||||||
_snd_chan, _recv_chan = trio.open_memory_channel(100)
|
|
||||||
_actor_state = {'some_state_stuff': None}
|
@dataclass
|
||||||
|
class MyProcessStateThing:
|
||||||
|
state: dict = field(default_factory=dict)
|
||||||
|
|
||||||
|
def update(self, msg: dict):
|
||||||
|
self.state.update(msg)
|
||||||
|
|
||||||
|
|
||||||
|
_actor_state = MyProcessStateThing()
|
||||||
|
|
||||||
|
|
||||||
async def update_local_state(msg: dict):
|
async def update_local_state(msg: dict):
|
||||||
|
"""Update process-local state from sent message and exit.
|
||||||
|
|
||||||
|
"""
|
||||||
|
actor = tractor.current_actor()
|
||||||
|
|
||||||
global _actor_state
|
global _actor_state
|
||||||
actor = tractor.current_actor()
|
|
||||||
|
|
||||||
print(f'Yo we got a message {msg}')
|
print(f'Yo we got a message {msg}')
|
||||||
|
|
||||||
# update the "actor state"
|
# update the "actor state"
|
||||||
_actor_state.update(msg)
|
_actor_state.update(msg)
|
||||||
|
|
||||||
print(f'New local "state" for {actor.uid} is {pformat(_actor_state)}')
|
print(f'New local "state" for {actor.uid} is {pformat(_actor_state.state)}')
|
||||||
|
|
||||||
# we're done so exit this task running in the subactor
|
# we're done so exit this task running in the subactor
|
||||||
|
|
||||||
|
@ -43,7 +56,7 @@ async def main():
|
||||||
):
|
):
|
||||||
await portal.run(update_local_state, msg={f'msg_{i}': count})
|
await portal.run(update_local_state, msg={f'msg_{i}': count})
|
||||||
|
|
||||||
# blocks here indefinitely synce we spawned "daemon actors using
|
# blocks here indefinitely synce we spawned "daemon actors" using
|
||||||
# .start_actor()`, you'll need to control-c to cancel.
|
# .start_actor()`, you'll need to control-c to cancel.
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue