Python如何實現非同步IO
阿新 • • 發佈:2020-10-15
前言
用阻塞 API 寫同步程式碼最簡單,但一個執行緒同一時間只能處理一個請求,有限的執行緒數導致無法實現萬級別的併發連線,過多的執行緒切換也搶走了 CPU 的時間,從而降低了每秒能夠處理的請求數量。為了達到高併發,你可能會選擇一個非同步框架,用非阻塞 API 把業務邏輯打亂到多個回撥函式,通過多路複用與事件迴圈的方式實現高併發。
磁碟 IO 為例,描述了多執行緒中使用阻塞方法讀磁碟,2 個執行緒間的切換方式。那麼,怎麼才能實現高併發呢?
把上圖中本來由核心實現的請求切換工作,交由使用者態的程式碼來完成就可以了,非同步化程式設計通過應用層程式碼實現了請求切換,降低了切換成本和記憶體佔用空間。非同步化依賴於 IO 多路複用機制,比如 Linux 的 epoll 或者 Windows 上的 iocp,同時,必須把阻塞方法更改為非阻塞方法,才能避免核心切換帶來的巨大消耗。Nginx、Redis 等高效能服務都依賴非同步化實現了百萬量級的併發。
下圖描述了非同步 IO 的非阻塞讀和非同步框架結合後,是如何切換請求的。
然而,寫非同步化程式碼很容易出錯。因為所有阻塞函式,都需要通過非阻塞的系統呼叫拆分成兩個函式。雖然這兩個函式共同完成一個功能,但呼叫方式卻不同。第一個函式由你顯式呼叫,第二個函式則由多路複用機制呼叫。
這種方式違反了軟體工程的內聚性原則,函式間同步資料也更復雜。特別是條件分支眾多、涉及大量系統呼叫時,非同步化的改造工作會非常困難。
Python如何實現非同步呼叫
from flask import Flask import time app = Flask(__name__) @app.route('/bar') def bar(): time.sleep(1) return '<h1>bar!</h1>' @app.route('/foo') def foo(): time.sleep(1) return '<h1>foo!</h1>' if __name__ == '__main__': app.run(host='127.0.0.1',port=5555,debug=True)
採用同步的方式呼叫
import requests import time starttime = time.time() print(requests.get('http://127.0.0.1:5555/bar').content) print(requests.get('http://127.0.0.1:5555/foo').content) print("消耗時間: ",time.time() -starttime)
b'<h1>bar!</h1>' b'<h1>foo!</h1>' 消耗時間: 2.015509605407715
取樣非同步的方式呼叫:
重點:
1.將阻塞io改為非阻塞io;
2.多路複用io監聽核心事件,事件觸發通過回撥函式;
3.使用者態程式碼採取事件迴圈的方式獲取事件,執行事件的回撥函式;
import selectors import socket import time # from asynrequest import ParserHttp class asynhttp: def __init__(self): self.selecter = selectors.DefaultSelector() def get(self,url,optiondict = None): global reqcount reqcount += 1 s = socket.socket() s.setblocking(False) try: s.connect(('127.0.0.1',5555)) except BlockingIOError: pass requset = 'GET %s HTTP/1.0\r\n\r\n' % url callback = lambda : self.send(s,requset) self.selecter.register(s.fileno(),selectors.EVENT_WRITE,callback) def send(self,s,requset): self.selecter.unregister(s.fileno()) s.send(requset.encode()) chunks = [] callback = lambda: self.recv(s,chunks) self.selecter.register(s.fileno(),selectors.EVENT_READ,callback) def recv(self,s,chunks): self.selecter.unregister(s.fileno()) chunk = s.recv(1024) if chunk: chunks.append(chunk) callback = lambda: self.recv(s,chunks) self.selecter.register(s.fileno(), selectors.EVENT_READ, callback) else: global reqcount reqcount -= 1 request_first,request_headers,request_content,_ = ParserHttp.parser(b''.join(chunks)) print("解析資料:",request_first,request_headers,request_content) print((b''.join(chunks)).decode()) return (b''.join(chunks)).decode() starttime = time.time() reqcount = 0 asynhttper = asynhttp() asynhttper.get('/bar') asynhttper.get('/foo') while reqcount: events = asynhttper.selecter.select() for event,mask in events: func = event.data func() print("消耗時間:" ,time.time() - starttime)
HTTP/1.0 200 OK Content-Type: text/html; charset=utf-8 Content-Length: 13 Server: Werkzeug/1.0.1 Python/3.7.7 Date: Thu, 15 Oct 2020 03:28:16 GMT <h1>bar!</h1> HTTP/1.0 200 OK Content-Type: text/html; charset=utf-8 Content-Length: 13 Server: Werkzeug/1.0.1 Python/3.7.7 Date: Thu, 15 Oct 2020 03:28:16 GMT <h1>foo!</h1> 消耗時間: 1.0127637386322021