Class: SplitIoClient::Cache::Repositories::Impressions::RedisRepository

Inherits:
SplitIoClient::Cache::Repository show all
Defined in:
lib/cache/repositories/impressions/redis_repository.rb

Constant Summary collapse

IMPRESSIONS_SLICE = 
1000

Instance Method Summary collapse

Methods inherited from SplitIoClient::Cache::Repository

#set_string, #string

Constructor Details

#initialize(adapter, config) ⇒ RedisRepository

Returns a new instance of RedisRepository.



8
9
10
11
 
# File 'lib/cache/repositories/impressions/redis_repository.rb', line 8

def initialize(adapter, config)
  @adapter = adapter
  @config = config
end

Instance Method Details

#add(split_name, data) ⇒ Object

Store impression data in Redis



14
15
16
17
18
 
# File 'lib/cache/repositories/impressions/redis_repository.rb', line 14

def add(split_name, data)
  @adapter.add_to_set(
    namespace_key("impressions.#{split_name}"), data.merge(split_name: split_name).to_json
  )
end

#add_bulk(key, treatments, time) ⇒ Object 



20
21
22
23
24
25
26
27
28
 
# File 'lib/cache/repositories/impressions/redis_repository.rb', line 20

def add_bulk(key, treatments, time)
  @adapter.redis.pipelined do
    treatments.each_slice(IMPRESSIONS_SLICE) do |treatments_slice|
      treatments_slice.each do |split_name, treatment|
        add(split_name, 'key_name' => key, 'treatment' => treatment, 'time' => time)
      end
    end
  end
end

#clear ⇒ Object

Get random impressions from redis in batches of size @config.impressions_queue_size, delete fetched impressions afterwards



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/cache/repositories/impressions/redis_repository.rb', line 32

def clear
  impressions = impression_keys.each_with_object([]) do |key, memo|
    members = @adapter.random_set_elements(key, @config.impressions_queue_size)
    members.each do |impression|
      parsed_impression = JSON.parse(impression)

      memo << {
        feature: parsed_impression['split_name'],
        impressions: parsed_impression.reject { |k, _| k == 'split_name' }
      }
    end

    @adapter.delete_from_set(key, members)
  end

  impressions
end