Python使用select实现异步通信

当一个服务器需要与多个客户端进行通信时，可以使用多进程或者多线程的服务器，也可以使用select模块，它可以实现异步通信。Python中的select模块包含了poll()和select(),select的原型为(rlist,wlist,xlist[,timeout]),其中rlist是等待读取的对象，wlist是等待写入的对象，xlist是等待异常的对象，最后一个是可选对象，指定等待的时间，单位是s.  select()方法的返回值是准备好的对象的三元组，若在timeout的时间内，没有对象准备好，那么返回值将是空的列表。

下面是使用select的服务器：

#!/usr/bin/env python
import socket,select
server=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
server.bind(('',10000))
server.listen(5)
inputs=[server]
while 1:
    rs,ws,es=select.select(inputs,[],[],1)
    for r in rs:
        if r is server:
            clientsock,clientaddr=r.accept();
            inputs.append(clientsock);
        else:
            data=r.recv(1024);
            if not data:
                inputs.remove(r);
            else:
                print data

再编写客户端程序，就可以测试select服务器了，客户端代码为：

#!/usr/bin/env python
import socket
host='127.0.0.1';
port=10000;
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect((host,port))
s.send('hello from client')
s.close();

poll实现服务器时，需要用到register()和unregister()方法，作用是加入和移除对象，poll()的返回值包括了文件描述符和事件，polling的事件常量有POLLIN，POLLPRI，POLLPOUT，POLLERR，POLLHUP，POLLVAL，分别表示读取数据，读取紧急数据，文件描述符已经准备好，文件描述符出错，连接丢失，无效请求。

下面是使用poll的服务器程序：

#!/usr/bin/env python
import socket,select
s=socket.socket()
host=""
port=10000
s.bind((host,port))
fdmap={s.fileno():s}
s.listen(5)
p=select.poll()
p.register(s.fileno(),select.POLLIN|select.POLLERR|select.POLLHUP)
while 1:
    events=p.poll(5000)
    if len(events)!=0:
        if events[0][1]==select.POLLIN:
            sock,addr=s.accept()
            buf=sock.recv(8196)
            if len(buf)!=0:
                print buf
                sock.close()
    print "no data"

再编写客户端程序，即可以测试服务器程序，客户端为：

#!/usr/bin/env python
import socket
port=10000
s=socket.socket()
host=socket.gethostname()
s.connect((host,port))
s.send("hello from the client")
s.close()

Python 的详细介绍：请点这里
Python 的下载地址：请点这里

推荐阅读：