-
Notifications
You must be signed in to change notification settings - Fork 0
/
sse.js
49 lines (44 loc) · 1.49 KB
/
sse.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
module.exports = (function() {
const connections = [];
return {
middleware: function(req, res, next) {
res.sseSetup = function() {
res.writeHead(200, {
'Content-Type': 'text/event-stream',
'Cache-Control': 'no-cache',
'Connection': 'keep-alive'
})
}
res.sseSend = function(data) {
res.write("data: " + JSON.stringify(data) + "\r\n\r\n");
res.flush();
}
next();
},
setupConnection: function(req, res, data) {
res.sseSetup();
res.sseSend(data);
connections.push(res);
req.on("close", function() {
// request closed unexpectedly
if (connections.indexOf(res) >= 0) {
connections.splice(connections.indexOf(res), 1);
}
});
req.on("end", function() {
if (connections.indexOf(res) >= 0) {
connections.splice(connections.indexOf(res), 1);
}
});
// prevent err_incomplete_chunked_encoding
setInterval(function() {
res.sseSend({ action: "keep-alive" });
}, 60000)
},
updateConnections: function(data) {
for (let i = 0; i < connections.length; i += 1) {
connections[i].sseSend(data);
}
}
}
})();