gandc/mtprotoproxy
1
0
Fork 0
mirror of https://github.com/alexbers/mtprotoproxy.git synced 2026-03-13 23:03:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "refactor task canceling a bit". The idea with the task cancelation doesn't work

Browse Source 
This reverts commit 444a1876b6.
This commit is contained in:
Alexander Bersenev
2018-06-27 01:05:08 +05:00
parent 32d3bffc7b
commit b74079c433
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mtprotoproxy.py
View File

@@ -801,22 +801,18 @@ async def handle_client(reader_clt, writer_clt):
update_stats(user, octets=len(data))
wr.write(data, extra)
await wr.drain()
except (OSError, asyncio.streams.IncompleteReadError) as e:
except (OSError, AttributeError, asyncio.streams.IncompleteReadError) as e:
# print_err(e)
pass
finally:
wr.transport.abort()
task_tg_to_clt = asyncio.ensure_future(connect_reader_to_writer(reader_tg, writer_clt, user))
task_clt_to_tg = asyncio.ensure_future(connect_reader_to_writer(reader_clt, writer_tg, user))
task_tg_to_clt = connect_reader_to_writer(reader_tg, writer_clt, user)
task_clt_to_tg = connect_reader_to_writer(reader_clt, writer_tg, user)
update_stats(user, curr_connects=1)
await asyncio.wait([task_tg_to_clt, task_clt_to_tg], return_when=asyncio.FIRST_COMPLETED)
update_stats(user, curr_connects=-1)
task_tg_to_clt.cancel()
task_clt_to_tg.cancel()
writer_tg.transport.abort()