+++ /dev/null
-/**
- * Copyright 2013,2014 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.
- **/
-
-module.exports = function(RED) {
- "use strict";
- var mongo = require('mongodb');
- var MongoClient = mongo.MongoClient;
-
- function MongoNode(n) {
- RED.nodes.createNode(this,n);
- this.hostname = n.hostname;
- this.port = n.port;
- this.db = n.db;
- this.name = n.name;
-
- var url = "mongodb://";
- if (this.credentials && this.credentials.user && this.credentials.password) {
- url += this.credentials.user+":"+this.credentials.password+"@";
- }
- url += this.hostname+":"+this.port+"/"+this.db;
-
- this.url = url;
- }
-
- RED.nodes.registerType("mongodb",MongoNode,{
- credentials: {
- user: {type:"text"},
- password: {type: "password"}
- }
- });
-
- function ensureValidSelectorObject(selector) {
- if (selector != null && (typeof selector != 'object' || Buffer.isBuffer(selector))) {
- return {};
- }
- return selector;
- }
-
-
- function MongoOutNode(n) {
- RED.nodes.createNode(this,n);
- this.collection = n.collection;
- this.mongodb = n.mongodb;
- this.payonly = n.payonly || false;
- this.upsert = n.upsert || false;
- this.multi = n.multi || false;
- this.operation = n.operation;
- this.mongoConfig = RED.nodes.getNode(this.mongodb);
-
- if (this.mongoConfig) {
- var node = this;
- MongoClient.connect(this.mongoConfig.url, function(err, db) {
- if (err) {
- node.error(err);
- } else {
- node.clientDb = db;
- var coll;
- if (node.collection) {
- coll = db.collection(node.collection);
- }
- node.on("input",function(msg) {
- if (!coll) {
- if (msg.collection) {
- coll = db.collection(msg.collection);
- } else {
- node.error("No collection defined");
- return;
- }
- }
- delete msg._topic;
- delete msg.collection;
- if (node.operation === "store") {
- if (node.payonly) {
- if (typeof msg.payload !== "object") {
- msg.payload = {"payload": msg.payload};
- }
- coll.save(msg.payload,function(err, item) {
- if (err) {
- node.error(err);
- }
- });
- } else {
- coll.save(msg,function(err, item) {
- if (err) {
- node.error(err);
- }
- });
- }
- } else if (node.operation === "insert") {
- if (node.payonly) {
- if (typeof msg.payload !== "object") {
- msg.payload = {"payload": msg.payload};
- }
- coll.insert(msg.payload, function(err, item) {
- if (err) {
- node.error(err);
- }
- });
- } else {
- coll.insert(msg, function(err,item) {
- if (err) {
- node.error(err);
- }
- });
- }
- } else if (node.operation === "update") {
- if (typeof msg.payload !== "object") {
- msg.payload = {"payload": msg.payload};
- }
- var query = msg.query || {};
- var payload = msg.payload || {};
- var options = {
- upsert: node.upsert,
- multi: node.multi
- };
-
- coll.update(query, payload, options, function(err, item) {
- if (err) {
- node.error(err + " " + payload);
- }
- });
- } else if (node.operation === "delete") {
- coll.remove(msg.payload, function(err, items) {
- if (err) {
- node.error(err);
- }
- });
- }
- });
- }
- });
- } else {
- this.error("missing mongodb configuration");
- }
-
- this.on("close", function() {
- if (this.clientDb) {
- this.clientDb.close();
- }
- });
- }
- RED.nodes.registerType("mongodb out",MongoOutNode);
-
- function MongoInNode(n) {
- RED.nodes.createNode(this,n);
- this.collection = n.collection;
- this.mongodb = n.mongodb;
- this.operation = n.operation || "find";
- this.mongoConfig = RED.nodes.getNode(this.mongodb);
-
- if (this.mongoConfig) {
- var node = this;
- MongoClient.connect(this.mongoConfig.url, function(err,db) {
- if (err) {
- node.error(err);
- } else {
- node.clientDb = db;
- var coll;
- if (node.collection) {
- coll = db.collection(node.collection);
- }
- node.on("input", function(msg) {
- if (!coll) {
- if (msg.collection) {
- coll = db.collection(msg.collection);
- } else {
- node.error("No collection defined");
- return;
- }
- }
- if (node.operation === "find") {
- msg.projection = msg.projection || {};
- var selector = ensureValidSelectorObject(msg.payload);
- coll.find(selector,msg.projection).sort(msg.sort).limit(msg.limit).toArray(function(err, items) {
- if (err) {
- node.error(err);
- } else {
- msg.payload = items;
- delete msg.projection;
- delete msg.sort;
- delete msg.limit;
- node.send(msg);
- }
- });
- } else if (node.operation === "count") {
- var selector = ensureValidSelectorObject(msg.payload);
- coll.count(selector, function(err, count) {
- if (err) {
- node.error(err);
- } else {
- msg.payload = count;
- node.send(msg);
- }
- });
- } else if (node.operation === "aggregate") {
- msg.payload = (msg.payload instanceof Array) ? msg.payload : [];
- coll.aggregate(msg.payload, function(err, result) {
- if (err) {
- node.error(err);
- } else {
- msg.payload = result;
- node.send(msg);
- }
- });
- }
- });
- }
- });
- } else {
- this.error("missing mongodb configuration");
- }
-
- this.on("close", function() {
- if (this.clientDb) {
- this.clientDb.close();
- }
- });
- }
- RED.nodes.registerType("mongodb in",MongoInNode);
-}
Code Review / sdnc / oam.git / blobdiff