写在前面
为提高web服务器的服务质量,一般通过多线程/多进程实现多任务来服务大量用户,但线程和进程往往要消耗较多的系统资源,而且如果线程/进程数达到一个较大的基数,服务器的性能便会下降,这是就必须尝试用单个任务能够服务更多的用户;
这次我们就通过用gevent创建协程的方式,实现单个任务服务更多的用户
最终的实现效果如图所示
用浏览器打开
127.0.0.1:8888
简单的目录结构
如果对协程还不熟悉,可以看我的另一篇简书文章[Python3简单实现多任务(线程/协程篇)],其中包含了协程的多种创建方式,还有gevent简单的使用范例~
服务端源码
代码语言:javascript复制import socket
import gevent
from gevent import monkey
import time
import random
import re
# 服务器类
class WIGS(object):
def __init__(self, port):
self.port = port
self.root_dir = "./HTML"
# 创建主套接字
self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 允许端口重用
self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
# 主套接字绑定端口
self.server_socket.bind(("", self.port))
# 主套接字转为被动模式
self.server_socket.listen(128)
pass
def run_forever(self):
self.create_new_socket()
pass
def create_new_socket(self):
while True:
new_client_socket, new_client_socket_addr = self.server_socket.accept()
gevent.spawn(self.deal_accept_data, new_client_socket)
def deal_accept_data(self, new_client_socket):
recv_data = new_client_socket.recv(1024)
print("已经接收到请求数据!")
# 接收到的请求为utf-8格式
recv_data = recv_data.decode("utf-8")
if not recv_data:
return
recv_data_list = recv_data.splitlines()
print("接收到的请求数据为",recv_data_list[0])
the_request_header = recv_data_list[0]
file_name = self.get_file_name(the_request_header)
print("请求的文件名为%s"%(file_name))
# 向客户端发送文件
self.send_html(file_name, new_client_socket)
new_client_socket.close()
def get_file_name(self, the_request_header):
"""GET /index.html HTTP/1.1"""
file_name = re.match(r"[^/] ([^ ] ).*", the_request_header).group(1)
if file_name == "/":
file_name = "/index.html"
return file_name
pass
#new_client_socket.close()
def send_html(self, file_name, new_client_socket):
try:
f = open(self.root_dir file_name, "rb")
except Exception as res:
print(res)
print("无法找到网页404")
else:
content = f.read()
respond_body = content
respond_header = "HTTP/1.1 200 OK rn"
respond_header = "Content-Type: text/html; charset=utf-8rn"
respond_header = respond_header "rn"
# 发送回应
new_client_socket.send(respond_header.encode("utf-8"))
new_client_socket.send(respond_body)
print("内容发送成功!")
pass
def main():
monkey.patch_all()
# 创建web服务器
port = int(input("请输入本地需要开启服务的端口号:"))
web_server = WIGS(port)
print("请在地址栏访问 <127.0.0.1:%d>"%(port))
# 启动web服务器
web_server.run_forever()
pass
if __name__ == "__main__":
main()
小结
在服务端服务大量的用户时,使用协程可以很好的节约资源,但稳定性不如多进程任务,建议和多任务配合使用;
gevent是很好用的框架,可以根据协程的耗时进度,自动在各协程之间跳转,虽然比如yied性能好,但用好了,可以极大提高协程的管理效率,人生苦短,我用gevent...
如果对以上的程序有疑问,可以给作者留言,作者会第一时间回复~