aboutsummaryrefslogtreecommitdiffstats
path: root/yardstick/network_services/traffic_profile/prox_profile.py
blob: be450c9f7e82c952a7f7c6ece9e44d741c3f1e20 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
# Copyright (c) 2016-2018 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.
""" Fixed traffic profile definitions """

from __future__ import absolute_import

import logging
import multiprocessing
import time

from yardstick.network_services.traffic_profile.base import TrafficProfile
from yardstick.network_services.vnf_generic.vnf.prox_helpers import ProxProfileHelper

LOG = logging.getLogger(__name__)


class ProxProfile(TrafficProfile):
    """
    This profile adds a single stream at the beginning of the traffic session
    """

    @staticmethod
    def sort_vpci(traffic_gen):
        """Return the list of external interfaces ordered by vpci and name

        :param traffic_gen: (ProxTrafficGen) traffic generator
        :return: list of ordered interfaces
        """
        def key_func(interface):
            return interface['virtual-interface']['vpci'], interface['name']

        return sorted(traffic_gen.vnfd_helper['vdu'][0]['external-interface'],
                      key=key_func)

    @staticmethod
    def fill_samples(samples, traffic_gen):
        vpci_if_name_ascending = ProxProfile.sort_vpci(traffic_gen)
        for vpci_idx, intf in enumerate(vpci_if_name_ascending):
            name = intf[1]
            # TODO: VNFDs KPIs values needs to be mapped to TRex structure
            xe_port = traffic_gen.resource_helper.sut.port_stats([vpci_idx])
            samples[name] = {
                "in_packets": xe_port[6],
                "out_packets": xe_port[7],
            }

    def __init__(self, tp_config):
        super(ProxProfile, self).__init__(tp_config)
        self.queue = None
        self.done = multiprocessing.Event()
        self.results = []

        # TODO: get init values from tp_config
        self.prox_config = tp_config["traffic_profile"]
        self.pkt_sizes = [int(x) for x in self.prox_config.get("packet_sizes", [])]
        self.pkt_size_iterator = iter(self.pkt_sizes)
        self.duration = int(self.prox_config.get("duration", 5))
        self.precision = float(self.prox_config.get('test_precision', 1.0))
        self.tolerated_loss = float(self.prox_config.get('tolerated_loss', 0.0))

        # TODO: is this ever a function of packet size?
        self.lower_bound = float(self.prox_config.get('lower_bound', 10.0))
        self.upper_bound = float(self.prox_config.get('upper_bound', 100.0))
        self.step_value = float(self.prox_config.get('step_value', 10.0))
        self._profile_helper = None

    def make_profile_helper(self, traffic_gen):
        if self._profile_helper is None:
            self._profile_helper = ProxProfileHelper.make_profile_helper(traffic_gen)
        return self._profile_helper

    def init(self, queue):
        self.pkt_size_iterator = iter(self.pkt_sizes)
        self.queue = queue
        self.queue.cancel_join_thread()

    def bounds_iterator(self, logger=None):
        if logger:
            logger.debug("Interval [%s, %s), step: %d", self.lower_bound,
                         self.upper_bound, self.step_value)

        test_value = self.lower_bound
        while test_value <= self.upper_bound:
            if logger:
                logger.info("Testing with value %s", test_value)
            yield test_value
            test_value += self.step_value

    @property
    def min_pkt_size(self):
        """Return the minimum required packet size for the test.

        Defaults to 64. Individual test must override this method if they have
        other requirements.

        Returns:
            int. The minimum required packet size for the test.
        """
        return 64

    def run_test_with_pkt_size(self, traffic_generator, pkt_size, duration):
        raise NotImplementedError

    def execute_traffic(self, traffic_generator):
        self.make_profile_helper(traffic_generator)

        try:
            pkt_size = next(self.pkt_size_iterator)
        except StopIteration:
            time.sleep(5)
            self.done.set()
            return

        # Adjust packet size upwards if it's less than the minimum
        # required packet size for the test.
        if pkt_size < self.min_pkt_size:
            pkt_size += self.min_pkt_size - 64

        duration = self.duration
        self.run_test_with_pkt_size(traffic_generator, pkt_size, duration)