External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-03-26 05:15:02 -02:30
Code Issues Packages Projects Releases Wiki Activity

Signal finished tasks to the task management system

Browse Source
This commit is contained in:
Matthew Jones
2014-03-12 17:52:47 -04:00
parent 99d42dc3fa
commit fc4f256723
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
awx/main/tasks.py
View File

@@ -23,6 +23,9 @@ import uuid
# Pexpect # Pexpect
import pexpect import pexpect
# ZMQ
import zmq
# Kombu # Kombu
from kombu import Connection, Exchange, Queue from kombu import Connection, Exchange, Queue
@@ -120,6 +123,13 @@ class BaseTask(Task):
logger.error('Failed to update %s after %d retries.', logger.error('Failed to update %s after %d retries.',
self.model._meta.object_name, retry_count) self.model._meta.object_name, retry_count)
def signal_finished(self, pk):
signal_context = zmq.Context()
signal_socket = signal_context.socket(zmq.REQ)
signal_socket.connect(settings.TASK_COMMAND_PORT)
signal_socket.send_json(dict(complete=pk))
signal_socket.recv()
def get_model(self, pk): def get_model(self, pk):
return self.model.objects.get(pk=pk) return self.model.objects.get(pk=pk)
@@ -342,6 +352,7 @@ class BaseTask(Task):
raise Exception("Task %s(pk:%s) was canceled" % (str(self.model.__class__), str(pk))) raise Exception("Task %s(pk:%s) was canceled" % (str(self.model.__class__), str(pk)))
else: else:
raise Exception("Task %s(pk:%s) encountered an error" % (str(self.model.__class__), str(pk))) raise Exception("Task %s(pk:%s) encountered an error" % (str(self.model.__class__), str(pk)))
self.signal_finished(pk)
class RunJob(BaseTask): class RunJob(BaseTask):
''' '''