limit concurrency of role assign to 10 calls

vatsimnetwork · May 12, 2023 · 5cf0c80 · 5cf0c80
1 parent 421ddc8
commit 5cf0c80
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 5 deletions.
diff --git a/assign_roles.js b/assign_roles.js
@@ -1,3 +1,7 @@
+const pLimit = require('p-limit');
+
+const limit = pLimit(10);
+
 async function assignRoles(member) {
     const MANAGED_ROLES = ["Supervisor", "Administrator", "New Member", "PPL", "IR", "CMEL", "ATPL", "Flight Instructor", "Flight Examiner", "No Military Rating", "M1", "M2","M3","M4"];
     const discordCidResponse = await fetch(`https://apiv2-dev.vatsim.net/v2/members/discord/${member.user.id}`).catch(error => console.trace(error));
@@ -73,5 +77,5 @@ async function assignRoles(member) {
 }
 
 module.exports = {
-    assignRoles: assignRoles
-}
+    assignRoles: (member) => limit(() => assignRoles(member))
+}
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -2,7 +2,8 @@
   "dependencies": {
     "discord.js": "^14.6.0",
     "dotenv": "^16.0.3",
-    "nodemon": "^2.0.20"
+    "nodemon": "^2.0.20",
+    "p-limit": "^3.1.0"
   },
   "scripts": {
     "start": "node bot.js",