132 lines
3.6 KiB
Python
132 lines
3.6 KiB
Python
# Copyright 2013 Cloudscaling Group, Inc
|
|
#
|
|
# 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.
|
|
|
|
"""Defines interface for DB access.
|
|
|
|
Functions in this module are imported into the ec2api.db namespace. Call these
|
|
functions from ec2api.db namespace, not the ec2api.db.api namespace.
|
|
|
|
**Related Flags**
|
|
|
|
:dbackend: string to lookup in the list of LazyPluggable backends.
|
|
`sqlalchemy` is the only supported backend right now.
|
|
|
|
:connection: string specifying the sqlalchemy connection to use, like:
|
|
`sqlite:///var/lib/ec2api/ec2api.sqlite`.
|
|
|
|
"""
|
|
|
|
from eventlet import tpool
|
|
from oslo_config import cfg
|
|
from oslo_db import api as db_api
|
|
from oslo_log import log as logging
|
|
|
|
|
|
tpool_opts = [
|
|
cfg.BoolOpt('use_tpool',
|
|
default=False,
|
|
deprecated_name='dbapi_use_tpool',
|
|
deprecated_group='DEFAULT',
|
|
help='Enable the experimental use of thread pooling for '
|
|
'all DB API calls'),
|
|
]
|
|
|
|
CONF = cfg.CONF
|
|
CONF.register_opts(tpool_opts, 'database')
|
|
|
|
_BACKEND_MAPPING = {'sqlalchemy': 'ec2api.db.sqlalchemy.api'}
|
|
|
|
|
|
class EC2DBAPI(object):
|
|
"""ec2's DB API wrapper class.
|
|
|
|
This wraps the oslo DB API with an option to be able to use eventlet's
|
|
thread pooling. Since the CONF variable may not be loaded at the time
|
|
this class is instantiated, we must look at it on the first DB API call.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.__db_api = None
|
|
|
|
@property
|
|
def _db_api(self):
|
|
if not self.__db_api:
|
|
ec2_db_api = db_api.DBAPI(CONF.database.backend,
|
|
backend_mapping=_BACKEND_MAPPING)
|
|
if CONF.database.use_tpool:
|
|
self.__db_api = tpool.Proxy(ec2_db_api)
|
|
else:
|
|
self.__db_api = ec2_db_api
|
|
return self.__db_api
|
|
|
|
def __getattr__(self, key):
|
|
return getattr(self._db_api, key)
|
|
|
|
|
|
IMPL = EC2DBAPI()
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def add_item(context, kind, data):
|
|
return IMPL.add_item(context, kind, data)
|
|
|
|
|
|
def add_item_id(context, kind, os_id):
|
|
return IMPL.add_item_id(context, kind, os_id)
|
|
|
|
|
|
def update_item(context, item):
|
|
IMPL.update_item(context, item)
|
|
|
|
|
|
def delete_item(context, item_id):
|
|
IMPL.delete_item(context, item_id)
|
|
|
|
|
|
def restore_item(context, kind, data):
|
|
return IMPL.restore_item(context, kind, data)
|
|
|
|
|
|
def get_items(context, kind):
|
|
return IMPL.get_items(context, kind)
|
|
|
|
|
|
def get_item_by_id(context, item_id):
|
|
return IMPL.get_item_by_id(context, item_id)
|
|
|
|
|
|
def get_items_by_ids(context, item_ids):
|
|
return IMPL.get_items_by_ids(context, item_ids)
|
|
|
|
|
|
def get_public_items(context, kind, item_ids=None):
|
|
return IMPL.get_public_items(context, kind, item_ids)
|
|
|
|
|
|
def get_items_ids(context, kind, os_ids):
|
|
return IMPL.get_items_ids(context, kind, os_ids)
|
|
|
|
|
|
def add_tags(context, tags):
|
|
return IMPL.add_tags(context, tags)
|
|
|
|
|
|
def delete_tags(context, item_ids, tag_pairs=None):
|
|
return IMPL.delete_tags(context, item_ids, tag_pairs)
|
|
|
|
|
|
def get_tags(context, kinds=None, item_ids=None):
|
|
return IMPL.get_tags(context, kinds, item_ids)
|