1 # Copyright (C) 2021 Wind River Systems, Inc.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
20 from urllib.parse import urlparse
21 from retry import retry
23 from o2common.service.unit_of_work import AbstractUnitOfWork
24 from o2common.config import config, conf
26 from o2ims.domain import commands
27 from o2ims.domain.subscription_obj import NotificationEventEnum
29 from o2common.helper import o2logging
30 logger = o2logging.get_logger(__name__)
34 cmd: commands.Register2SMO,
35 uow: AbstractUnitOfWork,
37 logger.info('In registry_to_smo')
39 logger.info('The Register2SMO notificationEventType is {}'.format(
40 data.notificationEventType))
42 ocloud = uow.oclouds.get(data.id)
45 logger.debug('O-Cloud Global UUID: {}'.format(ocloud.globalcloudId))
46 ocloud_dict = ocloud.serialize()
47 if data.notificationEventType == NotificationEventEnum.CREATE:
48 register_smo(uow, ocloud_dict)
51 def register_smo(uow, ocloud_data):
52 call_res = call_smo(ocloud_data)
53 logger.debug('Call SMO response is {}'.format(call_res))
54 # TODO: record the result for the smo register
57 # def retry(fun, max_tries=2):
58 # for i in range(max_tries):
61 # # await asyncio.sleep(5*i)
63 # logger.debug('retry function result: {}'.format(res))
69 @retry((ConnectionRefusedError), tries=2, delay=2)
70 def call_smo(reg_data: dict):
71 callback_data = json.dumps({
72 'consumerSubscriptionId': reg_data['globalcloudId'],
73 'notificationEventType': 'CREATE',
74 'objectRef': config.get_api_url(),
75 'postObjectState': reg_data
77 logger.info('URL: {}, data: {}'.format(
78 conf.DEFAULT.smo_register_url, callback_data))
80 o = urlparse(conf.DEFAULT.smo_register_url)
81 conn = http.client.HTTPConnection(o.netloc)
82 headers = {'Content-type': 'application/json'}
83 conn.request('POST', o.path, callback_data, headers)
84 resp = conn.getresponse()
85 data = resp.read().decode('utf-8')
86 # json_data = json.loads(data)
87 if resp.status == 202 or resp.status == 200:
88 logger.info('Registrer to SMO successed, response code {} {}, data {}'.
89 format(resp.status, resp.reason, data))
91 logger.error('Response code is: {}'.format(resp.status))