122 lines
4.1 KiB
Python
122 lines
4.1 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright (c) 2012 OpenStack, LLC.
|
|
#
|
|
# 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.
|
|
|
|
"""Volume-related Utilities and helpers."""
|
|
|
|
from cinder import flags
|
|
from cinder.openstack.common import log as logging
|
|
from cinder.openstack.common.notifier import api as notifier_api
|
|
from cinder.openstack.common import timeutils
|
|
from cinder import utils
|
|
|
|
|
|
FLAGS = flags.FLAGS
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def get_host_from_queue(queuename):
|
|
# This assumes the queue is named something like cinder-volume
|
|
# and does not have dot separators in the queue name
|
|
return queuename.split('@', 1)[0].split('.', 1)[1]
|
|
|
|
|
|
def notify_usage_exists(context, volume_ref, current_period=False):
|
|
""" Generates 'exists' notification for a volume for usage auditing
|
|
purposes.
|
|
|
|
Generates usage for last completed period, unless 'current_period'
|
|
is True."""
|
|
begin, end = utils.last_completed_audit_period()
|
|
if current_period:
|
|
audit_start = end
|
|
audit_end = timeutils.utcnow()
|
|
else:
|
|
audit_start = begin
|
|
audit_end = end
|
|
|
|
extra_usage_info = dict(audit_period_beginning=str(audit_start),
|
|
audit_period_ending=str(audit_end))
|
|
|
|
notify_about_volume_usage(context, volume_ref,
|
|
'exists', extra_usage_info=extra_usage_info)
|
|
|
|
|
|
def null_safe_str(s):
|
|
return str(s) if s else ''
|
|
|
|
|
|
def _usage_from_volume(context, volume_ref, **kw):
|
|
usage_info = dict(tenant_id=volume_ref['project_id'],
|
|
user_id=volume_ref['user_id'],
|
|
volume_id=volume_ref['id'],
|
|
volume_type=volume_ref['volume_type_id'],
|
|
display_name=volume_ref['display_name'],
|
|
launched_at=null_safe_str(volume_ref['launched_at']),
|
|
created_at=null_safe_str(volume_ref['created_at']),
|
|
status=volume_ref['status'],
|
|
snapshot_id=volume_ref['snapshot_id'],
|
|
size=volume_ref['size'])
|
|
|
|
usage_info.update(kw)
|
|
return usage_info
|
|
|
|
|
|
def notify_about_volume_usage(context, volume, event_suffix,
|
|
extra_usage_info=None, host=None):
|
|
if not host:
|
|
host = FLAGS.host
|
|
|
|
if not extra_usage_info:
|
|
extra_usage_info = {}
|
|
|
|
usage_info = _usage_from_volume(context, volume, **extra_usage_info)
|
|
|
|
notifier_api.notify(context, 'volume.%s' % host,
|
|
'volume.%s' % event_suffix,
|
|
notifier_api.INFO, usage_info)
|
|
|
|
|
|
def _usage_from_snapshot(context, snapshot_ref, **extra_usage_info):
|
|
usage_info = {
|
|
'tenant_id': snapshot_ref['project_id'],
|
|
'user_id': snapshot_ref['user_id'],
|
|
'volume_id': snapshot_ref['volume_id'],
|
|
'volume_size': snapshot_ref['volume_size'],
|
|
'snapshot_id': snapshot_ref['id'],
|
|
'display_name': snapshot_ref['display_name'],
|
|
'created_at': str(snapshot_ref['created_at']),
|
|
'status': snapshot_ref['status'],
|
|
'deleted': null_safe_str(snapshot_ref['deleted'])
|
|
}
|
|
|
|
usage_info.update(extra_usage_info)
|
|
return usage_info
|
|
|
|
|
|
def notify_about_snapshot_usage(context, snapshot, event_suffix,
|
|
extra_usage_info=None, host=None):
|
|
if not host:
|
|
host = FLAGS.host
|
|
|
|
if not extra_usage_info:
|
|
extra_usage_info = {}
|
|
|
|
usage_info = _usage_from_snapshot(context, snapshot, **extra_usage_info)
|
|
|
|
notifier_api.notify(context, 'snapshot.%s' % host,
|
|
'snapshot.%s' % event_suffix,
|
|
notifier_api.INFO, usage_info)
|