ruby_rate_limiter 2.0.8 → 2.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f812a1ae8b8449217f489de3234d4f26a077bdc4f8bd6653dbe29457c363ab23
-  data.tar.gz: f2941667395af0a013e11c87394ab355ed5c01ec789fdc294e5eb84ef4ac4c3b
+  metadata.gz: 79a5d1b8a3f50d3b30ef8a3915261f8eeecc12cb42cc87cebf308299a39453e3
+  data.tar.gz: 21b3689378e4d623e53f527271df257d32f4d7e049c83dee4ee2da5a38524437
 SHA512:
-  metadata.gz: ebd5d104e4172e0b32856dc26b815b04d0a49b89b35e7a9c8520ad474ab00e2dfb5c2771078a579e8c9075b49ed84a75c2b98e5e5ad672ec4b750ef576ea1eaf
-  data.tar.gz: e402b83a46be87121443283dc178a8eabed68056c44a23423f2c008d0bc5f9c7d8f5a870fb1c00ef629f5ad006501c42267fbb4ae01532030f434409662b8d1c
+  metadata.gz: cf15d23cd91b3f09fd4efdc3729d8aa9fd992851ece0c728444ae36927df349a0235f923348e6360859a84965b25cd23632d011a12383c45c604a8dc4a2d3f57
+  data.tar.gz: b5b7895627a9ae8b30e981932f1e3266b5ccc8ea0c3b3b88728d5be10b73033baf39af7e281953469e282c231519898feeda91ece31b1b7c422d9d5a3372f134

data/lib/ruby_rate_limiter/token_bucket.rb

@@ -1,12 +1,12 @@
 # lib/ruby_rate_limiter/token_bucket.rb
 require 'forwardable'
+require 'logger'
 require_relative 'storage/abstract_storage'
 require_relative 'storage/redis_storage'
 
 module RubyRateLimiter
   class TokenBucket
     extend Forwardable
-
     DEFAULT_BUCKET_SIZE = 10
     DEFAULT_REFILL_RATE = 1
     DEFAULT_STORAGE = RubyRateLimiter::Storage::RedisStorage.new
@@ -29,18 +29,21 @@ module RubyRateLimiter
       @bucket_size = bucket_size
       @refill_rate_per_second = refill_rate.to_f / TIME_UNITS[time_unit]
       @storage = storage
+
+      @logger = Logger.new(STDOUT)
+      @logger.level = Logger::DEBUG 
       initialize_bucket
     end
 
     def allow_request?
       refill_tokens
       tokens = get_bucket_size
-      puts "Allow request: tokens before = #{tokens}" # Debugging line
+      @logger.info "Allow request: tokens before = #{tokens}" 
 
       return false if tokens < 1
 
       update_bucket_size(tokens - 1)
-      puts "Allow request: tokens after = #{get_bucket_size}" # Debugging line
+      @logger.info "Allow request: tokens after = #{get_bucket_size}" 
       true
     end
 
@@ -50,19 +53,17 @@ module RubyRateLimiter
       if @storage.get("#{@user_id}_tokens").nil?
         update_bucket_size(@bucket_size)
         update_last_refill_time(Time.now.to_f)
-        puts "Initialized bucket: tokens = #{@bucket_size}, time = #{Time.now.to_f}"
+        @logger.info "Initialized bucket: tokens = #{@bucket_size}, time = #{Time.now.to_f}"
       end
     end
 
     def get_bucket_size
       size = (@storage.get("#{@user_id}_tokens") || @bucket_size).to_i
-      puts "get_bucket_size: #{@user_id}_tokens = #{size}" # Debugging line
       size
     end
 
     def get_last_refill_time
       last_refill = (@storage.get("#{@user_id}_last_refill") || Time.now.to_f).to_f
-      puts "get_last_refill_time: #{@user_id}_last_refill = #{last_refill}" # Debugging line
       last_refill
     end
 
@@ -80,18 +81,15 @@ module RubyRateLimiter
       elapsed_time = current_time - last_refill_time
     
       if elapsed_time < 0
-        puts "Warning: elapsed_time is negative. Adjusting to zero."
         elapsed_time = 0
       end
     
       new_tokens = (elapsed_time * @refill_rate_per_second).to_i
-      puts "Refill tokens: current_time = #{current_time}, last_refill_time = #{last_refill_time}, elapsed_time = #{elapsed_time}, new_tokens = #{new_tokens}" # Debugging line
-    
+
       if new_tokens > 0
         tokens = [get_bucket_size + new_tokens, @bucket_size].min
         update_bucket_size(tokens)
         update_last_refill_time(current_time)
-        puts "Refill tokens: updated tokens = #{tokens}, current_time = #{current_time}" # Debugging line
       end
     end
   end

data/lib/ruby_rate_limiter/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RateLimiter
-  VERSION = "2.0.8"
+  VERSION = "2.0.9"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_rate_limiter
 version: !ruby/object:Gem::Version
-  version: 2.0.8
+  version: 2.0.9
 platform: ruby
 authors:
 - Mutuba