99aaa0272c
This patchset adds one new method to the partitions API
- PATCH /v1/partitions/{partition}
The partitions storage layer is also updated to support a simpler
update interface:
- update(self, name, **kwargs) [with kwargs in ('hosts', 'weight')]
The partitions transport layer is cleared of cruft and reuses JSON
handling utilities from the queues transport layer.
This patch implementation uses jsonschema to achieve self-documenting,
strict format validation of the incoming input.
The partition storage controllers are modified accordingly, as are the
tests. More tests are added for the partitions transport layer to
verify the correctness and to ease future refactorings.
Change-Id: Ifa92b1225421196f95131c2b74e3c07b07c4cfd4
Implements: blueprint placement-service
Closes-Bug: 1230841
72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
# Copyright (c) 2013 Rackspace Hosting, 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.
|
|
|
|
import six
|
|
|
|
from marconi.proxy.storage import base
|
|
from marconi.proxy.storage import exceptions
|
|
|
|
|
|
class PartitionsController(base.PartitionsBase):
|
|
def __init__(self, *args, **kwargs):
|
|
super(PartitionsController, self).__init__(*args, **kwargs)
|
|
self._col = self.driver.db['partitions']
|
|
|
|
def list(self):
|
|
for entry in sorted(self._col.values(), key=lambda x: x['n']):
|
|
yield _normalize(entry)
|
|
|
|
def get(self, name):
|
|
entry = None
|
|
try:
|
|
entry = self._col[name]
|
|
except KeyError:
|
|
raise exceptions.PartitionNotFound(name)
|
|
|
|
return _normalize(entry)
|
|
|
|
def exists(self, name):
|
|
return name in self._col
|
|
|
|
def create(self, name, weight, hosts):
|
|
self._col[name] = {'n': name,
|
|
'w': weight,
|
|
'h': hosts}
|
|
|
|
def update(self, name, **kwargs):
|
|
key, value = kwargs.popitem()
|
|
assert key in ('weight', 'hosts'), "kwargs (hosts, weight)"
|
|
try:
|
|
self._col[name][key[0]] = value
|
|
except KeyError:
|
|
raise exceptions.PartitionNotFound(name)
|
|
|
|
def delete(self, name):
|
|
try:
|
|
del self._col[name]
|
|
except KeyError:
|
|
pass
|
|
|
|
def drop_all(self):
|
|
self._col = {}
|
|
|
|
|
|
def _normalize(entry):
|
|
return {
|
|
'name': six.text_type(entry['n']),
|
|
'hosts': entry['h'],
|
|
'weight': entry['w']
|
|
}
|