单进程、单线程非堵塞长连接http服务器

本章时前面的综合案例编写。关键点
1、设置非堵塞
2、参数的传递
3、返回头的设置

response_header += "Content-Length:%d\r\n" % len(response_body)

"""单进程、单线程非堵塞长连接http服务器"""
import socket
import re


def service_client(client_socket, request):
    """为客户端返回数据"""
    # 接收浏览器发送过来的请求 即 http 请求
    # request = new_socket.recv(1024).decode("utf-8")
    # 分割数据返回一个列表
    request_lines = request.splitlines()
    print(">>>>"*20)
    print(request_lines)
    ret = re.match(r"[^/]+(/[^ ]*)", request_lines[0])
    if ret:
        file_name = ret.group(1)
        # print("="*2, file_name)

    try:
        f = open("养狼计划.html", "rb")
    except:
        response = "HTTP/1.1 404 NOT FOUND\r\n"
        response += "\r\n"
        response += "----file not found----"
        client_socket.send(response.encode("utf-8"))
    else:
        html_content = f.read()
        f.close()
        # 返回 http格式的数据，给浏览器
        # 准备发送给浏览器的数据 ---header  \r\n 换行操作

        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


        # response = "HTTP/1.1 200 OK\r\n"
        # response += "\r\n"
        # html_content 读取出来的是字符串，不能直接相加
        # 将 response header 发送给浏览器
        client_socket.send(response)
        # 将response body 发送给浏览器
        # new_socket.send(html_content)

    # new_socket.close()
    # 关闭close之后，使用 Http 1.1 这是长连接，那么客户端要重新发起
    # 下一次的数据请求时，客户端要知道上一次服务器发送的数据要发送结束才行
    #  也就是接收服务器的数据要接收完毕，那么如何知道服务器的数据发送
    # 结束呢？此时 response header 里面添加一个数据表示数据长度。
    # "Content-Length: %d\r\n " % len(response_body)表示长度


def main():
    # 创建套接字
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 保证再次使用服务器不会出现被占用的问题 # SO_REUSEADDR
    tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    # 绑定
    tcp_server_socket.bind(("", 7790))
    # 变为监听套接字
    tcp_server_socket.listen(128)
    tcp_server_socket.setblocking(False)  # 设置非堵塞
    # 等待客户端的链接
    client_socket_list = list()
    while True:
        # new_socket, client_addr = tcp_server_socket.accept()
        # service_client(new_socket)

        try:
            new_socket, client_addr = tcp_server_socket.accept()
        # 为客户端服务
        except Exception as ret:
            pass
        else:
            new_socket.setblocking(False)
            client_socket_list.append(new_socket)

        for client_socket in client_socket_list:
            try:
                reve_data = client_socket.recv(1024).decode("utf-8")
            except Exception as ret:
                pass
            else:
                if reve_data:
                    service_client(client_socket, reve_data)
                else:
                    client_socket.close()
                    client_socket_list.remove(client_socket)
            # service_client(new_socket, client_socket)
    tcp_server_socket.close()


if __name__ == '__main__':
    main()