lib/atomic_sidekiq/atomic_operation/retrieve.rb in atomic-sidekiq-1.0.0 vs lib/atomic_sidekiq/atomic_operation/retrieve.rb in atomic-sidekiq-1.1.0

@@ -1,20 +1,24 @@
 module AtomicSidekiq
   module AtomicOperation
     class Retrieve < Base
+      RETRIEVE_SCRIPT = File.read(
+        File.join(File.dirname(__FILE__),
+                  "./lua_scripts/retrieve.lua")
+      )
+
       def perform(queues, expire_at)
         queues.each do |queue|
           res = retrieve_from_queue(queue, expire_at.to_i)
           return res if res
         end
         nil
       end
 
       private
 
-      RETRIEVE_SCRIPT = File.read(File.join(File.dirname(__FILE__), './lua_scripts/retrieve.lua'))
-
       def retrieve_from_queue(queue, expire_at)
+        in_flight_prefix = in_flight_keymaker.queue_prefix(queue)
         redis do |conn|
           conn.eval(RETRIEVE_SCRIPT, [queue, in_flight_prefix], [expire_at])
         end
       end
     end