经过一番研究,我决定使用 Node 原生的events module 来解决这个问题。本例使用ioredis而不是node_redis,但原理是一样的。
首先,我实例化了三个 redis 客户端。一个用于常规数据库工作,一个发布者和一个订阅者
/* redis.js */
const Redis = require('ioredis');
const redis = new Redis();
const publisher = new Redis();
const subscriber = new Redis();
// redis is the defaut export
// publisher and subscriber are "named" exports
const client = (module.exports = redis);
client.publisher = publisher;
client.subscriber = subscriber;
接下来,我们在节点中创建一个EventEmitter,每当订阅者从 redis 中的通道接收到消息时,它将发出一个事件。
/* emitter.js */
const EventEmitter = require('events');
const { subscriber } = require('./redis');
const eventEmitter = new EventEmitter();
subscriber.subscribe('my-channel', err => {
if (err) { return console.log('Unable to subscribe to my-event channel') };
console.log('Subscription to my-event channel successful');
});
subscriber.on('message', (channel, message) => {
eventEmitter.emit('my-event', message);
});
module.exports = eventEmitter;
这里我们有两条路线。第一个处理一个 PUT 请求,该请求在 redis 中设置一个字段,然后使用更新的哈希键将消息发布到通道。第二条路由处理保持打开状态的 GET 请求(例如,作为 SSE 连接的 EventSource)。它监听来自发射器的事件,然后从 redis 发送更新后的 key 的数据
/* route.js*/
const express = require('express');
const redis = require('./redis');
const { publisher } = require('./redis');
const { eventEmitter } = require('./emitter');
const router = express.Router();
router.put('/content', async (req, res) => {
const { key, field, content } = req.body;
try {
await redis.hset(key, field, content);
res.sendStatus(200);
return publisher.publish('my-channel', key);
} catch(err) {
res.status(500).send(err.message);
}
});
router.get('/content-stream', (req, res) => {
res.writeHead(200, {
'Content-Type': 'text/event-stream',
'Cache-Control': 'no-cache',
Connection: 'keep-alive'
});
res.write('\n');
const handleEvent = async key => {
try {
const query = await redis.hgetall(key);
res.write(`data: ${JSON.stringify(query)}\n\n`);
} catch(err) {
console.log('Something went wrong');
}
}
eventEmitter.addListener('my-event', handleEvent);
req.on('close', eventEmitter.removeListener('my-event', handleEvent));
module.exports = router;
这将有效地让您避免在每次连接时都实例化新的 redis 客户端。可能有更好的方法可以做到这一点,但这对我有用。