d5b24ef15c
The patch adds Shard support of the client. This will allow users to create and delete shards from Marconi. The API requires the Marconi instance to be and admin instance. However, there are no checks, restrictions, yet. This patch doesn't implement Shards listing. As soon as the MessageIterator lands, it'll be possible to reuse that code for the ShardsIterator. Partially-Implements blueprint: shard-management Partially-Implements blueprint: python-marconiclient-v1 Change-Id: I932c98d68a4a81b710dbea75a6bac123b7afaa08
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
# Copyright (c) 2014 Red Hat, 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 marconiclient.queues.v1 import core
|
|
|
|
|
|
class Shard(object):
|
|
|
|
def __init__(self, client, name,
|
|
weight=None, uri=None,
|
|
auto_create=True, **options):
|
|
self.client = client
|
|
|
|
self.uri = uri
|
|
self.name = name
|
|
self.weight = weight
|
|
self.options = options
|
|
|
|
if auto_create:
|
|
self.ensure_exists()
|
|
|
|
def ensure_exists(self):
|
|
"""Ensures shard exists
|
|
|
|
This method is not race safe,
|
|
the shard could've been deleted
|
|
right after it was called.
|
|
"""
|
|
req, trans = self.client._request_and_transport()
|
|
|
|
data = {'uri': self.uri,
|
|
'weight': self.weight,
|
|
'options': self.options}
|
|
|
|
core.shard_create(trans, req, self.name, data)
|
|
|
|
def delete(self):
|
|
req, trans = self.client._request_and_transport()
|
|
core.shard_delete(trans, req, self.name)
|