From 40b087fee8da6c3de437bac01405f0c2f0351288 Mon Sep 17 00:00:00 2001
From: Thomas <thomas-@users.noreply.github.com>
Date: Tue, 30 Nov 2021 11:41:53 +0100
Subject: [PATCH] Merge pull request #5952 from
 overleaf/tm-batched-update-ranges

Add BATCH_RANGE_END(/BATCH_RANGE_START) env vars for batchedUpdate scripts

GitOrigin-RevId: e74f30368194584a6fb398963e0c04f850a8b31d
---
 services/web/scripts/helpers/batchedUpdate.js | 21 +++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/services/web/scripts/helpers/batchedUpdate.js b/services/web/scripts/helpers/batchedUpdate.js
index de1dd6705f..aabc262b64 100644
--- a/services/web/scripts/helpers/batchedUpdate.js
+++ b/services/web/scripts/helpers/batchedUpdate.js
@@ -6,17 +6,34 @@ const BATCH_SIZE = parseInt(process.env.BATCH_SIZE, 10) || 1000
 let BATCH_LAST_ID
 if (process.env.BATCH_LAST_ID) {
   BATCH_LAST_ID = ObjectId(process.env.BATCH_LAST_ID)
+} else if (process.env.BATCH_RANGE_START) {
+  BATCH_LAST_ID = ObjectId(process.env.BATCH_RANGE_START)
+}
+let BATCH_RANGE_END
+if (process.env.BATCH_RANGE_END) {
+  BATCH_RANGE_END = ObjectId(process.env.BATCH_RANGE_END)
 }
 
 async function getNextBatch(collection, query, maxId, projection, options) {
+  const queryIdField = {}
   maxId = maxId || BATCH_LAST_ID
   if (maxId) {
     if (BATCH_DESCENDING) {
-      query._id = { $lt: maxId }
+      queryIdField.$lt = maxId
     } else {
-      query._id = { $gt: maxId }
+      queryIdField.$gt = maxId
     }
   }
+  if (BATCH_RANGE_END) {
+    if (BATCH_DESCENDING) {
+      queryIdField.$gt = BATCH_RANGE_END
+    } else {
+      queryIdField.$lt = BATCH_RANGE_END
+    }
+  }
+  if (queryIdField.$gt || queryIdField.$lt) {
+    query._id = queryIdField
+  }
   const entries = await collection
     .find(query, options)
     .project(projection)