redis-throttle 0.0.1 → 1.1.0

data/lib/redis/throttle/script.rb

@@ -1,50 +0,0 @@
-# frozen_string_literal: true
-
-require "digest"
-require "redis/errors"
-
-require_relative "./errors"
-
-class Redis
-  module Throttle
-    # Simple helper to run script by it's sha1 digest with fallbak to script
-    # load if it was not loaded yet.
-    class Script
-      # Redis error fired when script ID is unkown.
-      NOSCRIPT = "NOSCRIPT"
-      private_constant :NOSCRIPT
-
-      LUA_ERROR_MESSAGE = %r{
-        ERR\s
-        (?<message>Error\s(?:compiling|running)\sscript\s\(.*?\)):\s
-        (?:[^:]+:\d+:\s)+
-        (?<details>.+)
-      }x.freeze
-      private_constant :LUA_ERROR_MESSAGE
-
-      def initialize(source)
-        @source = -source.to_s
-        @digest = Digest::SHA1.hexdigest(@source)
-      end
-
-      def call(redis, keys: [], argv: [])
-        __call__(redis, :keys => keys, :argv => argv)
-      rescue Redis::CommandError => e
-        md = LUA_ERROR_MESSAGE.match(e.message.to_s)
-        raise unless md
-
-        raise LuaError, [md[:message], md[:details]].compact.join(": ")
-      end
-
-      private
-
-      def __call__(redis, keys:, argv:)
-        redis.evalsha(@digest, keys, argv)
-      rescue Redis::CommandError => e
-        raise unless e.message.include?(NOSCRIPT)
-
-        redis.eval(@source, keys, argv)
-      end
-    end
-  end
-end

data/lib/redis/throttle/threshold.lua

@@ -1,12 +0,0 @@
-local bucket, limit, period, now =
-  KEYS[1], tonumber(ARGV[1]), tonumber(ARGV[2]), tonumber(ARGV[3])
-
-if limit <= redis.call("LLEN", bucket) and now - redis.call("LINDEX", bucket, -1) < period then
-  return 1
-end
-
-redis.call("LPUSH", bucket, now)
-redis.call("LTRIM", bucket, 0, limit - 1)
-redis.call("EXPIRE", bucket, period)
-
-return 0

data/lib/redis/throttle/threshold.rb

@@ -1,38 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "./script"
-
-class Redis
-  module Throttle
-    class Threshold
-      SCRIPT = Script.new(File.read("#{__dir__}/threshold.lua"))
-      private_constant :SCRIPT
-
-      NOOP = ->(_) { nil }
-      private_constant :NOOP
-
-      # @param bucket [#to_s]
-      # @param limit [#to_i]
-      # @param period [#to_i]
-      def initialize(bucket, limit:, period:)
-        @bucket = bucket.to_s
-        @limit  = limit.to_i
-        @period = period.to_i
-      end
-
-      # @param redis [Redis, Redis::Namespace]
-      # @return [Boolean]
-      def acquire(redis)
-        SCRIPT
-          .call(redis, :keys => [@bucket], :argv => [@limit, @period, Time.now.to_i])
-          .zero?
-      end
-
-      # @param redis [Redis, Redis::Namespace]
-      # @return [void]
-      def reset(redis)
-        redis.del(@bucket)
-      end
-    end
-  end
-end

data/lib/redis/throttle/version.rb

@@ -1,8 +0,0 @@
-# frozen_string_literal: true
-
-class Redis
-  module Throttle
-    # Gem version.
-    VERSION = "0.0.1"
-  end
-end