X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fservice%2Fcommand%2Fregistration_handler.py;h=d08bf733c1bb8cc10d91812ee2b8acac5a591325;hb=defe8209b3628593c186487857fe02586d7e1503;hp=0a4395dbb347c22af505efa39f5b84f6c83589d0;hpb=3da89330f3837ac6cffd2cad4c4018c9f8c3327d;p=pti%2Fo2.git diff --git a/o2ims/service/command/registration_handler.py b/o2ims/service/command/registration_handler.py index 0a4395d..d08bf73 100644 --- a/o2ims/service/command/registration_handler.py +++ b/o2ims/service/command/registration_handler.py @@ -1,103 +1,107 @@ -# Copyright (C) 2021 Wind River Systems, Inc. -# -# 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 time -import json -# import asyncio -# import requests -import http.client -from urllib.parse import urlparse -from retry import retry - -from o2common.service.unit_of_work import AbstractUnitOfWork -from o2common.config import config -from o2ims.domain import commands -from o2ims.domain.subscription_obj import RegistrationStatusEnum - -from o2common.helper import o2logging -logger = o2logging.get_logger(__name__) - - -def registry_to_smo( - cmd: commands.Register2SMO, - uow: AbstractUnitOfWork, -): - logger.info('In registry_to_smo') - data = cmd.data - logger.info('The Register2SMO all is {}'.format(data.all)) - if data.all: - regs = uow.registrations.list() - for reg in regs: - reg_data = reg.serialize() - logger.debug('Registration: {}'.format(reg_data['registrationId'])) - - register_smo(uow, reg_data) - else: - with uow: - reg = uow.registrations.get(data.id) - if reg is None: - return - logger.debug('Registration: {}'.format(reg.registrationId)) - reg_data = reg.serialize() - register_smo(uow, reg_data) - - -def register_smo(uow, reg_data): - call_res = call_smo(reg_data) - logger.debug('Call SMO response is {}'.format(call_res)) - if call_res: - reg = uow.registrations.get(reg_data['registrationId']) - if reg is None: - return - reg.status = RegistrationStatusEnum.NOTIFIED - logger.debug('Updating Registration: {}'.format( - reg.registrationId)) - uow.registrations.update(reg) - uow.commit() - - -# def retry(fun, max_tries=2): -# for i in range(max_tries): -# try: -# time.sleep(5*i) -# # await asyncio.sleep(5*i) -# res = fun() -# logger.debug('retry function result: {}'.format(res)) -# return res -# except Exception: -# continue - - -@retry((ConnectionRefusedError), tries=2, delay=2) -def call_smo(reg_data: dict): - callback_data = json.dumps({ - 'consumerSubscriptionId': reg_data['registrationId'], - 'imsUrl': config.get_api_url() - }) - logger.info('URL: {}, data: {}'.format( - reg_data['callback'], callback_data)) - - o = urlparse(reg_data['callback']) - conn = http.client.HTTPConnection(o.netloc) - headers = {'Content-type': 'application/json'} - conn.request('POST', o.path, callback_data, headers) - resp = conn.getresponse() - data = resp.read().decode('utf-8') - # json_data = json.loads(data) - if resp.status == 202 or resp.status == 200: - logger.info('Registrer to SMO successed, response code {} {}, data {}'. - format(resp.status, resp.reason, data)) - return True - logger.error('Response code is: {}'.format(resp.status)) - return False +# Copyright (C) 2021 Wind River Systems, Inc. +# +# 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 time +import json +# import asyncio +# import requests +import http.client +from urllib.parse import urlparse +from retry import retry + +from o2common.service.unit_of_work import AbstractUnitOfWork +from o2common.config import config +from o2ims.domain import commands +from o2ims.domain.configuration_obj import ConfigurationTypeEnum, \ + RegistrationStatusEnum + +from o2common.helper import o2logging +logger = o2logging.get_logger(__name__) + + +def registry_to_smo( + cmd: commands.Register2SMO, + uow: AbstractUnitOfWork, +): + logger.info('In registry_to_smo') + data = cmd.data + logger.info('The Register2SMO all is {}'.format(data.all)) + if data.all: + confs = uow.configrations.list() + for conf in confs: + if conf.conftype != ConfigurationTypeEnum.SMO: + continue + reg_data = conf.serialize() + logger.debug('Configuration: {}'.format( + reg_data['configurationId'])) + + register_smo(uow, reg_data) + else: + with uow: + conf = uow.configurations.get(data.id) + if conf is None: + return + logger.debug('Configuration: {}'.format(conf.configurationId)) + conf_data = conf.serialize() + register_smo(uow, conf_data) + + +def register_smo(uow, reg_data): + call_res = call_smo(reg_data) + logger.debug('Call SMO response is {}'.format(call_res)) + if call_res: + reg = uow.configurations.get(reg_data['configurationId']) + if reg is None: + return + reg.status = RegistrationStatusEnum.NOTIFIED + logger.debug('Updating Configurations: {}'.format( + reg.configurationId)) + uow.configurations.update(reg) + uow.commit() + + +# def retry(fun, max_tries=2): +# for i in range(max_tries): +# try: +# time.sleep(5*i) +# # await asyncio.sleep(5*i) +# res = fun() +# logger.debug('retry function result: {}'.format(res)) +# return res +# except Exception: +# continue + + +@retry((ConnectionRefusedError), tries=2, delay=2) +def call_smo(reg_data: dict): + callback_data = json.dumps({ + 'consumerSubscriptionId': reg_data['configurationId'], + 'imsUrl': config.get_api_url() + }) + logger.info('URL: {}, data: {}'.format( + reg_data['callback'], callback_data)) + + o = urlparse(reg_data['callback']) + conn = http.client.HTTPConnection(o.netloc) + headers = {'Content-type': 'application/json'} + conn.request('POST', o.path, callback_data, headers) + resp = conn.getresponse() + data = resp.read().decode('utf-8') + # json_data = json.loads(data) + if resp.status == 202 or resp.status == 200: + logger.info('Registrer to SMO successed, response code {} {}, data {}'. + format(resp.status, resp.reason, data)) + return True + logger.error('Response code is: {}'.format(resp.status)) + return False