Upload the contribution of vstf as bottleneck network framework.

[bottlenecks.git] / vstf / vstf / agent / perf / vstfperf.py

diff --git a/vstf/vstf/agent/perf/vstfperf.py b/vstf/vstf/agent/perf/vstfperf.py
new file mode 100755 (executable)
index 0000000..224380f
--- /dev/null
+++ b/vstf/vstf/agent/perf/vstfperf.py
@@ -0,0 +1,105 @@
+#!/usr/bin/python
+# -*- coding: utf8 -*-
+# author: wly
+# date: 2015-09-08
+# see license for license details
+
+
+__doc__ = """
+operation: [start, stop, restart]
+action: [send, receive]
+tool: [pktgen, netperf, qperf, iperf, netmap]
+params:
+    protocol: [tcp_lat, udp_lat, tcp_bw, udp_bw]
+    namespace: None
+    src:[
+       { "iface":"eth0", "ip":"xxx.xxx.xxx.xxx", "mac":"FF:FF:FF:FF:FF:FF"}
+    ]
+    dst:[
+       { "iface":"eth0", "ip":"xxx.xxx.xxx.xxx", "mac":"FF:FF:FF:FF:FF:FF"}
+    ]
+    size: 64
+    threads: 1
+    ratep: 100000  (pps)
+    time: 100  (s)
+"""
+
+import sys
+import logging
+import vstf.common.constants as cst
+import vstf.common.decorator as deco
+import vstf.agent.perf.pktgen as vstf_pktgen
+import vstf.agent.perf.netmap as vstf_netmap
+import vstf.agent.perf.qperf as vstf_qperf
+import vstf.agent.perf.iperf as vstf_iperf
+import vstf.agent.perf.netperf as vstf_netperf
+
+LOG = logging.getLogger(__name__)
+
+
+class Vstfperf(object):
+    def __init__(self):
+        for tool in cst.TOOLS:
+            obj_name = 'vstf_' + tool
+            obj = getattr(sys.modules[__name__], obj_name)
+            cls_name = tool.title()
+            cls = getattr(obj, tool.title())
+            self.__dict__.update({tool: cls()})
+
+    @deco.check("operation", choices=cst.OPERATIONS)
+    @deco.check("action", choices=cst.ACTIONS)
+    @deco.check("tool", choices=cst.TOOLS)
+    @deco.check("params", defaults={})
+    def run(self, **kwargs):
+        print "_run in"
+        operation = kwargs.pop("operation")
+        tool = kwargs.pop("tool")
+        instance = getattr(self, tool)
+        action = kwargs.pop("action")
+        func_name = "%s_%s" % (action, operation)
+        func = getattr(instance, func_name)
+        LOG.info(kwargs['params'])
+        LOG.info(func)
+        ret = func(**kwargs['params'])
+        return ret
+
+    def force_clean(self):
+        LOG.info("%s %s start", self.__class__, self.force_clean.__name__)
+        for tool in cst.TOOLS:
+            instance = getattr(self, tool)
+            instance.force_clean()
+        return True
+
+
+def unit_test():
+    from vstf.common.log import setup_logging
+    setup_logging(level=logging.DEBUG, log_file="/var/log/vstf/vstf-vstfperf.log", clevel=logging.INFO)
+
+    perf = Vstfperf()
+    start = {
+        "operation": "start",
+        "action": "send",
+        "tool": "netperf",
+        "params": {
+            "namespace": "vnet_name1",
+            "protocol": "udp_lat",
+            "dst": [
+                {"ip": "192.168.1.102"}
+            ],
+            "size": 64,
+            "threads": 1,
+            "time": 100,
+        },
+    }
+    perf.run(**start)
+
+    stop = {
+        "operation": "stop",
+        "action": "send",
+        "tool": "netperf",
+    }
+    perf.run(**stop)
+
+
+if __name__ == '__main__':
+    unit_test()