long long runCount( const string& ns, const BSONObj &cmd, string &err, int &errCode ) {
// Lock 'ns'.
Client::Context cx(ns);
Collection* collection = cx.db()->getCollection(ns);
if (NULL == collection) {
err = "ns missing";
return -1;
}
BSONObj query = cmd.getObjectField("query");
BSONObj hintObj;
if (Object == cmd["hint"].type()) {
hintObj = cmd["hint"].Obj();
}
else if (String == cmd["hint"].type()) {
const std::string hint = cmd.getStringField("hint");
hintObj = BSON("$hint" << hint);
}
// count of all objects
if (query.isEmpty()) {
return applySkipLimit(collection->numRecords(), cmd);
}
Runner* rawRunner;
long long skip = cmd["skip"].numberLong();
long long limit = cmd["limit"].numberLong();
if (limit < 0) {
limit = -limit;
}
uassertStatusOK(getRunnerCount(collection, query, hintObj, &rawRunner));
auto_ptr<Runner> runner(rawRunner);
// Get a pointer to the current operation. We will try to copy the planSummary
// there so that it appears in db.currentOp() and the slow query log.
Client& client = cc();
CurOp* currentOp = client.curop();
// Have we copied the planSummary to 'currentOp' yet?
bool gotPlanSummary = false;
// Try to copy the plan summary to the 'currentOp'.
if (!gotPlanSummary) {
gotPlanSummary = setPlanSummary(runner.get(), currentOp);
}
try {
const ScopedRunnerRegistration safety(runner.get());
runner->setYieldPolicy(Runner::YIELD_AUTO);
long long count = 0;
Runner::RunnerState state;
while (Runner::RUNNER_ADVANCED == (state = runner->getNext(NULL, NULL))) {
// Try to copy the plan summary to the 'currentOp'. We need to try again
// here because we might not have chosen a plan until after the first
// call to getNext(...).
if (!gotPlanSummary) {
gotPlanSummary = setPlanSummary(runner.get(), currentOp);
}
if (skip > 0) {
--skip;
}
else {
++count;
// Fast-path. There's no point in iterating all over the runner if limit
// is set.
if (count >= limit && limit != 0) {
break;
}
}
}
// Try to copy the plan summary to the 'currentOp', if we haven't already. This
// could happen if, for example, the count is 0.
if (!gotPlanSummary) {
gotPlanSummary = setPlanSummary(runner.get(), currentOp);
}
// Emulate old behavior and return the count even if the runner was killed. This
// happens when the underlying collection is dropped.
return count;
}
catch (const DBException &e) {
err = e.toString();
errCode = e.getCode();
}
catch (const std::exception &e) {
err = e.what();
errCode = 0;
}
// Historically we have returned zero in many count assertion cases - see SERVER-2291.
log() << "Count with ns: " << ns << " and query: " << query
<< " failed with exception: " << err << " code: " << errCode
<< endl;
return -2;
}
long long runCount( const string& ns, const BSONObj &cmd, string &err, int &errCode ) {
// Lock 'ns'.
Client::Context cx(ns);
Collection* collection = cx.db()->getCollection(ns);
if (NULL == collection) {
err = "ns missing";
return -1;
}
BSONObj query = cmd.getObjectField("query");
const std::string hint = cmd.getStringField("hint");
const BSONObj hintObj = hint.empty() ? BSONObj() : BSON("$hint" << hint);
// count of all objects
if (query.isEmpty()) {
return applySkipLimit(collection->numRecords(), cmd);
}
Runner* rawRunner;
long long skip = cmd["skip"].numberLong();
long long limit = cmd["limit"].numberLong();
if (limit < 0) {
limit = -limit;
}
uassertStatusOK(getRunnerCount(collection, query, hintObj, &rawRunner));
auto_ptr<Runner> runner(rawRunner);
try {
const ScopedRunnerRegistration safety(runner.get());
runner->setYieldPolicy(Runner::YIELD_AUTO);
long long count = 0;
Runner::RunnerState state;
while (Runner::RUNNER_ADVANCED == (state = runner->getNext(NULL, NULL))) {
if (skip > 0) {
--skip;
}
else {
++count;
// Fast-path. There's no point in iterating all over the runner if limit
// is set.
if (count >= limit && limit != 0) {
break;
}
}
}
// Emulate old behavior and return the count even if the runner was killed. This
// happens when the underlying collection is dropped.
return count;
}
catch (const DBException &e) {
err = e.toString();
errCode = e.getCode();
}
catch (const std::exception &e) {
err = e.what();
errCode = 0;
}
// Historically we have returned zero in many count assertion cases - see SERVER-2291.
log() << "Count with ns: " << ns << " and query: " << query
<< " failed with exception: " << err << " code: " << errCode
<< endl;
return -2;
}
