virtual Status explain(OperationContext* txn,
const std::string& dbname,
const BSONObj& cmdObj,
ExplainCommon::Verbosity verbosity,
BSONObjBuilder* out) const {
CountRequest request;
Status parseStatus = parseRequest(dbname, cmdObj, &request);
if (!parseStatus.isOK()) {
return parseStatus;
}
request.explain = true;
// Acquire the db read lock.
AutoGetCollectionForRead ctx(txn, request.ns);
Collection* collection = ctx.getCollection();
PlanExecutor* rawExec;
Status getExecStatus = getExecutorCount(txn,
collection,
request,
PlanExecutor::YIELD_AUTO,
&rawExec);
if (!getExecStatus.isOK()) {
return getExecStatus;
}
scoped_ptr<PlanExecutor> exec(rawExec);
Explain::explainStages(exec.get(), verbosity, out);
return Status::OK();
}
Status CmdCount::explain(OperationContext* txn,
const std::string& dbname,
const BSONObj& cmdObj,
Explain::Verbosity verbosity,
BSONObjBuilder* out) const {
CountRequest request;
Status parseStatus = parseRequest(dbname, cmdObj, &request);
if (!parseStatus.isOK()) {
return parseStatus;
}
// Acquire the db read lock.
Client::ReadContext ctx(txn, request.ns);
Collection* collection = ctx.ctx().db()->getCollection(txn, request.ns);
PlanExecutor* rawExec;
Status getExecStatus = getExecutorCount(txn, collection, request, &rawExec);
if (!getExecStatus.isOK()) {
return getExecStatus;
}
scoped_ptr<PlanExecutor> exec(rawExec);
const ScopedExecutorRegistration safety(exec.get());
return Explain::explainStages(exec.get(), verbosity, out);
}
virtual bool run(OperationContext* txn,
const string& dbname,
BSONObj& cmdObj,
int, string& errmsg,
BSONObjBuilder& result,
bool fromRepl) {
CountRequest request;
Status parseStatus = parseRequest(dbname, cmdObj, &request);
if (!parseStatus.isOK()) {
return appendCommandStatus(result, parseStatus);
}
AutoGetCollectionForRead ctx(txn, request.ns);
Collection* collection = ctx.getCollection();
PlanExecutor* rawExec;
Status getExecStatus = getExecutorCount(txn,
collection,
request,
PlanExecutor::YIELD_AUTO,
&rawExec);
if (!getExecStatus.isOK()) {
return appendCommandStatus(result, getExecStatus);
}
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()) {
return appendCommandStatus(result, execPlanStatus);
}
// 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());
result.appendNumber("n", countStats->nCounted);
return true;
}
bool CmdCount::run(OperationContext* txn,
const string& dbname,
BSONObj& cmdObj,
int, string& errmsg,
BSONObjBuilder& result, bool) {
CountRequest request;
Status parseStatus = parseRequest(dbname, cmdObj, &request);
if (!parseStatus.isOK()) {
return appendCommandStatus(result, parseStatus);
}
// Acquire the db read lock.
Client::ReadContext ctx(txn, request.ns);
Collection* collection = ctx.ctx().db()->getCollection(txn, request.ns);
PlanExecutor* rawExec;
Status getExecStatus = getExecutorCount(txn, collection, request, &rawExec);
if (!getExecStatus.isOK()) {
return appendCommandStatus(result, getExecStatus);
}
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()) {
return appendCommandStatus(result, execPlanStatus);
}
// 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());
result.appendNumber("n", countStats->nCounted);
return true;
}
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;
}
