+
+ def _get_backend(self, vim_account_id: str):
+ keystone = self._build_keystone_client(vim_account_id)
+ if keystone.services.list('gnocchi'):
+ granularity = self._get_granularity(vim_account_id)
+ return GnocchiBackend(vim_account_id, granularity)
+ elif keystone.services.list('ceilometer'):
+ return CeilometerBackend(vim_account_id)
+
+ def _get_metric_type(self, metric_name: str, interface_name: str) -> MetricType:
+ if metric_name not in INTERFACE_METRICS:
+ return MetricType.INSTANCE
+ else:
+ if interface_name:
+ return MetricType.INTERFACE_ONE
+ return MetricType.INTERFACE_ALL
+
+
+class OpenstackBackend:
+ def collect_metric(self, metric_type: MetricType, metric_name: str, resource_id: str, interface_name: str):
+ pass
+
+ @staticmethod
+ def get_session(vim_account_id: str):
+ creds = CollectorUtils.get_credentials(vim_account_id)
+ verify_ssl = CollectorUtils.is_verify_ssl(creds)
+ auth = v3.Password(auth_url=creds.url,
+ username=creds.user,
+ password=creds.password,
+ project_name=creds.tenant_name,
+ project_domain_id='default',
+ user_domain_id='default')
+ return session.Session(auth=auth, verify=verify_ssl)
+
+
+class GnocchiBackend(OpenstackBackend):
+
+ def __init__(self, vim_account_id: str, granularity: int):
+ self.client = self._build_gnocchi_client(vim_account_id)
+ self.neutron = self._build_neutron_client(vim_account_id)
+ self.granularity = granularity
+
+ def _build_gnocchi_client(self, vim_account_id: str) -> gnocchi_client.Client:
+ sess = OpenstackBackend.get_session(vim_account_id)
+ return gnocchi_client.Client(session=sess)
+
+ def _build_neutron_client(self, vim_account_id: str) -> neutron_client.Client:
+ sess = OpenstackBackend.get_session(vim_account_id)
+ return neutron_client.Client(session=sess)
+
+ def collect_metric(self, metric_type: MetricType, metric_name: str, resource_id: str, interface_name: str):
+ if metric_type == MetricType.INTERFACE_ONE:
+ return self._collect_interface_one_metric(metric_name, resource_id, interface_name)
+
+ if metric_type == MetricType.INTERFACE_ALL:
+ return self._collect_interface_all_metric(metric_name, resource_id)
+
+ elif metric_type == MetricType.INSTANCE:
+ return self._collect_instance_metric(metric_name, resource_id)
+
+ else:
+ raise Exception('Unknown metric type %s' % metric_type.value)
+
+ def _collect_interface_one_metric(self, metric_name, resource_id, interface_name):
+ delta = 10 * self.granularity
+ start_date = datetime.datetime.now() - datetime.timedelta(seconds=delta)
+ ports = self.neutron.list_ports(name=interface_name, device_id=resource_id)
+ if not ports or not ports['ports']:
+ raise Exception(
+ 'Port not found for interface %s on instance %s' % (interface_name, resource_id))
+ port = ports['ports'][0]
+ port_uuid = port['id'][:11]
+ tap_name = 'tap' + port_uuid
+ interfaces = self.client.resource.search(resource_type='instance_network_interface',
+ query={'=': {'name': tap_name}})
+ measures = self.client.metric.get_measures(metric_name,
+ start=start_date,
+ resource_id=interfaces[0]['id'],
+ granularity=self.granularity)
+ return measures[-1][2] if measures else None
+
+ def _collect_interface_all_metric(self, openstack_metric_name, resource_id):
+ delta = 10 * self.granularity
+ start_date = datetime.datetime.now() - datetime.timedelta(seconds=delta)
+ total_measure = None
+ interfaces = self.client.resource.search(resource_type='instance_network_interface',
+ query={'=': {'instance_id': resource_id}})
+ for interface in interfaces:
+ try:
+ measures = self.client.metric.get_measures(openstack_metric_name,
+ start=start_date,
+ resource_id=interface['id'],
+ granularity=self.granularity)
+ if measures:
+ if not total_measure:
+ total_measure = 0.0
+ total_measure += measures[-1][2]
+
+ except gnocchiclient.exceptions.NotFound as e:
+ log.debug("No metric %s found for interface %s: %s", openstack_metric_name,
+ interface['id'], e)
+ return total_measure
+
+ def _collect_instance_metric(self, openstack_metric_name, resource_id):
+ delta = 10 * self.granularity
+ start_date = datetime.datetime.now() - datetime.timedelta(seconds=delta)
+ value = None
+ try:
+ measures = self.client.metric.get_measures(openstack_metric_name,
+ start=start_date,
+ resource_id=resource_id,
+ granularity=self.granularity)
+ if measures:
+ value = measures[-1][2]
+ except gnocchiclient.exceptions.NotFound as e:
+ log.debug("No metric %s found for instance %s: %s", openstack_metric_name, resource_id,
+ e)
+ return value
+
+
+class CeilometerBackend(OpenstackBackend):
+ def __init__(self, vim_account_id: str):
+ self.client = self._build_ceilometer_client(vim_account_id)
+
+ def _build_ceilometer_client(self, vim_account_id: str) -> ceilometer_client.Client:
+ sess = OpenstackBackend.get_session(vim_account_id)
+ return ceilometer_client.Client(session=sess)
+
+ def collect_metric(self, metric_type: MetricType, metric_name: str, resource_id: str, interface_name: str):
+ if metric_type != MetricType.INSTANCE:
+ raise NotImplementedError('Ceilometer backend only support instance metrics')
+ measures = self.client.samples.list(meter_name=metric_name, limit=1, q=[
+ {'field': 'resource_id', 'op': 'eq', 'value': resource_id}])
+ return measures[0].counter_volume if measures else None