Update sidekiq.rb

krisleech · Jan 10, 2023 · 25c1c25 · 25c1c25
1 parent 62d683c
commit 25c1c25
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/wisper/sidekiq.rb b/lib/wisper/sidekiq.rb
@@ -14,8 +14,9 @@ class Worker
       include ::Sidekiq::Worker
 
       def perform(yml)
-        (subscriber, event, args) = ::YAML.load(yml)
-        subscriber.public_send(event, *args)
+        (subscriber, event, args) = ::YAML.unsafe_load(yml)
+        kwargs = args.extract_options!
+        subscriber.public_send(event, *args, **kwargs)
       end
     end
 
@@ -26,13 +27,13 @@ def self.register
       end
     end
 
-    def broadcast(subscriber, publisher, event, args)
+    def broadcast(subscriber, publisher, event, *args, **kwargs)
       options = sidekiq_options(subscriber)
       schedule_options = sidekiq_schedule_options(subscriber, event)
 
       Worker.set(options).perform_in(
         schedule_options.fetch(:delay, 0),
-        ::YAML.dump([subscriber, event, args])
+        ::YAML.dump([subscriber, event, args.push(kwargs)])
       )
     end