Python通過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創建對一個多路文件復制程序，代碼如下：

　　?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 #!/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 %dn" % totalbytes return 0 if __name__=="__main__": main()

　　希望本文所述對大家的Python程序設計有所幫助。