107 lines
2.4 KiB
C++
107 lines
2.4 KiB
C++
// dbgrid/request.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/>.
|
|
*/
|
|
|
|
/* TODO
|
|
_ concurrency control.
|
|
_ connection pool
|
|
_ hostbyname_nonreentrant() problem
|
|
_ gridconfig object which gets config from the grid db.
|
|
connect to iad-sb-grid
|
|
_ limit() works right?
|
|
_ KillCursors
|
|
|
|
later
|
|
_ secondary indexes
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include "../grid/message.h"
|
|
#include "../db/dbmessage.h"
|
|
#include "connpool.h"
|
|
|
|
const char *tempHost = "localhost:27018";
|
|
|
|
void getMore(Message& m, MessagingPort& p) {
|
|
DbMessage d(m);
|
|
const char *ns = d.getns();
|
|
|
|
cout << "TEMP: getmore: " << ns << endl;
|
|
|
|
ScopedDbConnection dbcon(tempHost);
|
|
DBClientConnection &c = dbcon.conn();
|
|
|
|
Message response;
|
|
bool ok = c.port().call(m, response);
|
|
uassert("dbgrid: getmore: error calling db", ok);
|
|
p.reply(m, response, m.data->id);
|
|
|
|
dbcon.done();
|
|
}
|
|
|
|
void queryOp(Message& m, MessagingPort& p) {
|
|
DbMessage d(m);
|
|
const char *ns = d.getns();
|
|
|
|
cout << "TEMPns: " << ns << endl;
|
|
|
|
|
|
|
|
ScopedDbConnection dbcon(tempHost);
|
|
DBClientConnection &c = dbcon.conn();
|
|
|
|
Message response;
|
|
bool ok = c.port().call(m, response);
|
|
uassert("dbgrid: error calling db", ok);
|
|
p.reply(m, response, m.data->id);
|
|
|
|
dbcon.done();
|
|
}
|
|
|
|
void writeOp(int op, Message& m, MessagingPort& p) {
|
|
DbMessage d(m);
|
|
const char *ns = d.getns();
|
|
|
|
ScopedDbConnection dbcon(tempHost);
|
|
DBClientConnection &c = dbcon.conn();
|
|
|
|
c.port().say(m);
|
|
|
|
dbcon.done();
|
|
/*
|
|
while( d.moreJSObjs() ) {
|
|
BSONObj js = d.nextJsObj();
|
|
const char *ns = d.getns();
|
|
assert(*ns);
|
|
}
|
|
*/
|
|
}
|
|
|
|
void processRequest(Message& m, MessagingPort& p) {
|
|
int op = m.data->operation();
|
|
assert( op > dbMsg );
|
|
if( op == dbQuery ) {
|
|
queryOp(m,p);
|
|
}
|
|
else if( op == dbGetMore ) {
|
|
getMore(m,p);
|
|
}
|
|
else {
|
|
writeOp(op, m, p);
|
|
}
|
|
}
|