redis_batch 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ead79130daaa5dd59b528e53f0069b2d7520f2b40001e611ea0e1ce97baadc33
-  data.tar.gz: c63abeed3e5f550ea5ce84356ac4183e9042fec24e12d87c72a6210240846d57
+  metadata.gz: 21c8d0fc6fa834d58afff202ad63597688d3ec0943cba4f06d82e0033201b820
+  data.tar.gz: fe2a7c88467d02c85e205d726b53c6aff4f191d0c08aa7d88831c0414bdbcd6a
 SHA512:
-  metadata.gz: 9db50906c0891b09654a35d151096df5ebdd89b8fa6b854645d6eba5223840876f16fed94f3299bb12497f59dc92933169b4b67a73fb2609b82df810ab9d195e
-  data.tar.gz: 34ad2d47797b5e5fe7d73da4a300072e8a028ecb6f8a9fb971c42d7b67d5a0ee0cab5bf8e39ed763bc330bd0ec719a5a944e4d6d2df54525e69feeed284c19fb
+  metadata.gz: db65fe573cc1e50e53e870f620944dedb9d941309019ba06e068b2b49d8c9c6ff7b04b3352f560afeb2d487626fddd27cc26455c678685a021d29ee68d99198a
+  data.tar.gz: 1bbcf4b38252e74cd8e59ff705ee21c79ad9833f80933ca6b05f366da44a99225c0198e368bc691ee9022f85a56df2c847a61f5845a3a2a52cc60ca04a602f33

data/lib/redis_batch/lua.rb

@@ -0,0 +1,42 @@
+module RedisBatch
+  class Lua
+    def self.function_load(redis)
+      redis.call("FUNCTION", "LOAD", "REPLACE", redis_batch_lua)
+    end
+
+    def self.redis_batch_lua
+      <<~LUA
+        #!lua name=redis_batch
+
+        local function lmove(keys, args)
+          local count = tonumber(args[1])
+          local source = keys[1]
+          local target = keys[2]
+          local llen = redis.call("LLEN", source)
+          if llen < count then
+            count = llen
+          end
+          local values = {}
+          for i=1,count do
+            values[i] = redis.call("LMOVE", source, target, 'LEFT', 'RIGHT')
+          end 
+          return values
+        end
+
+        redis.register_function('rb_lmove', lmove)
+
+        local function lrestore(keys, args)
+          local source = keys[1]
+          local target = keys[2]
+          local llen = redis.call("LLEN", source)
+          for i=1,llen do
+            redis.call("LMOVE", source, target, 'RIGHT', 'LEFT')
+          end
+          return llen
+        end
+
+        redis.register_function('rb_lrestore', lrestore)
+      LUA
+    end
+  end
+end

data/lib/redis_batch/queue.rb

@@ -3,7 +3,7 @@ module RedisBatch
     def initialize(namespace = self.class.name)
       @namespace = namespace
       @client = Client.instance
-      @client.with  { |redis| Functions.function_load(redis) }
+      @client.with  { |redis| RedisBatch::Lua.function_load(redis) }
     end
 
     def add(*items)

data/lib/redis_batch.rb

@@ -1,8 +1,8 @@
 require 'redis'
-require './lib/redis_batch/configurable'
-require './lib/redis_batch/client'
-require './lib/redis_batch/functions'
-require './lib/redis_batch/queue'
+require 'redis_batch/configurable'
+require 'redis_batch/lua'
+require 'redis_batch/client'
+require 'redis_batch/queue'
 
 module RedisBatch
   Error = Class.new(StandardError)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: redis_batch
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Christian Avemark
@@ -34,6 +34,7 @@ files:
 - lib/redis_batch.rb
 - lib/redis_batch/client.rb
 - lib/redis_batch/configurable.rb
+- lib/redis_batch/lua.rb
 - lib/redis_batch/queue.rb
 homepage: https://github.com/avemark/redis_batch_rb
 licenses: