python socket select 一对多_python socket 网络编程selector用法 实用

Server端:

import socket

import selectors

class Server(object):

def init(self,sel,sock):

self.sel = sel

self.sock = sock

def 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',10011

server_obj = Server(sel,sock)

server_obj.run(host,port)

Client端:

import socket

import selectors

import random

sel = 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()#到此和服务器的请求基本处理完成,关闭套接字

return

print ('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.fileobj

callback = event.data#创建一个回调函数,方便触发事务执行

callback(sock)#调用刚创建的回调函数