python非阻塞式单进程服务器

深碍√TFBOYSˉ_ 2022-06-09 13:46 139阅读 0赞

python的单进程服务器一次只能处理一个客户端，显然是没有实用价值的，但是我们可以将单进程服务器变为非阻塞式的。  
利用socket中的setblocking()方法可以将tcp套接字转化为非阻塞式套接字，可处理多个客户端，且不会相互影响，因为是非阻塞式的，但是要注意，转化为非阻塞式之后，如果没有接收到客户端消息则会产生异常，所以要用到异常处理，以及每次接收到的客户端消息不断更新，需要使用列表进行存储。  
主要代码如下：

#-*- coding:utf-8 -*-
    
    from socket import *
    #建立套接字
    serSocket = socket(AF_INET, SOCK_STREAM)
    localAddr = ('', 7789)
    #为其绑定本机信息
    serSocket.bind(localAddr)
    #将tcp套接字转化为非阻塞式
    serSocket.setblocking(False)
    #将套接字变为被动监听
    serSocket.listen(5)
    #用来存储所有客户端的信息
    clientAddrList = []
    
    while True:
        #print("---wait for client---")
        try:
            newSocket, clientAddr = serSocket.accept()
        except:
            pass
        else:
            print('---new client [%s]'%(str(clientAddr)))
            #为了不令接收到的客户端套接字陷入阻塞状态，将客户端套接字也转为非阻塞
            newSocket.setblocking(False)
            clientAddrList.append((newSocket, clientAddr))
            #因为引用计数器的关系，每一次循环的执行都会使newSocket变为新的对象，newSocket无法在接下来使用，应将其存在列表里
        for newSocket, clientAddr in clientAddrList:
            try:
                recvDate = newSocket.recv(1204)
            except:
                pass
            else:
                if len(recvDate) > 0:
                    print('---massage from [%s] is: %s---'%(str(clientAddr), recvDate.decode('utf-8')))
                else:
                    print('---client was closed---')
                    newSocket.close()
                    clientAddrList.remove((newSocket, clientAddr))