基於select類型多路IO復用,實現簡單socket並發
阿新 • • 發佈:2018-05-23
清理 就是 ive class sockets true 簡單 Coding conn
還有很多缺限,如客戶斷開無限重復
以下轉至老師博客:
server:
#!/usr/bin/env python # -*- coding: utf-8 -*- __author__ = "alex" import select import socket import sys import queue server = socket.socket() server.setblocking(0) server_addr = (‘localhost‘,5000) print(‘starting up on %s port %s‘ % server_addr) server.bind(server_addr) server.listen(5) 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]
client:
__author__ = "alex" import socket import sys messages = [ b‘This is the message. ‘, b‘It will be sent ‘, b‘in parts.‘, ] server_address = (‘localhost‘, 5000) # 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() )
...
基於select類型多路IO復用,實現簡單socket並發