diff options
Diffstat (limited to 'yardstick/network_services/traffic_profile/base.py')
-rw-r--r-- | yardstick/network_services/traffic_profile/base.py | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/yardstick/network_services/traffic_profile/base.py b/yardstick/network_services/traffic_profile/base.py new file mode 100644 index 000000000..906498586 --- /dev/null +++ b/yardstick/network_services/traffic_profile/base.py @@ -0,0 +1,64 @@ +# Copyright (c) 2016-2017 Intel Corporation +# +# 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. +""" Base class for the generic traffic profile implementation """ + +from __future__ import absolute_import +from yardstick.common.utils import import_modules_from_package, itersubclasses + + +class TrafficProfile(object): + """ + This class defines the behavior + + """ + + @staticmethod + def get(tp_config): + """Get the traffic profile instance for the given traffic type + + :param tp_config: loaded YAML file + :return: + """ + profile_class = tp_config["traffic_profile"]["traffic_type"] + import_modules_from_package( + "yardstick.network_services.traffic_profile") + try: + return next(c for c in itersubclasses(TrafficProfile) + if c.__name__ == profile_class)(tp_config) + except StopIteration: + raise RuntimeError("No implementation for %s", profile_class) + + def __init__(self, tp_config): + # e.g. RFC2544 start_ip, stop_ip, drop_rate, + # IMIX = {"10K": 0.1, "100M": 0.5} + self.params = tp_config + + def execute(self, traffic_generator): + """ This methods defines the behavior of the traffic generator. + It will be called in a loop until the traffic generator exits. + + :param traffic_generator: TrafficGen instance + :return: None + """ + raise NotImplementedError() + + +class DummyProfile(TrafficProfile): + """ + This is an empty TrafficProfile implementation - if it is used, + the traffic will be completely handled by the Traffic Generator + implementation with no regard for the Traffic Profile. + """ + def execute(self, traffic_generator): + pass |