/
base.py
98 lines (82 loc) · 3.25 KB
/
base.py
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
# Copyright (c) 2013 Rackspace, 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.
from oslo.config import cfg
from stevedore import driver
from marconi.common import access
from marconi.common.cache import cache as oslo_cache
from marconi.common import decorators
from marconi.common import errors
from marconi.openstack.common import log
from marconi.proxy import transport # NOQA
_bootstrap_options = [
cfg.StrOpt('transport', default='wsgi',
help='Transport driver to use'),
cfg.StrOpt('storage', default='memory',
help='Storage driver to use'),
]
CFG = cfg.CONF
CFG.register_opts(_bootstrap_options, group="proxy:drivers")
LOG = log.getLogger(__name__)
class Bootstrap(object):
"""Defines the Marconi proxy bootstrapper.
The bootstrap loads up drivers per a given configuration, and
manages their lifetimes.
"""
def __init__(self, access_mode, config_file=None, cli_args=None):
default_file = None
if config_file is not None:
default_file = [config_file]
CFG(project='marconi', args=cli_args or [],
default_config_files=default_file)
log.setup('marconi_proxy')
form = 'marconi.proxy.{0}.transport'
lookup = {access.Access.public: 'public',
access.Access.admin: 'admin'}
self._transport_type = form.format(lookup[access_mode])
@decorators.lazy_property(write=False)
def storage(self):
LOG.debug(_(u'Loading Proxy Storage Driver'))
try:
mgr = driver.DriverManager('marconi.proxy.storage',
CFG['proxy:drivers'].storage,
invoke_on_load=True)
return mgr.driver
except RuntimeError as exc:
LOG.exception(exc)
raise errors.InvalidDriver(exc)
@decorators.lazy_property(write=False)
def cache(self):
LOG.debug(_(u'Loading Proxy Cache Driver'))
try:
mgr = oslo_cache.get_cache(CFG)
return mgr
except RuntimeError as exc:
LOG.exception(exc)
raise errors.InvalidDriver(exc)
@decorators.lazy_property(write=False)
def transport(self):
LOG.debug(_(u'Loading Proxy Transport Driver'))
try:
mgr = driver.DriverManager(self._transport_type,
CFG['proxy:drivers'].transport,
invoke_on_load=True,
invoke_args=[self.storage,
self.cache])
return mgr.driver
except RuntimeError as exc:
LOG.exception(exc)
raise errors.InvalidDriver(exc)
def run(self):
self.transport.listen()