从务实的角度来看,您的 solution 是过渡的方式,恕我直言。对于这样的问题,并发状态机可能值得一看。
对于这种特殊情况,虽然这肯定是最重要的,但我想借此机会展示转换如何处理并发。
更准确地说,我们可以利用它处理多个模型的能力来实现并发。
在下面的示例中,我扩展了Machine,使用 a) 一种在所有模型上触发事件的方法和 b) 使用一种初始化工作模型并将实际模型存储在内部变量中的方法(尽管这不是一个适当的堆栈)。
使用ignore_invalid_triggers,每个工作人员都会等待相应的事件,并在完成后将自己从ParallelMachine 模型列表中删除。当模型列表为空时,ParallelMachine 再次恢复实际模型状态。
from transitions import Machine
class WorkerModel:
def __init__(self, machine):
self.machine = machine
# I show myself out when I am done
def on_enter_Done(self):
self.machine.remove_model(self)
class ParallelMachine(Machine):
def __init__(self, *args, **kwargs):
self._pushed_models = []
super(ParallelMachine, self).__init__(*args, **kwargs)
# save actual models and initialise worker tasks
def add_worker(self, tasks):
self._pushed_models = self.models
self.models = []
for t in tasks:
self.add_model(WorkerModel(self), initial=t)
# trigger an event on ALL models (either workers or actual models)
def trigger(self, trigger, *args, **kwargs):
for m in self.models:
getattr(m, trigger)(*args, **kwargs)
if not self.models:
self.models = self._pushed_models
self._pushed_models = []
for m in self.models:
getattr(m, trigger)(*args, **kwargs)
class Model:
def __init__(self):
# the state list contains model states as well as worker states
states = ['Idle', 'Processing', 'Done', 'need_1', 'need_2', 'need_3']
# potentially got_1/2/3 all can trigger a transition to 'Done'
# but the model will only be triggered when the worker list
# has been emptied before
transitions = [['process', 'Idle', 'Processing'],
['got_1', ['Processing', 'need_1'], 'Done'],
['got_2', ['Processing', 'need_2'], 'Done'],
['got_3', ['Processing', 'need_3'], 'Done']]
# assigning machine to a model attribute prevents the need
# to pass it around
self.machine = ParallelMachine(model=self, states=states,
transitions=transitions, initial='Idle',
ignore_invalid_triggers=True)
def on_enter_Processing(self):
self.machine.add_worker(['need_1', 'need_2', 'need_3'])
def on_enter_Done(self):
print("I am done")
model = Model()
machine = model.machine
assert model.state == 'Idle'
# we use ParallelMachine.trigger instead of model.<event_name>
machine.trigger('process')
assert model.state == 'Processing'
machine.trigger('got_3')
machine.trigger('got_1')
# duplicated events do not matter
machine.trigger('got_3')
assert model.state == 'Processing'
machine.trigger('got_2')
# worker stack was empty and
assert model.state == 'Done'
这样全局状态可以是“Processing(need_1/need_2/done)”或“Processing(done/need_2/done)”,而不需要显式地为所有变体建模。
并发位于transitions 待办事项列表中,但需要进行一些更改。例如,回调应该在所有模型中都可用(例如,从Model 中删除on_enter_Done 将引发异常)。