mongodb-output.js

'use strict';

//
// Output plugin that stores logs in MongoDB database.
//

var assert = require('chai').assert;
var MongoClient = require('mongodb').MongoClient;

module.exports = function (conf) {

    console.log('Loading to database: ' + conf.get('db'));
    console.log('Logs to: ' + conf.get('logsCollection'));
    console.log('Errors to: ' + conf.get('errorsCollection'));

    return MongoClient.connect(conf.get('db'))
        .then(client => {
	    var db = client.db('logs');

            var logsCollection = db.collection(conf.get('logsCollection'));
            var errorsCollection = db.collection(conf.get('errorsCollection'));
            
            return {

                //
                // Emit an array of logs to the database.
                //
                emit: function (logs) {

		    if (logs && logs.length > 0) {
                    	logsCollection
				.insertMany(logs)
                        	.then(function () {
                            		//console.log("Added logs to database.");
                        	})
                        	.catch(function (err) {
                            		console.error("!! " + (err && err.stack || err));
                        	});
		    }

                    var errorLogs = logs.filter(function (log) {
                            return log.Level === 'Fatal' || log.Level === 'Error';
                        });
                    if (errorLogs.length > 0) {
                        errorsCollection.insertMany(errorLogs)
                            .then(function () {
                                //console.log("Added errors to database");
                            })
                            .catch(function (err) {
                                console.error("!! " + (err && err.stack || err));
                            });
                    }

                    /*
                    logs.forEach(function (log) {
                        console.log(log.Properties.UserName + " | " + log.RenderedMessage);
                    });
                    */			
                },

                //
                // Retreive logs that have been stored.
                //
                retrieveLogs: function (startDate, endDate) {
                    assert.instanceOf(startDate, Date);
                    assert.instanceOf(endDate, Date);

                    return logsCollection.find({
                            Timestamp: {
                                $gte: startDate,
                                $lte: endDate
                            },
                        })
                        .toArray()
                        ;
                },
            };
        });
};