long long runCount(OperationContext* txn,
const string& ns,
const BSONObj &cmd,
string &err,
int &errCode) {
AutoGetCollectionForRead ctx(txn, ns);
Collection* collection = ctx.getCollection();
if (NULL == collection) {
err = "ns missing";
return -1;
}
const NamespaceString nss(ns);
CountRequest request;
CmdCount* countComm = static_cast<CmdCount*>(Command::findCommand("count"));
Status parseStatus = countComm->parseRequest(nss.db().toString(), cmd, &request);
if (!parseStatus.isOK()) {
err = parseStatus.reason();
errCode = parseStatus.code();
return -1;
}
if (request.query.isEmpty()) {
return applySkipLimit(collection->numRecords(txn), cmd);
}
PlanExecutor* rawExec;
Status getExecStatus = getExecutorCount(txn,
collection,
request,
PlanExecutor::YIELD_AUTO,
&rawExec);
if (!getExecStatus.isOK()) {
err = getExecStatus.reason();
errCode = getExecStatus.code();
return -1;
}
scoped_ptr<PlanExecutor> exec(rawExec);
// Store the plan summary string in CurOp.
if (NULL != txn->getCurOp()) {
txn->getCurOp()->debug().planSummary = Explain::getPlanSummary(exec.get());
}
Status execPlanStatus = exec->executePlan();
if (!execPlanStatus.isOK()) {
err = execPlanStatus.reason();
errCode = execPlanStatus.code();
return -2;
}
// Plan is done executing. We just need to pull the count out of the root stage.
invariant(STAGE_COUNT == exec->getRootStage()->stageType());
CountStage* countStage = static_cast<CountStage*>(exec->getRootStage());
const CountStats* countStats =
static_cast<const CountStats*>(countStage->getSpecificStats());
return countStats->nCounted;
}
long long runCount(OperationContext* txn,
const string& ns,
const BSONObj &cmd,
string &err,
int &errCode) {
// Lock 'ns'.
Client::Context cx(txn, ns);
Collection* collection = cx.db()->getCollection(txn, ns);
const string& dbname = cx.db()->name();
if (NULL == collection) {
err = "ns missing";
return -1;
}
CountRequest request;
CmdCount* countComm = static_cast<CmdCount*>(Command::findCommand("count"));
Status parseStatus = countComm->parseRequest(dbname, cmd, &request);
if (!parseStatus.isOK()) {
err = parseStatus.reason();
errCode = parseStatus.code();
return -1;
}
if (request.query.isEmpty()) {
return applySkipLimit(collection->numRecords(txn), cmd);
}
PlanExecutor* rawExec;
Status getExecStatus = getExecutorCount(txn, collection, request, &rawExec);
if (!getExecStatus.isOK()) {
err = getExecStatus.reason();
errCode = parseStatus.code();
return -1;
}
scoped_ptr<PlanExecutor> exec(rawExec);
// Store the plan summary string in CurOp.
if (NULL != txn->getCurOp()) {
PlanSummaryStats stats;
Explain::getSummaryStats(exec.get(), &stats);
txn->getCurOp()->debug().planSummary = stats.summaryStr.c_str();
}
const ScopedExecutorRegistration safety(exec.get());
Status execPlanStatus = exec->executePlan();
if (!execPlanStatus.isOK()) {
err = execPlanStatus.reason();
errCode = execPlanStatus.code();
return -2;
}
// Plan is done executing. We just need to pull the count out of the root stage.
invariant(STAGE_COUNT == exec->getRootStage()->stageType());
CountStage* countStage = static_cast<CountStage*>(exec->getRootStage());
const CountStats* countStats =
static_cast<const CountStats*>(countStage->getSpecificStats());
return countStats->nCounted;
}
