3 (c) 2016 by Manuel Peuster <manuel.peuster@upb.de>
4 Steven Van Rossem <steven.vanrossem@intec.ugent.be>
10 from emuvim
.cli
import prometheus
12 pp
= pprint
.PrettyPrinter(indent
=4)
14 class ZeroRpcClient(object):
18 self
.c
= zerorpc
.Client()
19 # TODO connect to DCNetwork API
20 #self.c.connect("tcp://127.0.0.1:4242") # TODO hard coded for now. we'll change this later
21 self
.c
.connect("tcp://127.0.0.1:5151")
24 self
.compute_api
= zerorpc
.Client(heartbeat
=None, timeout
=120) # heartbeat=None, timeout=120
25 self
.compute_api
.connect("tcp://127.0.0.1:4242") # TODO hard coded for now. we'll change this later
29 def execute_command(self
, args
):
30 if getattr(self
, args
["command"]) is not None:
31 # call the local method with the same name as the command arg
32 getattr(self
, args
["command"])(args
)
34 print("Command not implemented.")
36 def setup_metric(self
, args
):
37 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
38 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
39 r
= self
.c
.setup_metric(
45 def stop_metric(self
, args
):
46 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
47 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
48 r
= self
.c
.stop_metric(
54 def setup_flow(self
, args
):
55 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
56 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
57 r
= self
.c
.setup_flow(
64 def stop_flow(self
, args
):
65 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
66 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
74 def prometheus_zrpc(self
, args
):
75 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
76 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
77 r
= self
.c
.prometheus(
78 args
.get("datacenter"),
84 def prometheus(self
, args
):
85 vnf_name
= self
._parse
_vnf
_name
(args
.get("vnf_name"))
86 vnf_interface
= self
._parse
_vnf
_interface
(args
.get("vnf_name"))
87 dc_label
= args
.get("datacenter")
88 query
= args
.get("query")
89 vnf_status
= self
.compute_api
.compute_status(dc_label
, vnf_name
)
90 uuid
= vnf_status
['id']
91 query
= query
.replace('<uuid>', uuid
)
93 r
= prometheus
.query_Prometheus(query
)
97 def _parse_vnf_name(self
, vnf_name_str
):
98 vnf_name
= vnf_name_str
.split(':')[0]
101 def _parse_vnf_interface(self
, vnf_name_str
):
103 vnf_interface
= vnf_name_str
.split(':')[1]
109 parser
= argparse
.ArgumentParser(description
='son-emu monitor')
112 choices
=['setup_metric', 'stop_metric', 'setup_flow', 'stop_flow','prometheus'],
113 help="setup/stop a metric/flow to be monitored or query Prometheus")
115 "--vnf_name", "-vnf", dest
="vnf_name",
116 help="vnf name:interface to be monitored")
118 "--metric", "-m", dest
="metric",
119 help="tx_bytes, rx_bytes, tx_packets, rx_packets")
121 "--cookie", "-c", dest
="cookie",
122 help="flow cookie to monitor")
124 "--query", "-q", dest
="query",
125 help="prometheus query")
127 "--datacenter", "-d", dest
="datacenter",
128 help="Data center where the vnf is deployed")
131 #print "This is the son-emu monitor CLI."
132 #print "Arguments: %s" % str(argv)
133 args
= vars(parser
.parse_args(argv
))
135 c
.execute_command(args
)