redis_queue 0.1.2 → 0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4e3d482340930676569479542795dbb405b03657
-  data.tar.gz: 3e85d5a06808cac114635c7173443b633861fb5d
+  metadata.gz: 068a0a6c1728f9f5f807aa9894d893594587f868
+  data.tar.gz: a534194a5007445263ef9a3326b74399ad6a1703
 SHA512:
-  metadata.gz: 1cdb29982089b13672c558c74fd2c970197a1551f42b93ae96cbe53a10494186d6c597453fe0e28d5fa041b841e1d7e6212f2beaff1cbc6436f52011b896b298
-  data.tar.gz: 563dbc1289fd709c0fe5ddb0477a78bed3a3f699d5fdb23b4a7f2046369210a8e05c5c73628c00923bb0c5969c3851915aa1be36de5bceefd9d269005e4478a6
+  metadata.gz: 87d378af57edf944f92bce2ec9b49af8728d6273d9e70666b71ae7f97f2054141df19f9fb7d9cc254cc7d6724e0da170b218245e27427ab940dd29d3572295b4
+  data.tar.gz: 0884f179db7552beacdd41b5247664516b995c40347c2bedd47cb42779af5764e2d04deb899fc41245089400833c88dda1430f6ba3272f84296334dabb07d416

data/lib/redis_connection.rb

@@ -1,8 +1,8 @@
 require 'redis'
 
 class RedisConnection
-  def initialize url
-    @url = url
+  def initialize args
+    @args = args
   end
 
   def run
@@ -20,6 +20,6 @@ class RedisConnection
   end
 
   def new_redis
-    ::Redis.connect(url: @url)
+    ::Redis.connect(@args)
   end
 end

data/lib/redis_queue.rb

@@ -1,23 +1,24 @@
 require_relative 'redis_connection'
 
 class RedisQueue
-  def initialize id=:messages, url='redis://localhost:6379/0'
-    @id = id
-    @redis_connection = RedisConnection.new(url)
+  def initialize args={id: :messages, url: 'redis://localhost:6379/0'}
+    @id             = args.delete(:id)
+    @redis          = RedisConnection.new(args)
+    @redis_blocking = RedisConnection.new(args)
   end
 
   def pop
-    @redis_connection.run do |redis|
+    @redis_blocking.run do |redis|
       redis.blpop(@id).last.tap { |msg| redis.sadd "#{@id}_in_use", msg }
     end
   end
 
   def push task
-    @redis_connection.run { |redis| redis.rpush @id, task }
+    @redis.run { |redis| redis.rpush @id, task }
   end
 
   def fail task
-    @redis_connection.run do |redis|
+    @redis.run do |redis|
       redis.pipelined do
         redis.sadd "#{@id}_failed", task
         redis.srem "#{@id}_in_use", task
@@ -26,7 +27,7 @@ class RedisQueue
   end
 
   def done task
-    @redis_connection.run do |redis|
+    @redis.run do |redis|
       redis.pipelined do
         redis.sadd "#{@id}_done", task
         redis.srem "#{@id}_in_use", task
@@ -35,7 +36,7 @@ class RedisQueue
   end
 
   def unpop task
-    @redis_connection.run do |redis|
+    @redis.run do |redis|
       redis.pipelined do
         redis.lpush @id, task
         redis.srem "#{@id}_in_use", task
@@ -45,16 +46,16 @@ class RedisQueue
 
   def reset
     init_from "#{@id}_in_use"
-    @redis_connection.run { |redis| redis.del "#{@id}_in_use" }
+    @redis.run { |redis| redis.del "#{@id}_in_use" }
   end
 
   def restart
     init_from "#{@id}_done"
-    @redis_connection.run { |redis| redis.del "#{@id}_done" }
+    @redis.run { |redis| redis.del "#{@id}_done" }
   end
 
   def init_from set
-    @redis_connection.run do |redis|
+    @redis.run do |redis|
       redis.eval "local vals = redis.call('smembers', '#{set}')
       for i = 1, table.getn(vals) do
         redis.call('rpush', '#{@id}', vals[i])
@@ -63,35 +64,35 @@ class RedisQueue
   end
 
   def size
-    @redis_connection.run { |redis| redis.llen @id }.to_i
+    @redis.run { |redis| redis.llen @id }.to_i
   end
 
   def done_size
-    @redis_connection.run { |redis| redis.scard "#{@id}_done" }.to_i
+    @redis.run { |redis| redis.scard "#{@id}_done" }.to_i
   end
 
   def failed_size
-    @redis_connection.run { |redis| redis.scard "#{@id}_failed" }.to_i
+    @redis.run { |redis| redis.scard "#{@id}_failed" }.to_i
   end
 
   def in_use_size
-    @redis_connection.run { |redis| redis.scard "#{@id}_in_use" }.to_i
+    @redis.run { |redis| redis.scard "#{@id}_in_use" }.to_i
   end
 
   def list
-    @redis_connection.run { |redis| redis.lrange @id, 0, -1 }
+    @redis.run { |redis| redis.lrange @id, 0, -1 }
   end
 
   def done_list
-    @redis_connection.run { |redis| redis.smembers "#{@id}_done" }
+    @redis.run { |redis| redis.smembers "#{@id}_done" }
   end
 
   def failed_list
-    @redis_connection.run { |redis| redis.smembers "#{@id}_failed" }
+    @redis.run { |redis| redis.smembers "#{@id}_failed" }
   end
 
   def in_use_list
-    @redis_connection.run { |redis| redis.smembers "#{@id}_in_use" }
+    @redis.run { |redis| redis.smembers "#{@id}_in_use" }
   end
 
   def print_stats
@@ -109,7 +110,7 @@ class RedisQueue
   end
 
   def clear
-    @redis_connection.run do |redis|
+    @redis.run do |redis|
       redis.del @id
       redis.del "#{@id}_in_use"
       redis.del "#{@id}_done"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: redis_queue
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: '0.2'
 platform: ruby
 authors:
 - Jose Ignacio Fernandez