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

How to compare grouped data between two dates using the mongodb aggregate framework

本文关键字:日期 两个 之间 数据 mongodb 何使用 框架 比较      更新时间:2023-09-26

考虑如下数据集

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

使用聚合框架和 $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"的文档的值不变,但将更改日期等于"2015-01-02"的文档的值符号
  • 最后,$group阶段将对所有值(正值和负值)求和。

生产:

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