1 ##############################################################################
2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
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 ##############################################################################
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
24 LOG = logging.getLogger(__name__)
29 super(ENV, self).__init__()
30 self.builder = builder.PluginManager()
32 def build_env(self, cfg_intent):
33 return self.builder.build(cfg_intent)
36 return self.builder.clean()
39 def create_images(cfg):
40 return ImageManager(cfg).create_all()
43 def clean_images(cfg):
44 return ImageManager(cfg).clean_all()
47 class Drivers(object):
49 super(Drivers, self).__init__()
50 self.dr_mgr = DriverPluginManager()
52 def install_drivers(self, drivers):
53 LOG.info("install drivers:%s", drivers)
55 ret = self.dr_mgr.load(drivers)
58 def clean_drivers(self):
59 return self.dr_mgr.clean()
61 def autoneg_on(self, iface, nspace):
62 return ethtool.autoneg_on(iface, nspace)
64 def autoneg_off(self, iface, nspace):
65 return ethtool.autoneg_off(iface, nspace)
67 def autoneg_query(self, iface, nspace):
68 return ethtool.autoneg_query(iface, nspace)
72 def affctl_load(self, policy):
73 return affctl.affctl_load(policy)
75 def affctl_list(self):
76 return affctl.affctl_list()
81 super(Perf, self).__init__()
82 self._vnstat = vnstat.VnStat()
83 self._vstfperf = vstfperf.Vstfperf()
86 def run_vnstat(self, device, namespace=None):
87 return self._vnstat.run_vnstat(device, namespace)
89 def kill_vnstat(self, pid, namespace=None):
90 return self._vnstat.kill_vnstat(pid, namespace)
92 def perf_run(self, **kwargs):
93 return self._vstfperf.run(**kwargs)
95 def run_cpuwatch(self, interval = 2):
96 return self._sar.start(interval)
98 def kill_cpuwatch(self, pid):
99 return self._sar.stop(pid)
101 def force_clean(self):
102 self._vstfperf.force_clean()
103 self._sar.force_clean()
104 self._vnstat.force_clean()
108 class EqualizerOps(GetPhyInfo, Optimize):
110 super(EqualizerOps, self).__init__()
113 class BaseAgent(coll.Collect,
118 commandline.CommandLine,
123 super(BaseAgent, self).__init__()
126 class softAgent(BaseAgent, Perf, EqualizerOps):
128 super(softAgent, self).__init__()
131 if __name__ == '__main__':