def _update(self, name, **kwargs):
names = ('uri', 'weight', 'flavor', 'options')
fields = common_utils.fields(kwargs,
names,
pred=lambda x: x is not None,
key_transform=lambda x: x[0])
assert fields, ('`weight`, `uri`, `flavor`, '
'or `options` not found in kwargs')
if 'o' in fields:
new_options = fields.get('o', None)
fields['o'] = self._packer(new_options)
pool_key = utils.pools_name_hash_key(name)
# (gengchc2): Pipeline ensures atomic inserts.
with self._client.pipeline() as pipe:
# (gengchc2): If flavor is changed, we need to change.pool key
# in pools subset.
if 'f' in fields:
flavor_old = self._get(name).get('flavor')
flavor_new = fields['f']
if flavor_old != flavor_new:
if flavor_new is not None:
new_subset_key = utils.pools_subset_key(flavor_new)
pipe.zadd(new_subset_key, 1, pool_key)
# (gengchc2) remove pool from flavor_old.pools subset
if flavor_old is not None:
old_subset_key = utils.pools_subset_key(flavor_old)
pipe.zrem(old_subset_key, pool_key)
pipe.hmset(pool_key, fields)
pipe.execute()
def _delete(self, name):
try:
pool = self.get(name)
flavor = pool.get("flavor", None)
# NOTE(gengchc2): If this is the only pool in the
# flavor and it's being used by a flavor, don't allow
# it to be deleted.
if flavor is not None:
flavor1 = {}
flavor1['name'] = flavor
pools_in_flavor = list(
self.get_pools_by_flavor(flavor=flavor1))
if self.flavor_ctl.exists(flavor)\
and len(pools_in_flavor) == 1:
raise errors.PoolInUseByFlavor(name, flavor)
pool_key = utils.pools_name_hash_key(name)
subset_key = utils.pools_subset_key(flavor)
set_key = utils.pools_set_key()
with self._client.pipeline() as pipe:
if flavor is not None:
pipe.zrem(subset_key, pool_key)
pipe.zrem(set_key, pool_key)
pipe.delete(pool_key)
pipe.execute()
except errors.PoolDoesNotExist:
pass
def _delete(self, name):
try:
pool = self.get(name)
flavor = pool.get("flavor", None)
# NOTE(gengchc2): If this is the only pool in the
# flavor and it's being used by a flavor, don't allow
# it to be deleted.
if flavor is not None:
flavor1 = {}
flavor1['name'] = flavor
pools_in_flavor = list(self.get_pools_by_flavor(
flavor=flavor1))
if self.flavor_ctl.exists(flavor)\
and len(pools_in_flavor) == 1:
raise errors.PoolInUseByFlavor(name, flavor)
pool_key = utils.pools_name_hash_key(name)
subset_key = utils.pools_subset_key(flavor)
set_key = utils.pools_set_key()
with self._client.pipeline() as pipe:
if flavor is not None:
pipe.zrem(subset_key, pool_key)
pipe.zrem(set_key, pool_key)
pipe.delete(pool_key)
pipe.execute()
except errors.PoolDoesNotExist:
pass
def _get(self, name, detailed=False):
pool_key = utils.pools_name_hash_key(name)
pool = self._client.hgetall(pool_key)
if pool is None or len(pool) == 0:
raise errors.PoolDoesNotExist(name)
return self._normalize(pool, detailed)
def _update(self, name, **kwargs):
names = ('uri', 'weight', 'flavor', 'options')
fields = common_utils.fields(kwargs, names,
pred=lambda x: x is not None,
key_transform=lambda x: x[0])
assert fields, ('`weight`, `uri`, `flavor`, '
'or `options` not found in kwargs')
if 'o' in fields:
new_options = fields.get('o', None)
fields['o'] = self._packer(new_options)
pool_key = utils.pools_name_hash_key(name)
# (gengchc2): Pipeline ensures atomic inserts.
with self._client.pipeline() as pipe:
# (gengchc2): If flavor is changed, we need to change.pool key
# in pools subset.
if 'f' in fields:
flavor_old = self._get(name).get('flavor')
flavor_new = fields['f']
if flavor_old != flavor_new:
if flavor_new is not None:
new_subset_key = utils.pools_subset_key(flavor_new)
pipe.zadd(new_subset_key, {pool_key: 1})
# (gengchc2) remove pool from flavor_old.pools subset
if flavor_old is not None:
old_subset_key = utils.pools_subset_key(flavor_old)
pipe.zrem(old_subset_key, pool_key)
pipe.hmset(pool_key, fields)
pipe.execute()
def _create(self, name, weight, uri, group=None, flavor=None,
options=None):
if group is not None:
raise errors.PoolRedisNotSupportGroup
flavor = flavor if flavor is not None else None
options = {} if options is None else options
pool_key = utils.pools_name_hash_key(name)
subset_key = utils.pools_subset_key(flavor)
set_key = utils.pools_set_key()
if self._exists(name):
self._update(name, weight=weight, uri=uri,
flavor=flavor, options=options)
return
pool = {
'pl': name,
'u': uri,
'w': weight,
'o': self._packer(options),
'f': flavor
}
# Pipeline ensures atomic inserts.
with self._client.pipeline() as pipe:
pipe.zadd(set_key, {pool_key: 1})
if flavor is not None:
pipe.zadd(subset_key, {pool_key: 1})
pipe.hmset(pool_key, pool)
pipe.execute()
def _get(self, name, detailed=False):
pool_key = utils.pools_name_hash_key(name)
pool = self._client.hgetall(pool_key)
if pool is None or len(pool) == 0:
raise errors.PoolDoesNotExist(name)
return self._normalize(pool, detailed)
def _list(self, marker=None, limit=10, detailed=False):
client = self._client
set_key = utils.pools_set_key()
marker_key = utils.pools_name_hash_key(marker)
rank = client.zrank(set_key, marker_key)
start = rank + 1 if rank is not None else 0
cursor = (f for f in client.zrange(set_key, start, start + limit - 1))
marker_next = {}
def normalizer(pools):
marker_next['next'] = pools['pl']
return self._normalize(pools, detailed=detailed)
yield utils.PoolsListCursor(self._client, cursor, normalizer)
yield marker_next and marker_next['next']
def _list(self, marker=None, limit=10, detailed=False):
client = self._client
set_key = utils.pools_set_key()
marker_key = utils.pools_name_hash_key(marker)
if marker_key:
rank = client.zrank(set_key, marker_key)
else:
rank = None
start = rank + 1 if rank is not None else 0
cursor = (f for f in client.zrange(set_key, start,
start + limit - 1))
marker_next = {}
def normalizer(pools):
marker_next['next'] = pools['pl']
return self._normalize(pools, detailed=detailed)
yield utils.PoolsListCursor(self._client, cursor, normalizer)
yield marker_next and marker_next['next']
def _create(self,
name,
weight,
uri,
group=None,
flavor=None,
options=None):
if group is not None:
raise errors.PoolRedisNotSupportGroup
flavor = flavor if flavor is not None else None
options = {} if options is None else options
pool_key = utils.pools_name_hash_key(name)
subset_key = utils.pools_subset_key(flavor)
set_key = utils.pools_set_key()
if self._exists(name):
self._update(name,
weight=weight,
uri=uri,
flavor=flavor,
options=options)
return
pool = {
'pl': name,
'u': uri,
'w': weight,
'o': self._packer(options),
'f': flavor
}
# Pipeline ensures atomic inserts.
with self._client.pipeline() as pipe:
pipe.zadd(set_key, 1, pool_key)
if flavor is not None:
pipe.zadd(subset_key, 1, pool_key)
pipe.hmset(pool_key, pool)
pipe.execute()
def _exists(self, name):
pool_key = utils.pools_name_hash_key(name)
return self._client.exists(pool_key)
def _exists(self, name):
pool_key = utils.pools_name_hash_key(name)
return self._client.exists(pool_key)