class RedisClient(object):
def __init__(self, key, startup_nodes):
"""
init cluster
"""
self.key = key
self.conn = StrictRedisCluster(startup_nodes=startup_nodes,
decode_responses=True)
def hdel(self, field):
"""
delete an item
:param field:
:return:
"""
self.conn.hdel(self.key, field)
def hexists(self, field):
"""
判断 key 中是否含有 field
:param field:
:return:
"""
return self.conn.hexists(self.key, field)
def hget(self, field):
"""
返回key中指定 field 中的 value
:param field:
:return:
"""
value = self.conn.hget(self.key, field)
if isinstance(value, bytes):
return value.decode('utf-8')
else:
return value if value else None
def hgetall(self):
"""
获取 {filed: value, field1: value1....}
:return:
"""
all_dict = self.conn.hgetall(self.key)
if not all_dict:
return
elif sys.version_info.major == 3:
return {
field.decode('utf-8'): value.decode('utf-8')
for field, value in all_dict.items()
}
else:
return all_dict
def hkeys(self):
"""
获取key中所有field
:return:
"""
field = self.conn.hkeys(self.key)
if isinstance(field, bytes):
return field.decode('utf-8')
else:
return field if field else None
def hlen(self):
"""
获取所有 filed 数量
:return:
"""
return self.conn.hlen(self.key)
def hset(self, field, value):
"""
设置 field: value
:param field:
:param value:
:return:
"""
self.conn.hset(self.key, field, value)
def hvals(self):
"""
获取所有values
:return:
"""
values = self.conn.hvals(self.key)
if not values:
return
elif sys.version_info.major == 3:
return [value.decode('utf-8') for value in values]
else:
return values
def change_key(self, key):
"""
替换 key
:param key:
:return:
"""
self.key = key
# ===============================================
def blpop(self, timeout):
self.conn.blpop(self.key, timeout=timeout)
def brpop(self, timeout):
self.conn.brpop(self.key, timeout=timeout)
def brpoplpush(self, dst, timeout):
self.conn.brpoplpush(self.key, dst=dst, timeout=timeout)
def lindex(self, i):
self.conn.lindex(self.key, index=i)
def llen(self):
self.conn.llen(self.key)
def lpop(self):
self.conn.lpop(self.key)
def lpush(self):
self.conn.lpush(self.key)
def lrange(self, start, stop):
self.conn.lrange(self.key, start, stop)
def lset(self, i, value):
self.conn.lset(self.key, index=i, value=value)
def rpop(self):
self.conn.rpop(self.key)
def rpoplpush(self, dst):
self.conn.rpoplpush(self.key, dst=dst)
def rpush(self, value):
self.conn.rpush(self.key, value)
skip_full_coverage_check=True)
nodes = r.connection_pool.nodes
slots = set(range(0, 16384))
node_key = 0
node_keys = {}
while slots:
node_key = node_key + 1
slot = nodes.keyslot(str(node_key))
if slot in slots:
slots.remove(slot)
node_keys[slot] = str(node_key)
workers = r.hkeys("Workers")
worker_count = len(workers)
print("%d workers" % worker_count)
p = r.pipeline()
for node_key in node_keys.viewvalues():
p.delete("{%s}:intersecting-workers" % node_key)
p.sadd("{%s}:intersecting-workers" % node_key, *workers)
p.execute()
print("created sets")
oversized_cas_names = []
