Python通过poll实现异步IO的方法_Python

Python通过poll实现异步IO的方法

2020-07-12 11:33raygtr Python

这篇文章主要介绍了Python通过poll实现异步IO的方法,实例分析了poll方法实现异步IO的相关技巧,需要的朋友可以参考下

本文实例讲述了Python通过poll实现异步IO的方法。分享给大家供大家参考。具体分析如下：

在使用poll（）后返回轮询对象，该对象支持以下方法：
pollObj.register(fd,[,eventmask])第一个参数是注册新的文件描述符fd，fd要么是一个整数文件描述符，要么可以带有一个获取文件描述符的fileno()方法的对象。eventmask是一些按位或标记，这些标记指示要处理的事件。

POLLIN：       用于读取数据
POLLPRI：      用于读取紧急数据
POLLOUT：      准备写入
POLLERR：      错误情况
POLLHUP：      保持状态
POLLNVAL：     无效请求

最后在循环中利用pollObj.poll()来进行对已注册的文件描述符进行轮询。返回一元祖（fd,event)。其中fd是文件描述符，event是指示时间的位掩码。至需要将event与对应的时间进行&测试即可。

利用poll创建对一个多路文件复制程序，代码如下：

									#!/usr/bin/env python

									import select

									BLKSIZE=8192

									def readwrite(fromfd,tofd):

									  readbuf = fromfd.read(BLKSIZE)

									  if readbuf:

									    tofd.write(readbuf)

									    tofd.flush()

									  return len(readbuf)

									def copyPoll(fromfd1,tofd1,fromfd2,tofd2):

									  #定义需要监听的事件

									  READ_ONLY = (select.POLLIN |

									       select.POLLPRI |

									      select.POLLHUP |

									      select.POLLERR )

									  totalbytes=0

									    if not (fromfd1 or fromfd2 or tofd1 or tofd2) :

									    return 0

									  fd_dict = {fromfd1.fileno():fromfd1,fromfd2.fileno():fromfd2}

									  #创建poll对象p

									  p=select.poll()

									  #利用poll对象p对需要监视的文件描述符进行注册

									  p.register(fromfd1,READ_ONLY)

									  p.register(fromfd2,READ_ONLY)

									  while True:

									  #轮询已经注册的文件描述符是否已经准备好

									    result = p.poll()

									    if len(result) != 0:

									      for fd,events in result:

									        if fd_dict[fd] is fromfd1:

									          if events & (select.POLLIN|select.POLLPRI):

									            bytesread = readwrite(fromfd1,tofd1)

									            totalbytes+=bytesread

									          elif events & (select.POLLERR):

									            p.unregister(fd_dict[fd])

									        if fd_dict[fd] is fromfd2:

									          if events & (select.POLLIN|select.POLLPRI):

									            bytesread = readwrite(fromfd2,tofd2)

									            totalbytes+=bytesread

									          elif events & (select.POLLERR):

									            p.unregister(fd_dict[fd])

									    if bytesread <= 0:  

									      break

									  return totalbytes

									def main():

									  fromfd1 = open("/etc/fstab","r")

									  fromfd2 = open("/root/VMwareTools-8.8.1-528969.tar.gz","r")

									  tofd1 = open("/root/fstab","w+")

									  tofd2 = open("/var/passwd","w+")

									  totalbytes = copyPoll(fromfd1,tofd1,fromfd2,tofd2)

									  print "Number of bytes copied %d\n" % totalbytes

									  return 0

									if __name__=="__main__":

									  main()