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()