websockets 和 socketio 的比较

websockets

websockets在 github 中的描述如下，简而言之，websockets是基于asyncio实现websocket。websockets的特点是使用简单并且保证高鲁棒性。

websockets is a library for building WebSocket servers and clients in Python with a focus on correctness and simplicity.

Built on top of asyncio, Python’s standard asynchronous I/O framework, it provides an elegant coroutine-based API.

websockets的优点是：

简单：使用websocket只需要了解msg = await ws.recv()和 await ws.send(msg)；websockets负责管理连接，以便可以专注于应用程序。
健壮：websockets为生产而构建。
质量：websockets经过严格测试。
性能：内存使用是可配置的。用 C 编写的扩展加速了昂贵的操作。它针对 Linux、macOS 和 Windows 进行了预编译，并针对每个系统和 Python 版本进行 package 。

websockets不适用于以下场景：

如果项目使用回调函数而不是协程。websockets创建是为了提供基于协程的 API 来管理 Python 中的 WebSocket 连接。
如果项目需要在 http/websocket 场景下使用，websockets并不是很适用，websockets主要是支持RFC6455规范，而在RFC7692规范中补充了关于websocket的压缩方案。websockets对 HTTP 的支持很少——只足以支撑 HTTP 的健康检查。

集成第三方应用

此外，在阅读websockets的文档中，发现websockets的确是主推其易用性和灵活性，很容易就可以上手构建服务端和客户端，但是它和第三方包的集成度比较低。

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-qMPF4roA-1666584023169)(https://cdn.jsdelivr.net/gh/ColorlessCube/images@main/blog/16347844013451634784401338.png)]

socketio

Socket.IO是一种传输协议，可在客户端（通常是 Web 浏览器）和服务器之间实现基于事件的实时双向通信。客户端和服务器组件的官方实现是用JavaScript编写的，python-socketio是Python实现。此外，Socket.IO还提供其他语言的支持。因此，通常开发场景中，websocket经常为前后端通信或跨平台通信，因此Socket.IO提供了很好的支持。

[外链图片转存中…(img-howPUkE3-1666584023173)]

此外，阅读Socket.IO的文档可以发现，server 端支持很多拓展功能。

namespace 和 room

Namespace：Socket.IO 协议支持多个逻辑连接，全部复用在同一个物理连接上。客户端可以通过在每个连接上指定不同的命名空间来打开多个连接。命名空间由客户端指定为跟在主机名和端口之后的路径名。例如，连接到 http://example.com:8000/chat 将打开到命名空间 /chat 的连接

Room：为了方便服务器向相关客户端组发送事件，应用程序可以将其客户端放入“房间”，然后将消息寻址到这些房间。

namespace和room的概念，使得Socket.IO能够更好地进行消息的分发。客户端能够通过连接到不同的namespace来完成不同主题信息的获取。服务器端可以通过将客户端放进不同的room，将客户端分组，根据分组分发不同的消息。

集成消息队列

使用分布式应用程序时，通常需要从多个进程访问Socket.IO的功能。有两个特定的用例：

一旦后台作业完成，使用工作队列的应用程序（如celery）可能需要向客户端发出事件。执行这项任务最方便的地方是处理这项工作的工作进程。
高度可用的应用程序可能希望使用Socket.IO服务器的水平扩展来处理大量并发客户端。

作为上述问题的解决方案，可以将 Socket.IO 服务器配置为连接到Redis或RabbitMQ等消息队列，与其他相关的Socket.IO服务器或辅助工作者进行通信。而Socket.IO提供了很好的集成方法，以Redis为例。

# socketio.Server class
mgr = socketio.RedisManager('redis://')
sio = socketio.Server(client_manager=mgr)# socketio.AsyncServer class
mgr = socketio.AsyncRedisManager('redis://')
sio = socketio.AsyncServer(client_manager=mgr)

集成 ASGI 服务器

Socket.IO可以与 ASGI 兼容的 Web 服务器部署应用程序，例如 Flask、Gunicorn。

sio = socketio.Server(async_mode='eventlet', cors_allowed_origins='*')
app = Flask(__name__)# 将 socketio 服务器集成到现有的 wsgi 应用中
app.wsgi_app = socketio.WSGIApp(sio, app.wsgi_app)

异步高可用性

Socket.IO支持通过gevent、eventlet来实现异步高可用性，可以通过以下方式设置。如果不采用上述方式，也可以使用标准多线程的方式来支持高并发场景。

sio = socketio.Server(async_mode='gevent')
sio = socketio.Server(async_mode='eventlet')
sio = socketio.Server(async_mode='threading')

总结

通过阅读官方文档，对Python的websockets包和socketio 包进行了简单说明。

可以看出，websockets的特点是易用性，专注于websocket的处理，能够快速搭建服务，并根据具体场景对服务进行改造，使其符合开发的预期。

而socketio则已经做了很多基础性工作，能够很好地与一些主流的技术集成，开发者只需要完成一些简单的配置即可。