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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
# 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
import threading
from oslo_db.sqlalchemy import session
from oslo_log import log as logging
from sqlalchemy import MetaData
from trove.common import cfg
from trove.common.i18n import _
from trove.db.sqlalchemy import mappers
_FACADE = None
_LOCK = threading.Lock()
LOG = logging.getLogger(__name__)
CONF = cfg.CONF
def configure_db(options, models_mapper=None):
facade = _create_facade(options)
if models_mapper:
models_mapper.map(facade)
else:
from trove.backup import models as backup_models
from trove.cluster import models as cluster_models
from trove.conductor import models as conductor_models
from trove.configuration import models as configurations_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.extensions.security_group import models as secgrp_models
from trove.guestagent import models as agent_models
from trove.instance import models as base_models
from trove.module import models as module_models
from trove.quota import models as quota_models
model_modules = [
base_models,
datastores_models,
dns_models,
mysql_models,
agent_models,
quota_models,
backup_models,
secgrp_models,
configurations_models,
conductor_models,
cluster_models,
module_models
]
models = {}
for module in model_modules:
models.update(module.persisted_models())
mappers.map(get_engine(), models)
def _create_facade(options):
global _LOCK, _FACADE
# TODO(mvandijk): Refactor this once oslo.db spec is implemented:
# https://specs.openstack.org/openstack/oslo-specs/specs/kilo/
# make-enginefacade-a-facade.html
if _FACADE is None:
with _LOCK:
if _FACADE is None:
conf = CONF.database
# pop the deprecated config option 'query_log'
if conf.query_log:
if conf.connection_debug < 50:
conf['connection_debug'] = 50
LOG.warning(('Configuration option "query_log" has been '
'depracated. Use "connection_debug" '
'instead. Setting connection_debug = '
'%(debug_level)s instead.'),
conf.get('connection_debug'))
# TODO(mvandijk): once query_log is removed,
# use enginefacade.from_config() instead
database_opts = dict(CONF.database)
database_opts.pop('query_log')
_FACADE = session.EngineFacade(
options['database']['connection'],
**database_opts
)
return _FACADE
def _check_facade():
if _FACADE is None:
LOG.exception("***The Database has not been setup!!!***")
raise RuntimeError(
_("***The Database has not been setup!!!***"))
def get_facade():
_check_facade()
return _FACADE
def get_engine(use_slave=False):
_check_facade()
return _FACADE.get_engine(use_slave=use_slave)
def get_session(**kwargs):
return get_facade().get_session(**kwargs)
def raw_query(model, **kwargs):
return get_session(**kwargs).query(model)
def clean_db():
engine = get_engine()
meta = MetaData()
meta.bind = engine
meta.reflect()
with contextlib.closing(engine.connect()) as con:
trans = con.begin() # pylint: disable=E1101
for table in reversed(meta.sorted_tables):
if table.name != "migrate_version":
con.execute(table.delete()) # pylint: disable=E1101
trans.commit()
def drop_db(options):
if options:
_create_facade(options)
engine = get_engine()
meta = MetaData()
meta.bind = engine
meta.reflect()
meta.drop_all()
|
