summaryrefslogtreecommitdiff
path: root/src/mongo/s/query/cluster_client_cursor_mock.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/mongo/s/query/cluster_client_cursor_mock.h')
-rw-r--r--src/mongo/s/query/cluster_client_cursor_mock.h5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/mongo/s/query/cluster_client_cursor_mock.h b/src/mongo/s/query/cluster_client_cursor_mock.h
index 81fb52f15b4..607250bba46 100644
--- a/src/mongo/s/query/cluster_client_cursor_mock.h
+++ b/src/mongo/s/query/cluster_client_cursor_mock.h
@@ -67,6 +67,8 @@ public:
bool isTailableAndAwaitData() const final;
+ BSONObj getOriginatingCommand() const final;
+
long long getNumReturnedSoFar() const final;
void queueResult(const ClusterQueryResult& result) final;
@@ -96,6 +98,9 @@ private:
std::queue<StatusWith<ClusterQueryResult>> _resultsQueue;
stdx::function<void(void)> _killCallback;
+ // Originating command object.
+ BSONObj _originatingCommand;
+
// Number of returned documents.
long long _numReturnedSoFar = 0;