Add exception handling for api side with trace log and

[pti/o2.git] / o2ims / service / command / registration_handler.py

diff --git a/o2ims/service/command/registration_handler.py b/o2ims/service/command/registration_handler.py
index d08bf73..aead4cc 100644 (file)
--- a/o2ims/service/command/registration_handler.py
+++ b/o2ims/service/command/registration_handler.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Wind River Systems, Inc.
+# Copyright (C) 2021-2022 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.
@@ -16,15 +16,19 @@
 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 o2common.config import config, conf
+from o2common.service.command.handler import get_https_conn_default
+from o2common.service.command.handler import get_http_conn
+from o2common.service.command.handler import get_https_conn_selfsigned
+from o2common.service.command.handler import post_data
+import ssl
 from o2ims.domain import commands
-from o2ims.domain.configuration_obj import ConfigurationTypeEnum, \
-    RegistrationStatusEnum
+from o2ims.domain.subscription_obj import NotificationEventEnum
 
 from o2common.helper import o2logging
 logger = o2logging.get_logger(__name__)
@@ -36,39 +40,31 @@ def registry_to_smo(
 ):
     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)
+    logger.info('The Register2SMO notificationEventType is {}'.format(
+        data.notificationEventType))
+    with uow:
+        ocloud = uow.oclouds.get(data.id)
+        if ocloud is None:
+            return
+        logger.debug('O-Cloud Global UUID: {}'.format(ocloud.globalCloudId))
+        ocloud_dict = ocloud.serialize()
+        if data.notificationEventType == NotificationEventEnum.CREATE:
+            register_smo(uow, ocloud_dict)
+
+
+class RegIMSToSMOExp(Exception):
+    def __init__(self, value):
+        self.value = value
 
 
-def register_smo(uow, reg_data):
-    call_res = call_smo(reg_data)
+def register_smo(uow, ocloud_data):
+    call_res, status = call_smo(ocloud_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()
+    if call_res is True:
+        logger.info('Register to smo success response is {}'.format(status))
+    else:
+        raise RegIMSToSMOExp('Register o2ims to SMO failed')
+    # TODO: record the result for the smo register
 
 
 # def retry(fun, max_tries=2):
@@ -85,23 +81,44 @@ def register_smo(uow, reg_data):
 
 @retry((ConnectionRefusedError), tries=2, delay=2)
 def call_smo(reg_data: dict):
+    smo_token = conf.DEFAULT.smo_token_data
+    smo_token_info = {
+        'iss': 'o2ims',
+        'aud': 'smo',
+        'smo_token_payload': smo_token,
+        'smo_token_type': 'jwt',
+        'smo_token_expiration': '',
+        'smo_token_algo': 'RS256'
+    }
+
     callback_data = json.dumps({
-        'consumerSubscriptionId': reg_data['configurationId'],
-        'imsUrl': config.get_api_url()
+        'globalCloudId': reg_data['globalCloudId'],
+        'oCloudId': reg_data['oCloudId'],
+        'IMS_EP': config.get_api_url(),
+        'smo_token_data': smo_token_info
     })
     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
+        conf.DEFAULT.smo_register_url, callback_data))
+    o = urlparse(conf.DEFAULT.smo_register_url)
+    if o.scheme == 'https':
+        conn = get_https_conn_default(o.netloc)
+    else:
+        conn = get_http_conn(o.netloc)
+
+    try:
+        return post_data(conn, o.path, callback_data)
+    except ssl.SSLCertVerificationError as e:
+        logger.info('Register to smo with trusted ca failed: {}'.format(e))
+        if 'self signed' in str(e):
+            conn = get_https_conn_selfsigned(o.netloc)
+            try:
+                return post_data(conn, o.path, callback_data)
+            except Exception as e:
+                logger.info(
+                    'Register to smo with self-signed ca failed: {}'.format(e))
+                # TODO: write the status to extension db table.
+                return False, None
+        return False, None
+    except Exception as e:
+        logger.critical('Register to smo except: {}'.format(e))
+        return False, None