feat(db-mongodb,drizzle): add atomic array operations for relationship fields

packages/db-mongodb/src/updateJobs.ts

@@ -55,9 +55,13 @@ export const updateJobs: UpdateJobs = async function updateMany(
 
   const $inc: Record<string, number> = {}
   const $push: Record<string, { $each: any[] } | any> = {}
+  const $addToSet: Record<string, { $each: any[] } | any> = {}
+  const $pull: Record<string, { $in: any[] } | any> = {}
 
   transform({
+    $addToSet,
     $inc,
+    $pull,
     $push,
     adapter: this,
     data,
@@ -73,6 +77,12 @@ export const updateJobs: UpdateJobs = async function updateMany(
   if (Object.keys($push).length) {
     updateOps.$push = $push
   }
+  if (Object.keys($addToSet).length) {
+    updateOps.$addToSet = $addToSet
+  }
+  if (Object.keys($pull).length) {
+    updateOps.$pull = $pull
+  }
   if (Object.keys(updateOps).length) {
     updateOps.$set = updateData
     updateData = updateOps

packages/db-mongodb/src/updateMany.ts

@@ -1,4 +1,4 @@
-import type { MongooseUpdateQueryOptions } from 'mongoose'
+import type { MongooseUpdateQueryOptions, UpdateQuery } from 'mongoose'
 
 import { flattenWhereToOperators, type UpdateMany } from 'payload'
 
@@ -70,7 +70,40 @@ export const updateMany: UpdateMany = async function updateMany(
     where,
   })
 
-  transform({ adapter: this, data, fields: collectionConfig.fields, operation: 'write' })
+  const $inc: Record<string, number> = {}
+  const $push: Record<string, { $each: any[] } | any> = {}
+  const $addToSet: Record<string, { $each: any[] } | any> = {}
+  const $pull: Record<string, { $in: any[] } | any> = {}
+
+  transform({
+    $addToSet,
+    $inc,
+    $pull,
+    $push,
+    adapter: this,
+    data,
+    fields: collectionConfig.fields,
+    operation: 'write',
+  })
+
+  const updateOps: UpdateQuery<any> = {}
+
+  if (Object.keys($inc).length) {
+    updateOps.$inc = $inc
+  }
+  if (Object.keys($push).length) {
+    updateOps.$push = $push
+  }
+  if (Object.keys($addToSet).length) {
+    updateOps.$addToSet = $addToSet
+  }
+  if (Object.keys($pull).length) {
+    updateOps.$pull = $pull
+  }
+  if (Object.keys(updateOps).length) {
+    updateOps.$set = data
+    data = updateOps
+  }
 
   try {
     if (typeof limit === 'number' && limit > 0) {

packages/db-mongodb/src/updateOne.ts

@@ -56,8 +56,19 @@ export const updateOne: UpdateOne = async function updateOne(
 
   const $inc: Record<string, number> = {}
   const $push: Record<string, { $each: any[] } | any> = {}
+  const $addToSet: Record<string, { $each: any[] } | any> = {}
+  const $pull: Record<string, { $in: any[] } | any> = {}
 
-  transform({ $inc, $push, adapter: this, data, fields, operation: 'write' })
+  transform({
+    $addToSet,
+    $inc,
+    $pull,
+    $push,
+    adapter: this,
+    data,
+    fields,
+    operation: 'write',
+  })
 
   const updateOps: UpdateQuery<any> = {}
 
@@ -67,6 +78,12 @@ export const updateOne: UpdateOne = async function updateOne(
   if (Object.keys($push).length) {
     updateOps.$push = $push
   }
+  if (Object.keys($addToSet).length) {
+    updateOps.$addToSet = $addToSet
+  }
+  if (Object.keys($pull).length) {
+    updateOps.$pull = $pull
+  }
   if (Object.keys(updateOps).length) {
     updateOps.$set = updateData
     updateData = updateOps