blob: 8b4197cfff997dd71aca9f9a31744ea6846aa109 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
# frozen_string_literal: true
module Gitlab
module ApplicationRateLimiter
class IncrementPerActionedResource < BaseStrategy
def initialize(resource_key)
@resource_key = resource_key
end
def increment(cache_key, expiry)
with_redis do |redis|
redis.pipelined do
redis.sadd(cache_key, resource_key)
redis.expire(cache_key, expiry)
redis.scard(cache_key)
end.last
end
end
def read(cache_key)
with_redis do |redis|
redis.scard(cache_key)
end
end
private
attr_accessor :resource_key
end
end
end
|