Add message router component

2025-03-01 10:36:34 +00:00 · 2016-10-29 21:46:19 +01:00
parent 1fd87bf664
commit cebddc0237
6 changed files with 223 additions and 128 deletions
--- a/red/runtime/nodes/router/index.js
+++ b/red/runtime/nodes/router/index.js
@@ -0,0 +1,112 @@
+/**
+ * Copyright 2016 IBM Corp.
+ *
+ * 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 util = require("util");
+
+var flows = require("../flows");
+var redUtil = require("../../util");
+
+var runtime;
+
+var routes = {};
+var sendQueue = [];
+
+
+
+function init(_runtime) {
+    runtime = _runtime;
+    wires = {};
+}
+
+function add(sourceNode, wires) {
+    routes[sourceNode.id] = wires;
+}
+function remove(sourceNode) {
+    delete routes[sourceNode.id];
+}
+function send(sourceNode, msg) {
+    if (msg === null || typeof msg === "undefined") {
+        return;
+    } else if (!util.isArray(msg)) {
+        msg = [msg];
+    }
+    var node;
+    var msgSent = false;
+    var nodeWires = routes[sourceNode.id];
+    if (nodeWires) {
+        var numOutputs = nodeWires.length;
+        var sendEvents = [];
+        var sentMessageId = null;
+        // for each output of node eg. [msgs to output 0, msgs to output 1, ...]
+        for (var i = 0; i < numOutputs; i++) {
+            var wires = nodeWires[i];
+            /* istanbul ignore else */
+            if (i < msg.length) {
+                var msgs = msg[i]; // msgs going to output i
+                if (msgs !== null && typeof msgs !== "undefined") {
+                    if (!util.isArray(msgs)) {
+                        msgs = [msgs];
+                    }
+                    var k = 0;
+                    // for each recipent node of that output
+                    for (var j = 0; j < wires.length; j++) {
+                        node = flows.get(wires[j]); // node at end of wire j
+                        if (node) {
+                            // for each msg to send eg. [[m1, m2, ...], ...]
+                            for (k = 0; k < msgs.length; k++) {
+                                var m = msgs[k];
+                                if (m !== null && m !== undefined) {
+                                    /* istanbul ignore else */
+                                    if (!sentMessageId) {
+                                        sentMessageId = m._msgid;
+                                    }
+                                    if (msgSent) {
+                                        var clonedmsg = redUtil.cloneMessage(m);
+                                        sendEvents.push({n:node,m:clonedmsg});
+                                    } else {
+                                        sendEvents.push({n:node,m:m});
+                                        msgSent = true;
+                                    }
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        /* istanbul ignore else */
+        if (!sentMessageId) {
+            sentMessageId = redUtil.generateId();
+        }
+        sourceNode.metric("send",{_msgid:sentMessageId});
+
+        for (i=0;i<sendEvents.length;i++) {
+            var ev = sendEvents[i];
+            /* istanbul ignore else */
+            if (!ev.m._msgid) {
+                ev.m._msgid = sentMessageId;
+            }
+            ev.n.receive(ev.m);
+        }
+    }
+}
+
+module.exports = {
+    init:init,
+    add:add,
+    remove: remove,
+    send:send
+}