RepoMirrors/nvchecker
1
0
Fork 0
mirror of https://github.com/lilydjwg/nvchecker synced 2025-02-17 02:46:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle graceful exit and timeout

Browse Source 
Timeout was refactored and the defaults work correctly here.
This commit is contained in:
Felix Yan 2018-11-29 06:39:14 +08:00
parent e3d62d4169
commit c1e5822f07
No known key found for this signature in database
GPG Key ID: 786C63F330D7CB92
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nvchecker/source/aiohttp_httpclient.py
View File

@ -2,6 +2,7 @@
# Copyright (c) 2013-2017 lilydjwg <lilydjwg@gmail.com>, et al.
import atexit
import asyncio
import aiohttp
connector = aiohttp.TCPConnector(limit=20)
@ -24,5 +25,9 @@ class BetterClientSession(aiohttp.ClientSession):
raise HTTPError(res.status, res.reason, res)
return res
session = BetterClientSession(connector=connector, read_timeout=10, conn_timeout=5)
atexit.register(session.close)
session = BetterClientSession(connector=connector)
@atexit.register
def cleanup():
loop = asyncio.get_event_loop()
loop.run_until_complete(session.close())