4 from kafka
import KafkaConsumer
5 from osm_policy_module
.core
.config
import Config
6 from osm_policy_module
.common
.lcm_client
import LcmClient
8 from osm_policy_module
.common
.alarm_config
import AlarmConfig
9 from osm_policy_module
.common
.mon_client
import MonClient
10 from osm_policy_module
.core
.database
import ScalingRecord
, ScalingAlarm
12 log
= logging
.getLogger(__name__
)
15 class PolicyModuleAgent
:
17 cfg
= Config
.instance()
19 kafka_server
= '{}:{}'.format(cfg
.get('policy_module', 'kafka_server_host'),
20 cfg
.get('policy_module', 'kafka_server_port'))
22 # Initialize Kafka consumer
23 log
.info("Connecting to Kafka server at %s", kafka_server
)
24 # TODO: Add logic to handle deduplication of messages when using group_id.
25 # See: https://stackoverflow.com/a/29836412
26 consumer
= KafkaConsumer(bootstrap_servers
=kafka_server
,
27 key_deserializer
=bytes
.decode
,
28 value_deserializer
=bytes
.decode
)
29 consumer
.subscribe(['lcm_pm', 'alarm_response'])
31 for message
in consumer
:
32 log
.info("Message arrived: %s", message
)
34 if message
.key
== 'configure_scaling':
35 content
= json
.loads(message
.value
)
36 log
.info("Creating scaling record in DB")
37 # TODO: Use transactions: http://docs.peewee-orm.com/en/latest/peewee/transactions.html
38 scaling_record
= ScalingRecord
.create(
39 nsr_id
=content
['ns_id'],
40 name
=content
['scaling_group_descriptor']['name'],
41 content
=json
.dumps(content
)
43 log
.info("Created scaling record in DB : nsr_id=%s, name=%s, content=%s",
44 scaling_record
.nsr_id
,
46 scaling_record
.content
)
47 alarm_configs
= self
._get
_alarm
_configs
(content
)
48 for config
in alarm_configs
:
49 mon_client
= MonClient()
50 log
.info("Creating alarm record in DB")
51 alarm_uuid
= mon_client
.create_alarm(
52 metric_name
=config
.metric_name
,
53 resource_uuid
=config
.resource_uuid
,
54 vim_uuid
=config
.vim_uuid
,
55 threshold
=config
.threshold
,
56 operation
=config
.operation
,
57 statistic
=config
.statistic
62 scaling_record
=scaling_record
64 if message
.key
== 'notify_alarm':
65 content
= json
.loads(message
.value
)
66 alarm_id
= content
['notify_details']['alarm_uuid']
67 alarm
= ScalingAlarm
.select().where(ScalingAlarm
.alarm_id
== alarm_id
).get()
69 lcm_client
= LcmClient()
70 log
.info("Sending scaling action message for ns: %s", alarm_id
)
71 lcm_client
.scale(alarm
.scaling_record
.nsr_id
, alarm
.scaling_record
.name
, alarm
.action
)
73 log
.exception("Error consuming message: ")
75 def _get_alarm_configs(self
, message_content
):
76 scaling_criterias
= message_content
['scaling_group_descriptor']['scaling_policy']['scaling_criteria']
78 for criteria
in scaling_criterias
:
80 scale_out_threshold
= criteria
['scale_out_threshold']
81 scale_in_threshold
= criteria
['scale_in_threshold']
82 scale_out_operation
= criteria
['scale_out_relational_operation']
83 scale_in_operation
= criteria
['scale_in_relational_operation']
84 statistic
= criteria
['monitoring_param']['aggregation_type']
87 if 'vdu_monitoring_param' in criteria
['monitoring_param']:
88 vim_uuid
= criteria
['monitoring_param']['vdu_monitoring_param']['vim_uuid']
89 resource_uuid
= criteria
['monitoring_param']['vdu_monitoring_param']['resource_id']
90 metric_name
= criteria
['monitoring_param']['vdu_monitoring_param']['name']
91 if 'vnf_metric' in criteria
['monitoring_param']:
94 if 'vdu_metric' in criteria
['monitoring_param']:
97 scale_out_alarm_config
= AlarmConfig(metric_name
,
104 scale_in_alarm_config
= AlarmConfig(metric_name
,
111 alarm_configs
.append(scale_in_alarm_config
)
112 alarm_configs
.append(scale_out_alarm_config
)