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()