vstf/vstf/agent/perf/vstfperf.py

   1 #!/usr/bin/python
   2 # -*- coding: utf8 -*-
   3 # author: wly
   4 # date: 2015-09-08
   5 # see license for license details
   6
   7
   8 __doc__ = """
   9 operation: [start, stop, restart]
  10 action: [send, receive]
  11 tool: [pktgen, netperf, qperf, iperf, netmap]
  12 params:
  13     protocol: [tcp_lat, udp_lat, tcp_bw, udp_bw]
  14     namespace: None
  15     src:[
  16        { "iface":"eth0", "ip":"xxx.xxx.xxx.xxx", "mac":"FF:FF:FF:FF:FF:FF"}
  17     ]
  18     dst:[
  19        { "iface":"eth0", "ip":"xxx.xxx.xxx.xxx", "mac":"FF:FF:FF:FF:FF:FF"}
  20     ]
  21     size: 64
  22     threads: 1
  23     ratep: 100000  (pps)
  24     time: 100  (s)
  25 """
  26
  27 import sys
  28 import logging
  29 import vstf.common.constants as cst
  30 import vstf.common.decorator as deco
  31 import vstf.agent.perf.pktgen as vstf_pktgen
  32 import vstf.agent.perf.netmap as vstf_netmap
  33 import vstf.agent.perf.qperf as vstf_qperf
  34 import vstf.agent.perf.iperf as vstf_iperf
  35 import vstf.agent.perf.netperf as vstf_netperf
  36
  37 LOG = logging.getLogger(__name__)
  38
  39
  40 class Vstfperf(object):
  41     def __init__(self):
  42         for tool in cst.TOOLS:
  43             obj_name = 'vstf_' + tool
  44             obj = getattr(sys.modules[__name__], obj_name)
  45             cls_name = tool.title()
  46             cls = getattr(obj, tool.title())
  47             self.__dict__.update({tool: cls()})
  48
  49     @deco.check("operation", choices=cst.OPERATIONS)
  50     @deco.check("action", choices=cst.ACTIONS)
  51     @deco.check("tool", choices=cst.TOOLS)
  52     @deco.check("params", defaults={})
  53     def run(self, **kwargs):
  54         print "_run in"
  55         operation = kwargs.pop("operation")
  56         tool = kwargs.pop("tool")
  57         instance = getattr(self, tool)
  58         action = kwargs.pop("action")
  59         func_name = "%s_%s" % (action, operation)
  60         func = getattr(instance, func_name)
  61         LOG.info(kwargs['params'])
  62         LOG.info(func)
  63         ret = func(**kwargs['params'])
  64         return ret
  65
  66     def force_clean(self):
  67         LOG.info("%s %s start", self.__class__, self.force_clean.__name__)
  68         for tool in cst.TOOLS:
  69             instance = getattr(self, tool)
  70             instance.force_clean()
  71         return True
  72
  73
  74 def unit_test():
  75     from vstf.common.log import setup_logging
  76     setup_logging(level=logging.DEBUG, log_file="/var/log/vstf/vstf-vstfperf.log", clevel=logging.INFO)
  77
  78     perf = Vstfperf()
  79     start = {
  80         "operation": "start",
  81         "action": "send",
  82         "tool": "netperf",
  83         "params": {
  84             "namespace": "vnet_name1",
  85             "protocol": "udp_lat",
  86             "dst": [
  87                 {"ip": "192.168.1.102"}
  88             ],
  89             "size": 64,
  90             "threads": 1,
  91             "time": 100,
  92         },
  93     }
  94     perf.run(**start)
  95
  96     stop = {
  97         "operation": "stop",
  98         "action": "send",
  99         "tool": "netperf",
 100     }
 101     perf.run(**stop)
 102
 103
 104 if __name__ == '__main__':
 105     unit_test()