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