79ff25be00bb389c4a55878f3677be1352240f04
3 (c) 2016 by Manuel Peuster <manuel.peuster@upb.de>
11 pp
= pprint
.PrettyPrinter(indent
=4)
13 class ZeroRpcClient(object):
17 self
.c
= zerorpc
.Client()
18 # TODO connect to DCNetwork API
19 #self.c.connect("tcp://127.0.0.1:4242") # TODO hard coded for now. we'll change this later
20 self
.c
.connect("tcp://127.0.0.1:5151")
23 self
.compute_api
= zerorpc
.Client(heartbeat
=None, timeout
=120) # heartbeat=None, timeout=120
24 self
.compute_api
.connect("tcp://127.0.0.1:4242") # TODO hard coded for now. we'll change this later
28 def execute_command(self
, args
):
29 if getattr(self
, args
["command"]) is not None:
30 # call the local method with the same name as the command arg
31 getattr(self
, args
["command"])(args
)
33 print "Command not implemented."
35 def setup_metric(self
, args
):
36 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
37 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
38 r
= self
.c
.setup_metric(
44 def stop_metric(self
, args
):
45 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
46 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
47 r
= self
.c
.stop_metric(
53 def setup_flow(self
, args
):
54 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
55 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
56 r
= self
.c
.setup_flow(
63 def stop_flow(self
, args
):
64 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
65 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
73 def prometheus_zrpc(self
, args
):
74 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
75 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
76 r
= self
.c
.prometheus(
77 args
.get("datacenter"),
83 def prometheus(self
, args
):
84 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
85 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
86 dc_label
= args
.get("datacenter")
87 query
= args
.get("query")
88 vnf_status
= self
.compute_api
.compute_status(dc_label
, vnf_name
)
89 uuid
= vnf_status
['id']
90 query
= query
.replace('<uuid>', uuid
)
92 r
= prometheus
.query_Prometheus(query
)
96 def _parse_vnf_name(self
, vnf_name_str
):
97 vnf_name
= vnf_name_str
.split(':')[0]
100 def _parse_vnf_interface(self
, vnf_name_str
):
102 vnf_interface
= vnf_name_str
.split(':')[1]
108 parser
= argparse
.ArgumentParser(description
='son-emu monitor')
111 choices
=['setup_metric', 'stop_metric', 'setup_flow', 'stop_flow','prometheus'],
112 help="setup/stop a metric/flow to be monitored or query Prometheus")
114 "--vnf_name", "-vnf", dest
="vnf_name",
115 help="vnf name:interface to be monitored")
117 "--metric", "-m", dest
="metric",
118 help="tx_bytes, rx_bytes, tx_packets, rx_packets")
120 "--cookie", "-c", dest
="cookie",
121 help="flow cookie to monitor")
123 "--query", "-q", dest
="query",
124 help="prometheus query")
126 "--datacenter", "-d", dest
="datacenter",
127 help="Data center where the vnf is deployed")
130 #print "This is the son-emu monitor CLI."
131 #print "Arguments: %s" % str(argv)
132 args
= vars(parser
.parse_args(argv
))
134 c
.execute_command(args
)