Manage start_failed

Signed-off-by: Yves Bastide <yves@botify.com>

Author: ybastide

simpleflow/history.py

@@ -647,6 +647,16 @@ def get_lambda():
                 'started_id': event.id,
                 'started_timestamp': event.timestamp,
             })
+        elif event.state == 'start_failed':
+            lambda_function = get_lambda()
+            lambda_function.update({
+                'state': event.state,
+                'cause': event.cause,
+                'message': getattr(event, 'message', ''),
+                'start_failed_id': event.id,
+                'start_failed_timestamp': event.timestamp,
+                'retry': lambda_function.get('retry', -1) + 1,
+            })
         elif event.state == 'completed':
             lambda_function = get_lambda()
             lambda_function.update({

simpleflow/swf/executor.py

@@ -389,11 +389,19 @@ def _get_future_from_lambda_function_event(self, event):
                 reason=event['reason'],
                 details=event.get('details'),
             ))
+        elif state == 'start_failed':
+            future.set_exception(exceptions.TaskFailed(
+                name=event['id'],
+                reason=event['cause'],
+                details=event.get('message'),
+            ))
         elif state == 'timed_out':
             future.set_exception(exceptions.TimeoutError(
                 event['timeout_type'],
                 None,
             ))
+        else:
+            logger.warning('Unknown state: %s', state)
 
         return future