src/emuvim/api/zerorpc/network.py

   1 """
   2 Distributed Cloud Emulator (dcemulator)
   3 """
   4
   5 import logging
   6 import threading
   7 import zerorpc
   8
   9
  10 logging.basicConfig(level=logging.INFO)
  11
  12
  13 class ZeroRpcApiEndpointDCNetwork(object):
  14     """
  15     Simple API endpoint that offers a zerorpc-based
  16     interface. This interface will be used by the
  17     default command line client.
  18     It can be used as a reference to implement
  19     REST interfaces providing the same semantics,
  20     like e.g. OpenStack compute API.
  21     """
  22
  23     def __init__(self, listenip, port, DCNetwork=None):
  24         if DCNetwork :
  25             self.connectDCNetwork(DCNetwork)
  26         self.ip = listenip
  27         self.port = port
  28         logging.debug("Created monitoring API endpoint %s(%s:%d)" % (
  29             self.__class__.__name__, self.ip, self.port))
  30
  31     def connectDCNetwork(self, net):
  32         self.net = net
  33         logging.info("Connected DCNetwork to API endpoint %s(%s:%d)" % (
  34             self.__class__.__name__, self.ip, self.port))
  35
  36     def start(self):
  37         thread = threading.Thread(target=self._api_server_thread, args=())
  38         thread.daemon = True
  39         thread.start()
  40         logging.debug("Started API endpoint %s(%s:%d)" % (
  41             self.__class__.__name__, self.ip, self.port))
  42
  43     def _api_server_thread(self):
  44         s = zerorpc.Server(DCNetworkApi(self.net))
  45         s.bind("tcp://%s:%d" % (self.ip, self.port))
  46         s.run()
  47
  48     def stop(self):
  49         logging.info("Stop the monitoring API endpoint")
  50         return
  51
  52
  53 class DCNetworkApi(object):
  54     """
  55         The networking and monitoring commands need the scope of the
  56         whole DC network to find the requested vnf. So this API is intended
  57         to work with a DCNetwork.
  58         Just pass through the corresponding request to the
  59         selected data center network. Do not implement provisioning
  60         logic here because will will have multiple API
  61         endpoint implementations at the end.
  62     """
  63
  64     def __init__(self, net):
  65         self.net = net
  66
  67     def network_action_start(self, vnf_src_name, vnf_dst_name, kwargs):
  68         # call DCNetwork method, not really datacenter specific API for now...
  69         # provided dc name needs to be part of API endpoint
  70         # no check if vnfs are really connected to this datacenter...
  71         logging.debug("RPC CALL: network chain start")
  72         try:
  73             c = self.net.setChain(
  74                 vnf_src_name, vnf_dst_name,
  75                 vnf_src_interface=kwargs.get('vnf_src_interface'),
  76                 vnf_dst_interface=kwargs.get('vnf_dst_interface'),
  77                 weight=kwargs.get('weight'),
  78                 match=kwargs.get('match'))
  79             return str(c)
  80         except Exception as ex:
  81             logging.exception("RPC error.")
  82             return ex.message
  83
  84     def network_action_stop(self, vnf_src_name, vnf_dst_name, vnf_src_interface=None, vnf_dst_interface=None, weight=None):
  85         # call DCNetwork method, not really datacenter specific API for now...
  86         # provided dc name needs to be part of API endpoint
  87         # no check if vnfs are really connected to this datacenter...
  88         logging.debug("RPC CALL: network chain stop")
  89         try:
  90             c = self.net.setChain(
  91                 vnf_src_name, vnf_dst_name,
  92                 vnf_src_interface=vnf_src_interface,
  93                 vnf_dst_interface=vnf_dst_interface,
  94                 cmd='del-flows',
  95                 weight=weight)
  96             return c
  97         except Exception as ex:
  98             logging.exception("RPC error.")
  99             return ex.message
 100
 101     # setup the rate measurement for a vnf interface
 102     def setup_metric(self, vnf_name, vnf_interface, metric):
 103         logging.debug("RPC CALL: setup metric")
 104         try:
 105             c = self.net.monitor_agent.setup_metric(vnf_name, vnf_interface, metric)
 106             return c
 107         except Exception as ex:
 108             logging.exception("RPC error.")
 109             return ex.message
 110
 111     # remove the rate measurement for a vnf interface
 112     def stop_metric(self, vnf_name, vnf_interface, metric):
 113         logging.debug("RPC CALL: stop metric")
 114         try:
 115             c = self.net.monitor_agent.stop_metric(vnf_name, vnf_interface, metric)
 116             return c
 117         except Exception as ex:
 118             logging.exception("RPC error.")
 119             return ex.message
 120
 121
 122