1
0
mirror of https://github.com/node-red/node-red.git synced 2023-10-10 13:36:53 +02:00
node-red/red/comms.js

133 lines
3.5 KiB
JavaScript
Raw Normal View History

2014-05-07 21:47:25 +02:00
/**
2015-02-03 23:02:26 +01:00
* Copyright 2014, 2015 IBM Corp.
2014-05-07 21:47:25 +02:00
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
**/
var ws = require("ws");
2015-02-03 23:02:26 +01:00
var log = require("./log");
2014-05-07 21:47:25 +02:00
var server;
var settings;
var wsServer;
var activeConnections = [];
2014-05-08 15:15:54 +02:00
var retained = {};
2014-05-07 21:47:25 +02:00
var heartbeatTimer;
var lastSentTime;
function init(_server,_settings) {
server = _server;
settings = _settings;
}
function start() {
2014-09-22 15:33:26 +02:00
if (!settings.disableEditor) {
var webSocketKeepAliveTime = settings.webSocketKeepAliveTime || 15000;
var path = settings.httpAdminRoot || "/";
path = path + (path.slice(-1) == "/" ? "":"/") + "comms";
wsServer = new ws.Server({server:server,path:path});
wsServer.on('connection',function(ws) {
activeConnections.push(ws);
ws.on('close',function() {
for (var i=0;i<activeConnections.length;i++) {
if (activeConnections[i] === ws) {
activeConnections.splice(i,1);
break;
}
2014-05-07 21:47:25 +02:00
}
2014-09-22 15:33:26 +02:00
});
ws.on('message', function(data,flags) {
var msg = null;
try {
msg = JSON.parse(data);
} catch(err) {
2015-02-03 23:02:26 +01:00
log.warn("comms received malformed message : "+err.toString());
2014-09-22 15:33:26 +02:00
return;
}
if (msg.subscribe) {
handleRemoteSubscription(ws,msg.subscribe);
}
});
ws.on('error', function(err) {
2015-02-03 23:02:26 +01:00
log.warn("comms error : "+err.toString());
2014-09-22 15:33:26 +02:00
});
2014-05-08 15:15:54 +02:00
});
2014-09-22 15:33:26 +02:00
wsServer.on('error', function(err) {
2015-02-03 23:02:26 +01:00
log.warn("comms server error : "+err.toString());
2014-05-07 21:47:25 +02:00
});
2014-09-22 15:33:26 +02:00
lastSentTime = Date.now();
heartbeatTimer = setInterval(function() {
var now = Date.now();
if (now-lastSentTime > webSocketKeepAliveTime) {
publish("hb",lastSentTime);
}
}, webSocketKeepAliveTime);
}
2014-05-07 21:47:25 +02:00
}
function stop() {
2014-09-22 15:33:26 +02:00
if (heartbeatTimer) {
clearInterval(heartbeatTimer);
}
if (wsServer) {
wsServer.close();
}
}
2014-05-08 15:15:54 +02:00
function publish(topic,data,retain) {
if (retain) {
retained[topic] = data;
} else {
delete retained[topic];
2014-05-08 15:15:54 +02:00
}
lastSentTime = Date.now();
2014-05-07 21:47:25 +02:00
activeConnections.forEach(function(conn) {
2014-05-08 15:15:54 +02:00
publishTo(conn,topic,data);
2014-05-07 21:47:25 +02:00
});
}
2014-05-08 15:15:54 +02:00
function publishTo(ws,topic,data) {
var msg = JSON.stringify({topic:topic,data:data});
try {
ws.send(msg);
} catch(err) {
2015-02-03 23:02:26 +01:00
log.warn("comms send error : "+err.toString());
2014-05-08 15:15:54 +02:00
}
}
function handleRemoteSubscription(ws,topic) {
var re = new RegExp("^"+topic.replace(/([\[\]\?\(\)\\\\$\^\*\.|])/g,"\\$1").replace(/\+/g,"[^/]+").replace(/\/#$/,"(\/.*)?")+"$");
for (var t in retained) {
if (re.test(t)) {
publishTo(ws,t,retained[t]);
}
}
}
2014-05-07 21:47:25 +02:00
module.exports = {
init:init,
start:start,
stop:stop,
2014-05-07 21:47:25 +02:00
publish:publish,
}