〖Python〗socket+select实现构建异步非阻塞模块

import socket
import selectclass Request(object):"""select监听的是有变化的对象&＃xff0c;当多个socket通信存在的话&＃xff0c;当发生变化时&＃xff0c;无法确定哪个是哪个。此时就需要唯一指定哪个通信处理的是什么数据&＃xff01;我们想到的办法是&＃xff0c;再另创建一个类&＃xff0c;用于接收当前的socket和信息&＃xff0c;实例化唯一的对象&＃xff01;巧妙之处就在于唯一化的处理&＃xff0c;生成的每个对象都是唯一的&＃xff01;&＃xff01;&＃xff01;"""def __init__(self,sock,info):#初始化self.sock &＃61; sock 通信self.info &＃61; info 信息def fileno(self): #转换成fileno对象return self.sock.fileno()class NoBlockIO(object):def __init__(self):self.sock_list &＃61; []self.conns &＃61; []def add_request(self,req_info):"""创建请求假设&＃xff1a;req_info &＃61; {&＃39;host&＃39;: &＃39;www.baidu.com&＃39;, &＃39;port&＃39;: 80, &＃39;path&＃39;: &＃39;/&＃39;},:return:"""sock &＃61; socket.socket() #创建socket对象sock.setblocking(False) #设置为非阻塞#建立连接&＃xff0c;因为是非阻塞状态&＃xff0c;连接可能会出异常所以需要做异常处理。#此处一定要注意&＃xff0c;不管报不报错&＃xff0c;连接的请求肯定会发出&＃xff01;try:sock.connect((req_info[&＃39;host&＃39;],req_info[&＃39;port&＃39;]))except BlockingIOError as e:pass#实例化唯一的通信对象obj &＃61; Request(sock,req_info) #将生成的sock对象和信息列表传递到Request类&＃xff0c;然后获取对应的文件句柄&＃xff0c;信息保留到self.sock_list.append(obj) #将得到的对象添加到sock_list 通信列表中self.conns.append(obj) #将得到的对象添加到conns 连接列表中def run(self):"""开始事件循环,检测&＃xff1a;连接成功&＃xff1f;数据是否返回&＃xff1f;注意此处操作的就全是Request类生成的对象&＃xff01;&＃xff01;&＃xff01;:return:"""while True:# select.select([request对象,])r,w,e &＃61; select.select(self.sock_list,self.conns,[],0.05)# w,是否连接成功for obj in w:# 检查obj:request对象是谁&＃xff0c;此时就需要唯一判断了&＃xff01;利用类可以封装对象的特性解决&＃xff01;# socket, {&＃39;host&＃39;: &＃39;www.baidu.com&＃39;, &＃39;port&＃39;: 80, &＃39;path&＃39;: &＃39;/&＃39;},data &＃61; "GET %s http/1.1\r\nhost:%s\r\n\r\n" %(obj.info[&＃39;path&＃39;],obj.info[&＃39;host&＃39;]) #发送GET请求的信息obj.sock.send(data.encode(&＃39;utf-8&＃39;)) #连接成功&＃xff0c;发送字节数据self.conns.remove(obj) #然后在连接列表中删除这个对象&＃xff0c;防止发送两次&＃xff0c;排除循环通信的可能# 数据返回&＃xff0c;接收到数据for obj in r:response &＃61; obj.sock.recv(8096) #成功接收到数据&＃xff0c;接收返回的信息#单个回调函数obj.info[&＃39;callback&＃39;](response) #指定对应对象执行回调函数&＃xff0c;参数为返回的信息#多个回调函数的情况&＃xff01;#for func in obj.info[&＃39;callback&＃39;] #指定对应对象执行回调函数&＃xff0c;参数为返回的信息# func(response)
self.sock_list.remove(obj) #在监听列表中移除这个对象# 所有请求已经返回if not self.sock_list: #所有请求都处理完&＃xff0c;就退出循环break

from .NBIO import NoBlockIO#自定义定义回调函数&＃xff0c;对返回的信息进行处理(执行完毕之后&＃xff0c;对数据进行什么操作)
def done1(response):print(response)def done2(response):print(response)#准备执行的信息&＃xff0c;域名&＃xff0c;端口号&＃xff0c;回调函数
url_list &＃61; [{&＃39;host&＃39;: &＃39;www.baidu.com&＃39;, &＃39;port&＃39;: 80, &＃39;path&＃39;: &＃39;/&＃39;,&＃39;callback&＃39;: done1},{&＃39;host&＃39;: &＃39;www.cnblogs.com&＃39;, &＃39;port&＃39;: 80, &＃39;path&＃39;: &＃39;/index.html&＃39;,&＃39;callback&＃39;: done2},{&＃39;host&＃39;: &＃39;www.bing.com&＃39;, &＃39;port&＃39;: 80, &＃39;path&＃39;: &＃39;/&＃39;,&＃39;callback&＃39;: done2},
]#实例化一个对象
noblockObj &＃61; NoBlockIO()#循环要处理的信息
for item in url_list:noblockObj.add_request(item) #为条信息创建请求

noblockObj.run() #执行
#谁先执行完就返回谁&＃xff0c;相对独立&＃xff0c;互不干扰&＃xff01;