perf(drizzle): remove unnecessary db.select call in updateOne operation (#11847)

This will improve performance when updating a single document in
postgres/drizzle, if the ID is known.

Previously, this resulted in 2 sequential operations:
- `db.select `to fetch the document by the ID
- `upsertRow` to update the document (multiple db operations)

This PR removes the unnecessary `db.select` call, as the document ID is
already known

Author: AlessioGr

packages/drizzle/src/updateOne.ts

@@ -19,50 +19,51 @@ export const updateOne: UpdateOne = async function updateOne(
     joins: joinQuery,
     locale,
     req,
+    returning,
     select,
     where: whereArg,
-    returning,
   },
 ) {
   const db = await getTransaction(this, req)
   const collection = this.payload.collections[collectionSlug].config
   const tableName = this.tableNameMap.get(toSnakeCase(collection.slug))
-  const whereToUse = whereArg || { id: { equals: id } }
   let idToUpdate = id
 
-  const { joins, selectFields, where } = buildQuery({
-    adapter: this,
-    fields: collection.flattenedFields,
-    locale,
-    tableName,
-    where: whereToUse,
-  })
+  if (!idToUpdate) {
+    const { joins, selectFields, where } = buildQuery({
+      adapter: this,
+      fields: collection.flattenedFields,
+      locale,
+      tableName,
+      where: whereArg,
+    })
 
-  // selectDistinct will only return if there are joins
-  const selectDistinctResult = await selectDistinct({
-    adapter: this,
-    chainedMethods: [{ args: [1], method: 'limit' }],
-    db,
-    joins,
-    selectFields,
-    tableName,
-    where,
-  })
+    // selectDistinct will only return if there are joins
+    const selectDistinctResult = await selectDistinct({
+      adapter: this,
+      chainedMethods: [{ args: [1], method: 'limit' }],
+      db,
+      joins,
+      selectFields,
+      tableName,
+      where,
+    })
 
-  if (selectDistinctResult?.[0]?.id) {
-    idToUpdate = selectDistinctResult?.[0]?.id
-    // If id wasn't passed but `where` without any joins, retrieve it with findFirst
-  } else if (whereArg && !joins.length) {
-    const table = this.tables[tableName]
+    if (selectDistinctResult?.[0]?.id) {
+      idToUpdate = selectDistinctResult?.[0]?.id
+      // If id wasn't passed but `where` without any joins, retrieve it with findFirst
+    } else if (whereArg && !joins.length) {
+      const table = this.tables[tableName]
 
-    const docsToUpdate = await (db as LibSQLDatabase)
-      .select({
-        id: table.id,
-      })
-      .from(table)
-      .where(where)
-      .limit(1)
-    idToUpdate = docsToUpdate?.[0]?.id
+      const docsToUpdate = await (db as LibSQLDatabase)
+        .select({
+          id: table.id,
+        })
+        .from(table)
+        .where(where)
+        .limit(1)
+      idToUpdate = docsToUpdate?.[0]?.id
+    }
   }
 
   const result = await upsertRow({
@@ -71,12 +72,12 @@ export const updateOne: UpdateOne = async function updateOne(
     data,
     db,
     fields: collection.flattenedFields,
+    ignoreResult: returning === false,
     joinQuery,
     operation: 'update',
     req,
     select,
     tableName,
-    ignoreResult: returning === false,
   })
 
   if (returning === false) {