用select模块实现的socket server

5290589

import logging　　import queue
　　import select
　　import socket
　　class Server(object):
　　def __init__(self, host, port, data_size=1024):
　　self.host = host
　　self.port = port
　　self.data_size = data_size
　　self.server = socket.socket()
　　self.server.setblocking(False)
　　self.server.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
　　self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
　　self.server.bind((self.host, self.port))
　　self.server.listen()
　　logging.critical("监听已经开启")
　　self.inputs = [self.server, ]
　　self.outputs = []
　　self.data_queue = {}
　　def run(self):
　　while True:
　　self.loop()
　　def loop(self):
　　"""调用一次select并处理
　　run方法中就是循环调用执行此方法
　　可以重构此方法：
　　调用执行父类中的这个方法后，加入一次循环中需要执行的其他步骤
　　"""
　　readable, writeable, exceptional = select.select(self.inputs, self.outputs, self.inputs)
　　logging.debug("select返回: {readable:%s, writeable:%s, exceptional:%s}"
　　% (readable, writeable, exceptional))
　　for r in readable:
　　if r is self.server:
　　conn, addr = r.accept()
　　logging.info("接收到新的客户端连接: {conn: %s, addr: %s}" % (conn, addr))
　　conn.setblocking(False)
　　self.inputs.append(conn)
　　self.data_queue[conn] = queue.Queue()
　　else:
　　try:
　　data = r.recv(self.data_size)
　　except ConnectionResetError as e:
　　logging.error("recv时捕获到异常: %s" % e)
　　self.clean(r)
　　if r in writeable: writeable.remove(r)
　　except Exception as e:
　　logging.critical("recv时捕获未知异常: %s" % e)
　　self.clean(r)
　　else:
　　if data:  # 如果有收到数据
　　logging.debug(data)
　　self.onrecv(r, data)
　　else:  # 可能是收到了空，就是客户端断开了
　　logging.info("客户端已断开: %s" % r)
　　self.clean(r)
　　# 只在writeable之前才从writeable列表中清除
　　if r in writeable: writeable.remove(r)
　　for w in writeable:
　　try:
　　data = self.data_queue[w].get_nowait()
　　except KeyError as e:  # 客户端突然断开，这个连接可能会同时出现在读和写列表中，而读中已经将它删掉了
　　logging.error("获取消息队列时捕获到异常: %s" % e)
　　# self.clean(w)  # 这里就是被删了才出的异常
　　except queue.Empty:  # 如果队列空了才说明消息都发完了，从读列表中remove
　　# if w in self.outputs: self.outputs.remove(w)
　　self.outputs.remove(w)  # 这里应该不用判断，一定在列表里。因为如果不在会包KeyError
　　except Exception as e:
　　logging.critical("获取消息队列时捕获未知异常: %s" % e)
　　self.clean(w)
　　else:
　　if data:
　　try:
　　w.sendall(data)  # 可能一次发不完，所以send之后不从读列表中remove。下次会发现队列为空
　　except ConnectionResetError as e:
　　logging.error("send时捕获到异常: %s" % e)
　　except Exception as e:
　　logging.critical("send时捕获未知异常: %s" % e)
　　for e in exceptional:
　　logging.critical("异常列表有返回: %s" % e)
　　self.clean(e)
　　def clean(self, conn):
　　"""清理客户端连接信息
　　连接断开时处理以下4步
　　1. 从读列表中去掉，不再去监听这个连接
　　2. 从写列表中去掉，如果还有没发出的消息，那么也不再发了
　　3. 关闭这个连接
　　4. 从消息队列字典中中删除这个队列，可能还有未发送的消息
　　可以重构此方法：
　　如果还有其他在子类中定义的列表或字典需要清理，
　　那么重构此方法，调用执行父类的次方法后，添加自定义的内容
　　"""
　　if conn in self.inputs: self.inputs.remove(conn)
　　if conn in self.outputs: self.outputs.remove(conn)
　　conn.close()
　　if conn in self.data_queue: del self.data_queue[conn]
　　def onrecv(self, conn, data):
　　"""收到消息后调用执行此方法
　　需要重构此方法
　　否则是调用send_data方法，原样发回
　　"""
　　self.send_data(conn, data)
　　def send_data(self, conn, data):
　　"""发送数据"""
　　if conn not in self.outputs: self.outputs.append(conn)
　　self.data_queue[conn].put(data)
　　if __name__ == '__main__':
　　Server('localhost', 9999).run()