python使用select实现异步通信

来源:互联网 发布:青海广电网络投诉电话 编辑:程序博客网 时间:2024/06/15 15:06

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

下面是使用select的服务器:

#!/usr/bin/env pythonimport socket,selectserver=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 pythonimport sockethost='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 pythonimport socket,selects=socket.socket()host=""port=10000s.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 pythonimport socketport=10000s=socket.socket()host=socket.gethostname()s.connect((host,port))s.send("hello from the client")s.close()




2 0