Python网络编程笔记二

 使用select模块实现IO多路复用服务端

import socket
import select
#windows上只支持select.select，不支持poll epoll

HOST = "127.0.0.1"
PORT = 9999
BUFFSIZE = 1024
ADDR = (HOST, PORT)
#创建服务器端监听套接字
tcpSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
tcpSocket.bind(ADDR)
tcpSocket.listen(5)
tcpSocket.setblocking(0)


#监听读事件
inputs = [tcpSocket]
#监听写事件
outputs = []
#监听异常事件
exceptions = []

messageDict = {}

while True:
    #每当有客户端连接时，返回的r_list中包含监听的服务器套接字tcpSocket
    #w_list中始终与outputs相同
    r_list, w_list, e_list = select.select(inputs, outputs, exceptions, 1)
    print(len(inputs))
    for r in r_list:
        if r == tcpSocket:
            request, addr = r.accept()
            request.setblocking(0)
            #将客户端套接字加入到监听列表中，接下来每当客户端发送数据过来，r_list中即可包含该套接字
            inputs.append(request)
            messageDict[request] = []
        #如果套接字不等于tcpSocket，说明是客户端发送数据
        else:
            try:
                dataBytes = r.recv(1024)
            #说明客户端断开连接
            #ConnectionResetError
            except Exception as e:
                #将客户端套接字移除出监听列表
                inputs.remove(r)
                del messageDict[r]
                r.close()
            else:
                dataStr = str(dataBytes, encoding="utf-8")
                messageDict[r].append(dataStr)
                # print("接收到来自", r.getpeername(), " : ", dataStr)
                # r.sendall(bytes("已经收到您发送的" + dataStr, encoding="utf-8"))
                #添加到outputs，下次再循环时w_list中包含数据
                outputs.append(r)

    for w in w_list:
        messageList = messageDict[w]
        recvData = messageList[0]
        del messageList[0]
        print("接收到来自", w.getpeername(), " : ", recvData)
        w.sendall(bytes("已经收到您发送的" + recvData, encoding="utf-8"))
        outputs.remove(w)
tcpSocket.close()