mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 13:27:47 +00:00
Code Issues Releases Wiki Activity

Join net worker threads before stopping. Fixes #62

Browse Source
This commit is contained in:
Dan 2018-04-29 20:30:32 +02:00
parent 33c417942c
commit de332471df
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pyrogram/session/session.py
View File

@ -120,6 +120,8 @@ class Session:
self.next_salt_thread = None self.next_salt_thread = None
self.next_salt_thread_event = Event() self.next_salt_thread_event = Event()
self.net_worker_list = []
self.is_connected = Event() self.is_connected = Event()
def start(self): def start(self):
@ -128,7 +130,14 @@ class Session:
self.connection.connect() self.connection.connect()
for i in range(self.NET_WORKERS): for i in range(self.NET_WORKERS):
Thread(target=self.net_worker, name="NetWorker#{}".format(i + 1)).start() self.net_worker_list.append(
Thread(
target=self.net_worker,
name="NetWorker#{}".format(i + 1)
)
)
self.net_worker_list[-1].start()
Thread(target=self.recv, name="RecvThread").start() Thread(target=self.recv, name="RecvThread").start()
@ -190,6 +199,11 @@ class Session:
for i in range(self.NET_WORKERS): for i in range(self.NET_WORKERS):
self.recv_queue.put(None) self.recv_queue.put(None)
for i in self.net_worker_list:
i.join()
self.net_worker_list.clear()
for i in self.results.values(): for i in self.results.values():
i.event.set() i.event.set()