Initial code drop from Cisco

[nfvbench.git] / nfvbench / chain_runner.py

diff --git a/nfvbench/chain_runner.py b/nfvbench/chain_runner.py
new file mode 100644 (file)
index 0000000..2e222de
--- /dev/null
+++ b/nfvbench/chain_runner.py
@@ -0,0 +1,82 @@
+#!/usr/bin/env python
+# Copyright 2016 Cisco Systems, Inc.  All rights reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+#
+
+from log import LOG
+from service_chain import ServiceChain
+import traceback
+from traffic_client import TrafficClient
+
+
+class ChainRunner(object):
+    """Run selected chain, collect results and analyse them."""
+
+    def __init__(self, config, clients, cred, specs, factory, notifier=None):
+        self.config = config
+        self.clients = clients
+        self.specs = specs
+        self.factory = factory
+        self.chain_name = self.config.service_chain
+
+        try:
+            TORClass = factory.get_tor_class(self.config.tor.type, self.config.no_tor_access)
+        except AttributeError:
+            raise Exception("Requested TOR class '{}' was not found.".format(self.config.tor.type))
+
+        self.clients['tor'] = TORClass(self.config.tor.switches)
+        self.clients['traffic'] = TrafficClient(config, notifier)
+        self.chain = ServiceChain(config, clients, cred, specs, factory, notifier)
+
+        LOG.info('ChainRunner initialized.')
+
+    def run(self):
+        """
+        Run a chain, collect and analyse results.
+
+        :return: dictionary
+        """
+        self.clients['traffic'].start_traffic_generator()
+        self.clients['traffic'].set_macs()
+
+        return self.chain.run()
+
+    def close(self):
+        try:
+            if not self.config.no_cleanup:
+                LOG.info('Cleaning up...')
+            else:
+                LOG.info('Clean up skipped.')
+
+            for client in ['traffic', 'tor']:
+                try:
+                    self.clients[client].close()
+                except Exception as e:
+                    traceback.print_exc()
+                    LOG.error(e)
+
+            self.chain.close()
+        except Exception:
+            traceback.print_exc()
+            LOG.error('Cleanup not finished.')
+
+    def get_version(self):
+        versions = {
+            'Traffic Generator': self.clients['traffic'].get_version(),
+            'TOR': self.clients['tor'].get_version(),
+        }
+
+        versions.update(self.chain.get_version())
+
+        return versions