利用Python發(fā)送 10 萬(wàn)個(gè) http 請(qǐng)求
前言:
假如有一個(gè)文件,里面有 10 萬(wàn)個(gè) url,需要對(duì)每個(gè) url
發(fā)送 http
請(qǐng)求,并打印請(qǐng)求結(jié)果的狀態(tài)碼,如何編寫(xiě)代碼盡可能快的完成這些任務(wù)呢?
Python 并發(fā)編程有很多方法,多線(xiàn)程的標(biāo)準(zhǔn)庫(kù) threading
,concurrency
,協(xié)程 asyncio
,當(dāng)然還有 grequests
這種異步庫(kù),每一個(gè)都可以實(shí)現(xiàn)上述需求,下面一一用代碼實(shí)現(xiàn)一下,本文的代碼可以直接運(yùn)行,給你以后的并發(fā)編程作為參考:
1.隊(duì)列+多線(xiàn)程
定義一個(gè)大小為 400 的隊(duì)列,然后開(kāi)啟 200 個(gè)線(xiàn)程,每個(gè)線(xiàn)程都是不斷的從隊(duì)列中獲取 url 并訪(fǎng)問(wèn)。
主線(xiàn)程讀取文件中的 url 放入隊(duì)列中,然后等待隊(duì)列中所有的元素都被接收和處理完畢。
代碼如下:
from threading import Thread import sys from queue import Queue import requests concurrent = 200 def doWork(): while True: url = q.get() status, url = getStatus(url) doSomethingWithResult(status, url) q.task_done() def getStatus(ourl): try: res = requests.get(ourl) return res.status_code, ourl except: return "error", ourl def doSomethingWithResult(status, url): print(status, url) q = Queue(concurrent * 2) for i in range(concurrent): t = Thread(target=doWork) t.daemon = True t.start() try: for url in open("urllist.txt"): q.put(url.strip()) q.join() except KeyboardInterrupt: sys.exit(1)
運(yùn)行結(jié)果如下:
有沒(méi)有 get 到新技能?
2.線(xiàn)程池
如果使用線(xiàn)程池,推薦使用更高級(jí)的 concurrent.futures 庫(kù):
import concurrent.futures import requests out = [] CONNECTIONS = 100 TIMEOUT = 5 urls = [] with open("urllist.txt") as reader: for url in reader: urls.append(url.strip()) def load_url(url, timeout): ans = requests.get(url, timeout=timeout) return ans.status_code with concurrent.futures.ThreadPoolExecutor(max_workers=CONNECTIONS) as executor: future_to_url = (executor.submit(load_url, url, TIMEOUT) for url in urls) for future in concurrent.futures.as_completed(future_to_url): try: data = future.result() except Exception as exc: data = str(type(exc)) finally: out.append(data) print(data)
3.協(xié)程 + aiohttp
協(xié)程也是并發(fā)非常常用的工具了,
import asyncio from aiohttp import ClientSession, ClientConnectorError async def fetch_html(url: str, session: ClientSession, **kwargs) -> tuple: try: resp = await session.request(method="GET", url=url, **kwargs) except ClientConnectorError: return (url, 404) return (url, resp.status) async def make_requests(urls: set, **kwargs) -> None: async with ClientSession() as session: tasks = [] for url in urls: tasks.append( fetch_html(url=url, session=session, **kwargs) ) results = await asyncio.gather(*tasks) for result in results: print(f'{result[1]} - {str(result[0])}') if __name__ == "__main__": import sys assert sys.version_info >= (3, 7), "Script requires Python 3.7+." with open("urllist.txt") as infile: urls = set(map(str.strip, infile)) asyncio.run(make_requests(urls=urls))
4.grequests[1]
這是個(gè)第三方庫(kù),目前有 3.8K 個(gè)星,就是 Requests + Gevent[2],
讓異步 http 請(qǐng)求變得更加簡(jiǎn)單。Gevent
的本質(zhì)還是協(xié)程。
使用前:
pip install grequests
使用起來(lái)那是相當(dāng)?shù)暮?jiǎn)單:
import grequests urls = [] with open("urllist.txt") as reader: for url in reader: urls.append(url.strip()) rs = (grequests.get(u) for u in urls) for result in grequests.map(rs): print(result.status_code, result.url)
注意 :grequests.map(rs) 是并發(fā)執(zhí)行的。
運(yùn)行結(jié)果如下:
也可以加入異常處理:
>>> def exception_handler(request, exception): ... print("Request failed") >>> reqs = [ ... grequests.get('http://httpbin.org/delay/1', timeout=0.001), ... grequests.get('http://fakedomain/'), ... grequests.get('http://httpbin.org/status/500')] >>> grequests.map(reqs, exception_handler=exception_handler) Request failed Request failed [None, None, <Response [500]>]
最后的話(huà):
今天分享了并發(fā) http 請(qǐng)求的幾種實(shí)現(xiàn)方式,有人說(shuō)異步(協(xié)程)性能比多線(xiàn)程好,其實(shí)要分場(chǎng)景看的,沒(méi)有一種方法適用所有的場(chǎng)景,筆者就曾做過(guò)一個(gè)實(shí)驗(yàn),也是請(qǐng)求 url,當(dāng)并發(fā)數(shù)量超過(guò) 500 時(shí),協(xié)程明顯變慢。所以,不能說(shuō)哪個(gè)一定比哪個(gè)好,需要?jiǎng)澐智闆r。
到此這篇關(guān)于利用Python發(fā)送 10 萬(wàn)個(gè) http 請(qǐng)求的文章就介紹到這了,更多相關(guān)利用Python發(fā)送 http 請(qǐng)求內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Python高級(jí)特性之閉包與裝飾器實(shí)例詳解
這篇文章主要介紹了Python高級(jí)特性之閉包與裝飾器,結(jié)合實(shí)例形式詳細(xì)分析了Python閉包與裝飾器的相關(guān)原理、使用方法及相關(guān)操作注意事項(xiàng),需要的朋友可以參考下2019-11-11Python使用Paramiko庫(kù)實(shí)現(xiàn)SSH管理詳解
paramiko 是一個(gè)用于在Python中實(shí)現(xiàn)SSHv2協(xié)議的庫(kù),它支持對(duì)遠(yuǎn)程服務(wù)器進(jìn)行加密的通信,本文主要介紹了如何使用Paramiko庫(kù)實(shí)現(xiàn)SSH管理,感興趣的小伙伴可以學(xué)習(xí)一下2023-11-11基于Python實(shí)現(xiàn)傻瓜式GIF制作工具
有沒(méi)有什么內(nèi)容形式,比小視頻更小,比普通圖片更豐富?有,GIF動(dòng)態(tài)圖就是其中一種形式。本文將為大家介紹如何通過(guò)Python實(shí)現(xiàn)一個(gè)傻瓜式的gif生成工具,感興趣的可以了解一下2021-12-12Python與數(shù)據(jù)庫(kù)交互:入門(mén)指南
這篇文章主要介紹了Python與數(shù)據(jù)庫(kù)交互:入門(mén)指南的相關(guān)資料,需要的朋友可以參考下2023-11-11pytorch實(shí)現(xiàn)textCNN的具體操作
這篇文章主要介紹了pytorch實(shí)現(xiàn)textCNN的具體操作流程,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2021-05-05python使用ctypes調(diào)用擴(kuò)展模塊的實(shí)例方法
在本篇文章里小編給大家整理的是一篇關(guān)于python使用ctypes調(diào)用擴(kuò)展模塊的實(shí)例方法內(nèi)容,需要的朋友們可以學(xué)習(xí)參考下。2020-01-01