LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-10 19:46:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Better attempt at joining update worker threads

Browse Source
This commit is contained in:
Lonami Exo 2018-03-01 13:31:39 +01:00
parent 3d68c879dd
commit 771c573db1
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
telethon/update_state.py
View File

@ -62,23 +62,23 @@ class UpdateState:
"""
self.stop_workers()
self._workers = n
if n is None:
while self._updates:
self._updates.get()
else:
if n is not None:
self.setup_workers()
workers = property(fget=get_workers, fset=set_workers)
def stop_workers(self):
"""Raises "StopIterationException" on the worker threads to stop them,
and also clears all of them off the list
"""
Raises "StopIterationException" on the worker threads to stop
them, and also clears all the workers/updates from the lists.
"""
if self._workers:
with self._updates_lock:
# Insert at the beginning so the very next poll causes an error
# on all the worker threads
# TODO Should this reset the pts and such?
while self._updates:
self._updates.get()
for _ in range(self._workers):
self._updates.put(StopIteration())