epoll实现服务器多进程多任务问答机制

#coding=utf-8
import socket
import  re
import select
import threading
import multiprocessing
import gevent
from gevent import  monkey
monkey.patch_all()  # 遇到阻塞自动切换协程，程序启动时执行monkey.patch_all()解决


def service_client(new_socket, request):
    '''为这个客户端返回数据'''

    # 1. 接收浏览器发过来的请求，即http请求
    # GET / HTTP/1.1
    # ...
    # request = new_socket.recv(1024).decode('UTF-8') # 解码
    #print('>>>'*40)
    #print(request)
    request_lines = request.splitlines()  # 转换成列表
    print('')
    print('>'*30)
    print(request_lines)

    # GET /index.html HTTP/1.1
    # get post put del
    ret = re.match(r'[^/]+(/[^ ]*)', request_lines[0])
    if ret:
        file_name = ret.group(1)
        print('*'*50, file_name)
        if file_name == '/':
            file_name = '/index.html'
    # 2. 返回http格式的数据给浏览器

    try:
        f = open('./html' + file_name, 'rb')

    except:
        response = 'HTTP/1.1 404 NOT FOUND\r\n'
        response += '\r\n'
        response += '----file not find----'
        new_socket.send(response.encode('utf-8'))

    else:
        html_content = f.read()
        f.close()
        # 2.1 准备发送给浏览器的数据 -- header,body

        response_body = html_content
        response_header = 'HTTP/1.1 200 OK\r\n'
        response_header += 'Content-Length:%d\r\n' % len(response_body)
        response_header += '\r\n'  # 空行

        response = response_header.encode('utf-8') + response_body

        # 2.2 将response header + body 发送给浏览器
        new_socket.send(response)


    # 关闭套接字
    # new_socket.close()


def main():
    '''整体控制'''
    # 1. 创建套接字
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    tcp_server_socket.setblocking(False)  # 设置套接字为非堵塞

    # 2. 绑定本地信息
    tcp_server_socket.bind(('', 7890))

    # 3. 开启监听
    tcp_server_socket.listen(128)
    tcp_server_socket.setblocking(False)  # 将套接字变为非堵塞

    # 创建一个epoll对象
    epl = select.epoll()

    # 将监听套接字对于的fd注册到epoll中
    epl.register(tcp_server_socket.fileno(), select.EPOLLIN)  # fileno得到文件描述符

    fd_event_dict = dict()

    while True:
        # 默认会堵塞，直到OS监测到数据到来，通过事件通知方式告诉这个程序，此时才会解堵塞，返回列表
        fd_event_list = epl.poll()

        # 元组[（fd, event),(套接字对应得文件描述符，这个文件描述符到底是什么事件 例如：可以调用recv接收等)]
        for fd, event in fd_event_list:
            if fd == tcp_server_socket.fileno():
                new_client_socket, new_client_addr = tcp_server_socket.accept()
                epl.register(new_client_socket.fileno(), select.EPOLLIN)
                fd_event_dict[new_client_socket.fileno()] = new_client_socket
            elif event == select.EPOLLIN:
                # 判断已经连接的客户端是否有数据发送过来
                client_socket = fd_event_dict[fd]
                recv_data = client_socket.recv(1024).decode('utf-8')
                if recv_data:
                    service_client(client_socket, recv_data)
                else:
                    client_socket.close()
                    epl.unregister(fd)
                    del fd_event_dict[fd]

    # 关闭监听套接字
    tcp_server_socket.close()

if __name__ == '__main__':
    main()