From bea3a10529b5a10509e4573fdda4573e56b773e0 Mon Sep 17 00:00:00 2001 From: edmaas Date: Sun, 26 Feb 2017 15:42:44 +0100 Subject: [PATCH] added a topology which will stop when a certain method is called or a SIGTERM/SIGINT is received. To be used by son-profiling --- src/emuvim/examples/profiling.py | 109 +++++++++++++++++++++++++++++++ 1 file changed, 109 insertions(+) create mode 100644 src/emuvim/examples/profiling.py diff --git a/src/emuvim/examples/profiling.py b/src/emuvim/examples/profiling.py new file mode 100644 index 0000000..ec743e1 --- /dev/null +++ b/src/emuvim/examples/profiling.py @@ -0,0 +1,109 @@ +""" +Copyright (c) 2015 SONATA-NFV +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. + +Neither the name of the SONATA-NFV [, ANY ADDITIONAL AFFILIATION] +nor the names of its contributors may be used to endorse or promote +products derived from this software without specific prior written +permission. + +This work has been performed in the framework of the SONATA project, +funded by the European Commission under Grant number 671517 through +the Horizon 2020 and 5G-PPP programmes. The authors would like to +acknowledge the contributions of their colleagues of the SONATA +partner consortium (www.sonata-nfv.eu). +""" +""" +A simple topology with a single data center for usage in son-profile. + +""" + +import logging +from mininet.log import setLogLevel +from emuvim.dcemulator.net import DCNetwork +from emuvim.api.rest.rest_api_endpoint import RestApiEndpoint +from emuvim.api.sonata import SonataDummyGatekeeperEndpoint +from mininet.node import RemoteController +import argparse +import sys +import signal + +logging.basicConfig() +LOG = logging.getLogger("sonata-profiling") +LOG.setLevel(logging.DEBUG) +logging.getLogger("werkzeug").setLevel(logging.WARNING) + +""" + Catches SIGINT and SIGTERM to shut the topology down gracefully. +""" +class GracefulKiller: + def __init__(self, to_be_killed): + signal.signal(signal.SIGINT, self.exit_gracefully) + signal.signal(signal.SIGTERM, self.exit_gracefully) + self.to_be_killed = to_be_killed + + def exit_gracefully(self, signum, frame): + self.to_be_killed.stop_it() + + +""" + A simple topology with only one data center which will stop when another thread tells it to or when a time limit is reached. + :args: an argument list which may contain the time limit +""" +class Profiling: + + stop_now = False + + def __init__(self): + GracefulKiller(self) + # create topology + self.net = DCNetwork(controller=RemoteController, monitor=False, enable_learning=False) + self.dc = self.net.addDatacenter("dc1") + + # add the command line interface endpoint to each DC (REST API) + self.rapi1 = RestApiEndpoint("0.0.0.0", 5001) + self.rapi1.connectDCNetwork(self.net) + self.rapi1.connectDatacenter(self.dc) + # run API endpoint server (in another thread, don't block) + self.rapi1.start() + + # add the SONATA dummy gatekeeper to each DC + self.sdkg1 = SonataDummyGatekeeperEndpoint("0.0.0.0", 5000, deploy_sap=False) + self.sdkg1.connectDatacenter(self.dc) + # run the dummy gatekeeper (in another thread, don't block) + self.sdkg1.start() + + + self.net.start() + LOG.info("Started topology") + while(not self.stop_now): + pass + self.net.stop() + LOG.info("Stopped topology") + + def stop_it(self): + self.stop_now = True + + +def main(args): + setLogLevel('info') # set Mininet loglevel + p = Profiling() + + +if __name__ == '__main__': + parser = argparse.ArgumentParser(description="Run a simple topology") + parser.add_argument('--time', '-t', metavar='seconds', type=float, help='a time limit', default=-1, required=False, dest='time') + arg_list = vars(parser.parse_args(sys.argv[1:])) + main(arg_list) -- 2.25.1