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