python socket 异步通信

原创

mob64ca12da2d62 2023-09-29 21:25:25 ©著作权

文章标签 客户端 python 服务器 文章分类 Python 后端开发

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

Python Socket 异步通信实现

引言

Python提供了socket库来进行网络通信。在网络通信中，往往需要处理多个客户端的请求，这时就需要使用异步通信来提高效率。本文将介绍如何使用Python实现socket异步通信，并提供代码示例。

流程图

下面是整个流程的示意图：

gantt
    title Python Socket 异步通信流程

    section 建立服务器
    创建socket对象           :a1, 0, 1
    绑定服务器地址和端口       :a2, 2, 3
    监听客户端连接请求         :a3, 4, 5
    接受客户端连接            :a4, 6, 7
    设置socket为非阻塞模式      :a5, 8, 9

    section 处理客户端请求
    接收客户端数据            :b1, 10, 11
    处理客户端请求            :b2, 12, 13
    发送响应给客户端           :b3, 14, 15

    section 关闭服务器
    关闭客户端连接            :c1, 16, 17
    关闭服务器连接            :c2, 18, 19

步骤说明

建立服务器

创建socket对象（socket.AF_INET, socket.SOCK_STREAM）：使用IPv4和TCP协议创建socket对象。
```
import socket

server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
```
绑定服务器地址和端口（socket.bind(address)）：指定服务器的地址和端口号。
```
server_address = ('localhost', 8888)
server_socket.bind(server_address)
```
监听客户端连接请求（socket.listen(backlog)）：开始监听客户端连接请求。
```
server_socket.listen(5)
```
接受客户端连接（socket.accept()）：等待客户端连接，并返回一个新的socket对象和客户端地址。
```
client_socket, client_address = server_socket.accept()
```
设置socket为非阻塞模式（socket.setblocking(flag)）：将socket设置为非阻塞模式，以支持异步通信。
```
server_socket.setblocking(False)
```

处理客户端请求

接收客户端数据（socket.recv(bufsize)）：接收客户端发送的数据。
```
data = client_socket.recv(1024)
```
处理客户端请求：根据接收到的数据进行相应的处理逻辑。
发送响应给客户端（socket.sendall(bytes)）：将处理结果发送给客户端。
```
response = 'Hello, World!'
client_socket.sendall(response.encode())
```

关闭服务器

关闭客户端连接（socket.close()）：关闭客户端的socket连接。
```
client_socket.close()
```
关闭服务器连接（socket.close()）：关闭服务器的socket连接。
```
server_socket.close()
```

完整代码示例

下面是一个完整的示例代码：

import socket

# 建立服务器
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_address = ('localhost', 8888)
server_socket.bind(server_address)
server_socket.listen(5)
server_socket.setblocking(False)

# 处理客户端请求
while True:
    try:
        client_socket, client_address = server_socket.accept()

        data = client_socket.recv(1024)
        # 处理客户端请求
        response = 'Hello, World!'
        client_socket.sendall(response.encode())

        client_socket.close()
    except BlockingIOError:
        pass

# 关闭服务器
server_socket.close()