forked from goodboy/tractor
1
0
Fork 0

Use a global dataclass instead, cuz we like "objects"?

actor_state_via_messages
Tyler Goodlet 2021-01-24 15:18:52 -05:00
parent 7f8c5cdfe6
commit 47d7b603db
1 changed files with 18 additions and 5 deletions

View File

@ -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.