Merge pull request #74 from stevenvanrossem/master

[osm/vim-emu.git] / src / emuvim / dcemulator / monitoring.py

diff --git a/src/emuvim/dcemulator/monitoring.py b/src/emuvim/dcemulator/monitoring.py
index 094c09b..50393ea 100755 (executable)
--- a/src/emuvim/dcemulator/monitoring.py
+++ b/src/emuvim/dcemulator/monitoring.py
@@ -4,6 +4,7 @@ import urllib2
 import logging\r
 from mininet.node import  OVSSwitch\r
 import ast\r
+import time\r
 logging.basicConfig(level=logging.INFO)\r
 \r
 """\r
@@ -18,9 +19,38 @@ class DCNetworkMonitor():
         self.port = '8080'\r
         self.REST_api = 'http://{0}:{1}'.format(self.ip,self.port)\r
 \r
+        self.previous_measurement = 0\r
+        self.previous_monitor_time = 0\r
+        self.switch_dpid = 0\r
+        self.metric_key = None\r
+        self.mon_port = None\r
+\r
+    # first set some parameters, before measurement can start\r
+    def setup_rate_measurement(self, vnf_name, vnf_interface=None, direction='tx', metric='packets'):\r
+        # check if port is specified (vnf:port)\r
+        if vnf_interface is None:\r
+            # take first interface by default\r
+            connected_sw = self.net.DCNetwork_graph.neighbors(vnf_name)[0]\r
+            link_dict = self.net.DCNetwork_graph[vnf_name][connected_sw]\r
+            vnf_interface = link_dict[0]['src_port_id']\r
+\r
+        for connected_sw in self.net.DCNetwork_graph.neighbors(vnf_name):\r
+            link_dict = self.net.DCNetwork_graph[vnf_name][connected_sw]\r
+            for link in link_dict:\r
+                # logging.info("{0},{1}".format(link_dict[link],vnf_interface))\r
+                if link_dict[link]['src_port_id'] == vnf_interface:\r
+                    # found the right link and connected switch\r
+                    # logging.info("{0},{1}".format(link_dict[link]['src_port_id'], vnf_source_interface))\r
+                    self.mon_port = link_dict[link]['dst_port']\r
+                    break\r
 \r
-    def get_rate(self, vnf_name, direction='tx'):\r
         try:\r
+            # default port direction to monitor\r
+            if direction is None:\r
+                direction = 'tx'\r
+            if metric is None:\r
+                metric = 'packets'\r
+\r
             vnf_switch = self.net.DCNetwork_graph.neighbors(str(vnf_name))\r
 \r
             if len(vnf_switch) > 1:\r
@@ -33,27 +63,51 @@ class DCNetworkMonitor():
                 vnf_switch = vnf_switch[0]\r
             next_node = self.net.getNodeByName(vnf_switch)\r
 \r
-            if not isinstance( next_node, OVSSwitch ):\r
+            if not isinstance(next_node, OVSSwitch):\r
                 logging.info("vnf: {0} is not connected to switch".format(vnf_name))\r
                 return\r
 \r
-            mon_port = self.net.DCNetwork_graph[vnf_name][vnf_switch]['dst_port']\r
-            switch_dpid = x = int(str(next_node.dpid),16)\r
+            self.previous_measurement = 0\r
+            self.previous_monitor_time = 0\r
 \r
-            ret = self.REST_cmd('stats/port', switch_dpid)\r
-            port_stat_dict = ast.literal_eval(ret)\r
-            for port_stat in port_stat_dict[str(switch_dpid)]:\r
-                if port_stat['port_no'] == mon_port:\r
-                    return port_stat\r
-                    break\r
-\r
-            return ret\r
+            #self.switch_dpid = x = int(str(next_node.dpid), 16)\r
+            self.switch_dpid = int(str(next_node.dpid), 16)\r
+            self.metric_key = '{0}_{1}'.format(direction, metric)\r
 \r
         except Exception as ex:\r
             logging.exception("get_txrate error.")\r
             return ex.message\r
 \r
 \r
+    # call this function repeatedly for streaming measurements\r
+    def get_rate(self, vnf_name, vnf_interface=None, direction='tx', metric='packets'):\r
+\r
+            key = self.metric_key\r
+\r
+            ret = self.REST_cmd('stats/port', self.switch_dpid)\r
+            port_stat_dict = ast.literal_eval(ret)\r
+            for port_stat in port_stat_dict[str(self.switch_dpid)]:\r
+                if port_stat['port_no'] == self.mon_port:\r
+                    port_uptime = port_stat['duration_sec'] + port_stat['duration_nsec'] * 10 ** (-9)\r
+                    this_measurement = port_stat[key]\r
+\r
+                    if self.previous_monitor_time <= 0 or self.previous_monitor_time >= port_uptime:\r
+                        self.previous_measurement = port_stat[key]\r
+                        self.previous_monitor_time = port_uptime\r
+                        # do first measurement\r
+                        time.sleep(1)\r
+                        byte_rate = self.get_rate(vnf_name, vnf_interface, direction, metric)\r
+                        return byte_rate\r
+                    else:\r
+                        time_delta = (port_uptime - self.previous_monitor_time)\r
+                        byte_rate = (this_measurement - self.previous_measurement) / float(time_delta)\r
+                        #logging.info('uptime:{2} delta:{0} rate:{1}'.format(time_delta,byte_rate,port_uptime))\r
+\r
+                    self.previous_measurement = this_measurement\r
+                    self.previous_monitor_time = port_uptime\r
+                    return byte_rate\r
+\r
+            return ret\r
 \r
     def REST_cmd(self, prefix, dpid):\r
         url = self.REST_api + '/' + str(prefix) + '/' + str(dpid)\r