写在前面

为提高web服务器的服务质量,一般通过多线程/多进程实现多任务来服务大量用户,但线程和进程往往要消耗较多的系统资源,而且如果线程/进程数达到一个较大的基数,服务器的性能便会下降,这是就必须尝试用单个任务能够服务更多的用户;

这次我们就通过用gevent创建协程的方式,实现单个任务服务更多的用户

最终的实现效果如图所示

用浏览器打开 127.0.0.1:8888

简单的目录结构

如果对协程还不熟悉,可以看我的另一篇简书文章[Python3简单实现多任务(线程/协程篇)],其中包含了协程的多种创建方式,还有gevent简单的使用范例~

服务端源码

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...

如果对以上的程序有疑问,可以给作者留言,作者会第一时间回复~