7b45f7536b
Synced from commit 5fb343faee1442921e2d610b1a5222e67418c4cd. Due to changes in oslo.db API, this sync requires a bit more work on Neutron side. Sync includes the following commits: 5b7e61c Dispose db connections pool on disconnect d1988b9 Set sql_mode callback on connect instead of checkout a1a8280 Fix excessive logging from db.sqlalchemy.session 9933bdd Get mysql_sql_mode parameter from config 96a2217 Prevent incorrect usage of _wrap_db_error() 20a7510 Add from_config() method to EngineFacade fea119e Drop special case for MySQL traditional mode, update unit tests dda24eb Introduce mysql_sql_mode option, remove old warning 0b5af67 Introduce a method to set any MySQL session SQL mode 8dccc7b Handle ibm_db_sa DBDuplicateEntry integrity errors 5b9e9f4 Fix doc build errors in db.sqlalchemy ac84a40 Update log translation domains 86707cd Remove None for dict.get() 0545121 Fix duplicating of SQL queries in logs fcf517d Update oslo log messages with translation domains 630d395 Don't use cfg.CONF in oslo.db ce69e7f Don't store engine instances in oslo.db Change-Id: I0e1d86878d3eb924b01e04dced0f90b4e57757d8
90 lines
2.4 KiB
Python
90 lines
2.4 KiB
Python
# Copyright 2011 VMware, Inc.
|
|
# 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.
|
|
|
|
from oslo.config import cfg
|
|
import sqlalchemy as sql
|
|
|
|
from neutron.db import model_base
|
|
from neutron.openstack.common.db.sqlalchemy import session
|
|
from neutron.openstack.common import log as logging
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
BASE = model_base.BASEV2
|
|
|
|
cfg.CONF.import_opt('connection',
|
|
'neutron.openstack.common.db.options',
|
|
group='database')
|
|
|
|
_FACADE = None
|
|
|
|
|
|
def _create_facade_lazily():
|
|
global _FACADE
|
|
|
|
if _FACADE is None:
|
|
_FACADE = session.EngineFacade.from_config(
|
|
cfg.CONF.database.connection, cfg.CONF, sqlite_fk=True)
|
|
|
|
return _FACADE
|
|
|
|
|
|
def configure_db():
|
|
"""Configure database.
|
|
|
|
Establish the database, create an engine if needed, and register
|
|
the models.
|
|
"""
|
|
register_models()
|
|
|
|
|
|
def clear_db(base=BASE):
|
|
unregister_models(base)
|
|
|
|
|
|
def get_engine():
|
|
"""Helper method to grab engine."""
|
|
facade = _create_facade_lazily()
|
|
return facade.get_engine()
|
|
|
|
|
|
def get_session(autocommit=True, expire_on_commit=False):
|
|
"""Helper method to grab session."""
|
|
facade = _create_facade_lazily()
|
|
return facade.get_session(autocommit=autocommit,
|
|
expire_on_commit=expire_on_commit)
|
|
|
|
|
|
def register_models(base=BASE):
|
|
"""Register Models and create properties."""
|
|
try:
|
|
facade = _create_facade_lazily()
|
|
engine = facade.get_engine()
|
|
base.metadata.create_all(engine)
|
|
except sql.exc.OperationalError as e:
|
|
LOG.info(_("Database registration exception: %s"), e)
|
|
return False
|
|
return True
|
|
|
|
|
|
def unregister_models(base=BASE):
|
|
"""Unregister Models, useful clearing out data before testing."""
|
|
try:
|
|
facade = _create_facade_lazily()
|
|
engine = facade.get_engine()
|
|
base.metadata.drop_all(engine)
|
|
except Exception:
|
|
LOG.exception(_("Database exception"))
|