src/emuvim/cli/monitor.py

   1 """
   2 son-emu monitor CLI
   3 (c) 2016 by Manuel Peuster <manuel.peuster@upb.de>
   4 Steven Van Rossem <steven.vanrossem@intec.ugent.be>
   5 """
   6
   7 import argparse
   8 import pprint
   9 import zerorpc
  10 from emuvim.cli import prometheus
  11
  12 pp = pprint.PrettyPrinter(indent=4)
  13
  14 class ZeroRpcClient(object):
  15
  16     def __init__(self):
  17         # network zerorpc
  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")
  22
  23         # compute zerorpc
  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
  26
  27         self.cmds = {}
  28
  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)
  33         else:
  34             print("Command not implemented.")
  35
  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(
  40             vnf_name,
  41             vnf_interface,
  42             args.get("metric"))
  43         pp.pprint(r)
  44
  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(
  49             vnf_name,
  50             vnf_interface,
  51             args.get("metric"))
  52         pp.pprint(r)
  53
  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(
  58             vnf_name,
  59             vnf_interface,
  60             args.get("metric"),
  61             args.get("cookie"))
  62         pp.pprint(r)
  63
  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"))
  67         r = self.c.stop_flow(
  68             vnf_name,
  69             vnf_interface,
  70             args.get("metric"),
  71             args.get("cookie"))
  72         pp.pprint(r)
  73
  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"),
  79             vnf_name,
  80             vnf_interface,
  81             args.get("query"))
  82         pp.pprint(r)
  83
  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)
  92
  93         r = prometheus.query_Prometheus(query)
  94         pp.pprint(r)
  95
  96
  97     def _parse_vnf_name(self, vnf_name_str):
  98         vnf_name = vnf_name_str.split(':')[0]
  99         return vnf_name
 100
 101     def _parse_vnf_interface(self, vnf_name_str):
 102         try:
 103             vnf_interface = vnf_name_str.split(':')[1]
 104         except:
 105             vnf_interface = None
 106
 107         return vnf_interface
 108
 109 parser = argparse.ArgumentParser(description='son-emu monitor')
 110 parser.add_argument(
 111     "command",
 112     choices=['setup_metric', 'stop_metric', 'setup_flow', 'stop_flow','prometheus'],
 113     help="setup/stop a metric/flow to be monitored or query Prometheus")
 114 parser.add_argument(
 115     "--vnf_name", "-vnf", dest="vnf_name",
 116     help="vnf name:interface to be monitored")
 117 parser.add_argument(
 118     "--metric", "-m", dest="metric",
 119     help="tx_bytes, rx_bytes, tx_packets, rx_packets")
 120 parser.add_argument(
 121     "--cookie", "-c", dest="cookie",
 122     help="flow cookie to monitor")
 123 parser.add_argument(
 124     "--query", "-q", dest="query",
 125     help="prometheus query")
 126 parser.add_argument(
 127     "--datacenter", "-d", dest="datacenter",
 128     help="Data center where the vnf is deployed")
 129
 130 def main(argv):
 131     #print "This is the son-emu monitor CLI."
 132     #print "Arguments: %s" % str(argv)
 133     args = vars(parser.parse_args(argv))
 134     c = ZeroRpcClient()
 135     c.execute_command(args)