server/services/estimated-chats-service.js

   1 'use strict';
   2
   3 var ChatMessage = require('gitter-web-persistence').ChatMessage;
   4 var _ = require('lodash');
   5 var Promise = require('bluebird');
   6 var mongoReadPrefs = require('gitter-web-persistence-utils/lib/mongo-read-prefs');
   7
   8 function getEstimatedMessageCountForRoomId(roomId) {
   9   return ChatMessage.count({ toTroupeId: roomId })
  10     .read(mongoReadPrefs.secondaryPreferred)
  11     .exec();
  12 }
  13
  14 function getEstimatedMessageCountForRoomIds(roomIds) {
  15   if (!roomIds || !roomIds.length) return {};
  16   if (roomIds.length === 1) {
  17     var singleId = roomIds[0];
  18     return getEstimatedMessageCountForRoomId(singleId).then(function(count) {
  19       var hash = {};
  20       hash[singleId] = count;
  21       return hash;
  22     });
  23   }
  24
  25   return ChatMessage.aggregate([
  26     {
  27       $match: { toTroupeId: { $in: roomIds } }
  28     },
  29     {
  30       $group: {
  31         _id: '$toTroupeId',
  32         count: { $sum: 1 }
  33       }
  34     }
  35   ])
  36     .read(mongoReadPrefs.secondaryPreferred)
  37     .exec()
  38     .then(function(results) {
  39       if (!results || !results.length) return {};
  40       return _.reduce(
  41         results,
  42         function(memo, result) {
  43           memo[result._id] = result.count;
  44           return memo;
  45         },
  46         {}
  47       );
  48     });
  49 }
  50
  51 module.exports = {
  52   getEstimatedMessageCountForRoomId: Promise.method(getEstimatedMessageCountForRoomId),
  53   getEstimatedMessageCountForRoomIds: Promise.method(getEstimatedMessageCountForRoomIds)
  54 };