webapps/routes/fraud.js

var _ = require('underscore');
var async = require('async');
var Db = require('mongodb').Db;
var Server = require('mongodb').Server;
var Deferred = require('jquery-deferred').Deferred;
var config = require('../config');

// Create deferred for mongo connection
var mongoCxn = Deferred();

// Connect to the mongo database
var db = new Db('frauddetect', new Server(config.fraud.mongo.host, config.fraud.mongo.port));
db.open(function(err, mongoclient){
    console.log('Mongo Database connection opened');
    mongoCxn.resolve();
});

/**
 * Returns a map with keys of the collection names
 * and values of their counts
*/
function getAlertCount(req, res) {
    
    // Ensure mongo connection has been made
    if ( mongoCxn.state() === "pending" ) {
        return mongoCxn.done(getAlertCount.bind(this, req, res));
    }
    
    // Alerts
    var alertNames = ['ccAlerts', 'avgAlerts', 'binAlerts'];

    // Get the counts of every alert collection
    async.map(
        
        alertNames,
        
        function(key,cb) {
            if (req.query.since) {
                console.log('since was specified in getAlertCount');
                db.collection(key).find({ time: { $gt: req.query.since*1 } }).count(cb);
            } else {
                db.collection(key).count(cb);
            }
        },

        function(err, response) {
            // Create object in the form: { ccAlerts: [NUM], avgAlerts: [NUM], binAlerts: [NUM] }
            response = _.object(alertNames, response);
            res.send(response);
        }
        
    );
}

/**
 * Retrieves a recent document from the txStats collection
*/
function getRecentStats(req, res) {
    // Ensure mongo connection has been made
    if ( mongoCxn.state() === "pending" ) {
        return mongoCxn.done(getAlertCount.bind(this, req, res));
    }
    
    // Alerts
    var colName = 'txStats';

    // Get random recent stats
    db.collection(colName).find().sort({"time":-1}).skip(2000).limit(5).toArray(function(err, items) {
        var randIndex = Math.floor(Math.random() * 5);
        res.send(items[randIndex]);
    });
    
}

exports.getAlertCount = getAlertCount;
exports.getRecentStats = getRecentStats;