vstf/vstf/agent/softagent.py

   1 ##############################################################################
   2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10
  11 import logging
  12 import time
  13 from vstf.agent.env.basic.image_manager import ImageManager
  14 from vstf.agent.env.basic.source_manager import SourceCodeManager
  15 from vstf.agent.env.basic import commandline
  16 from vstf.agent.env.basic.device_manager import DeviceManager
  17 from vstf.agent.env.basic import collect as coll
  18 from vstf.agent.perf import netns, vnstat, vstfperf, sar, ethtool, affctl
  19 from vstf.agent.env import builder
  20 from vstf.agent.equalizer.get_info import GetPhyInfo
  21 from vstf.agent.equalizer.optimize import Optimize
  22 from vstf.agent.env.driver_plugins.manager import DriverPluginManager
  23
  24 LOG = logging.getLogger(__name__)
  25
  26
  27 class ENV(object):
  28     def __init__(self):
  29         super(ENV, self).__init__()
  30         self.builder = builder.PluginManager()
  31
  32     def build_env(self, cfg_intent):
  33         return self.builder.build(cfg_intent)
  34
  35     def clean_env(self):
  36         return self.builder.clean()
  37
  38     @staticmethod
  39     def create_images(cfg):
  40         return ImageManager(cfg).create_all()
  41
  42     @staticmethod
  43     def clean_images(cfg):
  44         return ImageManager(cfg).clean_all()
  45
  46
  47 class Drivers(object):
  48     def __init__(self):
  49         super(Drivers, self).__init__()
  50         self.dr_mgr = DriverPluginManager()
  51
  52     def install_drivers(self, drivers):
  53         LOG.info("install drivers:%s", drivers)
  54         self.dr_mgr.clean()
  55         ret = self.dr_mgr.load(drivers)
  56         return ret
  57
  58     def clean_drivers(self):
  59         return self.dr_mgr.clean()
  60
  61     def autoneg_on(self, iface, nspace):
  62         return ethtool.autoneg_on(iface, nspace)
  63
  64     def autoneg_off(self, iface, nspace):
  65         return ethtool.autoneg_off(iface, nspace)
  66
  67     def autoneg_query(self, iface, nspace):
  68         return ethtool.autoneg_query(iface, nspace)
  69
  70
  71 class Cpu(object):
  72     def affctl_load(self, policy):
  73         return affctl.affctl_load(policy)
  74
  75     def affctl_list(self):
  76         return affctl.affctl_list()
  77
  78
  79 class Perf(object):
  80     def __init__(self):
  81         super(Perf, self).__init__()
  82         self._vnstat = vnstat.VnStat()
  83         self._vstfperf = vstfperf.Vstfperf()
  84         self._sar = sar.Sar()
  85
  86     def run_vnstat(self, device, namespace=None):
  87         return self._vnstat.run_vnstat(device, namespace)
  88
  89     def kill_vnstat(self, pid, namespace=None):
  90         return self._vnstat.kill_vnstat(pid, namespace)
  91
  92     def perf_run(self, **kwargs):
  93         return self._vstfperf.run(**kwargs)
  94
  95     def run_cpuwatch(self, interval = 2):
  96         return self._sar.start(interval)
  97
  98     def kill_cpuwatch(self, pid):
  99         return self._sar.stop(pid)
 100
 101     def force_clean(self):
 102         self._vstfperf.force_clean()
 103         self._sar.force_clean()
 104         self._vnstat.force_clean()
 105         return True
 106
 107
 108 class EqualizerOps(GetPhyInfo, Optimize):
 109     def __init__(self):
 110         super(EqualizerOps, self).__init__()
 111
 112
 113 class BaseAgent(coll.Collect,
 114                 ENV,
 115                 Cpu,
 116                 Drivers,
 117                 DeviceManager,
 118                 commandline.CommandLine,
 119                 netns.NetnsManager,
 120                 SourceCodeManager
 121                 ):
 122     def __init__(self):
 123         super(BaseAgent, self).__init__()
 124
 125
 126 class softAgent(BaseAgent, Perf, EqualizerOps):
 127     def __init__(self):
 128         super(softAgent, self).__init__()
 129
 130
 131 if __name__ == '__main__':
 132     softAgent()
 133