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
|
# Copyright (c) 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.
import abc
import six
from yardstick.common import exceptions
@six.add_metaclass(abc.ABCMeta)
class Payload(object):
"""Base Payload class to transfer data through the MQ service"""
REQUIRED_FIELDS = {'version'}
def __init__(self, **kwargs):
"""Init method
:param kwargs: (dictionary) attributes and values of the object
:returns: Payload object
"""
if not all(req_field in kwargs for req_field in self.REQUIRED_FIELDS):
_attrs = set(kwargs) - self.REQUIRED_FIELDS
missing_attributes = ', '.join(str(_attr) for _attr in _attrs)
raise exceptions.PayloadMissingAttributes(
missing_attributes=missing_attributes)
for name, value in kwargs.items():
setattr(self, name, value)
self._fields = set(kwargs.keys())
def obj_to_dict(self):
"""Returns a dictionary with the attributes of the object"""
return {field: getattr(self, field) for field in self._fields}
@classmethod
def dict_to_obj(cls, _dict):
"""Returns a Payload object built from the dictionary elements"""
return cls(**_dict)
class TrafficGeneratorPayload(Payload):
"""Base traffic generator payload class"""
REQUIRED_FIELDS = {
'version', # (str) version of the payload transmitted.
'iteration', # (int) iteration index during the traffic injection,
# starting from 1.
'kpi' # (dict) collection of KPIs collected from the traffic
# injection. The content will depend on the generator and the
# traffic type.
}
class RunnerPayload(Payload):
"""Base runner payload class"""
REQUIRED_FIELDS = {
'version', # (str) version of the payload transmitted.
'data' # (dict) generic container of data to be used if needed.
}
|