基于 Redis的Pub/Sub 实现 Websocket 推送

背景

微信小程序的生态越来越完善，而在技术上，小程序目前只支持两种通信协议：HTTPS 和 WebSocket，所以在需要使用双工通信的时候，除了 WebSocket 也没有别的选择。最近恰好有个这样的需求，所以我也花了点时间研究了一下。

项目上实现的目标就是小程序和服务器建立 WebSocket 建立连接，在服务端收到来自于第三方的事件推送之后，主动推送给客户端而不是靠客户端轮询来获取消息（这里就不介绍 WebSocket 的基础知识了）。因为我们项目组成员大多都是 PHP 开发，所以也是考虑用 PHP 来实现。

实现

这里会遇到的问题就是，用 PHP 的库来开一个 WebSocket 服务端口的时候，由于要保持连接，并接收的后续连接，所以服务本身是处于监听端口的状态。而如果程序同时要订阅来自 Redis 的事件，同样也需要监听 Redis 的消息。那么要如何实现呢？这里先直接抛出我所使用的两个库：

Ratchet：一个 PHP 实现的异步 WebSocket 服务器

predis-async: PHP 实现的异步 Redis 客户端

仔细看上面的描述，除了 PHP 实现 外，他们还有一个相同的关键词：异步。没错，这里的异步和 node.js 描述的异步差不多是同一回事。实现异步的基础就是：EventLoop。这里我也不具体描述 EventLoop 到底是怎么一回事儿。其实 Ratchet 提供的 examples 里也有一个借用 zeromq 实现 push的例子：

上面的例子中可以看出来，实现主动推送的核心点也是在于共享了同一个 loop 实例。

实现上和上面的代码示例类似：

通过上面的实现，就可以监听 Pub 到 pub.* 的消息并主动推送给通过 WebSocket 连接到后端的客户端了。

其他 Ratchet 的 WebSocket hander 一旦被实例化，在所有新进入的连接中都是共享的，所以一定要处理好各个连接之间的身份认证、数据隔离等关系。新连接的认证可以在 onOpen 方法被调用时处理。

由于小程序也不支持 Cookie，我推荐使用 JWT 做身份认证。虽然 JWT 本身并不完美，但是一个不错的解决问题的方式。

本文只是提供一种 PHP 实现的思路，实际并没有经过大量连接的考验。同样也有很多其他的思路来解决这个问题，比如基于 openresty 的信号量或者 golang 的线程机制来实现。

文章来源网络，如有侵权请联系小编