python之select模組
阿新 • • 發佈:2018-11-09
1、select模組簡介
2、select 多併發socket 例子
2.1、select socket server
#_*_coding:utf-8_*_ import select import socket import sys import queue server = socket.socket() server.setblocking(0) server_addr = ('localhost',10000) print('starting up on %s port %s' % server_addr) server.bind(server_addr) server.listen(View Code5) inputs = [server, ] #自己也要監測呀,因為server本身也是個fd outputs = [] message_queues = {} while True: print("waiting for next event...") readable, writeable, exeptional = select.select(inputs,outputs,inputs) #如果沒有任何fd就緒,那程式就會一直阻塞在這裡 for s in readable: #每個s就是一個socket if s is server: #別忘記,上面我們server自己也當做一個fd放在了inputs列表裡,傳給了select,如果這個s是server,代表server這個fd就緒了, #就是有活動了, 什麼情況下它才有活動? 當然 是有新連線進來的時候 呀 #新連線進來了,接受這個連線 conn, client_addr = s.accept() print("new connection from",client_addr) conn.setblocking(0) inputs.append(conn)#為了不阻塞整個程式,我們不會立刻在這裡開始接收客戶端發來的資料, 把它放到inputs裡, 下一次loop時,這個新連線 #就會被交給select去監聽,如果這個連線的客戶端發來了資料 ,那這個連線的fd在server端就會變成就續的,select就會把這個連線返回,返回到 #readable 列表裡,然後你就可以loop readable列表,取出這個連線,開始接收資料了, 下面就是這麼幹 的 message_queues[conn] = queue.Queue() #接收到客戶端的資料後,不立刻返回 ,暫存在佇列裡,以後傳送 else: #s不是server的話,那就只能是一個 與客戶端建立的連線的fd了 #客戶端的資料過來了,在這接收 data = s.recv(1024) if data: print("收到來自[%s]的資料:" % s.getpeername()[0], data) message_queues[s].put(data) #收到的資料先放到queue裡,一會返回給客戶端 if s not in outputs: outputs.append(s) #為了不影響處理與其它客戶端的連線 , 這裡不立刻返回資料給客戶端 else:#如果收不到data代表什麼呢? 代表客戶端斷開了呀 print("客戶端斷開了",s) if s in outputs: outputs.remove(s) #清理已斷開的連線 inputs.remove(s) #清理已斷開的連線 del message_queues[s] ##清理已斷開的連線 for s in writeable: try : next_msg = message_queues[s].get_nowait() except queue.Empty: print("client [%s]" %s.getpeername()[0], "queue is empty..") outputs.remove(s) else: print("sending msg to [%s]"%s.getpeername()[0], next_msg) s.send(next_msg.upper()) for s in exeptional: print("handling exception for ",s.getpeername()) inputs.remove(s) if s in outputs: outputs.remove(s) s.close() del message_queues[s]
2.2、select socket client
#_*_coding:utf-8_*_ import socket import sys messages = [ b'This is the message. ', b'It will be sent ', b'in parts.', ] server_address = ('localhost', 10000) # Create a TCP/IP socket socks = [ socket.socket(socket.AF_INET, socket.SOCK_STREAM), socket.socket(socket.AF_INET, socket.SOCK_STREAM), ] # Connect the socket to the port where the server is listening print('connecting to %s port %s' % server_address) for s in socks: s.connect(server_address) for message in messages: # Send messages on both sockets for s in socks: print('%s: sending "%s"' % (s.getsockname(), message) ) s.send(message) # Read responses on both sockets for s in socks: data = s.recv(1024) print( '%s: received "%s"' % (s.getsockname(), data) ) if not data: print(sys.stderr, 'closing socket', s.getsockname() )View Code
3、selectors模組
預設用epoll,沒有epoll用select。
import selectors import socket sel = selectors.DefaultSelector() def accept(sock, mask): conn, addr = sock.accept() # Should be ready print('accepted', conn, 'from', addr,mask) conn.setblocking(False) sel.register(conn, selectors.EVENT_READ, read) #新連線註冊read回撥函式 def read(conn, mask): data = conn.recv(1024) # Should be ready if data: print('echoing', repr(data), 'to', conn) conn.send(data) # Hope it won't block else: print('closing', conn) sel.unregister(conn) conn.close() sock = socket.socket() sock.bind(('localhost', 9999)) sock.listen(100) sock.setblocking(False) sel.register(sock, selectors.EVENT_READ, accept) while True: events = sel.select() #預設阻塞,有活動連線就返回活動的連線列表 for key, mask in events: callback = key.data #accept callback(key.fileobj, mask) #key.fileobj= 檔案控制代碼View Code