osm_mon/collector/vnf_collectors/juju.py

   1 # Copyright 2018 Whitestack, LLC
   2 # *************************************************************
   3
   4 # This file is part of OSM Monitoring module
   5 # All Rights Reserved to Whitestack, LLC
   6
   7 # Licensed under the Apache License, Version 2.0 (the "License"); you may
   8 # not use this file except in compliance with the License. You may obtain
   9 # a copy of the License at
  10
  11 #         http://www.apache.org/licenses/LICENSE-2.0
  12
  13 # Unless required by applicable law or agreed to in writing, software
  14 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  15 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  16 # License for the specific language governing permissions and limitations
  17 # under the License.
  18
  19 # For those usages not covered by the Apache License, Version 2.0 please
  20 # contact: bdiaz@whitestack.com or glavado@whitestack.com
  21 ##
  22 import asyncio
  23 import logging
  24 from typing import List
  25
  26 from n2vc.vnf import N2VC
  27
  28 from osm_mon.collector.metric import Metric
  29 from osm_mon.collector.vnf_collectors.base import BaseCollector
  30 from osm_mon.collector.vnf_metric import VnfMetric
  31 from osm_mon.core.common_db import CommonDbClient
  32 from osm_mon.core.config import Config
  33 from osm_mon.core.exceptions import VcaDeploymentInfoNotFound
  34
  35 log = logging.getLogger(__name__)
  36
  37
  38 class VCACollector(BaseCollector):
  39     def __init__(self, config: Config):
  40         super().__init__(config)
  41         self.common_db = CommonDbClient(config)
  42         self.loop = asyncio.get_event_loop()
  43         self.n2vc = N2VC(server=config.get('vca', 'host'),
  44                          user=config.get('vca', 'user'),
  45                          secret=config.get('vca', 'secret'),
  46                          ca_cert=config.get('vca', 'cacert'))
  47
  48     def collect(self, vnfr: dict) -> List[Metric]:
  49         nsr_id = vnfr['nsr-id-ref']
  50         vnf_member_index = vnfr['member-vnf-index-ref']
  51         vnfd = self.common_db.get_vnfd(vnfr['vnfd-id'])
  52
  53         # Populate extra tags for metrics
  54         tags = {}
  55         tags['ns_name'] = self.common_db.get_nsr(nsr_id)['name']
  56         if vnfr['_admin']['projects_read']:
  57             tags['project_id'] = vnfr['_admin']['projects_read'][0]
  58         else:
  59             tags['project_id'] = None
  60
  61         metrics = []
  62         for vdur in vnfr['vdur']:
  63             # This avoids errors when vdur records have not been completely filled
  64             if 'name' not in vdur:
  65                 continue
  66             vdu = next(
  67                 filter(lambda vdu: vdu['id'] == vdur['vdu-id-ref'], vnfd['vdu'])
  68             )
  69             if 'vdu-configuration' in vdu and 'metrics' in vdu['vdu-configuration']:
  70                 try:
  71                     vca_deployment_info = self.get_vca_deployment_info(nsr_id, vnf_member_index, vdur['vdu-id-ref'],
  72                                                                        vdur['count-index'])
  73                 except VcaDeploymentInfoNotFound as e:
  74                     log.warning(repr(e))
  75                     continue
  76                 measures = self.loop.run_until_complete(self.n2vc.GetMetrics(vca_deployment_info['model'],
  77                                                                              vca_deployment_info['application']))
  78                 log.debug('Measures: %s', measures)
  79                 for measure_list in measures.values():
  80                     for measure in measure_list:
  81                         log.debug("Measure: %s", measure)
  82                         metric = VnfMetric(nsr_id, vnf_member_index, vdur['name'], measure['key'],
  83                                            float(measure['value'], tags))
  84                         metrics.append(metric)
  85         if 'vnf-configuration' in vnfd and 'metrics' in vnfd['vnf-configuration']:
  86             try:
  87                 vca_deployment_info = self.get_vca_deployment_info(nsr_id, vnf_member_index)
  88             except VcaDeploymentInfoNotFound as e:
  89                 log.warning(repr(e))
  90                 return metrics
  91             measures = self.loop.run_until_complete(self.n2vc.GetMetrics(vca_deployment_info['model'],
  92                                                                          vca_deployment_info['application']))
  93             log.debug('Measures: %s', measures)
  94             for measure_list in measures.values():
  95                 for measure in measure_list:
  96                     log.debug("Measure: %s", measure)
  97                     metric = VnfMetric(nsr_id, vnf_member_index, '', measure['key'], float(measure['value'], tags))
  98                     metrics.append(metric)
  99         return metrics
 100
 101     def get_vca_deployment_info(self, nsr_id, vnf_member_index, vdu_id=None, vdu_count=0):
 102         nsr = self.common_db.get_nsr(nsr_id)
 103         for vca_deployment in nsr["_admin"]["deployed"]["VCA"]:
 104             if vca_deployment:
 105                 if vdu_id is None:
 106                     if vca_deployment['member-vnf-index'] == vnf_member_index and vca_deployment['vdu_id'] is None:
 107                         return vca_deployment
 108                 else:
 109                     if vca_deployment['member-vnf-index'] == vnf_member_index and \
 110                             vca_deployment['vdu_id'] == vdu_id and vca_deployment['vdu_count_index'] == vdu_count:
 111                         return vca_deployment
 112         raise VcaDeploymentInfoNotFound(
 113             "VCA deployment info for nsr_id {}, index {}, vdu_id {} and vdu_count_index {} not found.".format(
 114                 nsr_id,
 115                 vnf_member_index,
 116                 vdu_id,
 117                 vdu_count))