1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
# Copyright 2011 OpenStack Foundation
# 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.
import contextlib
from sqlalchemy import create_engine
from sqlalchemy import MetaData
from sqlalchemy.orm import sessionmaker
from trove.common import cfg
from trove.openstack.common import log as logging
from trove.openstack.common.gettextutils import _
from trove.db.sqlalchemy import mappers
_ENGINE = None
_MAKER = None
LOG = logging.getLogger(__name__)
CONF = cfg.CONF
def configure_db(options, models_mapper=None):
global _ENGINE
if not _ENGINE:
_ENGINE = _create_engine(options)
if models_mapper:
models_mapper.map(_ENGINE)
else:
from trove.instance import models as base_models
from trove.datastore import models as datastores_models
from trove.dns import models as dns_models
from trove.extensions.mysql import models as mysql_models
from trove.guestagent import models as agent_models
from trove.quota import models as quota_models
from trove.backup import models as backup_models
from trove.extensions.security_group import models as secgrp_models
from trove.configuration import models as configurations_models
from trove.conductor import models as conductor_models
model_modules = [
base_models,
datastores_models,
dns_models,
mysql_models,
agent_models,
quota_models,
backup_models,
secgrp_models,
configurations_models,
conductor_models,
]
models = {}
for module in model_modules:
models.update(module.persisted_models())
mappers.map(_ENGINE, models)
def _create_engine(options):
engine_args = {
"pool_recycle": CONF.sql_idle_timeout,
"echo": CONF.sql_query_log
}
LOG.info(_("Creating SQLAlchemy engine with args: %s") % engine_args)
return create_engine(options['sql_connection'], **engine_args)
def get_session(autocommit=True, expire_on_commit=False):
"""Helper method to grab session."""
global _MAKER, _ENGINE
if not _MAKER:
if not _ENGINE:
msg = "***The Database has not been setup!!!***"
LOG.exception(msg)
raise RuntimeError(msg)
_MAKER = sessionmaker(bind=_ENGINE,
autocommit=autocommit,
expire_on_commit=expire_on_commit)
return _MAKER()
def raw_query(model, autocommit=True, expire_on_commit=False):
return get_session(autocommit, expire_on_commit).query(model)
def clean_db():
global _ENGINE
meta = MetaData()
meta.reflect(bind=_ENGINE)
with contextlib.closing(_ENGINE.connect()) as con:
trans = con.begin()
for table in reversed(meta.sorted_tables):
if table.name != "migrate_version":
con.execute(table.delete())
trans.commit()
def drop_db(options):
meta = MetaData()
engine = _create_engine(options)
meta.bind = engine
meta.reflect()
meta.drop_all()
|
