Class: ActiveRecordApi::Request::TokenCache

Inherits:
Object
  • Object
show all
Includes:
ActiveAttr::Model
Defined in:
lib/active_record_api/request/token_cache.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#debug ⇒ Object

Returns the value of attribute debug.



6
7
8
 
# File 'lib/active_record_api/request/token_cache.rb', line 6

def debug
  @debug
end

Instance Method Details

#acquire_lock ⇒ Object 



67
68
69
70
71
72
 
# File 'lib/active_record_api/request/token_cache.rb', line 67

def acquire_lock
  @redis_client.set 'auth_token_lock', unique_key, nx: true, ex: 60
  current_lock_value = @redis_client.get 'auth_token_lock'
  Rails.logger.info "#{current_lock_value} == #{unique_key}" if debug
  current_lock_value == unique_key
end

#debug_log(message) ⇒ Object 



63
64
65
 
# File 'lib/active_record_api/request/token_cache.rb', line 63

def debug_log(message)
  Rails.logger.info message if debug
end

#fetch_token(token_proc) ⇒ Object 



19
20
21
 
# File 'lib/active_record_api/request/token_cache.rb', line 19

def fetch_token(token_proc)
  retrieve_token_from_redis(token_proc)
end

#flush ⇒ Object 



8
9
10
11
12
13
14
15
16
17
 
# File 'lib/active_record_api/request/token_cache.rb', line 8

def flush
  return unless defined? AUTH_REDIS_POOL
  debug_log 'flushing token cache'
  Honeybadger.notify(StandardError.new('token cache has been flushed'))
  AUTH_REDIS_POOL.with do |client|
    @redis_client = client
    @redis_client.del('auth_token')
    release_lock
  end
end

#release_lock ⇒ Object 



74
75
76
 
# File 'lib/active_record_api/request/token_cache.rb', line 74

def release_lock
  @redis_client.del 'auth_token_lock'
end

#retrieve_token_from_redis(token_proc) ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/active_record_api/request/token_cache.rb', line 23

def retrieve_token_from_redis(token_proc)
  debug_log 'looking?'
  unless defined? AUTH_REDIS_POOL
    debug_log 'no auth redis pool'
    Honeybadger.notify(StandardError.new('no auth redis pool for token cache'))
    return token_proc.call()
  end
  debug_log 'looking for token in cache'
  AUTH_REDIS_POOL.with do |client|
    @redis_client = client
    @token = nil
    @retry = 0
    while @token.nil? && @retry < 5
      @retry += 1
      @token = @redis_client.get('auth_token')
      debug_log "retry ##{@retry}"
      debug_log "found token: #{@token}"
      if @token.nil?
        if acquire_lock
          begin
            debug_log "acquired lock"
            Honeybadger.notify(StandardError.new('acquired lock and fetching token for cache'))
            @token = token_proc.call()
            raise "no token returned! #{@token}" if @token.nil?
            @redis_client.set 'auth_token', @token
          ensure
            debug_log 'releasing lock'
            release_lock
          end
        else
          debug_log 'waiting for token to be cached'
          sleep 3
        end
      end
    end
  end
  raise "no token found after waiting #{@retry} times" if @token.nil?
  @token
end

#unique_key ⇒ Object 



78
79
80
 
# File 'lib/active_record_api/request/token_cache.rb', line 78

def unique_key
  @unique_key ||= rand.to_s
end