Python57_epoll实现HTTP

2019-09-29  本文已影响0人  jxvl假装

ps:epoll是Linux中使用的

IO多路复用

就是我们说的select, pll, epoll,有些地方也称这种IO方式为event drive IO

select/epoll的好处就在与单个process就可以同时处理多个网络连接的IO

它的基本原理就是select, poll, epoll这个fucntion会不断的轮询所负责的所有socket,当某个socket有数据达到了,通知用户进程

epoll

通过以上两个方面保证程序的效率

实例

Linux环境

from socket import *
import select
import re


def service_client(new_socket, request):
    # request = new_socket.recv(1024).decode("utf-8")
    request_lines = request.splitlines()

    for line in request_lines:
        print(line)

    file_name = None
    ret = re.search(r"/\w*", request_lines[0])
    print(ret.group())  # 测试用,输出匹配到的内容
    if ret:
        file_name = ret.group()
        if file_name == "/":
            file_name = "/index.html"

    try:
        f = open("." + file_name, "rb")
    except:
        response = "HTTP/1.1 404 NOT FOUND\r\n\r\n"
        new_socket.send(response.encode("utf-8"))
    else:
        html_content = f.read()
        f.close()

        response_body = html_content
        response_header = "HTTP/1.1 200 OK\r\nContent-Length:{}\r\n\r\n".format(
            len(response_body))  # header中添加Content_Length,让浏览器知道本只传输内容的长度
        response = response_header.encode("utf-8") + response_body
        new_socket.send(response)
        # new_socket.close()    # 如果此处有关闭,就又变成了短连接,
        # 但是如果不close,浏览器就不知道请求的数据传送完了没有,就一直“转圈”
        # 如何让浏览器知道本次传输已经完成?在header里面添加"Content-Length:%d"%(本次传输内容(body)的字节数)


def main():
    tcp_server = socket(AF_INET, SOCK_STREAM)
    tcp_server.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
    tcp_server.bind(("", 8080))
    tcp_server.listen(128)
    tcp_server.setblocking(False)  # 设置为非阻塞

    epl = select.epoll()  # 创建epoll对象,对应于应用程序和内核共用的内存
    # 然后后面想方设法把监听套接字扔进去
    epl.register(tcp_server.fileno(), select.EPOLLIN)   # 将监听套接字对应的文件描述符注册到(扔到)epoll中,第二个参数表示让操作系统检测的时候检测的是“收”,当收到数据的时候进行通知
   
    fd_event_dict = {}  # 为了在poll返回时能够用文件描述符找到对应的socket,用字典进行存储
    while True:
        # time.sleep(0.5)  # 为了验证程序,使程序的速度慢一点
        fd_event_list = epl.poll()  # poll默认会阻塞,直到os监测到数据到来,透过事件通知的方式告诉这个程序,此时才会解阻塞。
        # 返回的是一个列表,当epoll里面有很多个套接字的时候,
        # 当多个套接字都有数据到来,则对他们解阻塞,并将其以列表的方式返回
        # 列表中的元素形式为元组(套接字对应文件描述符fd,这个文件描述符到底是什么事件event(例如可以调用recv接收等))
        for fd, event in fd_event_list:
            if fd == tcp_server.fileno():   # 如果是监听套接字
                new_socket, client_addr = tcp_server.accept()   # 连接新的客户端
                epl.register(new_socket.fileno(), select.EPOLLIN)   # 将新的socket注册到epoll
                fd_event_dict[new_socket.fileno()] = new_socket
            elif event == select.EPOLLIN:   
            # 如果新的套接字有反应,事件类型一定是EPOLLIN
                recv_data = fd_event_dict[fd].recv(1024).decode("utf-8")
                if recv_data:
                    service_client(fd_event_dict[fd], recv_data)
                else:
                    fd_event_dict[fd].close()
                    epl.unregister(fd)  # 注销fd
                    del fd_event_dict[fd]


if __name__ == "__main__":
    main()
上一篇下一篇

猜你喜欢

热点阅读