def __init__(self):
super(ServiceResource, self).__init__()
self.__context = zmq.Context()
self.__monitor = self.__context.socket(zmq.PUSH)
self.__monitor.connect('tcp://collector:' + str(STATUS_PORT))
self.__sockets = {}
self.__redis_conn = redis.Redis('redis', 6379)
self.__util = ServiceUtil()
def __init__(self):
super(ServiceResource, self).__init__()
self.__context = zmq.Context()
self.__monitor = self.__context.socket(zmq.PUSH)
self.__monitor.connect('tcp://collector:' + str(STATUS_PORT))
self.__sockets = {}
self.__redis_conn = redis.Redis('redis', 6379)
self.__util = ServiceUtil()
class ServicesResource(Resource):
"""
List of available services.
"""
def __init__(self):
self.__redis_conn = redis.Redis('redis', 6379)
self.__util = ServiceUtil()
def get(self):
"""
List all registered services.
:return:
"""
registered_services = self.__redis_conn.hgetall('endpoints')
services = []
for key in registered_services.keys():
services.append(self.__util.get_service_details(key))
return services, 200
class ServicesResource(Resource):
"""
List of available services.
"""
def __init__(self):
self.__redis_conn = redis.Redis('redis', 6379)
self.__util = ServiceUtil()
def get(self):
"""
List all registered services.
:return:
"""
registered_services = self.__redis_conn.hgetall('endpoints')
services = []
for key in registered_services.keys():
services.append(self.__util.get_service_details(key))
return services, 200
def __init__(self):
self.__redis_conn = redis.Redis('redis', 6379)
self.__util = ServiceUtil()
class ServiceResource(Resource):
"""
Sample API for calling new worker pool
"""
def __init__(self):
super(ServiceResource, self).__init__()
self.__context = zmq.Context()
self.__monitor = self.__context.socket(zmq.PUSH)
self.__monitor.connect('tcp://collector:' + str(STATUS_PORT))
self.__sockets = {}
self.__redis_conn = redis.Redis('redis', 6379)
self.__util = ServiceUtil()
def get(self, name):
"""
Details of this service
:return: Details about this service.
"""
return self.__util.get_service_details(name), 200
def post(self, name):
"""
POST a new job to a service
:return:
"""
logging.debug('Service name: ' + name)
# Stream the input data to file server (data cache server).
req = client.post(INPUT_DATA_SERVER_LOCATION + 'data',
json=request.stream.read(),
stream=True)
logging.debug('File server response Data = ' + str(req.json()))
# This is a unique file ID for this input data
file_id = req.json()['fileId']
# ID of this new job
job_id = self.submit_to_worker(file_id, name)
# Save the key-value pair (Job ID to input file ID)
# - This will be used for deletion.
input_file_location = INPUT_DATA_SERVER_LOCATION + 'data/' + file_id
self.__redis_conn.hset(name='input-file', key=job_id, value=input_file_location)
current_status = {
'job_id': job_id,
'status': 'queued'
}
logging.debug('Sending status: ' + str(current_status))
# send status of job to monitor
self.__monitor.send_json(current_status)
# Job created
return current_status, 202
def submit_to_worker(self, input_data_location, service_name):
# Generate unique job ID
job_id = str(uuid.uuid4())
# The worker will get location of data, not actual data.
task = {
'job_id': job_id,
'data': INPUT_DATA_SERVER_LOCATION + 'data/' + input_data_location
}
logging.debug('Task JSON = ' + str(task))
registered = self.__redis_conn.hgetall('endpoints')
logging.debug('Endpoints = ' + str(registered))
if service_name not in registered.keys():
raise ValueError('No such service: ' + service_name)
send_port = registered[service_name]
logging.debug('target Port = ' + str(send_port))
if service_name not in self.__sockets.keys():
send_socket = self.__context.socket(zmq.PUSH)
send_socket.bind('tcp://*:' + str(send_port))
self.__sockets[service_name] = send_socket
else:
send_socket = self.__sockets[service_name]
logging.debug('Sending data....')
send_socket.send_json(task)
return job_id
class ServiceResource(Resource):
"""
Sample API for calling new worker pool
"""
def __init__(self):
super(ServiceResource, self).__init__()
self.__context = zmq.Context()
self.__monitor = self.__context.socket(zmq.PUSH)
self.__monitor.connect('tcp://collector:' + str(STATUS_PORT))
self.__sockets = {}
self.__redis_conn = redis.Redis('redis', 6379)
self.__util = ServiceUtil()
def get(self, name):
"""
Details of this service
:return: Details about this service.
"""
return self.__util.get_service_details(name), 200
def post(self, name):
"""
POST a new job to a service
:return:
"""
logging.debug('Service name: ' + name)
# Stream the input data to file server (data cache server).
req = client.post(INPUT_DATA_SERVER_LOCATION + 'data',
json=request.stream.read(),
stream=True)
logging.debug('File server response Data = ' + str(req.json()))
# This is a unique file ID for this input data
file_id = req.json()['fileId']
# ID of this new job
job_id = self.submit_to_worker(file_id, name)
# Save the key-value pair (Job ID to input file ID)
# - This will be used for deletion.
input_file_location = INPUT_DATA_SERVER_LOCATION + 'data/' + file_id
self.__redis_conn.hset(name='input-file',
key=job_id,
value=input_file_location)
current_status = {'job_id': job_id, 'status': 'queued'}
logging.debug('Sending status: ' + str(current_status))
# send status of job to monitor
self.__monitor.send_json(current_status)
# Job created
return current_status, 202
def submit_to_worker(self, input_data_location, service_name):
# Generate unique job ID
job_id = str(uuid.uuid4())
# The worker will get location of data, not actual data.
task = {
'job_id': job_id,
'data': INPUT_DATA_SERVER_LOCATION + 'data/' + input_data_location
}
logging.debug('Task JSON = ' + str(task))
registered = self.__redis_conn.hgetall('endpoints')
logging.debug('Endpoints = ' + str(registered))
if service_name not in registered.keys():
raise ValueError('No such service: ' + service_name)
send_port = registered[service_name]
logging.debug('target Port = ' + str(send_port))
if service_name not in self.__sockets.keys():
send_socket = self.__context.socket(zmq.PUSH)
send_socket.bind('tcp://*:' + str(send_port))
self.__sockets[service_name] = send_socket
else:
send_socket = self.__sockets[service_name]
logging.debug('Sending data....')
send_socket.send_json(task)
return job_id
def __init__(self):
self.__redis_conn = redis.Redis('redis', 6379)
self.__util = ServiceUtil()
