2018-11-06 04:45:35 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2017-02-07 17:06:08 +00:00
|
|
|
module Gitlab
|
|
|
|
module EtagCaching
|
|
|
|
class Store
|
2017-03-23 11:23:13 +00:00
|
|
|
EXPIRY_TIME = 20.minutes
|
2017-07-11 03:35:47 +00:00
|
|
|
SHARED_STATE_NAMESPACE = 'etag:'.freeze
|
2017-02-07 17:06:08 +00:00
|
|
|
|
|
|
|
def get(key)
|
2017-07-11 03:35:47 +00:00
|
|
|
Gitlab::Redis::SharedState.with { |redis| redis.get(redis_shared_state_key(key)) }
|
2017-02-07 17:06:08 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def touch(key, only_if_missing: false)
|
|
|
|
etag = generate_etag
|
|
|
|
|
2017-07-11 03:35:47 +00:00
|
|
|
Gitlab::Redis::SharedState.with do |redis|
|
|
|
|
redis.set(redis_shared_state_key(key), etag, ex: EXPIRY_TIME, nx: only_if_missing)
|
2017-02-07 17:06:08 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
etag
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
def generate_etag
|
|
|
|
SecureRandom.hex
|
|
|
|
end
|
|
|
|
|
2017-07-11 03:35:47 +00:00
|
|
|
def redis_shared_state_key(key)
|
2017-06-12 07:09:51 +00:00
|
|
|
raise 'Invalid key' if !Rails.env.production? && !Gitlab::EtagCaching::Router.match(key)
|
|
|
|
|
2017-07-11 03:35:47 +00:00
|
|
|
"#{SHARED_STATE_NAMESPACE}#{key}"
|
2017-02-07 17:06:08 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|