Python select實現異步IO

在Python中使用select與poll比起在C中使用簡單得多。select函數的參數是3個列表，包含整數文件描述符，或者帶有可返回文件描述符的fileno()方法對象。第一個參數是需要等待輸入的對象，第二個指定等待輸出的對象，第三個參數指定異常情況的對象。第四個參數則為設置超時時間，是一個浮點數。指定以秒為單位的超時值。select函數將會返回一組文件描述符，包括輸入，輸出以及異常。

在linux下利用select實現多路IO的文件復制程序：

1. #!/usr/bin/env python
   
2. 
   
3. 
   
4. import select
   
5. #導入select模塊
6. 
   
7. BLKSIZE=8192
   
8. 
   
9. def readwrite(fromfd,tofd):
   
10. readbuf = fromfd.read(BLKSIZE)
    
11. if readbuf:
    
12. tofd.write(readbuf)
    
13. tofd.flush()
    
14. return len(readbuf)
    
15. 
    
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:
    
26. 
    
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")
    
41. 
    
42. tofd1 = open("/root/fstab","w+")
    
43. tofd2 = open("/root/passwd","w+")
    
44. 
    
45. totalbytes = copy2file(fromfd1,tofd1,fromfd2,tofd2)
    
46. 
    
47. print "Number of bytes copied %d\n" % totalbytes
    
48. return 0
    
49. 
    
50. 
    
51. 
    
52. if __name__=="__main__":
    
53. main()