From 973e6f31d71588a315660ccb510e5918ac4cd94a Mon Sep 17 00:00:00 2001
From: Alexandre Bourdin <abourdin@users.noreply.github.com>
Date: Wed, 31 Mar 2021 16:16:10 +0200
Subject: [PATCH] Merge pull request #3871 from
 overleaf/ab-referral-id-script-timeout

Increase timeout for initial query of referral ID generation script

GitOrigin-RevId: f60d67ed8a8862c771a1dc33332230d73145ae51
---
 .../regenerate_duplicate_referral_ids.js       | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/services/web/scripts/regenerate_duplicate_referral_ids.js b/services/web/scripts/regenerate_duplicate_referral_ids.js
index 8db85fdf87..31ab76f3d9 100644
--- a/services/web/scripts/regenerate_duplicate_referral_ids.js
+++ b/services/web/scripts/regenerate_duplicate_referral_ids.js
@@ -11,14 +11,16 @@ async function main() {
 
   await waitForDb()
 
-  const duplicates = await db.users.aggregate(
-    [
-      { $match: { referal_id: { $exists: true } } },
-      { $group: { _id: '$referal_id', count: { $sum: 1 } } },
-      { $match: { count: { $gt: 1 } } }
-    ],
-    { allowDiskUse: true }
-  )
+  const duplicates = await db.users
+    .aggregate(
+      [
+        { $match: { referal_id: { $exists: true } } },
+        { $group: { _id: '$referal_id', count: { $sum: 1 } } },
+        { $match: { count: { $gt: 1 } } }
+      ],
+      { allowDiskUse: true, readPreference: 'secondary' }
+    )
+    .maxTimeMS(600000)
 
   const duplicateReferralIds = []
   let duplicate