osm_mon/collector/infra_collectors/base_osinfra.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 logging
  23 from typing import List
  24
  25 from keystoneclient.v3 import client as keystone_client
  26 from novaclient import client as nova_client
  27
  28 from osm_mon.collector.infra_collectors.base_vim import BaseVimInfraCollector
  29 from osm_mon.collector.metric import Metric
  30 from osm_mon.collector.utils.openstack import OpenstackUtils
  31 from osm_mon.core.common_db import CommonDbClient
  32 from osm_mon.core.config import Config
  33
  34 log = logging.getLogger(__name__)
  35
  36
  37 class BaseOpenStackInfraCollector(BaseVimInfraCollector):
  38     def __init__(self, config: Config, vim_account_id: str):
  39         super().__init__(config, vim_account_id)
  40         self.conf = config
  41         self.common_db = CommonDbClient(config)
  42         self.vim_account = self.common_db.get_vim_account(vim_account_id)
  43         self.keystone = self._build_keystone_client(self.vim_account)
  44         self.nova = self._build_nova_client(self.vim_account)
  45
  46     def collect(self) -> List[Metric]:
  47         metrics = []
  48         vim_status = self.is_vim_ok()
  49         if self.vim_account['_admin']['projects_read']:
  50             vim_project_id = self.vim_account['_admin']['projects_read'][0]
  51         else:
  52             vim_project_id = ''
  53         vim_tags = {
  54             'vim_account_id': self.vim_account['_id'],
  55             'project_id': vim_project_id
  56         }
  57         vim_status_metric = Metric(vim_tags, 'vim_status', vim_status)
  58         metrics.append(vim_status_metric)
  59         vnfrs = self.common_db.get_vnfrs(vim_account_id=self.vim_account['_id'])
  60         for vnfr in vnfrs:
  61             nsr_id = vnfr['nsr-id-ref']
  62             ns_name = self.common_db.get_nsr(nsr_id)['name']
  63             vnf_member_index = vnfr['member-vnf-index-ref']
  64             if vnfr['_admin']['projects_read']:
  65                 vnfr_project_id = vnfr['_admin']['projects_read'][0]
  66             else:
  67                 vnfr_project_id = ''
  68             for vdur in vnfr['vdur']:
  69                 if 'vim-id' not in vdur:
  70                     log.debug("Field vim-id is not present in vdur")
  71                     continue
  72                 resource_uuid = vdur['vim-id']
  73                 tags = {
  74                     'vim_account_id': self.vim_account['_id'],
  75                     'resource_uuid': resource_uuid,
  76                     'nsr_id': nsr_id,
  77                     'ns_name': ns_name,
  78                     'vnf_member_index': vnf_member_index,
  79                     'vdur_name': vdur.get("name", ""),
  80                     'project_id': vnfr_project_id
  81                 }
  82                 try:
  83                     vm = self.nova.servers.get(resource_uuid)
  84                     vm_status = (1 if vm.status == 'ACTIVE' else 0)
  85                     vm_status_metric = Metric(tags, 'vm_status', vm_status)
  86                 except Exception as e:
  87                     log.warning("VM status is not OK: %s" % e)
  88                     vm_status_metric = Metric(tags, 'vm_status', 0)
  89                 metrics.append(vm_status_metric)
  90
  91         return metrics
  92
  93     def is_vim_ok(self) -> bool:
  94         try:
  95             self.nova.servers.list()
  96             return True
  97         except Exception as e:
  98             log.warning("VIM status is not OK: %s" % e)
  99             return False
 100
 101     def _build_keystone_client(self, vim_account: dict) -> keystone_client.Client:
 102         sess = OpenstackUtils.get_session(vim_account)
 103         return keystone_client.Client(session=sess, timeout=10)
 104
 105     def _build_nova_client(self, vim_account: dict) -> nova_client.Client:
 106         sess = OpenstackUtils.get_session(vim_account)
 107         return nova_client.Client("2", session=sess, timeout=10)