class DaemonProtocol(object):
def __init__(self, illuminant_uri):
self.daemon_uri = None
self.illuminant_uri = illuminant_uri
self.illuminant_proxy = ServerProxy(illuminant_uri)
self.methods = {
name: method
for name, method in DaemonProtocol.__dict__.items()
if not name.startswith('_')
}
def _dispatch(self, method, params):
"""
Dispatch all methods not included in DaemonProtocol will be dispatch to Illuminant
"""
if method in self.methods:
return self.methods[method](self, *params)
else:
return getattr(self.illuminant_proxy, method)(*params)
def _ready(self, uri):
self.daemon_uri = uri
# METHODS FOR ILLUMINANT TO CONTROL SELL
def launch(self, package_name):
pass
def execute(self, command):
pass
def terminate(self):
pass
def is_alive(self):
pass
def is_busy(self):
pass
# METHODS FOR ILLUMINANT TO MAINTAIN SERVICES
def registerService(self, caller_id, service, service_api, caller_api):
"""
This will register service to illuminant, not Master
"""
# map the port
_, port = service_api[9:].split(':')
subprocess.call(['iptables', '-t', 'nat', '-A', 'PREROUTING', '-p', 'tcp', '--dport', '30000', '-j', 'REDIRECT',
'--to-ports', port])
return self.illuminant_proxy.regCell(service, service_api, self.daemon_uri)
def unregisterService(self, caller_id, service, service_api):
"""
This will unregister service from illuminant, not Master
"""
# cancel port mapping
subprocess.call(['iptables', '-t', 'nat', '-A', 'PREROUTING', '-p', 'tcp', '--dport', '30000', '-j', 'REDIRECT',
'--to-ports', '30000'])
return self.illuminant_proxy.unregCell(service, service_api, self.daemon_uri)
# THESE METHODS INTERRUPT ORIGINAL CALLS AND DO NOTHING
def registerPublisher(self, caller_id, topic, topic_type, caller_api):
return interrupted_response([])
def registerSubscriber(self, caller_id, topic, topic_type, caller_api):
return interrupted_response([])
def unregisterPublisher(self, caller_id, topic, caller_api):
return interrupted_response(0)
def unregisterSubscriber(self, caller_id, topic, caller_api):
return interrupted_response(0)
