policyhandler/deploy_handler.py

""" send notification to deploy-handler"""

# org.onap.dcae
# ================================================================================
# Copyright (c) 2017 AT&T Intellectual Property. All rights reserved.
# ================================================================================
# 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.
# ============LICENSE_END=========================================================
#
# ECOMP is a trademark and service mark of AT&T Intellectual Property.

import logging
import json
import requests

from .config import Config
from .discovery import DiscoveryClient
from .onap.audit import REQUEST_X_ECOMP_REQUESTID, Audit, AuditHttpCode

POOL_SIZE = 1

class DeployHandler(object):
    """ deploy-handler """
    _logger = logging.getLogger("policy_handler.deploy_handler")
    _lazy_inited = False

    _requests_session = None
    _config = None
    _url = None
    _url_path = None
    _target_entity = None

    @staticmethod
    def _lazy_init():
        """ set static properties """
        if DeployHandler._lazy_inited:
            return
        DeployHandler._lazy_inited = True

        DeployHandler._requests_session = requests.Session()
        DeployHandler._requests_session.mount(
            'https://',
            requests.adapters.HTTPAdapter(pool_connections=POOL_SIZE, pool_maxsize=POOL_SIZE)
        )
        DeployHandler._requests_session.mount(
            'http://',
            requests.adapters.HTTPAdapter(pool_connections=POOL_SIZE, pool_maxsize=POOL_SIZE)
        )

        DeployHandler._target_entity = Config.config["deploy_handler"]
        DeployHandler._url = DiscoveryClient.get_service_url(DeployHandler._target_entity)
        DeployHandler._url_path = (DeployHandler._url or "") + '/policy'
        DeployHandler._logger.info("DeployHandler url(%s)", DeployHandler._url)

    @staticmethod
    def policy_update(audit, latest_policies, removed_policies=None,
                      errored_policies=None, catch_up=False):
        """post policy_updated message to deploy-handler"""
        if not latest_policies and not removed_policies and not catch_up:
            return

        latest_policies = latest_policies or {}
        removed_policies = removed_policies or {}
        errored_policies = errored_policies or {}

        DeployHandler._lazy_init()
        msg = {
            "catch_up" : catch_up,
            "latest_policies" : latest_policies,
            "removed_policies" : removed_policies,
            "errored_policies" : errored_policies
        }
        sub_aud = Audit(aud_parent=audit, targetEntity=DeployHandler._target_entity,
                        targetServiceName=DeployHandler._url_path)
        headers = {REQUEST_X_ECOMP_REQUESTID : sub_aud.request_id}

        msg_str = json.dumps(msg)
        headers_str = json.dumps(headers)

        DeployHandler._logger.info(
            "catch_up(%s) latest_policies[%s], removed_policies[%s], errored_policies[%s]",
            catch_up, len(latest_policies), len(removed_policies), len(errored_policies))
        log_line = "post to deployment-handler {0} msg={1} headers={2}".format(
            DeployHandler._url_path, msg_str, headers_str)

        DeployHandler._logger.info(log_line)
        sub_aud.metrics_start(log_line)

        if not DeployHandler._url:
            error_msg = "no url found to {0}".format(log_line)
            DeployHandler._logger.error(error_msg)
            sub_aud.set_http_status_code(AuditHttpCode.SERVICE_UNAVAILABLE_ERROR.value)
            audit.set_http_status_code(AuditHttpCode.SERVICE_UNAVAILABLE_ERROR.value)
            sub_aud.metrics(error_msg)
            return

        res = None
        try:
            res = DeployHandler._requests_session.post(
                DeployHandler._url_path, json=msg, headers=headers
            )
        except requests.exceptions.RequestException as ex:
            error_msg = "failed to post to deployment-handler {0} {1} msg={2} headers={3}" \
                .format(DeployHandler._url_path, str(ex), msg_str, headers_str)
            DeployHandler._logger.exception(error_msg)
            sub_aud.set_http_status_code(AuditHttpCode.SERVICE_UNAVAILABLE_ERROR.value)
            audit.set_http_status_code(AuditHttpCode.SERVICE_UNAVAILABLE_ERROR.value)
            sub_aud.metrics(error_msg)
            return

        sub_aud.set_http_status_code(res.status_code)
        audit.set_http_status_code(res.status_code)

        sub_aud.metrics(
            "response from deployment-handler to post {0}: {1} msg={2} text={3} headers={4}" \
            .format(DeployHandler._url_path, res.status_code, msg_str, res.text,
                    res.request.headers))

        if res.status_code == requests.codes.ok:
            return res.json()