mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +00:00
9f2b0c769f
We should take adventage of the recipient field being denormalized into the Stream model. We don't need to make queries to figure out a stream's recipient id, so we take advantage of that to eliminate some of those redundant queries and simplify StreamRecipientMap.
71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
from typing import (Dict, List)
|
|
|
|
from django.db import connection
|
|
from zerver.models import Recipient
|
|
|
|
class StreamRecipientMap:
|
|
'''
|
|
This class maps stream_id -> recipient_id and vice versa.
|
|
It is useful for bulk operations. Call the populate_* methods
|
|
to initialize the data structures. You should try to avoid
|
|
excessive queries by finding ids up front, but you can call
|
|
this repeatedly, and it will only look up new ids.
|
|
|
|
You should ONLY use this class for READ operations.
|
|
|
|
Note that this class uses raw SQL, because we want to highly
|
|
optimize page loads.
|
|
'''
|
|
def __init__(self) -> None:
|
|
self.recip_to_stream = dict() # type: Dict[int, int]
|
|
self.stream_to_recip = dict() # type: Dict[int, int]
|
|
|
|
def populate_with(self, *, stream_id: int, recipient_id: int) -> None:
|
|
# We use * to enforce using named arguments when calling this function,
|
|
# to avoid confusion about the ordering of the two integers.
|
|
self.recip_to_stream[recipient_id] = stream_id
|
|
self.stream_to_recip[stream_id] = recipient_id
|
|
|
|
def populate_for_recipient_ids(self, recipient_ids: List[int]) -> None:
|
|
recipient_ids = sorted([
|
|
recip_id for recip_id in recipient_ids
|
|
if recip_id not in self.recip_to_stream
|
|
])
|
|
|
|
if not recipient_ids:
|
|
return
|
|
|
|
# see comment at the top of the class
|
|
id_list = ', '.join(str(recip_id) for recip_id in recipient_ids)
|
|
query = '''
|
|
SELECT
|
|
zerver_recipient.id as recipient_id,
|
|
zerver_recipient.type_id as stream_id
|
|
FROM
|
|
zerver_recipient
|
|
WHERE
|
|
zerver_recipient.type = %d
|
|
AND
|
|
zerver_recipient.id in (%s)
|
|
''' % (Recipient.STREAM, id_list)
|
|
|
|
self._process_query(query)
|
|
|
|
def _process_query(self, query: str) -> None:
|
|
cursor = connection.cursor()
|
|
cursor.execute(query)
|
|
rows = cursor.fetchall()
|
|
cursor.close()
|
|
for recip_id, stream_id in rows:
|
|
self.recip_to_stream[recip_id] = stream_id
|
|
self.stream_to_recip[stream_id] = recip_id
|
|
|
|
def recipient_id_for(self, stream_id: int) -> int:
|
|
return self.stream_to_recip[stream_id]
|
|
|
|
def stream_id_for(self, recip_id: int) -> int:
|
|
return self.recip_to_stream[recip_id]
|
|
|
|
def recipient_to_stream_id_dict(self) -> Dict[int, int]:
|
|
return self.recip_to_stream
|