如何使用mongodb聚合框架比较两个日期之间的分组数据

时间:2015-06-19 20:09:47

标签: javascript mongodb aggregation-framework

考虑如下数据集

Sector Value Date
a      2     1/1/2015
b      5     1/1/2015
c      8     1/1/2015 
a      3     1/1/2015
a      6     1/1/2015
a      1     1/1/2015
c      1     1/1/2015

Sector Value Date
a      2     2/1/2015
b      10    2/1/2015
b      5     2/1/2015
c      8     2/1/2015

有没有办法使用MongoDB聚合框架来查找相关两个日期的每个分组扇区之间的差异

换句话说,可以得到这样的输出:

a    -> (2+3+6) - 2 = 10 
b    -> 5 - (10+5) = -10
c    -> (8+1) - 8 = 1

1 个答案:

答案 0 :(得分:2)

使用聚合框架和$cond运算符更改2015年2月1日的值符号,这是一个简单的总和。

考虑到测试集合:

{ name: 'a',      value: 2,     date: ISODate('2015-01-01') },
{ name: 'b',      value: 5,     date: ISODate('2015-01-01') },
{ name: 'c',      value: 8,     date: ISODate('2015-01-01') },
{ name: 'a',      value: 3,     date: ISODate('2015-01-01') },
{ name: 'a',      value: 6,     date: ISODate('2015-01-01') },
{ name: 'a',      value: 1,     date: ISODate('2015-01-01') },
{ name: 'c',      value: 1,     date: ISODate('2015-01-01') },
{ name: 'a',      value: 2,     date: ISODate('2015-01-02') },
{ name: 'b',      value: 10,    date: ISODate('2015-01-02') },
{ name: 'b',      value: 5,     date: ISODate('2015-01-02') },
{ name: 'c',      value: 8,     date: ISODate('2015-01-02') }

以下是如何达到预期效果:

db.collection.aggregate([
  { $match: { date: { $in: [ ISODate('2015-01-01'), ISODate('2015-01-02') ] }}},
  { $project: { name: 1,
                value: { $cond: [ {$eq: [ "$date", ISODate('2015-01-01')]},
                                  "$value",
                                  { $subtract: [0, "$value"] } ]}}},
  { $group: { _id: "$name", total: { $sum: "$value" }}}
])
  • $match阶段仅保留给定两天的值
  • $project阶段将保持日期等于' 2015-01-01'的文档的值,但会更改值 sign 对于日期等于' 2015-01-02'
  • 的文件
  • 最后,$group阶段将对所有值(正面和负面)求和。

产:

{ "_id" : "c", "total" : 1 }
{ "_id" : "b", "total" : -10 }
{ "_id" : "a", "total" : 10 }