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; }