SERVER-19569 CursorResponse::toBSON() ability to return first batch

1 files changed, 7 insertions, 2 deletions

diff --git a/src/mongo/db/query/cursor_response.h b/src/mongo/db/query/cursor_response.h
index e606dcd1559..ac15477e28c 100644
--- a/src/mongo/db/query/cursor_response.h
+++ b/src/mongo/db/query/cursor_response.h
@@ -66,6 +66,11 @@ void appendGetMoreResponseObject(long long cursorId,
                                  BSONObjBuilder* builder);
 
 struct CursorResponse {
+    enum class ResponseType {
+        InitialResponse,
+        SubsequentResponse,
+    };
+
     /**
      * Constructs from values for each of the fields.
      */
@@ -82,8 +87,8 @@ struct CursorResponse {
     /**
      * Converts this response to its raw BSON representation.
      */
-    BSONObj toBSON() const;
-    void addToBSON(BSONObjBuilder* builder) const;
+    BSONObj toBSON(ResponseType responseType) const;
+    void addToBSON(ResponseType responseType, BSONObjBuilder* builder) const;
 
     const NamespaceString nss;
     const CursorId cursorId;