src/emuvim/cli/monitor.py

   1 """
   2 son-emu monitor CLI
   3 (c) 2016 by Manuel Peuster <manuel.peuster@upb.de>
   4 """
   5
   6 import argparse
   7 import pprint
   8 from tabulate import tabulate
   9 import zerorpc
  10 import time
  11
  12
  13 pp = pprint.PrettyPrinter(indent=4)
  14
  15 class ZeroRpcClient(object):
  16
  17     def __init__(self):
  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         self.cmds = {}
  23
  24     def execute_command(self, args):
  25         if getattr(self, args["command"]) is not None:
  26             # call the local method with the same name as the command arg
  27             getattr(self, args["command"])(args)
  28         else:
  29             print "Command not implemented."
  30
  31     def setup_metric(self, args):
  32         vnf_name = self._parse_vnf_name(args.get("vnf_name"))
  33         vnf_interface = self._parse_vnf_interface(args.get("vnf_name"))
  34         r = self.c.setup_metric(
  35             vnf_name,
  36             vnf_interface,
  37             args.get("metric"))
  38         pp.pprint(r)
  39
  40     def stop_metric(self, args):
  41         vnf_name = self._parse_vnf_name(args.get("vnf_name"))
  42         vnf_interface = self._parse_vnf_interface(args.get("vnf_name"))
  43         r = self.c.stop_metric(
  44             vnf_name,
  45             vnf_interface,
  46             args.get("metric"))
  47         pp.pprint(r)
  48
  49     def setup_flow(self, args):
  50         vnf_name = self._parse_vnf_name(args.get("vnf_name"))
  51         vnf_interface = self._parse_vnf_interface(args.get("vnf_name"))
  52         r = self.c.setup_flow(
  53             vnf_name,
  54             vnf_interface,
  55             args.get("metric"),
  56             args.get("cookie"))
  57         pp.pprint(r)
  58
  59     def stop_flow(self, args):
  60         vnf_name = self._parse_vnf_name(args.get("vnf_name"))
  61         vnf_interface = self._parse_vnf_interface(args.get("vnf_name"))
  62         r = self.c.stop_flow(
  63             vnf_name,
  64             vnf_interface,
  65             args.get("metric"),
  66             args.get("cookie"))
  67         pp.pprint(r)
  68
  69     def prometheus(self, args):
  70         vnf_name = self._parse_vnf_name(args.get("vnf_name"))
  71         vnf_interface = self._parse_vnf_interface(args.get("vnf_name"))
  72         r = self.c.prometheus(
  73             args.get("datacenter"),
  74             vnf_name,
  75             vnf_interface,
  76             args.get("query"))
  77         pp.pprint(r)
  78
  79
  80     def _parse_vnf_name(self, vnf_name_str):
  81         vnf_name = vnf_name_str.split(':')[0]
  82         return vnf_name
  83
  84     def _parse_vnf_interface(self, vnf_name_str):
  85         try:
  86             vnf_interface = vnf_name_str.split(':')[1]
  87         except:
  88             vnf_interface = None
  89
  90         return vnf_interface
  91
  92 parser = argparse.ArgumentParser(description='son-emu monitor')
  93 parser.add_argument(
  94     "command",
  95     choices=['setup_metric', 'stop_metric', 'setup_flow', 'stop_flow','prometheus'],
  96     help="setup/stop a metric/flow to be monitored")
  97 parser.add_argument(
  98     "--vnf_name", "-vnf", dest="vnf_name",
  99     help="vnf name:interface to be monitored")
 100 parser.add_argument(
 101     "--metric", "-m", dest="metric",
 102     help="tx_bytes, rx_bytes, tx_packets, rx_packets")
 103 parser.add_argument(
 104     "--cookie", "-c", dest="cookie",
 105     help="flow cookie to monitor")
 106 parser.add_argument(
 107     "--query", "-q", dest="query",
 108     help="prometheus query")
 109 parser.add_argument(
 110     "--datacenter", "-d", dest="datacenter",
 111     help="Data center where the vnf is deployed")
 112
 113 def main(argv):
 114     #print "This is the son-emu monitor CLI."
 115     #print "Arguments: %s" % str(argv)
 116     args = vars(parser.parse_args(argv))
 117     c = ZeroRpcClient()
 118     c.execute_command(args)