Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify decision task handling in SWF backend

Browse Source
This commit is contained in:
Jean-Baptiste Barth 2015-10-26 18:06:04 +01:00
parent a0e484fa6d
commit d650f71d9c
2 changed files with 9 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
moto/swf/models/__init__.py
View File

@ -180,16 +180,16 @@ class SWFBackend(BaseBackend):
# aren't distributed. # aren't distributed.
# #
# TODO: handle long polling (case 2) for decision tasks # TODO: handle long polling (case 2) for decision tasks
decision_candidates = [] candidates = []
for _, wfe in domain.workflow_executions.iteritems(): for _task_list, tasks in domain.decision_task_lists.iteritems():
if wfe.task_list == task_list: if _task_list == task_list:
decision_candidates += wfe.scheduled_decision_tasks candidates += filter(lambda t: t.state == "SCHEDULED", tasks)
if any(decision_candidates): if any(candidates):
# TODO: handle task priorities (but not supported by boto for now) # TODO: handle task priorities (but not supported by boto for now)
decision = min(decision_candidates, key=lambda d: d.scheduled_at) task = min(candidates, key=lambda d: d.scheduled_at)
wfe = decision.workflow_execution wfe = task.workflow_execution
wfe.start_decision_task(decision.task_token, identity=identity) wfe.start_decision_task(task.task_token, identity=identity)
return decision return task
else: else:
return None return None

7
moto/swf/models/workflow_execution.py
View File

@ -182,13 +182,6 @@ class WorkflowExecution(object):
self.domain.decision_tasks self.domain.decision_tasks
) )
@property
def scheduled_decision_tasks(self):
return filter(
lambda t: t.state == "SCHEDULED",
self.decision_tasks
)
@property @property
def activity_tasks(self): def activity_tasks(self):
return filter( return filter(