本文共 2081 字,大约阅读时间需要 6 分钟。
Server端:
import socket
import selectorsclass Server(object):
def init(self,sel,sock):self.sel = selself.sock = sockdef run(self,host,port): self.sock.bind((host,port)) self.sock.listen(50) self.sock.setblocking(False)#设置非阻塞 self.sel.register(sock,selectors.EVENT_READ,self.accept) while True: events = self.sel.select()#默认是阻塞,有活动连接就返回活动连接列表 for key,mask in events: callback = key.data#创建一个回调函数并获取 callback(key.fileobj,mask)#调用回调函数def accept(self,sock,mask): conn,addr = sock.accept()#已经就绪,等待接收 print ('连接来自于{}'.format(addr)) conn.setblocking(False) # sock.send(str('thanks').decode()) sel.register(conn,selectors.EVENT_READ,self.read)#注册事件def read(self,conn,mask): data = conn.recv(1024)#就绪,等待接收数据 if data:#判断是否有数据过来,有就执行 print ('来自客户端:',data) conn.send(data) else: print ('准备关闭连接',conn) sel.unregister(conn) conn.close()
if name == 'main':
sel = selectors.DefaultSelector()#默认的选择方式sock = socket.socket()sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR,1)host,port = '127.0.0.1',10011server_obj = Server(sel,sock)server_obj.run(host,port)Client端:
import socket
import selectorsimport randomsel = selectors.DefaultSelector()
def write(sock):
sock.send(str(random.randint(0,99)).encode('utf-8'))#发送必须是一个byts的数据,需要转码sel.unregister(sock)#当发送成功后必须取消注册,用与接下来的接收(读)到数据的注册sel.register(sock,selectors.EVENT_READ,read)#注册一个读的事件def read(sock):
data = sock.recv(1024)if not data:#判断接收到的数据是否为空数据sel.unregister(sock)#当接收完成后,依然是取消注册sock.close()#到此和服务器的请求基本处理完成,关闭套接字returnprint ('receiver server %s' %data)for i in range(1000):#开1000个客户端测试
client = socket.socket()client.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)#防止编辑器的每次运行端口冲突问题client.connect(('localhost',10011))sel.register(client,selectors.EVENT_WRITE,write)#注册一个写的事件,接下来处理发送(写)事件的处理while True:
all_event = sel.select()print (allevent)for event, in all_event:sock = event.fileobjcallback = event.data#创建一个回调函数,方便触发事务执行callback(sock)#调用刚创建的回调函数转载于:https://blog.51cto.com/9694110/2054243