python多进程、线程实现简易http服务器

一、多进程版本

import re
import socket
import multiprocessing

#发送请求的数据文件
def webService(new_socket):
    """为客户端返回数据"""
    #1、接收浏览器发送过来的请求，即http请求
    #GET / HTTP1.1
    request = new_socket.recv(1024).decode("utf-8")
    request_lines = request.splitlines() #按行切割
    print("")
    print(">" * 100)
    print(request_lines)

    #提取需要的文件名
    #GET /index.html HTTP/1.1
    file_name = ""
    ret = re.match(r"[^/]+(/[^ ]*)", request_lines[0])
    if ret:
        file_name = ret.group(1)
        if file_name == "/":
            file_name = "./index.html"
    else:
        return

    try:
        f = open("." + file_name, "rb")
    except:
        response = "HTTP/1.1 404 NOT FOUND\r\n"
        response += "\r\n"
        response += "-------file not found-------"
        new_socket.send(response.encode("utf-8"))
    else:
        # 2、返回http格式的数据给浏览器
        # 2.1、准备发送给浏览器的数据 --header--
        response = "HTTP/1.1 200 OK\r\n"
        response += "\r\n"  # 换行
        # 2.2、准备发送给浏览器的数据 --body----
        html_content = f.read()
        f.close()
        # 2.3、将response header发送给浏览器
        new_socket.send(response.encode("utf-8"))
        # 2.4、将response body发送给浏览器
        new_socket.send(html_content)

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


def main():
    """用来完成整体的控制"""
    #1、创建套接字
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    #2、绑定
    tcp_server_socket.bind(("", 7890))

    #3、变为监听套接字
    tcp_server_socket.listen(128)

    while True:
        #4、等待新客户端的连接
        new_socket, client_addr = tcp_server_socket.accept()

        #5、为这个客户端服务
        p = multiprocessing.Process(target=webService, args=(new_socket,))
        p.start()
        
        new_socket.close()

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


if __name__ == '__main__':
    main()

二、多线程版本

import re
import socket
import threading

#发送请求的数据文件
def webService(new_socket):
    """为客户端返回数据"""
    #1、接收浏览器发送过来的请求，即http请求
    #GET / HTTP1.1
    request = new_socket.recv(1024).decode("utf-8")
    request_lines = request.splitlines() #按行切割
    print("")
    print(">" * 100)
    print(request_lines)

    #提取需要的文件名
    #GET /index.html HTTP/1.1
    file_name = ""
    ret = re.match(r"[^/]+(/[^ ]*)", request_lines[0])
    if ret:
        file_name = ret.group(1)
        if file_name == "/":
            file_name = "./index.html"
    else:
        return

    try:
        f = open("." + file_name, "rb")
    except:
        response = "HTTP/1.1 404 NOT FOUND\r\n"
        response += "\r\n"
        response += "-------file not found-------"
        new_socket.send(response.encode("utf-8"))
    else:
        # 2、返回http格式的数据给浏览器
        # 2.1、准备发送给浏览器的数据 --header--
        response = "HTTP/1.1 200 OK\r\n"
        response += "\r\n"  # 换行
        # 2.2、准备发送给浏览器的数据 --body----
        html_content = f.read()
        f.close()
        # 2.3、将response header发送给浏览器
        new_socket.send(response.encode("utf-8"))
        # 2.4、将response body发送给浏览器
        new_socket.send(html_content)

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


def main():
    """用来完成整体的控制"""
    #1、创建套接字
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    #2、绑定
    tcp_server_socket.bind(("", 7890))

    #3、变为监听套接字
    tcp_server_socket.listen(128)

    while True:
        #4、等待新客户端的连接
        new_socket, client_addr = tcp_server_socket.accept()

        #5、为这个客户端服务
        t = threading.Thread(target=webService, args=(new_socket,))
        t.start()

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


if __name__ == '__main__':
    main()

三、两个版本区别

Linux中一切皆文件，一个新的socket其实就是对应一个文件，拥有了一个文件描述符。

注：两个版本区别：多进程版本在主线程中需要再次关闭一次new_socket，因为多进程版本之间资源不共享，子进程中拷贝了一份new_socket的文件信息，使用结束后，子进程需要关闭自己的连接，其次执行结束，主进程也需要关闭自己的new_sokcet；多线程版本中，线程间共享数据，主线程不需要再次关闭。