单线程非阻塞长链接

轮询的方法接收数据，效率低，套接字越多，需要的时间越长

代码：

import socket
import re


def client_msg(new_client,recv_data):
    # 接收客户端请求
    # recv_data = new_client.recv(1024).decode("utf-8")
    # 把读出的数据分行
    recv_data_lines = recv_data.splitlines()
    # 正则匹配GET后面的请求的页面
    # [^/]表示非“/”字符 +表示至少1个或多个，[^ ]只要不是空格，*表示0个或多个
    ret = re.match(r"[^/]+(/[^ ]*)",recv_data_lines[0])

    # ret = re.search(r"/[^ ]*",recv_data_lines[0])  # 查找第一次“/”后面非空格的字符串
    file_name = ret.group(1)
    # 打开文件
    try:
        f = open("./html" + file_name,"rb")
    except:
        # 创建返回数据 HEADER
        response = "HTTP/1.1 404 NOT FOUND\r\n"
        response += "\r\n"
        response += "-----NOT FOUND----"
        # 发送数据到客户端
        new_client.send(response.encode("utf-8"))
    else:
        # 读出数据
        html_content = f.read()
        # 把读出的数据放到BODY体里面
        response_body = html_content

        # 创建返回数据 HEADER
        response_header = "HTTP/1.1 200 ok\r\n"
        # 告诉浏览器发送过去的长度
        response_header += "Content-Length:%d\r\n" % len(response_body)
        response_header += "\r\n"
        # 把HEADER和BODY拼接在一起
        response = response_header.encode("utf-8") + response_body
        # 发送数据到客户端
        new_client.send(response)
       
    # 关闭套接字 长链接不要关闭此套接字 等客户端关闭
    # new_client.close()


def main():
    # 创建套接字
    tcp_server_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    # 设置当服务器先调用close() 即服务器4次回收之后资源能够立即释放，这样就保证了下次运行程序时，可以立即使用
    # SO_REUSEADDR重复使用地址
    tcp_server_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
    # 绑定端口
    tcp_server_socket.bind(("",7890))
    # 设置监听状态
    tcp_server_socket.listen(128)
    # 设置非堵塞
    tcp_server_socket.setblocking(False)

    # 新建socket存储列表
    new_client_list = list()

    # 新建为客户端服务的套接字
    while True:
        try:
            new_client,client_addr = tcp_server_socket.accept()
        except Exception as ret:
           pass
        else:
            # 新套接字设置非堵塞
            new_client.setblocking(False)
            # 把新套接字放入客户端服务套接字列表
            new_client_list.append(new_client)
        # 遍历套接字列表，循环查看有没有数据过来 ----轮询
        for new_client_socket in new_client_list:
            try:
                # 如果有数据，就接收并解码
                recv_data = new_client_socket.recv(1024).decode("utf-8")
            except Exception as ret:
                # 如果没数据就抛出异常 pass处理
                pass
            else:
                # 接收的如果是客户的请求数据
                if recv_data:
                    # 调用客户端服务
                    client_msg(new_client_socket,recv_data)
                # 如果是客户端调用close() 则关闭套接字移出列表
                else:
                    new_client_socket.close()
                    # 移出列表节省资源
                    new_client_list.remove(new_client_socket)

    # 关闭套接字
    tcp_server_socket.close()


if __name__ == '__main__':
    main()