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