vstf/vstf/agent/softagent.py

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