|
| 1 | +# -*- coding: utf-8 -*- |
| 2 | + |
| 3 | +import time |
| 4 | +import requests |
| 5 | +import functools |
| 6 | +from concurrent import futures |
| 7 | + |
| 8 | + |
| 9 | +def time_count(func): |
| 10 | + @functools.wraps(func) |
| 11 | + def wrapper(*args, **kwargs): |
| 12 | + start = time.time() |
| 13 | + result = func(*args, **kwargs) |
| 14 | + end = time.time() |
| 15 | + print "time", end - start |
| 16 | + return result |
| 17 | + return wrapper |
| 18 | + |
| 19 | + |
| 20 | +urls = ['https://ele.me', |
| 21 | + 'https://baidu.com', |
| 22 | + 'https://jd.com', |
| 23 | + 'https://v2ex.com', |
| 24 | + 'https://windard.com', |
| 25 | + 'https://taobao.com', |
| 26 | + 'https://zhihu.com', |
| 27 | + 'https://vip.com', |
| 28 | + 'https://t.tt'] |
| 29 | + |
| 30 | + |
| 31 | +@time_count |
| 32 | +def main(): |
| 33 | + executor = futures.ThreadPoolExecutor() |
| 34 | + roads = [] |
| 35 | + results = [] |
| 36 | + for url in urls: |
| 37 | + future = executor.submit(requests.get, url) |
| 38 | + roads.append(future) |
| 39 | + |
| 40 | + for future in futures.as_completed(roads): |
| 41 | + result = future.result() |
| 42 | + results.append(result.status_code) |
| 43 | + |
| 44 | + executor.shutdown() |
| 45 | + return results |
| 46 | + |
| 47 | + |
| 48 | +@time_count |
| 49 | +def sync_main(): |
| 50 | + with futures.ThreadPoolExecutor() as executor: |
| 51 | + roads = executor.map(requests.get, urls) |
| 52 | + results = [result.status_code for result in roads] |
| 53 | + return results |
| 54 | + |
| 55 | + |
| 56 | +@time_count |
| 57 | +def async_main(): |
| 58 | + results = [] |
| 59 | + for url in urls: |
| 60 | + results.append(requests.get(url).status_code) |
| 61 | + return results |
| 62 | + |
| 63 | + |
| 64 | +if __name__ == '__main__': |
| 65 | + print main() |
| 66 | + print async_main() |
| 67 | + print sync_main() |
0 commit comments