socket 多进程 python

原创

mob64ca12f8a724 2024-02-01 12:04:41 ©著作权

文章标签 客户端子进程父进程 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12f8a724的原创作品，请联系作者获取转载授权，否则将追究法律责任

实现 socket 多进程 Python

概述

在本文中，我将向你展示如何使用 Python 实现 socket 多进程通信。首先，让我们先了解一下整个实现流程，然后我会逐步指导你完成每一步需要做的事情。

实现流程

下面是整个实现流程的步骤表格：

步骤	描述
步骤 1	创建服务器端 socket 监听连接
步骤 2	接受客户端连接请求并创建子进程
步骤 3	子进程处理客户端请求
步骤 4	父进程继续监听其他客户端连接请求
步骤 5	关闭连接

接下来，让我们逐步讲解每一步需要做的事情，并提供相应的代码示例。

步骤 1：创建服务器端 socket 监听连接

首先，我们需要创建一个服务器端的 socket，用于监听客户端的连接请求。下面是实现这一步需要的代码：

import socket

# 创建 socket 对象
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 绑定 IP 地址和端口
server_socket.bind(('localhost', 8888))

# 监听连接
server_socket.listen(5)

上述代码中，我们使用 socket.socket() 方法创建了一个 socket 对象，并指定了地址族为 socket.AF_INET，传输层协议为 socket.SOCK_STREAM。接着，我们使用 bind() 方法将服务器端的 IP 地址和端口绑定到 socket 对象上，并使用 listen() 方法开始监听连接。

步骤 2：接受客户端连接请求并创建子进程

在这一步中，我们将接受客户端的连接请求，并为每个连接创建一个子进程来处理客户端的请求。下面是实现这一步需要的代码：

import os

while True:
    # 接受客户端连接请求
    client_socket, address = server_socket.accept()

    # 创建子进程
    pid = os.fork()

    if pid == 0:
        # 子进程处理客户端请求
        client_handler(client_socket)
        client_socket.close()
        os._exit(0)
    else:
        # 父进程继续监听其他客户端连接请求
        client_socket.close()

上述代码中，我们使用 accept() 方法接受客户端的连接请求，并返回一个新的 socket 对象和客户端的地址。接着，我们使用 os.fork() 方法创建一个子进程，并根据返回的子进程 ID 判断当前代码是在父进程还是子进程中。如果是子进程，我们调用 client_handler() 方法来处理客户端的请求，并在处理完成后关闭连接并退出子进程。如果是父进程，我们关闭连接并继续监听其他客户端连接请求。

步骤 3：子进程处理客户端请求

在这一步中，子进程将处理客户端的请求。你可以根据具体的需求来编写相应的代码。下面是一个简单的示例：

def client_handler(client_socket):
    # 接收客户端发送的数据
    data = client_socket.recv(1024)

    # 处理数据
    response = "Hello, " + data.decode()

    # 发送响应给客户端
    client_socket.send(response.encode())

上述代码中，我们首先使用 recv() 方法接收客户端发送的数据，然后根据具体的需求处理数据，并将处理后的响应使用 send() 方法发送给客户端。