Python select实现异步IO


在Python中使用select与poll比起在C中使用简单得多。select函数的参数是3个列表，包含整数文件描述符，或者带有可返回文件描述符的fileno()方法对象。第一个参数是需要等待输入的对象，第二个指定等待输出的对象，第三个参数指定异常情况的对象。第四个参数则为设置超时时间，是一个浮点数。指定以秒为单位的超时值。select函数将会返回一组文件描述符，包括输入，输出以及异常。

在linux下利用select实现多路IO的文件复制程序：

  1. #!/usr/bin/env python


  4. import select
  5. #导入select模块

  7. BLKSIZE=8192

  9. def readwrite(fromfd,tofd):
  10.     readbuf = fromfd.read(BLKSIZE)
  11.     if readbuf:
  12.         tofd.write(readbuf)
  13.         tofd.flush()
  14.     return len(readbuf)

  16. def copy2file(fromfd1,tofd1,fromfd2,tofd2):
  17.         ''' using select to choice fds'''
  18.     totalbytes=0
  19.         if not (fromfd1 or fromfd2 or tofd1 or tofd2) :
  20. #检查所有文件描述符是否合法
  21.                 return 0
  22.     while True:
  23. #开始利用select对输入所有输入的文件描述符进行监视
  24.         rs,ws,es = select.select([fromfd1,fromfd2],[],[])
  25.         for r in rs:

  27.             if r is fromfd1:
  28. #当第一个文件描述符可读时，读入数据
  29.                 bytesread = readwrite(fromfd1,tofd1)            
  30.                 totalbytes += bytesread
  31.             if r is fromfd2:
  32.                 bytesread = readwrite(fromfd2,tofd2)
  33.                 totalbytes += bytesread
  34.         if (bytesread <= 0):
  35.             break
  36.     return totalbytes
  37. def main():
  38.     
  39.     fromfd1 = open("/etc/fstab","r")
  40.     fromfd2 = open("/etc/passwd","r")

  42.     tofd1 = open("/root/fstab","w+")
  43.     tofd2 = open("/root/passwd","w+")

  45.     totalbytes = copy2file(fromfd1,tofd1,fromfd2,tofd2)
  46.     
  47.     print "Number of bytes copied %d\n" % totalbytes
  48.     return 0
  49.     


  52. if __name__=="__main__":
  53.     main()

相关内容