SERVER-62674 Add stats tracking for multiple collections

2 files changed, 23 insertions, 0 deletions

diff --git a/src/mongo/db/stats/top.cpp b/src/mongo/db/stats/top.cpp
index 1674fe394d6..513b7fae754 100644
--- a/src/mongo/db/stats/top.cpp
+++ b/src/mongo/db/stats/top.cpp
@@ -88,6 +88,18 @@ void Top::record(OperationContext* opCtx,
     _record(opCtx, coll, logicalOp, lockType, micros, readWriteType);
 }
 
+void Top::record(OperationContext* opCtx,
+                 const std::set<NamespaceString>& nssSet,
+                 LogicalOp logicalOp,
+                 LockType lockType,
+                 long long micros,
+                 bool command,
+                 Command::ReadWriteType readWriteType) {
+    for (const auto& nss : nssSet) {
+        record(opCtx, nss.ns(), logicalOp, lockType, micros, command, readWriteType);
+    }
+}
+
 void Top::_record(OperationContext* opCtx,
                   CollectionData& c,
                   LogicalOp logicalOp,
diff --git a/src/mongo/db/stats/top.h b/src/mongo/db/stats/top.h
index 5aa3034688b..358dc5d7164 100644
--- a/src/mongo/db/stats/top.h
+++ b/src/mongo/db/stats/top.h
@@ -104,6 +104,17 @@ public:
                 bool command,
                 Command::ReadWriteType readWriteType);
 
+    /**
+     * Same as the above, but for multiple namespaces.
+     */
+    void record(OperationContext* opCtx,
+                const std::set<NamespaceString>& nssSet,
+                LogicalOp logicalOp,
+                LockType lockType,
+                long long micros,
+                bool command,
+                Command::ReadWriteType readWriteType);
+
     void append(BSONObjBuilder& b);
 
     void cloneMap(UsageMap& out) const;