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                 bidirectional=kwargs.get('bidirectional'))
  80             return str(c)
  81         except Exception as ex:
  82             logging.exception("RPC error.")
  83             return ex.message
  84
  85     def network_action_stop(self, vnf_src_name, vnf_dst_name, kwargs):
  86         # call DCNetwork method, not really datacenter specific API for now...
  87         # provided dc name needs to be part of API endpoint
  88         # no check if vnfs are really connected to this datacenter...
  89         logging.debug("RPC CALL: network chain stop")
  90         try:
  91             c = self.net.setChain(
  92                 vnf_src_name, vnf_dst_name,
  93                 vnf_src_interface=kwargs.get('vnf_src_interface'),
  94                 vnf_dst_interface=kwargs.get('vnf_dst_interface'),
  95                 cmd='del-flows',
  96                 weight=kwargs.get('weight'),
  97                 match=kwargs.get('match'),
  98                 bidirectional=kwargs.get('bidirectional'))
  99             return c
 100         except Exception as ex:
 101             logging.exception("RPC error.")
 102             return ex.message
 103
 104     # setup the rate measurement for a vnf interface
 105     def setup_metric(self, vnf_name, vnf_interface, metric):
 106         logging.debug("RPC CALL: setup metric")
 107         try:
 108             c = self.net.monitor_agent.setup_metric(vnf_name, vnf_interface, metric)
 109             return c
 110         except Exception as ex:
 111             logging.exception("RPC error.")
 112             return ex.message
 113
 114     # remove the rate measurement for a vnf interface
 115     def stop_metric(self, vnf_name, vnf_interface, metric):
 116         logging.debug("RPC CALL: stop metric")
 117         try:
 118             c = self.net.monitor_agent.stop_metric(vnf_name, vnf_interface, metric)
 119             return c
 120         except Exception as ex:
 121             logging.exception("RPC error.")
 122             return ex.message
 123
 124
 125