set chaining via VLANs

[osm/vim-emu.git] / src / emuvim / cli / monitor.py

diff --git a/src/emuvim/cli/monitor.py b/src/emuvim/cli/monitor.py
index 3b667f7..123abe5 100755 (executable)
--- a/src/emuvim/cli/monitor.py
+++ b/src/emuvim/cli/monitor.py
@@ -46,6 +46,27 @@ class ZeroRpcClient(object):
             args.get("metric"))\r
         pp.pprint(r)\r
 \r
+    def setup_flow(self, args):\r
+        vnf_name = self._parse_vnf_name(args.get("vnf_name"))\r
+        vnf_interface = self._parse_vnf_interface(args.get("vnf_name"))\r
+        r = self.c.setup_flow(\r
+            vnf_name,\r
+            vnf_interface,\r
+            args.get("metric"),\r
+            args.get("cookie"))\r
+        pp.pprint(r)\r
+\r
+    def prometheus(self, args):\r
+        vnf_name = self._parse_vnf_name(args.get("vnf_name"))\r
+        vnf_interface = self._parse_vnf_interface(args.get("vnf_name"))\r
+        r = self.c.prometheus(\r
+            args.get("datacenter"),\r
+            vnf_name,\r
+            vnf_interface,\r
+            args.get("query"))\r
+        pp.pprint(r)\r
+\r
+\r
     def _parse_vnf_name(self, vnf_name_str):\r
         vnf_name = vnf_name_str.split(':')[0]\r
         return vnf_name\r
@@ -58,7 +79,7 @@ class ZeroRpcClient(object):
 \r
         return vnf_interface\r
 \r
-parser = argparse.ArgumentParser(description='son-emu network')\r
+parser = argparse.ArgumentParser(description='son-emu monitor')\r
 parser.add_argument(\r
     "command",\r
     help="Action to be executed")\r
@@ -68,7 +89,15 @@ parser.add_argument(
 parser.add_argument(\r
     "--metric", "-m", dest="metric",\r
     help="tx_bytes, rx_bytes, tx_packets, rx_packets")\r
-\r
+parser.add_argument(\r
+    "--cookie", "-c", dest="cookie",\r
+    help="flow cookie to monitor")\r
+parser.add_argument(\r
+    "--query", "-q", dest="query",\r
+    help="prometheus query")\r
+parser.add_argument(\r
+    "--datacenter", "-d", dest="datacenter",\r
+    help="Data center where the vnf is deployed")\r
 \r
 def main(argv):\r
     #print "This is the son-emu monitor CLI."\r