Class: RailsRateLimit::Stores::Memory

Inherits:
Base
  • Object
show all
Defined in:
lib/rails_rate_limit/stores/memory.rb

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from Base

resolve

Constructor Details

#initialize ⇒ Memory

Returns a new instance of Memory.



12
13
14
15
 
# File 'lib/rails_rate_limit/stores/memory.rb', line 12

def initialize
  @store = {}
  @mutex = Mutex.new
end

Class Method Details

.instance ⇒ Object 



7
8
9
 
# File 'lib/rails_rate_limit/stores/memory.rb', line 7

def instance
  @instance ||= new
end

Instance Method Details

#clear ⇒ Object 



47
48
49
50
51
 
# File 'lib/rails_rate_limit/stores/memory.rb', line 47

def clear
  @mutex.synchronize do
    @store.clear
  end
end

#count_requests(key, period) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/rails_rate_limit/stores/memory.rb', line 17

def count_requests(key, period)
  now = current_time
  min_time = now - period
  key = cache_key(key)

  @mutex.synchronize do
    cleanup_old_requests(key, min_time)
    @store[key]&.size || 0
  end
rescue StandardError => e
  RailsRateLimit.logger&.error("RailsRateLimit::Stores::Memory#count_requests error: #{e.message}")
  0
end

#record_request(key, period) ⇒ Object 



31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/rails_rate_limit/stores/memory.rb', line 31

def record_request(key, period)
  now = current_time
  min_time = now - period
  key = cache_key(key)

  @mutex.synchronize do
    cleanup_old_requests(key, min_time)
    @store[key] ||= []
    @store[key] << now
    @store[key].size
  end
rescue StandardError => e
  RailsRateLimit.logger&.error("RailsRateLimit::Stores::Memory#record_request error: #{e.message}")
  0
end