class Rack::MiniProfiler::MemcacheStore

Constants

EXPIRES_IN_SECONDS
MAX_RETRIES

Public Class Methods

new(args = nil) click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 8
def initialize(args = nil)
  require 'dalli' unless defined? Dalli
  args ||= {}
  @prefix             = args[:prefix]     || "MPMemcacheStore"
  @prefix             += "-#{Rack::MiniProfiler::VERSION}"
  @client             = args[:client]     || Dalli::Client.new
  @expires_in_seconds = args[:expires_in] || EXPIRES_IN_SECONDS
end

Public Instance Methods

allowed_tokens() click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 58
def allowed_tokens

  token_info = @client.get("#{@prefix}-tokens")
  key1, key2, cycle_at = nil


  if token_info
     key1, key2, cycle_at = Marshal::load(token_info)

     key1 = nil unless key1 && key1.length == 32
     key2 = nil unless key2 && key2.length == 32

     if key1 && cycle_at && (cycle_at > Time.now)
        return [key1,key2].compact
     end
  end

  timeout = Rack::MiniProfiler::AbstractStore::MAX_TOKEN_AGE

  # cycle keys
  key2 = key1
  key1 = SecureRandom.hex
  cycle_at = Time.now + timeout

  @client.set("#{@prefix}-tokens", Marshal::dump([key1, key2, cycle_at]))

  [key1, key2].compact
end
flush_tokens() click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 54
def flush_tokens
  @client.set("#{@prefix}-tokens", nil)
end
get_unviewed_ids(user) click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 50
def get_unviewed_ids(user)
  @client.get("#{@prefix}-#{user}-v") || []
end
load(id) click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 21
def load(id)
  raw = @client.get("#{@prefix}#{id}")
  Marshal::load(raw) if raw
end
save(page_struct) click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 17
def save(page_struct)
  @client.set("#{@prefix}#{page_struct[:id]}", Marshal::dump(page_struct), @expires_in_seconds)
end
set_all_unviewed(user, ids) click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 46
def set_all_unviewed(user, ids)
  @client.set("#{@prefix}-#{user}-v", ids, @expires_in_seconds)
end
set_unviewed(user, id) click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 26
def set_unviewed(user, id)
  @client.add("#{@prefix}-#{user}-v", [], @expires_in_seconds)
  MAX_RETRIES.times do
    break if @client.cas("#{@prefix}-#{user}-v", @expires_in_seconds) do |ids|
      ids << id unless ids.include?(id)
      ids
    end
  end
end
set_viewed(user, id) click to toggle source
# File lib/mini_profiler/storage/memcache_store.rb, line 36
def set_viewed(user, id)
  @client.add("#{@prefix}-#{user}-v", [], @expires_in_seconds)
  MAX_RETRIES.times do
    break if @client.cas("#{@prefix}-#{user}-v", @expires_in_seconds) do |ids|
      ids.delete id
      ids
    end
  end
end