socket服务端
之前笔记里面记录的比较乱，最后我写了一个类，试着封装成一个模块的样子。
使用的时候通过继承生成一个子类，然后调用run执行。
你应该需要重构其中的部分方法，另外可能还需要在子类中创建新的方法。
至少需要重构onrecv方法，接收到数据后的处理。
另外要发数据，调用send_data接口，把conn连接和bytes类型的data传入。
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()测试客户端
下面是测试并发的客户端程序，通过多线程实现并发。
import socket
import threading
HOST = 'localhost'
PORT = 9999
def client(i):
client = socket.socket()
client.connect((HOST, PORT))
for j in range(500):
msg = "hello %s %s" % (i, j)
client.send(msg.encode('utf-8'))
data = client.recv(1024)
print('Received:', data.decode('utf-8'))
client.close()
if __name__ == '__main__':
for i in range(50):
t = threading.Thread(target=client, args=(i,))
t.start()