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
|
// strategy_simple.cpp
#include "stdafx.h"
#include "request.h"
#include "../client/connpool.h"
#include "../db/commands.h"
namespace mongo {
class SingleStrategy : public Strategy {
virtual void queryOp( Request& r ){
QueryMessage q( r.d() );
bool lateAssert = false;
log(3) << "single query: " << q.ns << " " << q.query << endl;
try {
if ( q.ntoreturn == 1 && strstr(q.ns, ".$cmd") ) {
BSONObjBuilder builder;
bool ok = runCommandAgainstRegistered(q.ns, q.query, builder);
if ( ok ) {
BSONObj x = builder.done();
replyToQuery(0, r.p(), r.m(), x);
return;
}
string commandName = q.query.firstElement().fieldName();
if ( commandName == "count" ){
string dbName = q.ns;
dbName = dbName.substr( 0 , dbName.size() - 5 );
string collection = q.query.firstElement().valuestrsafe();
DBConfig * conf = grid.getDBConfig( dbName , false );
if ( conf && conf->isPartitioned() && conf->sharded( dbName + "." + collection ) ){
uassert( "can't handle sharded count yet" , 0 );
}
}
log() << "don't know what i should do with command: " << commandName << " " << q.query << endl;
}
lateAssert = true;
doQuery( r , r.singleServerName() );
}
catch ( AssertionException& e ) {
assert( !lateAssert );
BSONObjBuilder err;
err.append("$err", string("mongos: ") + (e.msg.empty() ? "assertion during query" : e.msg));
BSONObj errObj = err.done();
replyToQuery(QueryResult::ResultFlag_ErrSet, r.p() , r.m() , errObj);
return;
}
}
virtual void getMore( Request& r ){
const char *ns = r.getns();
log(3) << "single getmore: " << ns << endl;
ScopedDbConnection dbcon( r.singleServerName() );
DBClientBase& _c = dbcon.conn();
// TODO
DBClientConnection &c = dynamic_cast<DBClientConnection&>(_c);
Message response;
bool ok = c.port().call( r.m() , response);
uassert("dbgrid: getmore: error calling db", ok);
r.reply( response );
dbcon.done();
}
virtual void writeOp( int op , Request& r ){
const char *ns = r.getns();
log(3) << "single write: " << ns << endl;
doWrite( op , r , r.singleServerName() );
}
};
Strategy * SINGLE = new SingleStrategy();
}
|
