SERVER-70753 Make shards support persisting sampled write queries

2 files changed, 30 insertions, 0 deletions

diff --git a/src/mongo/s/analyze_shard_key_common.idl b/src/mongo/s/analyze_shard_key_common.idl
index c89776591e3..af352a8ff1e 100644
--- a/src/mongo/s/analyze_shard_key_common.idl
+++ b/src/mongo/s/analyze_shard_key_common.idl
@@ -49,6 +49,15 @@ enums:
       kCount: "count"
       kDistinct: "distinct"
 
+  SampledWriteCommandName:
+    description: "The command name of a sampled write query."
+    type: string
+    values:
+      kInsert: "insert"
+      kUpdate: "update"
+      kDelete: "delete"
+      kFindAndModify: "findAndModify"
+
 structs:
   QueryAnalyzerConfiguration:
     description: "The query analyzer configuration for a collection as configured via the
diff --git a/src/mongo/s/analyze_shard_key_documents.idl b/src/mongo/s/analyze_shard_key_documents.idl
index 190d67d8cde..a3728d6c1cf 100644
--- a/src/mongo/s/analyze_shard_key_documents.idl
+++ b/src/mongo/s/analyze_shard_key_documents.idl
@@ -80,3 +80,24 @@ structs:
       cmd:
         type: object
         description: "The command object for the read."
+
+  SampledWriteQueryDocument:
+    description: "Represents a document storing a sampled write query."
+    strict: false
+    fields:
+      _id:
+        type: uuid
+        description: "The unique sample id for the write."
+        cpp_name: sampleId
+      ns:
+        type: namespacestring
+        description: "The namespace of the collection for the write."
+      collectionUuid:
+        type: uuid
+        description: "The UUID of the collection for the write."
+      cmdName:
+        type: SampledWriteCommandName
+        description: "The command name for the write."
+      cmd:
+        type: object
+        description: "The command object for the write."