多进程并发网络通信---基于os.fork实现

并发网络通信

from socket import *
import os
import signal

# 定义常量
HOST = "127.0.0.1"
PORT = 8899
ADDR = (HOST, PORT)


# 详细的处理请求过程
def handle(sock):
    while True:
        data = sock.recv(1024)
        # 客户端断开，服务端会收到空字符
        if not data:
            break
        print(data.decode())
        sock.send(b"ok")
    sock.close()
    pass


# 处理僵尸进程
signal.signal(signal.SIGCHLD, signal.SIG_IGN)
# 创建TCP服务器
sockfd = socket(AF_INET, SOCK_STREAM)
sockfd.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
sockfd.bind(ADDR)
sockfd.listen(5)
sock = None
while True:
    try:
        sock, addr = sockfd.accept()
    except KeyboardInterrupt:
        os._exit()
    except Exception:
        continue
    #     创建子进程处理客户端请求
    pid = os.fork()
    if pid == 0:
        # 子进程处理，子进程空间也会有sockfd
        sockfd.close()
        handle(sock)
        os._exit()  # 销毁子进程
    else:
        # 父进程结束此次循环，继续等待下一个客户端请求
        sock.close()
        continue