roypur/stream-api
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

cancel tasks in loop

Browse Source
This commit is contained in:
Roy Olav Purser
2021-05-14 21:55:56 +02:00
parent 55de3febd9
commit 0c171e84c8
1 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
stream.py
View File

@@ -160,9 +160,6 @@ class UpstreamHandler():
futures.append(task) futures.append(task)
done = False done = False
for future in asyncio.as_completed(futures): for future in asyncio.as_completed(futures):
if done:
future.cancel()
else:
try: try:
result = await future result = await future
resp = result.resp resp = result.resp
@@ -179,6 +176,11 @@ class UpstreamHandler():
self.upstream_safe = urllib.parse.quote(new_url) self.upstream_safe = urllib.parse.quote(new_url)
self.proxy = result.current_proxy self.proxy = result.current_proxy
proxies[self.provider] = result.current_proxies proxies[self.provider] = result.current_proxies
break
for future in futures:
if not future.done():
future.cancel()
async def meta(self): async def meta(self):
data = [] data = []
try: try: