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
|
// shardkey.cpp
/**
* Copyright (C) 2008 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 "stdafx.h"
#include "shard.h"
#include "../db/jsobj.h"
#include "../util/unittest.h"
/**
TODO: this only works with numbers right now
this is very temporary, need to make work with anything
*/
namespace mongo {
ShardKey::ShardKey( BSONObj fieldsAndOrder ) : _fieldsAndOrder( fieldsAndOrder ){
if ( _fieldsAndOrder.nFields() > 0 ){
_init();
}
else {
_fieldName = "";
}
}
void ShardKey::init( BSONObj fieldsAndOrder ){
_fieldsAndOrder = fieldsAndOrder.copy();
_init();
}
void ShardKey::_init(){
_fieldName = _fieldsAndOrder.firstElement().fieldName();
uassert( "shard key only supports 1 field right now" , 1 == _fieldsAndOrder.nFields() );
uassert( "shard key has to be a number right now" , _fieldsAndOrder.firstElement().isNumber() );
}
void ShardKey::globalMin( BSONObjBuilder& b ){
uassert( "not valid yet" , _fieldName.size() );
b << _fieldName << (int)(-0xfffffff);
}
void ShardKey::globalMax( BSONObjBuilder& b ){
uassert( "not valid yet" , _fieldName.size() );
b << _fieldName << (int)(0xfffffff);
}
int ShardKey::compare( const BSONObj& lObject , const BSONObj& rObject ) const {
uassert( "not valid yet" , _fieldName.size() );
BSONElement lElement = lObject[ _fieldsAndOrder.firstElement().fieldName() ];
uassert( "left key doesn't have the shard key" , ! lElement.eoo() );
uassert( "left key isn't number" , lElement.isNumber() );
BSONElement rElement = rObject[ _fieldsAndOrder.firstElement().fieldName() ];
uassert( "right key doesn't have the shard key" , ! rElement.eoo() );
uassert( "right key isn't number" , rElement.isNumber() );
if ( lElement.number() < rElement.number() )
return -1;
if ( lElement.number() > rElement.number() )
return 1;
return 0;
}
void ShardKey::middle( BSONObjBuilder & b , BSONObj & lObject , BSONObj & rObject ){
BSONElement lElement = lObject[ _fieldsAndOrder.firstElement().fieldName() ];
uassert( "left key doesn't have the shard key" , ! lElement.eoo() );
uassert( "left key isn't number" , lElement.isNumber() );
BSONElement rElement = rObject[ _fieldsAndOrder.firstElement().fieldName() ];
uassert( "right key doesn't have the shard key" , ! rElement.eoo() );
uassert( "right key isn't number" , rElement.isNumber() );
b.append( _fieldName.c_str() , ( lElement.number() + rElement.number() ) / 2 );
}
string ShardKey::toString() const {
return _fieldsAndOrder.toString();
}
class ShardKeyUnitTest : public UnitTest {
public:
void run(){
ShardKey k( BSON( "key" << 1 ) );
BSONObj min = k.globalMin();
BSONObj max = k.globalMax();
log(3) << "globalMin: " << min << endl;
log(3) << "globalMax: " << max << endl;
assert( k.compare( min , max ) < 0 );
assert( k.compare( max , min ) > 0 );
assert( k.compare( min , min ) == 0 );
}
} shardKeyTest;
} // namespace mongo
|
