class Dispatcher(threading.Thread):
# FIXME: quick and dirty
def __init__(self):
super(Dispatcher, self).__init__()
self.active = True
self.socket = RedisQueue(REDIS_CHANNEL)
# cache nodes
self._nodes = dict()
def register(self, nodes):
# FIXME: currently only register to dictionary
self._nodes.update({node.name: node for node in nodes})
def run(self):
while self.active:
try:
data = self.socket.get()
if data is not None:
if data.get('halt', False):
self.active = False
continue
func = '_load_{module_name}'.format(**data['data']['invocation'])
getattr(self._nodes[data['host']], func)(data['data'])
# FIXME: replace with log
print('dispatcher thread id is', self, threading.currentThread(), os.getpid())
except (AttributeError, KeyError, TypeError) as err:
# FIXME: replace with log
print(err)
time.sleep(0.1)
def close(self):
self.socket.put({'halt': True})
self.join()
def redis_put(data):
socket = RedisQueue(REDIS_CHANNEL)
socket.put(data)