WIP

mongodb · Oct 12, 2024 · d2d8a49 · d2d8a49
1 parent 9782fe1
commit d2d8a49
Showing 1 changed file with 20 additions and 15 deletions.
diff --git a/x/mongo/driver/operation.go b/x/mongo/driver/operation.go
@@ -1439,26 +1439,31 @@ func (op Operation) addEncryptCommandFields(dst []byte, desc description.Selecte
 	if err != nil {
 		return 0, nil, err
 	}
-	if op.Batches == nil {
-		return 0, dst, nil
-	}
-	maxBatchCount := int(desc.MaxBatchCount)
-	maxDocumentSize := int(desc.MaxDocumentSize)
 	var n int
-	if maxBatchCount > 1 {
+	if op.Batches != nil {
+		maxBatchCount := int(desc.MaxBatchCount)
+		maxDocumentSize := int(desc.MaxDocumentSize)
 		n, dst, err = op.Batches.AppendBatchArray(dst, maxBatchCount, cryptMaxBsonObjectSize, maxDocumentSize)
 		if err != nil {
 			return 0, nil, err
 		}
-	}
-	if n == 0 {
-		n, dst, err = op.Batches.AppendBatchArray(dst, 1, maxDocumentSize, maxDocumentSize)
-		if err != nil {
-			return 0, nil, err
-		}
-		if n == 0 {
-			return 0, nil, ErrDocumentTooLarge
-		}
+		/*
+			if maxBatchCount > 1 {
+				n, dst, err = op.Batches.AppendBatchArray(dst, maxBatchCount, cryptMaxBsonObjectSize, maxDocumentSize)
+				if err != nil {
+					return 0, nil, err
+				}
+			}
+			if n == 0 {
+				n, dst, err = op.Batches.AppendBatchArray(dst, 1, maxDocumentSize, maxDocumentSize)
+				if err != nil {
+					return 0, nil, err
+				}
+				if n == 0 {
+					return 0, nil, ErrDocumentTooLarge
+				}
+			}
+		*/
 	}
 	dst, err = bsoncore.AppendDocumentEnd(dst, idx)
 	if err != nil {