python--IO多路复用（TCP结合IO多路复用实例代码）

一、select方法

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Date  : 2021/8/9
# @Name  : ZhouZongXin

"""
IO多路复用（TCP结合IO多路复用实例代码,实现服务端代码收发）
"""
from socket import *
from select import select

# 全局变量
HOST = "0.0.0.0"
PORT = 8888
ADDR = (HOST, PORT)

# TCP是一个IO对象(UDP是一个IO对象,open打开一个文件也是io对象)
tcp_socket = socket()
tcp_socket.bind(ADDR)
tcp_socket.listen(5)

# 设置为非阻塞
tcp_socket.setblocking(False)

# io监控列表
rlist = [tcp_socket]
wlist = []
xlist = []

# 循环监听
while True:
    # 对关注的IO进行监控
    rs, ws, xs = select(rlist, wlist, xlist)  # 读事件， 写事件， 异常事件
    # 对返回值RS 分情况讨论 监听套接字  客户端连接套接字
    for r in rs:
        if r is tcp_socket:
            # 处理客户端连接
            connfd, addr = r.accept()
            print("Connect from", addr)
            connfd.setblocking(False)  # 设置非阻塞
            rlist.append(connfd)  # 添加到监控列表
        else:
            # 收消息
            data = r.recv(1024)
            if not data:
                # 客户端退出
                rlist.remove(r)  # 移除消息
                r.close()
                continue
            print(data.decode())
            wlist.append(r)  # 当有客户端发消息接受，不回复，然后把套接字放入写列表，下方循环进行发送

    for w in ws:
        w.send(b"OK")  # 发送消息
        wlist.remove(w)  # 发送完消息后要移除，如果不移除会不断的写入