gevent模块可以在单线程内启动协程,并且检测到阻塞会自动切换到其他协程,提升效率
测试下服务端单线程内利用协程,客户端开启500个线程连接的效果。
服务端
from gevent import monkey, spawn # gevent模块下,spawn提交协程
monkey.patch_all() # 必须放到socket模块之前,放在文件最上,检测识别遇到的阻塞
from socket import *
def communicate(conn):
while True:
try:
data = conn.recv(1024)
conn.send(data.upper())
except ConnectionResetError:
break
conn.close()
def server(ip, port):
server = socket(AF_INET, SOCK_STREAM)
server.setsockopt(SOL_SOCKET,SO_REUSEADDR,1)
server.bind((ip, port))
server.listen(500)
while True:
conn, addr = server.accept()
spawn(communicate, conn) #创建一个协程对象,第一个参数是函数名,第二个参数是给目标函数传参
server.close()
if __name__ == '__main__':
server('127.0.0.1', 8090) # 放在if __name__ == '__main__':之下
from socket import *
from threading import Thread, currentThread
def client():
client = socket(AF_INET, SOCK_STREAM)
client.connect(('127.0.0.1', 8090))
while True:
client.send(('%s hello' % currentThread().getName()).encode('utf-8'))
data = client.recv(1024)
print(data.decode('utf-8'))
client.close()
if __name__ == '__main__':
for i in range(500): # 开启500个线程
t = Thread(target=client)
t.start()