multiple number of mammoth users

TheBLVD · Nov 8, 2023 · e70683f · e70683f
1 parent 3ad025b
commit e70683f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.env.production.sample b/.env.production.sample
@@ -86,3 +86,5 @@ ACCOUNT_RELAY_KEY=xxxxxx
 # JWT Decode MAMMOTH AUTHENTICATION
 MAMMOTH_SECRET_KEY=xxxxxx
 
+# Multiple number of users to load test the ForYouMammothFeed Scheduler
+FOR_YOU_LOAD_TEST_MULTIPLIER=1
diff --git a/app/workers/scheduler/for_you_mammoth_scheduler.rb b/app/workers/scheduler/for_you_mammoth_scheduler.rb
@@ -11,9 +11,11 @@ class Scheduler::ForYouMammothScheduler
 
   sidekiq_options retry: 0
 
+  LOAD_TEST_MULTIPLIER = ENV['FOR_YOU_LOAD_TEST_MULTIPLIER'] || 1
+
   def perform
     users = mammoth_users.wait
-    users.each do |acct|
+    ([users] * LOAD_TEST_MULTIPLIER).inject(&:zip).flatten.each do |acct|
       UpdateForYouWorker.perform_async({ acct: acct, rebuild: false })
     end
   end