blob: 9f18b1820c86288bcecac76d8b28b05bfce15bd4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/**
* Copyright (c) 2011 10gen Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License, version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "pch.h"
#include "accumulator.h"
#include "db/pipeline/document.h"
#include "db/pipeline/expression_context.h"
#include "db/pipeline/value.h"
namespace mongo {
const char AccumulatorAvg::subTotalName[] = "subTotal";
const char AccumulatorAvg::countName[] = "count";
intrusive_ptr<const Value> AccumulatorAvg::evaluate(
const intrusive_ptr<Document> &pDocument) const {
if (!pCtx->getInRouter()) {
Super::evaluate(pDocument);
++count;
}
else {
/*
If we're in the router, we expect an object that contains
both a subtotal and a count. This is what getValue() produced
below.
*/
intrusive_ptr<const Value> prhs(
vpOperand[0]->evaluate(pDocument));
assert(prhs->getType() == Object);
intrusive_ptr<Document> pShardDoc(prhs->getDocument());
intrusive_ptr<const Value> pSubTotal(
pShardDoc->getValue(subTotalName));
assert(pSubTotal.get());
BSONType subTotalType = pSubTotal->getType();
if ((totalType == NumberLong) || (subTotalType == NumberLong))
totalType = NumberLong;
if ((totalType == NumberDouble) || (subTotalType == NumberDouble))
totalType = NumberDouble;
if (subTotalType == NumberInt) {
int v = pSubTotal->getInt();
longTotal += v;
doubleTotal += v;
}
else if (subTotalType == NumberLong) {
long long v = pSubTotal->getLong();
longTotal += v;
doubleTotal += v;
}
else {
double v = pSubTotal->getDouble();
doubleTotal += v;
}
intrusive_ptr<const Value> pCount(pShardDoc->getValue(countName));
count += pCount->getLong();
}
return Value::getZero();
}
intrusive_ptr<Accumulator> AccumulatorAvg::create(
const intrusive_ptr<ExpressionContext> &pCtx) {
intrusive_ptr<AccumulatorAvg> pA(new AccumulatorAvg(pCtx));
return pA;
}
intrusive_ptr<const Value> AccumulatorAvg::getValue() const {
if (!pCtx->getInShard()) {
double avg = 0;
if (count) {
if (totalType != NumberDouble)
avg = static_cast<double>(longTotal / count);
else
avg = doubleTotal / count;
}
return Value::createDouble(avg);
}
intrusive_ptr<Document> pDocument(Document::create());
intrusive_ptr<const Value> pSubTotal;
if (totalType == NumberInt)
pSubTotal = Value::createInt((int)longTotal);
else if (totalType == NumberLong)
pSubTotal = Value::createLong(longTotal);
else
pSubTotal = Value::createDouble(doubleTotal);
pDocument->addField(subTotalName, pSubTotal);
intrusive_ptr<const Value> pCount(Value::createLong(count));
pDocument->addField(countName, pCount);
return Value::createDocument(pDocument);
}
AccumulatorAvg::AccumulatorAvg(
const intrusive_ptr<ExpressionContext> &pTheCtx):
AccumulatorSum(),
count(0),
pCtx(pTheCtx) {
}
const char *AccumulatorAvg::getOpName() const {
return "$avg";
}
}
|