modify Collector, CollectorDataset for LinuxNetAgent

[ti2/ti2.git] / linkpair / collect / collector.py

diff --git a/linkpair/collect/collector.py b/linkpair/collect/collector.py
index 2c24791..8e46891 100755 (executable)
--- a/linkpair/collect/collector.py
+++ b/linkpair/collect/collector.py
@@ -16,6 +16,7 @@ from linkpair.collect.utils import CollectUtils
 from linkpair.collect.collector_dataset import CollectorDataset
 from linkpair.collect.agent.commandrunner import CommandRunner
 from linkpair.collect.agent.linux_agent import LinuxAgent
+from linkpair.collect.agent.linuxnet_agent import LinuxNetAgent
 from linkpair.collect.agent.namespace_agent import NamespaceAgent
 from linkpair.collect.agent.veth_agent import VethAgent
 from linkpair.collect.agent.ovs_agent import OVSAgent
@@ -37,6 +38,7 @@ class Collector(object):
         self._devices = {}
         self._ports = {}
         self._os_info = {}
+        self._linuxnet = {}
         self._linkpairs = []
         self._port_to_br = {}
         self._iface_to_nss = {}
@@ -61,7 +63,7 @@ class Collector(object):
 
     def run(self):
         cd = CollectorDataset(
-            self._devices, self._ports, self._os_info, self._linkpairs,
+            self._devices, self._ports, self._os_info, self._linuxnet, self._linkpairs,
             self._port_to_br, self._iface_to_nss, self._veth_peer)
         cu = CollectUtils(
             cd, self._u, self._formatter)
@@ -70,6 +72,11 @@ class Collector(object):
                 self._runner, cu, self._remote_desc, cd, self._formatter)
             self._u.debug_out("OSAgent running...")
             os_agent.run()
+        if self._opts.disable_linuxnet_agent == False:
+            linuxnet_agent = LinuxNetAgent(
+                self._runner, cu, self._remote_desc, cd, self._formatter)
+            self._u.debug_out("LinuxNetAgent running...")
+            linuxnet_agent.run()
         if self._opts.disable_namespace_agent == False:
             namespace_agent = NamespaceAgent(
                 self._runner, cu, self._remote_desc, cd, self._formatter)