128 lines
4.0 KiB
Python
128 lines
4.0 KiB
Python
# -*- encoding: utf-8 -*-
|
|
#
|
|
# Copyright © 2012 New Dream Network, LLC (DreamHost)
|
|
#
|
|
# Author: Doug Hellmann <doug.hellmann@dreamhost.com>
|
|
#
|
|
# 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.
|
|
"""Storage backend management
|
|
"""
|
|
|
|
|
|
import urlparse
|
|
|
|
from oslo.config import cfg
|
|
from stevedore import driver
|
|
|
|
from ceilometer.openstack.common import log
|
|
from ceilometer import utils
|
|
from ceilometer import service
|
|
|
|
|
|
LOG = log.getLogger(__name__)
|
|
|
|
STORAGE_ENGINE_NAMESPACE = 'ceilometer.storage'
|
|
|
|
STORAGE_OPTS = [
|
|
cfg.StrOpt('database_connection',
|
|
secret=True,
|
|
default=None,
|
|
help='DEPRECATED - Database connection string',
|
|
),
|
|
]
|
|
|
|
|
|
cfg.CONF.register_opts(STORAGE_OPTS)
|
|
cfg.CONF.import_opt('connection',
|
|
'ceilometer.openstack.common.db.sqlalchemy.session',
|
|
group='database')
|
|
|
|
|
|
def register_opts(conf):
|
|
"""Register any options for the storage system."""
|
|
p = get_engine(conf)
|
|
p.register_opts(conf)
|
|
|
|
|
|
def get_engine(conf):
|
|
"""Load the configured engine and return an instance."""
|
|
if conf.database_connection:
|
|
conf.set_override('connection', conf.database_connection,
|
|
group='database')
|
|
engine_name = urlparse.urlparse(conf.database.connection).scheme
|
|
LOG.debug('looking for %r driver in %r',
|
|
engine_name, STORAGE_ENGINE_NAMESPACE)
|
|
mgr = driver.DriverManager(STORAGE_ENGINE_NAMESPACE,
|
|
engine_name,
|
|
invoke_on_load=True)
|
|
return mgr.driver
|
|
|
|
|
|
def get_connection(conf):
|
|
"""Return an open connection to the database."""
|
|
engine = get_engine(conf)
|
|
engine.register_opts(conf)
|
|
db = engine.get_connection(conf)
|
|
return db
|
|
|
|
|
|
class SampleFilter(object):
|
|
"""Holds the properties for building a query from a meter/sample filter.
|
|
|
|
:param user: The sample owner.
|
|
:param project: The sample project.
|
|
:param start: Earliest timestamp to include.
|
|
:param end: Only include samples with timestamp less than this.
|
|
:param resource: Optional filter for resource id.
|
|
:param meter: Optional filter for meter type using the meter name.
|
|
:param source: Optional source filter.
|
|
:param metaquery: Optional filter on the metadata
|
|
"""
|
|
def __init__(self, user=None, project=None, start=None, end=None,
|
|
resource=None, meter=None, source=None, metaquery={}):
|
|
self.user = user
|
|
self.project = project
|
|
self.start = utils.sanitize_timestamp(start)
|
|
self.end = utils.sanitize_timestamp(end)
|
|
self.resource = resource
|
|
self.meter = meter
|
|
self.source = source
|
|
self.metaquery = metaquery
|
|
|
|
|
|
class EventFilter(object):
|
|
"""Properties for building an Event query.
|
|
|
|
:param start: UTC start datetime (mandatory)
|
|
:param end: UTC end datetime (mandatory)
|
|
:param event_name: the name of the event. None for all.
|
|
:param traits: the trait filter dict, all of which are optional
|
|
{'key': <key>,
|
|
't_string': <value>,
|
|
't_int': <value>,
|
|
't_datetime': <value>
|
|
't_float': <value>}
|
|
currently, only one trait dict is supported.
|
|
"""
|
|
|
|
def __init__(self, start, end, event_name=None, traits={}):
|
|
self.start = utils.sanitize_timestamp(start)
|
|
self.end = utils.sanitize_timestamp(end)
|
|
self.event_name = event_name
|
|
self.traits = traits
|
|
|
|
|
|
def dbsync():
|
|
service.prepare_service()
|
|
get_connection(cfg.CONF).upgrade()
|