3 from typing
import List
5 from osm_mon
.collector
.infra_collectors
.onos
import OnosInfraCollector
6 from osm_mon
.collector
.infra_collectors
.openstack
import OpenstackInfraCollector
7 from osm_mon
.collector
.infra_collectors
.vmware
import VMwareInfraCollector
8 from osm_mon
.collector
.infra_collectors
.vio
import VIOInfraCollector
9 from osm_mon
.collector
.metric
import Metric
10 from osm_mon
.collector
.utils
import CollectorUtils
11 from osm_mon
.collector
.vnf_collectors
.juju
import VCACollector
12 from osm_mon
.collector
.vnf_collectors
.openstack
import OpenstackCollector
13 from osm_mon
.collector
.vnf_collectors
.vio
import VIOCollector
14 from osm_mon
.collector
.vnf_collectors
.vmware
import VMwareCollector
15 from osm_mon
.core
.common_db
import CommonDbClient
16 from osm_mon
.core
.config
import Config
18 log
= logging
.getLogger(__name__
)
21 "openstack": OpenstackCollector
,
22 "vmware": VMwareCollector
,
25 VIM_INFRA_COLLECTORS
= {
26 "openstack": OpenstackInfraCollector
,
27 "vmware": VMwareInfraCollector
,
28 "vio": VIOInfraCollector
30 SDN_INFRA_COLLECTORS
= {
31 "onos": OnosInfraCollector
35 class CollectorService
:
36 def __init__(self
, config
: Config
):
38 self
.common_db
= CommonDbClient(self
.conf
)
39 self
.queue
= multiprocessing
.Queue()
41 def _collect_vim_metrics(self
, vnfr
: dict, vim_account_id
: str):
42 # TODO(diazb) Add support for aws
43 vim_type
= CollectorUtils
.get_vim_type(vim_account_id
)
44 if vim_type
in VIM_COLLECTORS
:
45 collector
= VIM_COLLECTORS
[vim_type
](self
.conf
, vim_account_id
)
46 metrics
= collector
.collect(vnfr
)
47 for metric
in metrics
:
48 self
.queue
.put(metric
)
50 log
.debug("vimtype %s is not supported.", vim_type
)
52 def _collect_vim_infra_metrics(self
, vim_account_id
: str):
53 vim_type
= CollectorUtils
.get_vim_type(vim_account_id
)
54 if vim_type
in VIM_INFRA_COLLECTORS
:
55 collector
= VIM_INFRA_COLLECTORS
[vim_type
](self
.conf
, vim_account_id
)
56 metrics
= collector
.collect()
57 for metric
in metrics
:
58 self
.queue
.put(metric
)
60 log
.debug("vimtype %s is not supported.", vim_type
)
62 def _collect_sdnc_infra_metrics(self
, sdnc_id
: str):
63 common_db
= CommonDbClient(self
.conf
)
64 sdn_type
= common_db
.get_sdnc(sdnc_id
)['type']
65 if sdn_type
in SDN_INFRA_COLLECTORS
:
66 collector
= SDN_INFRA_COLLECTORS
[sdn_type
](self
.conf
, sdnc_id
)
67 metrics
= collector
.collect()
68 for metric
in metrics
:
69 self
.queue
.put(metric
)
71 log
.debug("sdn_type %s is not supported.", sdn_type
)
73 def _collect_vca_metrics(self
, vnfr
: dict):
74 log
.debug('_collect_vca_metrics')
75 log
.debug('vnfr: %s', vnfr
)
76 vca_collector
= VCACollector(self
.conf
)
77 metrics
= vca_collector
.collect(vnfr
)
78 for metric
in metrics
:
79 self
.queue
.put(metric
)
81 def collect_metrics(self
) -> List
[Metric
]:
82 vnfrs
= self
.common_db
.get_vnfrs()
85 nsr_id
= vnfr
['nsr-id-ref']
86 vnf_member_index
= vnfr
['member-vnf-index-ref']
87 vim_account_id
= self
.common_db
.get_vim_account_id(nsr_id
, vnf_member_index
)
88 p
= multiprocessing
.Process(target
=self
._collect
_vim
_metrics
,
89 args
=(vnfr
, vim_account_id
))
92 p
= multiprocessing
.Process(target
=self
._collect
_vca
_metrics
,
96 vims
= self
.common_db
.get_vim_accounts()
98 p
= multiprocessing
.Process(target
=self
._collect
_vim
_infra
_metrics
,
102 sdncs
= self
.common_db
.get_sdncs()
104 p
= multiprocessing
.Process(target
=self
._collect
_sdnc
_infra
_metrics
,
108 for process
in processes
:
109 process
.join(timeout
=10)
111 while not self
.queue
.empty():
112 metrics
.append(self
.queue
.get())