Python实现多线程/多进程的TCP服务器

  

为了实现Python多线程/多进程的TCP服务器,我们需要采用以下步骤:

1. 创建TCP Server

首先我们需要创建一个TCP服务器。可以使用Python的标准库socket来进行创建。我们需要指定服务器的IP地址和端口号,然后进行绑定和监听。

import socket

# TCP服务器IP和端口号配置
TCP_IP = 'localhost'
TCP_PORT = 9001

# 创建一个socket
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 绑定服务器IP和端口号
server_socket.bind((TCP_IP, TCP_PORT))

# 开始监听客户端请求
server_socket.listen(5)

print(f'TCP server is running on {TCP_IP}:{TCP_PORT}')

2. 实现多线程/多进程

实现多线程

使用threading模块,创建多个线程来处理客户端请求。当有新的连接请求到达时,服务器会为其创建一个新线程处理,并同时处理其他客户端的请求。

import threading

# TCP服务器IP和端口号配置
TCP_IP = 'localhost'
TCP_PORT = 9001

# 创建一个socket
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 绑定服务器IP和端口号
server_socket.bind((TCP_IP, TCP_PORT))

# 开始监听客户端请求
server_socket.listen(5)

# 线程处理函数
def handle_client(conn, addr):
    print(f'New connection from {addr}')

    # 处理客户端请求
    while True:
        data = conn.recv(1024)
        if not data:
            break
        print(f'Received {data.decode()} from {addr}')
        conn.send(data)

    # 关闭连接
    conn.close()

# 开始循环监听客户端请求
while True:
    # 等待连接请求
    conn, addr = server_socket.accept()

    # 创建新线程处理客户端请求
    t = threading.Thread(target=handle_client, args=(conn, addr))
    t.start()

实现多进程

使用multiprocessing模块,创建多个进程来处理客户端请求。当有新的连接请求到达时,服务器会为其创建一个新进程处理,并同时处理其他客户端的请求。

import multiprocessing

# TCP服务器IP和端口号配置
TCP_IP = 'localhost'
TCP_PORT = 9001

# 创建一个socket
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

# 绑定服务器IP和端口号
server_socket.bind((TCP_IP, TCP_PORT))

# 开始监听客户端请求
server_socket.listen(5)

# 进程处理函数
def handle_client(conn, addr):
    print(f'New connection from {addr}')

    # 处理客户端请求
    while True:
        data = conn.recv(1024)
        if not data:
            break
        print(f'Received {data.decode()} from {addr}')
        conn.send(data)

    # 关闭连接
    conn.close()

# 开始循环监听客户端请求
while True:
    # 等待连接请求
    conn, addr = server_socket.accept()

    # 创建新进程处理客户端请求
    p = multiprocessing.Process(target=handle_client, args=(conn, addr))
    p.start()

示例说明

以下是一个简单的客户端,可以使用telnet命令或其他方式连接服务器:

import socket

# TCP服务器IP和端口号配置
TCP_IP = 'localhost'
TCP_PORT = 9001

# 连接服务器
client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client_socket.connect((TCP_IP, TCP_PORT))

# 发送数据
client_socket.send(b'Hello, World!')

# 接收数据
data = client_socket.recv(1024)
print(f'Received {data.decode()}')

# 关闭连接
client_socket.close()

我们可以使用多个客户端同时连接服务器,并发送数据,观察服务器的输出。在多线程情况下,每个客户端的请求都会在不同的线程中处理,而在多进程情况下,每个客户端的请求会在不同的进程中处理。

相关文章