python中的异步网络框架socketserver _socketserver

简介SocketServer是Python基础库中提供的高级应用库, 主要用来编写异步服务程序. 最近工作原因需要写虚拟的外围服务程序, 正好了解到了这个, 所以写篇文章记录一下使用心得
在python3中SocketServer模块被改名为了socketserver模块
一个小例子从一个小例子开始讲起, 一个小型回显服务器
server.py
#_*_ coding: utf-8 _*_import SocketServerclass FooHandler(SocketServer.BaseRequestHandler):def handle(self):# 'self.request'为客户端socket, 可以调用recv, send方法接收或发送数据data = https://www.isolves.com/it/cxkf/kj/2020-07-18/self.request.recv(30).strip()print('From (%s:%s) msg:%s' % (self.client_address[0], self.client_address[1], data))if __name__ == '__main__':server = SocketServer.TCPServer(('127.0.0.1', 8000), FooHandler)server.serve_forever()【python中的异步网络框架socketserver】运行python server.py后输出如下

文章插图
输出结果
可以看到我们只使用了不到20行的代码便实现了一个小型的异步服务器
需要注意这个程序客户端发起一次数据后客户会话就会关闭
在实际实验中, 可能我们在启动服务程序的时候经常会发生以下异常, 大多数原因是因为服务程序结束后, listen_socket进入TIME_WAIT状态, 导致第二次启动的时候会提示端口已经被占用了

Traceback (most recent call last):File "server.py", line 12, in <module>server = SocketServer.ThreadingTCPServer(('127.0.0.1', 8000), FooHandler)File "/usr/lib/python2.7/SocketServer.py", line 417, in __init__self.server_bind()File "/usr/lib/python2.7/SocketServer.py", line 431, in server_bindself.socket.bind(self.server_address)File "/usr/lib/python2.7/socket.py", line 228, in methreturn getattr(self._sock,name)(*args)socket.error: [Errno 98] Address already in use

了解了异常产生的原因后, 其实就可以找到对应的解决方法了. 将服务listen_socket的SO_REUSEADDR属性打开即可, 由于SocketServer并没有直接将listen_socket暴露出来, 所以可能不方便直接操作listen_socket, 但是官方提供了一种方法, 就是写一个继承SocketServer.TCPServer的子类, 然后将子类的allow_reuse_address设置为True
修改一下我们的程序server.py, 代码如下

#_*_ coding: utf-8 _*_import SocketServerclass FooHandler(SocketServer.BaseRequestHandler):def handle(self):# 'self.request'为客户端socket, 可以调用recv, send方法接收或发送数据data = https://www.isolves.com/it/cxkf/kj/2020-07-18/self.request.recv(30).strip()print('From (%s:%s) msg:%s' % (self.client_address[0], self.client_address[1], data))class SimpleServer(SocketServer.TCPServer):# 开启监听套接字的SO_REUSEADDR属性, 此属性在'SocketServer.TCPServer'中是默认关闭的allow_reuse_address = Trueif __name__ == '__main__':server = SimpleServer(('127.0.0.1', 8000), FooHandler)server.serve_forever()

SocketServer.BaseRequestHandler一共有三个方法可以override

setup 处理前回调, 用于给连接socket设置一些必要的属性, 比如socket.setblocking(0)
handle 处理时回调, 定义服务器与客户端交互的逻辑
finish 处理后回调, 用于定义关闭连接后的清理操作

ThreadingTCPServer之前例子中编写的服务器程序只能用于同时处理一个连接, 当使用多个客户一起连接时会发现后续的客户端会被阻塞住, 如果想要处理多个客户请求时, 需要使用到SocketServer.ThreadingTCPServer
thread_server.py

import SocketServerclass FooHandler(SocketServer.BaseRequestHandler):def handle(self):# 'self.request'为客户端socket, 可以调用recv, send方法接收或发送数据data = https://www.isolves.com/it/cxkf/kj/2020-07-18/self.request.recv(30).strip()print('From (%s:%s) msg:%s' % (self.client_address[0], self.client_address[1], data))if __name__ == '__main__':server = ThreadingTCPServer(('127.0.0.1', 8000), FooHandler)server.serve_forever()

如果喜欢我的文章点赞关注下哦！