summaryrefslogtreecommitdiff
path: root/src/mongo/db/pipeline/document_source_merge_cursors.cpp
diff options
context:
space:
mode:
authorHenrik Edin <henrik.edin@mongodb.com>2018-09-24 14:14:47 -0400
committerHenrik Edin <henrik.edin@mongodb.com>2018-10-03 16:40:40 -0400
commit9b3e7447b40c106ec4b83b176084525c4bb45133 (patch)
tree59f491d32afcdd20cf2807ad5989ce096e434719 /src/mongo/db/pipeline/document_source_merge_cursors.cpp
parent509f243d99e693e826807a26db703095120bbd73 (diff)
downloadmongo-9b3e7447b40c106ec4b83b176084525c4bb45133.tar.gz
SERVER-37297 Move aggregation pipeline stages that require sharding out of the pipeline lib.
Diffstat (limited to 'src/mongo/db/pipeline/document_source_merge_cursors.cpp')
-rw-r--r--src/mongo/db/pipeline/document_source_merge_cursors.cpp140
1 files changed, 0 insertions, 140 deletions
diff --git a/src/mongo/db/pipeline/document_source_merge_cursors.cpp b/src/mongo/db/pipeline/document_source_merge_cursors.cpp
deleted file mode 100644
index 795755893de..00000000000
--- a/src/mongo/db/pipeline/document_source_merge_cursors.cpp
+++ /dev/null
@@ -1,140 +0,0 @@
-/**
- * Copyright (C) 2018 MongoDB 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/>.
- *
- * As a special exception, the copyright holders give permission to link the
- * code of portions of this program with the OpenSSL library under certain
- * conditions as described in each individual source file and distribute
- * linked combinations including the program with the OpenSSL library. You
- * must comply with the GNU Affero General Public License in all respects
- * for all of the code used other than as permitted herein. If you modify
- * file(s) with this exception, you may extend this exception to your
- * version of the file(s), but you are not obligated to do so. If you do not
- * wish to do so, delete this exception statement from your version. If you
- * delete this exception statement from all source files in the program,
- * then also delete it in the license file.
- */
-
-#include "mongo/platform/basic.h"
-
-#include "mongo/db/pipeline/document_source_merge_cursors.h"
-#include "mongo/db/pipeline/document_source_sort.h"
-#include "mongo/db/query/find_common.h"
-#include "mongo/executor/task_executor_pool.h"
-#include "mongo/s/grid.h"
-
-namespace mongo {
-
-REGISTER_DOCUMENT_SOURCE(mergeCursors,
- LiteParsedDocumentSourceDefault::parse,
- DocumentSourceMergeCursors::createFromBson);
-
-constexpr StringData DocumentSourceMergeCursors::kStageName;
-
-DocumentSourceMergeCursors::DocumentSourceMergeCursors(
- executor::TaskExecutor* executor,
- AsyncResultsMergerParams armParams,
- const boost::intrusive_ptr<ExpressionContext>& expCtx,
- boost::optional<BSONObj> ownedParamsSpec)
- : DocumentSource(expCtx),
- _armParamsObj(std::move(ownedParamsSpec)),
- _executor(executor),
- _armParams(std::move(armParams)) {}
-
-std::size_t DocumentSourceMergeCursors::getNumRemotes() const {
- if (_armParams) {
- return _armParams->getRemotes().size();
- }
- return _blockingResultsMerger->getNumRemotes();
-}
-
-bool DocumentSourceMergeCursors::remotesExhausted() const {
- if (_armParams) {
- // We haven't started iteration yet.
- return false;
- }
- return _blockingResultsMerger->remotesExhausted();
-}
-
-void DocumentSourceMergeCursors::populateMerger() {
- invariant(!_blockingResultsMerger);
- invariant(_armParams);
- _blockingResultsMerger.emplace(pExpCtx->opCtx, std::move(*_armParams), _executor);
- _armParams = boost::none;
-}
-
-std::unique_ptr<RouterStageMerge> DocumentSourceMergeCursors::convertToRouterStage() {
- invariant(!_blockingResultsMerger, "Expected conversion to happen before execution");
- return stdx::make_unique<RouterStageMerge>(pExpCtx->opCtx, _executor, std::move(*_armParams));
-}
-
-DocumentSource::GetNextResult DocumentSourceMergeCursors::getNext() {
- if (!_blockingResultsMerger) {
- populateMerger();
- }
-
- auto next = uassertStatusOK(_blockingResultsMerger->next(pExpCtx->opCtx, _execContext));
- if (next.isEOF()) {
- return GetNextResult::makeEOF();
- }
- return Document::fromBsonWithMetaData(*next.getResult());
-}
-
-Value DocumentSourceMergeCursors::serialize(
- boost::optional<ExplainOptions::Verbosity> explain) const {
- invariant(!_blockingResultsMerger);
- invariant(_armParams);
- return Value(Document{{kStageName, _armParams->toBSON()}});
-}
-
-boost::intrusive_ptr<DocumentSource> DocumentSourceMergeCursors::createFromBson(
- BSONElement elem, const boost::intrusive_ptr<ExpressionContext>& expCtx) {
- uassert(17026,
- "$mergeCursors stage expected an object as argument",
- elem.type() == BSONType::Object);
- auto ownedObj = elem.embeddedObject().getOwned();
- auto armParams = AsyncResultsMergerParams::parse(IDLParserErrorContext(kStageName), ownedObj);
- return new DocumentSourceMergeCursors(
- Grid::get(expCtx->opCtx)->getExecutorPool()->getArbitraryExecutor(),
- std::move(armParams),
- expCtx,
- std::move(ownedObj));
-}
-
-boost::intrusive_ptr<DocumentSourceMergeCursors> DocumentSourceMergeCursors::create(
- executor::TaskExecutor* executor,
- AsyncResultsMergerParams params,
- const boost::intrusive_ptr<ExpressionContext>& expCtx) {
- return new DocumentSourceMergeCursors(executor, std::move(params), expCtx);
-}
-
-void DocumentSourceMergeCursors::detachFromOperationContext() {
- if (_blockingResultsMerger) {
- _blockingResultsMerger->detachFromOperationContext();
- }
-}
-
-void DocumentSourceMergeCursors::reattachToOperationContext(OperationContext* opCtx) {
- if (_blockingResultsMerger) {
- _blockingResultsMerger->reattachToOperationContext(opCtx);
- }
-}
-
-void DocumentSourceMergeCursors::doDispose() {
- if (_blockingResultsMerger) {
- _blockingResultsMerger->kill(pExpCtx->opCtx);
- }
-}
-
-} // namespace mongo