python socket epoll_python epoll方式tcp连接回发消息

#-*- coding:utf-8 -*-

importsocketimportselectclasstestserver():def __init__(self):

self.serversocket=socket.socket(socket.AF_INET, socket.SOCK_STREAM)

self.serversocket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR,1)

self.serversocket.bind(('0.0.0.0', 8889))

self.serversocket.listen(128)

self.serversocket.setblocking(0)defrun(self):

response= "connecting status: 200"epoll=select.epoll()

epoll.register(self.serversocket.fileno(), select.EPOLLIN)try:

connections={}

requests={}

responses={}

endflag= '\n\r\n'

whileTrue:

events= epoll.poll(1)for fid, event inevents:if fid ==self.serversocket.fileno():

connection, address=self.serversocket.accept()

connection.setblocking(0)

epoll.register(connection.fileno(), select.EPOLLIN)

connections[connection.fileno()]=connection

requests[connection.fileno()]= ''responses[connection.fileno()]=response.encode()elif event &select.EPOLLIN:try:

requests[fid]= connections[fid].recv(1024)if len(str(requests[fid].decode())) ==0:

connections[fid].shutdown(socket.SHUT_RDWR)break

else:print("2 | ------ :" + str(requests[fid].decode()) + "\n")

byteswritten=connections[fid].send(responses[fid])if endflag inrequests[fid]:

epoll.modify(fid, select.EPOLLOUT)

connections[fid].setsockopt(socket.IPPROTO_TCP, socket.TCP_CORK,1)print('-' * 40 + '\n' + requests[fid].decode()[:-2])except:continue

elif event &select.EPOLLOUT:

byteswritten=connections[fid].send(responses[fid])

responses[fid]=responses[fid][byteswritten:]if len(responses[fid]) ==0:

connections[fid].setsockopt(socket.IPPROTO_TCP, socket.TCP_CORK, 0)

epoll.modify(fid, 0)

connections[fid].shutdown(socket.SHUT_RDWR)elif event &select.EPOLLHUP:

epoll.unregister(fid)

connections[fid].close()delconnections[fid]except:print("server excepted ...")

epoll.unregister(self.serversocket.fileno())

self.run()finally:print("server closed ...")if __name__ == "__main__":

serv=testserver()

serv.run()