TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 12:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow to specify a limit to message cache size

Browse Source
This commit is contained in:
KurimuzonAkuma 2024-03-07 12:39:50 +03:00
parent 0b83b1d55b
commit fd5d19dab1
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pyrogram/client.py
View File

@ -190,6 +190,10 @@ class Client(Methods):
A value that is too high may result in network related issues. A value that is too high may result in network related issues.
Defaults to 1. Defaults to 1.
max_message_cache_size (``int``, *optional*):
Set the maximum size of the message cache.
Defaults to 10000.
storage_engine (:obj:`~pyrogram.storage.Storage`, *optional*): storage_engine (:obj:`~pyrogram.storage.Storage`, *optional*):
Pass an instance of your own implementation of session storage engine. Pass an instance of your own implementation of session storage engine.
Useful when you want to store your session in databases like Mongo, Redis, etc. Useful when you want to store your session in databases like Mongo, Redis, etc.
@ -217,6 +221,7 @@ class Client(Methods):
UPDATES_WATCHDOG_INTERVAL = 15 * 60 UPDATES_WATCHDOG_INTERVAL = 15 * 60
MAX_CONCURRENT_TRANSMISSIONS = 1 MAX_CONCURRENT_TRANSMISSIONS = 1
MAX_MESSAGE_CACHE_SIZE = 10000
mimetypes = MimeTypes() mimetypes = MimeTypes()
mimetypes.readfp(StringIO(mime_types)) mimetypes.readfp(StringIO(mime_types))
@ -251,6 +256,7 @@ class Client(Methods):
sleep_threshold: int = Session.SLEEP_THRESHOLD, sleep_threshold: int = Session.SLEEP_THRESHOLD,
hide_password: bool = False, hide_password: bool = False,
max_concurrent_transmissions: int = MAX_CONCURRENT_TRANSMISSIONS, max_concurrent_transmissions: int = MAX_CONCURRENT_TRANSMISSIONS,
max_message_cache_size: int = MAX_MESSAGE_CACHE_SIZE,
storage_engine: Storage = None, storage_engine: Storage = None,
init_connection_params: "raw.base.JSONValue" = None init_connection_params: "raw.base.JSONValue" = None
): ):
@ -284,6 +290,7 @@ class Client(Methods):
self.sleep_threshold = sleep_threshold self.sleep_threshold = sleep_threshold
self.hide_password = hide_password self.hide_password = hide_password
self.max_concurrent_transmissions = max_concurrent_transmissions self.max_concurrent_transmissions = max_concurrent_transmissions
self.max_message_cache_size = max_message_cache_size
self.init_connection_params = init_connection_params self.init_connection_params = init_connection_params
self.executor = ThreadPoolExecutor(self.workers, thread_name_prefix="Handler") self.executor = ThreadPoolExecutor(self.workers, thread_name_prefix="Handler")
@ -320,7 +327,7 @@ class Client(Methods):
self.me: Optional[User] = None self.me: Optional[User] = None
self.message_cache = Cache(10000) self.message_cache = Cache(self.max_message_cache_size)
# Sometimes, for some reason, the server will stop sending updates and will only respond to pings. # Sometimes, for some reason, the server will stop sending updates and will only respond to pings.
# This watchdog will invoke updates.GetState in order to wake up the server and enable it sending updates again # This watchdog will invoke updates.GetState in order to wake up the server and enable it sending updates again