mirror of https://github.com/zulip/zulip.git
Extract adding clients to the various dicts to a function.
(imported from commit 667e5736710c23becdf4ebe5688a31421ac632f5)
This commit is contained in:
parent
af347acf99
commit
7c829c4024
|
@ -286,6 +286,11 @@ def get_client_descriptors_for_user(user_profile_id):
|
||||||
def get_client_descriptors_for_realm_all_streams(realm_id):
|
def get_client_descriptors_for_realm_all_streams(realm_id):
|
||||||
return realm_clients_all_streams.get(realm_id, [])
|
return realm_clients_all_streams.get(realm_id, [])
|
||||||
|
|
||||||
|
def add_to_client_dicts(client):
|
||||||
|
user_clients.setdefault(client.user_profile_id, []).append(client)
|
||||||
|
if client.all_public_streams:
|
||||||
|
realm_clients_all_streams.setdefault(client.realm_id, []).append(client)
|
||||||
|
|
||||||
def allocate_client_descriptor(user_profile_id, realm_id, event_types, client_type,
|
def allocate_client_descriptor(user_profile_id, realm_id, event_types, client_type,
|
||||||
apply_markdown, all_public_streams, lifespan_secs,
|
apply_markdown, all_public_streams, lifespan_secs,
|
||||||
narrow=[]):
|
narrow=[]):
|
||||||
|
@ -295,9 +300,7 @@ def allocate_client_descriptor(user_profile_id, realm_id, event_types, client_ty
|
||||||
client = ClientDescriptor(user_profile_id, realm_id, EventQueue(id), event_types, client_type,
|
client = ClientDescriptor(user_profile_id, realm_id, EventQueue(id), event_types, client_type,
|
||||||
apply_markdown, all_public_streams, lifespan_secs, narrow)
|
apply_markdown, all_public_streams, lifespan_secs, narrow)
|
||||||
clients[id] = client
|
clients[id] = client
|
||||||
user_clients.setdefault(user_profile_id, []).append(client)
|
add_to_client_dicts(client)
|
||||||
if all_public_streams:
|
|
||||||
realm_clients_all_streams.setdefault(realm_id, []).append(client)
|
|
||||||
return client
|
return client
|
||||||
|
|
||||||
def do_gc_event_queues(to_remove, affected_users, affected_realms):
|
def do_gc_event_queues(to_remove, affected_users, affected_realms):
|
||||||
|
@ -381,9 +384,7 @@ def load_event_queues():
|
||||||
for client in clients.itervalues():
|
for client in clients.itervalues():
|
||||||
# Put code for migrations due to event queue data format changes here
|
# Put code for migrations due to event queue data format changes here
|
||||||
|
|
||||||
user_clients.setdefault(client.user_profile_id, []).append(client)
|
add_to_client_dicts(client)
|
||||||
if client.all_public_streams:
|
|
||||||
realm_clients_all_streams.setdefault(client.realm_id, []).append(client)
|
|
||||||
|
|
||||||
logging.info('Tornado loaded %d event queues in %.3fs'
|
logging.info('Tornado loaded %d event queues in %.3fs'
|
||||||
% (len(clients), time.time() - start))
|
% (len(clients), time.time() - start))
|
||||||
|
|
Loading…
Reference in New Issue