def sort_users_cache(self):
user_count = 0
users_dict = {}
users_sort = []
# Create temp uuid key to prevent multiple threads modifying same key
temp_suffix = 'temp_' + uuid.uuid4().hex
temp_users_sorted_key = 'users_sorted_' + temp_suffix
users_page_index_key = 'users_page_index_' + temp_suffix
try:
for user_id in cache_db.set_elements(self.get_cache_key('users')):
user = User.get_user(self, id=user_id)
if not user:
continue
name_id = '%s_%s' % (user.name, user_id)
if user.type == CERT_CLIENT:
user_count += 1
users_dict[name_id] = (user_id, user.type)
users_sort.append(name_id)
cache_db.set(self.get_cache_key('user_count'), str(user_count))
cur_page = 0
user_count = 0
client_count = 0
for name_id in sorted(users_sort):
if users_dict[name_id][1] == CERT_CLIENT:
page = client_count / USER_PAGE_COUNT
if page != cur_page:
cur_page = page
cache_db.dict_set(self.get_cache_key(users_page_index_key),
str(cur_page), str(user_count))
client_count += 1
user_count += 1
cache_db.list_rpush(self.get_cache_key(temp_users_sorted_key),
users_dict[name_id][0])
cache_db.lock_acquire(self.get_cache_key('sort'))
try:
cache_db.rename(self.get_cache_key(users_page_index_key),
self.get_cache_key('users_page_index'))
cache_db.rename(self.get_cache_key(temp_users_sorted_key),
self.get_cache_key('users_sorted'))
cache_db.set(self.get_cache_key('users_page_total'),
str(cur_page))
finally:
cache_db.lock_release(self.get_cache_key('sort'))
except:
cache_db.remove(self.get_cache_key(users_page_index_key))
cache_db.remove(self.get_cache_key(temp_users_sorted_key))
raise
def sort_servers_cache(cls):
servers_dict = {}
servers_sort = []
# Create temp uuid key to prevent multiple threads modifying same key
temp_sorted_key = 'servers_sorted_temp_' + uuid.uuid4().hex
try:
for server_id_type in cache_db.set_elements('servers'):
server_id, server_type = server_id_type.split('_', 1)
server = Server.get_server(id=server_id, type=server_type)
if not server:
continue
name_id = '%s_%s' % (server.name, server_id)
servers_dict[name_id] = server_id_type
servers_sort.append(name_id)
for name_id in sorted(servers_sort):
cache_db.list_rpush(temp_sorted_key, servers_dict[name_id])
cache_db.rename(temp_sorted_key, 'servers_sorted')
except:
cache_db.remove(temp_sorted_key)
raise
def sort_orgs_cache(cls):
org_count = 0
orgs_dict = {}
orgs_sort = []
# Create temp uuid key to prevent multiple threads modifying same key
temp_orgs_sorted_key = 'orgs_sorted_temp_' + uuid.uuid4().hex
try:
for org_id in cache_db.set_elements('orgs'):
org = Organization.get_org(id=org_id)
if not org:
continue
name_id = '%s_%s' % (org.name, org_id)
org_count += 1
orgs_dict[name_id] = org_id
orgs_sort.append(name_id)
cache_db.set('org_count', str(org_count))
for name_id in sorted(orgs_sort):
cache_db.list_rpush(temp_orgs_sorted_key, orgs_dict[name_id])
cache_db.rename(temp_orgs_sorted_key, 'orgs_sorted')
except:
cache_db.remove(temp_orgs_sorted_key)
raise
