add final_counter to EOF websocket

This commit is contained in:
Jake McDermott
2018-06-08 22:49:00 -04:00
parent ce411a21c6
commit 60a38a196a
2 changed files with 3 additions and 3 deletions

View File

@@ -162,7 +162,7 @@ class CallbackBrokerWorker(ConsumerMixin):
if body.get('event') == 'EOF':
try:
final_line_count = body.get('final_line_count', 0)
final_counter = body.get('final_counter', 0)
logger.info('Event processing is finished for Job {}, sending notifications'.format(job_identifier))
# EOF events are sent when stdout for the running task is
# closed. don't actually persist them to the database; we
@@ -170,7 +170,7 @@ class CallbackBrokerWorker(ConsumerMixin):
# approximation for when a job is "done"
emit_channel_notification(
'jobs-summary',
dict(group_name='jobs', unified_job_id=job_identifier, final_line_count=final_line_count)
dict(group_name='jobs', unified_job_id=job_identifier, final_counter=final_counter)
)
# Additionally, when we've processed all events, we should
# have all the data we need to send out success/failure

View File

@@ -989,7 +989,7 @@ class OutputEventFilter(object):
if value:
self._emit_event(value)
self._buffer = StringIO()
self._event_callback(dict(event='EOF', final_line_count=self._start_line))
self._event_callback(dict(event='EOF', final_counter=self._counter - 1))
def _emit_event(self, buffered_stdout, next_event_data=None):
next_event_data = next_event_data or {}