co-limit 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4686cb64b1936e1c6c1420557f947a27263f498f7f0f1f90074e79a27dd400b8
-  data.tar.gz: 7373d4fb997a7d3d2293525e1e8096e21bfb70b3b92ad5f91917467883561ebf
+  metadata.gz: d0298c1dc7c99033e8919babcecbfdc0a25fd25ff05a8ebdc53437b898bc7ee7
+  data.tar.gz: 63b245b43b37eaa0340298d9b4317a078868fc37862b01d414764e671ef4c84b
 SHA512:
-  metadata.gz: d2bc5af6b8acd79aa457d872302f597f55479d324390b1267a3e35c8d3e8469621851dc995f243b3aa6847c5c6bcb5d7787a8d2b93e95c3eca6738f52f7bc19e
-  data.tar.gz: 55ce9a1e9e90dff76b8e56aac61a65e736c1e77bf6d519ebc7f85fdf6c0d3b7d6eaa98107f9719aaf7be3d978af79958ba0ae3eb07d3f448a5c7fd4363a47904
+  metadata.gz: 6b959ef55df27bea377e777d83e2acaad02f166eff5ff3e6140253ab54fb5e4fd0e8d6bcf40f2ea440a20cef7099cae8f3ac468d960ce090b6fa1fbb220b4406
+  data.tar.gz: 9a3b26e770dda72c4096c20ed601d672faaf4ee26bff9c6cfa155c8ba8319e977c7c0477acc27417b77252b92e250e64782da291d02de5f5c97629ca8e03ad8a

data/lib/limit/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Limit
-  VERSION = '0.1.1'
+  VERSION = '0.1.3'
 end

data/lib/limit.rb

@@ -43,8 +43,8 @@ module Limit
 
     def initialize(identifier_prefix:, limit_calculator:, host: nil, port: nil, password: nil)
 
-    # @param identifier_prefix: [String] A namespace prefix for redis keys for this limiter instance
-    # @param limit_calculator: [Lambda] A method that takes a key(String) and returns hash: {max_requests: Integer, window_seconds: Integer}
+      # @param identifier_prefix: [String] A namespace prefix for redis keys for this limiter instance
+      # @param limit_calculator: [Lambda] A method that takes a key(String) and returns hash: {max_requests: Integer, window_seconds: Integer}
 
       unless identifier_prefix.is_a?(String) && !identifier_prefix.empty?
         raise ArgumentError, 'identifier_prefix must be a non-empty String'
@@ -54,6 +54,7 @@ module Limit
 
       # Will be using the same connection across all instance unless wanted to connect to diff instance of redis
 
+      #TODO: Connection creation can be lazy
       @redis = if host && port && password
                  self.class.send(:create_connection, host: host, port: port, password: password)
                else
@@ -78,7 +79,7 @@ module Limit
       raise NotImplementedError "#{self.class.name} must implement the allowed? method"
     end
 
-    def get_key(prefix)
+    def get_key(key)
       raise NotImplementedError "#{self.class.name} must implement the get_key() method"
     end
 
@@ -98,9 +99,19 @@ module Limit
     end
 
     def redis_pipeline(&block)
+      attempts ||= 0
       begin
         @redis.pipelined { |pipe| block.call(pipe) }
-      rescue Redis::CommandError => e
+      rescue Redis::CannotConnectError, Redis::TimeoutError, Redis::ConnectionError => e
+        if attempts < 3
+          @logger.warn(e.message)
+          @redis = self.class.connection
+          attempts += 1
+          retry
+        else
+          @logger.error("Error connecting to Redis: #{e.message}")
+        end
+      rescue Redis::BaseError => e
         @logger.error(e.message)
       end
     end
@@ -136,9 +147,9 @@ module Limit
       results[0] <= max_requests
     end
 
-    def get_key(prefix, window_seconds)
+    def get_key(key, window_seconds)
       time_window = (Time.now.to_i / window_seconds) * window_seconds
-      "#{@identifier_prefix}:#{prefix}:#{time_window.to_s}"
+      "#{@identifier_prefix}:#{key}:#{time_window.to_s}"
     end
   end
 
@@ -167,8 +178,8 @@ module Limit
       results[2] <= max_requests
     end
 
-    def get_key(prefix)
-      "#{@identifier_prefix}:#{prefix}"
+    def get_key(key)
+      "#{@identifier_prefix}:#{key}"
     end
   end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: co-limit
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - CosmicOppai
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-05-01 00:00:00.000000000 Z
+date: 2025-08-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redis