6ad360741f
adding the volume notifications added unit tests for notifications added context for notify with rabbit blueprint cinder-notifications Change-Id: I5aef1e718eebb9a61e0670d524fcd5f438dee016
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
#!/usr/bin/env python
|
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright (c) 2011 Openstack, LLC.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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.
|
|
|
|
"""Cron script to generate usage notifications for volumes existing during
|
|
the audit period.
|
|
|
|
Together with the notifications generated by volumes
|
|
create/delete/resize, over that time period, this allows an external
|
|
system consuming usage notification feeds to calculate volume usage
|
|
for each tenant.
|
|
|
|
Time periods are specified as 'hour', 'month', 'day' or 'year'
|
|
|
|
hour = previous hour. If run at 9:07am, will generate usage for 8-9am.
|
|
month = previous month. If the script is run April 1, it will generate
|
|
usages for March 1 through March 31.
|
|
day = previous day. if run on July 4th, it generates usages for July 3rd.
|
|
year = previous year. If run on Jan 1, it generates usages for
|
|
Jan 1 through Dec 31 of the previous year.
|
|
"""
|
|
|
|
import gettext
|
|
import os
|
|
import sys
|
|
import traceback
|
|
|
|
# If ../cinder/__init__.py exists, add ../ to Python search path, so that
|
|
# it will override what happens to be installed in /usr/(local/)lib/python...
|
|
POSSIBLE_TOPDIR = os.path.normpath(os.path.join(os.path.abspath(sys.argv[0]),
|
|
os.pardir,
|
|
os.pardir))
|
|
if os.path.exists(os.path.join(POSSIBLE_TOPDIR, 'cinder', '__init__.py')):
|
|
sys.path.insert(0, POSSIBLE_TOPDIR)
|
|
|
|
gettext.install('cinder', unicode=1)
|
|
from cinder import context
|
|
from cinder import db
|
|
from cinder import flags
|
|
from cinder.openstack.common import log as logging
|
|
from cinder.openstack.common import rpc
|
|
from cinder import utils
|
|
import cinder.volume.utils
|
|
|
|
|
|
FLAGS = flags.FLAGS
|
|
|
|
|
|
def output(msg):
|
|
if not FLAGS.silent:
|
|
print msg
|
|
|
|
|
|
if __name__ == '__main__':
|
|
rpc.register_opts(FLAGS)
|
|
admin_context = context.get_admin_context()
|
|
utils.default_flagfile()
|
|
flags.FLAGS(sys.argv)
|
|
logging.setup()
|
|
begin, end = utils.last_completed_audit_period()
|
|
output("Starting volume usage audit")
|
|
output("Creating usages for %s until %s" % (str(begin), str(end)))
|
|
volumes = db.volume_get_active_by_window(admin_context,
|
|
begin,
|
|
end)
|
|
output("Found %d volumes" % len(volumes))
|
|
for volume_ref in volumes:
|
|
try:
|
|
cinder.volume.utils.notify_usage_exists(
|
|
admin_context, volume_ref)
|
|
except Exception, e:
|
|
output(traceback.format_exc(e))
|
|
output("Volume usage audit completed")
|