node-red/nodes/storage/66-mongodb.js

/**
 * Copyright 2013 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 RED = require("../../red/red");
var mongo = require('mongodb');

function MongoNode(n) {
    RED.nodes.createNode(this,n);
    this.hostname = n.hostname;
    this.port = n.port;
    this.db = n.db;
}
RED.nodes.registerType("mongodb",MongoNode);


function MongoOutNode(n) {
    RED.nodes.createNode(this,n);
    this.collection = n.collection;
    this.mongodb = n.mongodb;
    this.mongoConfig = RED.nodes.getNode(this.mongodb);

    if (this.mongoConfig) {
        var node = this;

        this.clientDb = new mongo.Db(node.mongoConfig.db, new mongo.Server(node.mongoConfig.hostname, node.mongoConfig.port, {}), {w: 1});
        this.clientDb.open(function(err,cli) {
                if (err) { node.error(err); }
                else {
                    node.clientDb.collection(node.collection,function(err,coll) {
                            if (err) { node.error(err); }
                            else {
                                node.on("input",function(msg) {
                                        delete msg._topic;
                                        coll.save(msg,function(err,item){if (err){node.error(err);}});
                                });
                            }
                    });
                }
        });
    } else {
        this.error("missing mongodb configuration");
    }
}

RED.nodes.registerType("mongodb out",MongoOutNode);

MongoOutNode.prototype.close = function() {
    if (this.clientDb) {
        this.clientDb.close();
    }
}