Run redis calls inside block (#91)

arturictus · May 25, 2023 · 32a8543 · 32a8543
1 parent 815bed7
commit 32a8543
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 11 deletions.
diff --git a/lib/sidekiq_alive/redis/base.rb b/lib/sidekiq_alive/redis/base.rb
@@ -16,13 +16,7 @@ def delete(key)
       end
 
       def ttl(...)
-        redis.ttl(...)
-      end
-
-      private
-
-      def redis
-        Sidekiq.redis { |r| r }
+        Sidekiq.redis { |redis| redis.ttl(...) }
       end
     end
   end

diff --git a/lib/sidekiq_alive/redis/redis_client_gem.rb b/lib/sidekiq_alive/redis/redis_client_gem.rb
@@ -8,19 +8,19 @@ module Redis
     # https://github.com/redis-rb/redis-client
     class RedisClientGem < Base
       def set(key, time:, ex:)
-        redis.call("SET", key, time, ex: ex)
+        Sidekiq.redis { |redis| redis.call("SET", key, time, ex: ex) }
       end
 
       def get(key)
-        redis.call("GET", key)
+        Sidekiq.redis { |redis| redis.call("GET", key) }
       end
 
       def match(key)
-        redis.scan("MATCH", key).map { |key| key }
+        Sidekiq.redis { |redis| redis.scan("MATCH", key).map { |key| key } }
       end
 
       def delete(key)
-        redis.call("DEL", key)
+        Sidekiq.redis { |redis| redis.call("DEL", key) }
       end
     end
   end

diff --git a/lib/sidekiq_alive/redis/redis_gem.rb b/lib/sidekiq_alive/redis/redis_gem.rb
@@ -30,6 +30,12 @@ def match(key)
       def delete(key)
         redis.del(key)
       end
+
+      private
+
+      def redis
+        Sidekiq.redis { |redis| redis }
+      end
     end
   end
 end