vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Add a test for swift path computation

Browse source
This commit is contained in:
Jake Moshenko 2016-01-15 11:15:40 -05:00
parent c6d7eba98d
commit 909e7d45b7
7 changed files with 74 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

14
storage/__init__.py
View file

@ -13,27 +13,27 @@ STORAGE_DRIVER_CLASSES = {
'SwiftStorage': SwiftStorage,
}
def get_storage_driver(storage_params):
def get_storage_driver(metric_queue, storage_params):
""" Returns a storage driver class for the given storage configuration
(a pair of string name and a dict of parameters). """
driver = storage_params[0]
parameters = storage_params[1]
driver_class = STORAGE_DRIVER_CLASSES.get(driver, FakeStorage)
return driver_class(**parameters)
return driver_class(metric_queue, **parameters)
class Storage(object):
def __init__(self, app=None):
def __init__(self, app=None, metric_queue=None):
self.app = app
if app is not None:
self.state = self.init_app(app)
if app is not None and metric_queue is not None:
self.state = self.init_app(app, metric_queue)
else:
self.state = None
def init_app(self, app):
def init_app(self, app, metric_queue):
storages = {}
for location, storage_params in app.config.get('DISTRIBUTED_STORAGE_CONFIG').items():
storages[location] = get_storage_driver(storage_params)
storages[location] = get_storage_driver(metric_queue, storage_params)
preference = app.config.get('DISTRIBUTED_STORAGE_PREFERENCE', None)
if not preference: