当一个服务器需要与多个客户端进行通信时,可以使用多进程或者多线程的服务器,也可以使用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()