select版tcp服务器（python实现）-蒲公英云

select版tcp服务器（python实现）

红太狼 2022-06-09 14:15 312阅读 0赞

在编写了单进程非阻塞式服务器之后，还有另外种写服务器的方法，便是利用select。
select是对底层操作系统的一个访问操作，因而效率较高，比单进程非阻塞中的for循环遍历效率要高，可以利用select进行选择，选择出来可以读取信息的套接字、可以发送信息的套接字、以及产生的异常（分别是三个返回值）。

readable, writable, exceptionable = select([], [], [])

以上即为select的使用方法，程序执行到该语句后进行阻塞等待，接收到新的套接字之后便解阻塞。
程序思路便是利用select检测、选择出能读取的套接字（包括服务器套接字、客户端套接字），将接收到消息的客户端套接字存入列表（列表中本来只有服务器套接字），之后进行for循环遍历，读取套接字中的信息或者进行与客户端的连接。
代码如下：

#-*- coding:utf-8 -*-
from socket import *
from select import select
import sys
def main():
    serSocket = socket(AF_INET, SOCK_STREAM)
    localAddr = ('', 7788)
    serSocket.bind(localAddr)
    serSocket.setblocking(False)
    serSocket.listen(100)
    inputs = [serSocket]
    running = True
    while True:
        readable, writable, exceptionable = select(inputs, [], [])
        for sock in readable:
            if sock == serSocket:
                clientSocket, clientAddr = serSocket.accept()
                print('newClient[%s]'%str(clientAddr))
                inputs.append(clientSocket)
            elif sock == sys.stdin:
                cmd = sys.stdin.readline()
                running = False
                break
            else:
                massage = sock.recv(1024)
                if massage:
                    print('massage from [%s] is %s'%(str(sock), massage.decode('utf-8')))
                else:
                    print('[%s] was closed'%(str(sock)))
                    inputs.remove(sock)
                    sock.close()
        if not running:
            break
    serSocket.close()
if __name__ == '__main__':
    main()