| """ |
| Distributed Cloud Emulator (dcemulator) |
| (c) 2015 by Manuel Peuster <manuel.peuster@upb.de> |
| """ |
| |
| import logging |
| import threading |
| import zerorpc |
| |
| logging.basicConfig(level=logging.DEBUG) |
| |
| |
| class ZeroRpcApiEndpoint(object): |
| """ |
| Simple API endpoint that offers a zerorpc-based |
| interface. This interface will be used by the |
| default command line client. |
| It can be used as a reference to implement |
| REST interfaces providing the same semantics, |
| like e.g. OpenStack compute API. |
| """ |
| |
| def __init__(self, listenip, port): |
| self.dcs = {} |
| self.ip = listenip |
| self.port = port |
| logging.debug("Created API endpoint %s(%s:%d)" % ( |
| self.__class__.__name__, self.ip, self.port)) |
| |
| def connectDatacenter(self, dc): |
| self.dcs[dc.name] = dc |
| logging.info("Connected DC(%s) to API endpoint %s(%s:%d)" % ( |
| dc.name, self.__class__.__name__, self.ip, self.port)) |
| |
| def start(self): |
| thread = threading.Thread(target=self._api_server_thread, args=()) |
| thread.daemon = True |
| thread.start() |
| logging.debug("Started API endpoint %s(%s:%d)" % ( |
| self.__class__.__name__, self.ip, self.port)) |
| |
| def _api_server_thread(self): |
| s = zerorpc.Server(MultiDatacenterApi(self.dcs)) |
| s.bind("tcp://%s:%d" % (self.ip, self.port)) |
| s.run() |
| |
| |
| class MultiDatacenterApi(object): |
| """ |
| Just pass through the corresponding request to the |
| selected data center. Do not implement provisioning |
| logic here because will will have multiple API |
| endpoint implementations at the end. |
| """ |
| |
| def __init__(self, dcs): |
| self.dcs = dcs |
| |
| def compute_action_start(self, dc_name, compute_name): |
| # TODO what to return UUID / given name / internal name ? |
| logging.debug("RPC CALL: compute start") |
| if dc_name in self.dcs: |
| return self.dcs[dc_name].addCompute(compute_name) |
| |
| def compute_action_stop(self, dc_name, compute_name): |
| logging.info("RPC CALL: compute stop") |
| if dc_name in self.dcs: |
| return self.dcs[dc_name].removeCompute(compute_name) |
| |
| def compute_list(self, dc_name): |
| logging.info("RPC CALL: compute list") |
| if dc_name in self.dcs: |
| return [(c.name, c.IP()) for c in self.dcs[dc_name].listCompute()] |
| |
| def compute_status(self, dc_name, compute_name): |
| logging.info("RPC CALL: compute status") |
| # TODO implement |