python 3下基于select模型的事件驱动机制程序

importselect
importsocket
importqueue
fromtimeimportsleep

classTCPServer:
def__init__(self,server,server_address,inputs,outputs,message_queues):
#CreateaTCP/IP
self.server=server
self.server.setblocking(False)

#Bindthesockettotheport
self.server_address=server_address
print('startingupon%sport%s'%self.server_address)
self.server.bind(self.server_address)

#Listenforincomingconnections
self.server.listen(5)

#Socketsfromwhichweexpecttoread
self.inputs=inputs

#Socketstowhichweexpecttowrite
#处理要发送的消息
self.outputs=outputs
#Outgoingmessagequeues(socket:Queue)
self.message_queues=message_queues

defhandler_recever(self,readable):
#Handleinputs
#循环判断是否有客户端连接进来,当有客户端连接进来时select将触发
forsinreadable:
#判断当前触发的是不是服务端对象,当触发的对象是服务端对象时,说明有新客户端连接进来了
#表示有新用户来连接
ifsisself.server:
#A"readable"socketisreadytoacceptaconnection
connection,client_address=s.accept()
self.client_address=client_address
print('connectionfrom',client_address)
#thisisconnectionnotserver
connection.setblocking(0)
#将客户端对象也加入到监听的列表中,当客户端发送消息时select将触发
self.inputs.append(connection)

#Givetheconnectionaqueuefordatawewanttosend
#为连接的客户端单独创建一个消息队列，用来保存客户端发送的消息
self.message_queues[connection]=queue.Queue()
else:
#有老用户发消息,处理接受
#由于客户端连接进来时服务端接收客户端连接请求，将客户端加入到了监听列表中(input_list),客户端发送消息将触发
#所以判断是否是客户端对象触发
data=s.recv(1024)
#客户端未断开
ifdata!=b'':
#Areadableclientsockethasdata
print('received"%s"from%s'%(data,s.getpeername()))
#将收到的消息放入到相对应的socket客户端的消息队列中
self.message_queues[s].put(data)
#Addoutputchannelforresponse
#将需要进行回复操作socket放到output列表中,让select监听
ifsnotinself.outputs:
self.outputs.append(s)
else:
#客户端断开了连接,将客户端的监听从input列表中移除
#Interpretemptyresultasclosedconnection
print('closing',s.getpeername())#获取客户端的socket信息
#Stoplisteningforinputontheconnection
ifsinself.outputs:
self.outputs.remove(s)
self.inputs.remove(s)
s.close()

#Removemessagequeue
#移除对应socket客户端对象的消息队列
delself.message_queues[s]
return"gotit"

defhandler_send(self,writable):
#Handleoutputs
#如果现在没有客户端请求,也没有客户端发送消息时,开始对发送消息列表进行处理,是否需要发送消息
#存储哪个客户端发送过消息
forsinwritable:
try:
#如果消息队列中有消息,从消息队列中获取要发送的消息
message_queue=self.message_queues.get(s)
send_data=''
ifmessage_queueisnotNone:
send_data=message_queue.get_nowait()
exceptqueue.Empty:
#客户端连接断开了
self.outputs.remove(s)
else:
#print"sending%sto%s"%(send_data,s.getpeername)
#print"sendsomething"
ifmessage_queueisnotNone:
s.send(send_data)
else:
print("clienthasclosed")
#delmessage_queues[s]
#writable.remove(s)
#print"Client%sdisconnected"%(client_address)
return"gotit"

defhandler_exception(self,exceptional):
##Handle"exceptionalconditions"
#处理异常的情况
forsinexceptional:
print('exceptionconditionon',s.getpeername())
#Stoplisteningforinputontheconnection
self.inputs.remove(s)
ifsinself.outputs:
self.outputs.remove(s)
s.close()

#Removemessagequeue
delself.message_queues[s]
return"gotit"

defevent_loop(tcpserver,inputs,outputs):
whileinputs:
#Waitforatleastoneofthesocketstobereadyforprocessing
print('waitingforthenextevent')
#开始select监听,对input_list中的服务器端server进行监听
#当socket调用send,recv等函数时,就会再次调用此函数,这时返回的第二个参数就会有值
readable,writable,exceptional=select.select(inputs,outputs,inputs)
ifreadableisnotNone:
tcp_recever=tcpserver.handler_recever(readable)
iftcp_recever=='gotit':
print("serverhavereceived")
ifwritableisnotNone:
tcp_send=tcpserver.handler_send(writable)
iftcp_send=='gotit':
print("serverhavesend")
ifexceptionalisnotNone:
tcp_exception=tcpserver.handler_exception(exceptional)
iftcp_exception=='gotit':
print("serverhaveexception")

sleep(0.8)

if__name__=='__main__':
server_address=('localhost',8090)
server=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
inputs=[server]
outputs=[]
message_queues={}
tcpserver=TCPServer(server,server_address,inputs,outputs,message_queues)
#开启事件循环
event_loop(tcpserver,inputs,outputs)

importsocket

messages=['Thisisthemessage','Itwillbesent','inparts',]

server_address=('localhost',8090)

#CreateaTCP/IPsocket

socks=[socket.socket(socket.AF_INET,socket.SOCK_STREAM),socket.socket(socket.AF_INET,socket.SOCK_STREAM),]

#Connectthesockettotheportwheretheserverislistening

print('connectingto%sport%s'%server_address)
#连接到服务器
forsinsocks:
s.connect(server_address)

forindex,messageinenumerate(messages):
#Sendmessagesonbothsockets
forsinsocks:
print('%s:sending"%s"'%(s.getsockname(),message+str(index)))
send_data=message+str(index)
s.sendall(bytes(send_data,encoding='utf-8'))
#Readresponsesonbothsockets

forsinsocks:
data=s.recv(1024)
print('%s:received"%s"'%(s.getsockname(),data))
ifdata!="":
print('closingsocket',s.getsockname())
s.close()