blob: e20e2b6a6aed4ead71a0e6065d0ad3a8c4b3c247 (
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
|
// @file querypattern.cpp - Query pattern matching for selecting similar plans given similar queries.
/* Copyright 2011 10gen Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "querypattern.h"
namespace mongo {
QueryPattern::QueryPattern( const FieldRangeSet &frs, const BSONObj &sort ) {
for( map<string,FieldRange>::const_iterator i = frs.ranges().begin(); i != frs.ranges().end(); ++i ) {
if ( i->second.equality() ) {
_fieldTypes[ i->first ] = QueryPattern::Equality;
}
else if ( i->second.empty() ) {
// This case generally results from an upper and lower bound that are inconsistent for a single key index.
_fieldTypes[ i->first ] = QueryPattern::UpperAndLowerBound;
}
else if ( i->second.nontrivial() ) {
bool upper = i->second.max().type() != MaxKey;
bool lower = i->second.min().type() != MinKey;
if ( upper && lower )
_fieldTypes[ i->first ] = QueryPattern::UpperAndLowerBound;
else if ( upper )
_fieldTypes[ i->first ] = QueryPattern::UpperBound;
else if ( lower )
_fieldTypes[ i->first ] = QueryPattern::LowerBound;
}
}
setSort( sort );
}
/** for testing only - speed unimportant */
bool QueryPattern::operator==( const QueryPattern &other ) const {
bool less = operator<( other );
bool more = other.operator<( *this );
assert( !( less && more ) );
return !( less || more );
}
/** for testing only - speed unimportant */
bool QueryPattern::operator!=( const QueryPattern &other ) const {
return !operator==( other );
}
string typeToString( enum QueryPattern::Type t ) {
switch (t) {
case QueryPattern::Equality:
return "Equality";
case QueryPattern::LowerBound:
return "LowerBound";
case QueryPattern::UpperBound:
return "UpperBound";
case QueryPattern::UpperAndLowerBound:
return "UpperAndLowerBound";
}
return "";
}
string QueryPattern::toString() const {
BSONObjBuilder b;
for( map<string,Type>::const_iterator i = _fieldTypes.begin(); i != _fieldTypes.end(); ++i ) {
b << i->first << typeToString( i->second );
}
return BSON( "query" << b.done() << "sort" << _sort ).toString();
}
void QueryPattern::setSort( const BSONObj sort ) {
_sort = normalizeSort( sort );
}
BSONObj QueryPattern::normalizeSort( const BSONObj &spec ) {
if ( spec.isEmpty() )
return spec;
int direction = ( spec.firstElement().number() >= 0 ) ? 1 : -1;
BSONObjIterator i( spec );
BSONObjBuilder b;
while( i.moreWithEOO() ) {
BSONElement e = i.next();
if ( e.eoo() )
break;
b.append( e.fieldName(), direction * ( ( e.number() >= 0 ) ? -1 : 1 ) );
}
return b.obj();
}
} // namespace mongo
|