summaryrefslogtreecommitdiff
path: root/lib/gitlab/etag_caching/store.rb
blob: 2395e7be0263688a0a1b3a50a06411c9e18aa0c7 (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
32
33
34
35
36
# frozen_string_literal: true

module Gitlab
  module EtagCaching
    class Store
      EXPIRY_TIME = 20.minutes
      SHARED_STATE_NAMESPACE = 'etag:'.freeze

      def get(key)
        Gitlab::Redis::SharedState.with { |redis| redis.get(redis_shared_state_key(key)) }
      end

      def touch(key, only_if_missing: false)
        etag = generate_etag

        Gitlab::Redis::SharedState.with do |redis|
          redis.set(redis_shared_state_key(key), etag, ex: EXPIRY_TIME, nx: only_if_missing)
        end

        etag
      end

      private

      def generate_etag
        SecureRandom.hex
      end

      def redis_shared_state_key(key)
        raise 'Invalid key' if !Rails.env.production? && !Gitlab::EtagCaching::Router.match(key)

        "#{SHARED_STATE_NAMESPACE}#{key}"
      end
    end
  end
end