def start(self):
self.update_status = UpdateStatus(self.update_publisher)
self.update_status.start()
self._running = True
while self._running:
# logger.debug("drain event")
if not self.update_status.is_alive():
self.update_status.join()
self.update_status = UpdateStatus()
self.update_status.start()
try:
connection.Connection.get_instance().drain_events()
except KeyboardInterrupt as e:
self.stop()
raise e
except Exception as e:
logger.exception(e)
logger.debug("Try to recover connection")
connection.Connection.get_instance().reconnect()
self.reconnect_message_connection()
self.update_status.set_publisher(self.update_publisher)
class ComputeServer():
def __init__(self, configurator):
self.daemon = True
self._running = False
self.configurator = configurator
# routing_key = "nokkhum_compute."+self.ip.replace('.', ":")+".rpc_request"
# self._consumer = consumer.ConsumerFactory().get_consumer(routing_key)
self.reconnect_message_connection()
from .processors import processor_controller
self.processor_controller = processor_controller.ProcessorController()
def process_msg(self, body, message):
logger.debug("get command: %s" % body)
# message.ack()
if 'method' not in body:
logger.debug("ignore message: %s" % body)
elif body['method'] == 'get_machine_specification':
self.update_status.get_machine_specification()
logger.debug(" self.rpc._publisher.routing_key_list %s" %
self.rpc._publisher.routing_key_list)
self.rpc._publisher.routing_key_list.clear()
elif body['method'] == 'update_system_configuration':
settings = body['args']['settings']
UpdateConfiguration().update(settings)
else:
if 'message_id' in body:
respons = None
if body['method'] == 'list_processors':
respons = self.processor_controller.list_processors()
elif body['method'] == 'get_processor_attributes':
respons = self.processor_controller.get_processor_attributes(
body['args']['processor_id'])
elif body['method'] == 'start_processor':
respons = self.processor_controller.start_processor(
body['args']['processor_id'], body['args']['attributes'])
elif body['method'] == 'stop_processor':
respons = self.processor_controller.stop_processor(
body['args']['processor_id'])
respons['message_id'] = body['message_id']
self.rpc.reply(respons, body['reply_to'])
logger.debug("reply_to %s " % body['reply_to'])
logger.debug("response command: %s" % respons)
logger.debug("success process command")
message.ack()
def reconnect_message_connection(self):
from nokkhum.messaging import connection
if connection.Connection.get_instance() is None:
logger.debug("initial message connection")
connection.Connection(self.configurator.settings.get('amq.url'))
ip = "127.0.0.1"
try:
ip = netifaces.ifaddresses(self.configurator.settings.get(
'nokkhum.compute.interface')).setdefault(netifaces.AF_INET)[0]['addr']
except Exception as e:
logger.exception(e)
ip = netifaces.ifaddresses('lo').setdefault(
netifaces.AF_INET)[0]['addr']
self.update_publisher = connection.Connection.get_instance(
).publisher_factory.get_publisher("nokkhum_compute.update_status")
self.rpc = connection.Connection.get_instance(
).get_rpc_factory().get_default_rpc_server(ip)
self.rpc.register_callback(self.process_msg)
def start(self):
self.update_status = UpdateStatus(self.update_publisher)
self.update_status.start()
self._running = True
while self._running:
# logger.debug("drain event")
if not self.update_status.is_alive():
self.update_status.join()
self.update_status = UpdateStatus()
self.update_status.start()
try:
connection.Connection.get_instance().drain_events()
except KeyboardInterrupt as e:
self.stop()
raise e
except Exception as e:
logger.exception(e)
logger.debug("Try to recover connection")
connection.Connection.get_instance().reconnect()
self.reconnect_message_connection()
self.update_status.set_publisher(self.update_publisher)
# import time
# print("sleep 30")
# time.sleep(30)
def stop(self):
logger.debug("start to stop all processors")
self.processor_controller.stop_all()
logger.debug("end to stop all processors")
if self.update_status:
self.update_status.stop()
self._running = False
try:
connection.Connection.get_instance().release()
except:
pass
