本文實例講述了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 %d\n" % totalbytes return 0 if __name__ = = "__main__" : main() |
希望本文所述對大家的Python程序設計有所幫助。