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
19
 
# 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, bucketing_key, treatments_labels_change_numbers, time) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
 
# File 'lib/cache/repositories/impressions/redis_repository.rb', line 21

def add_bulk(key, bucketing_key, treatments_labels_change_numbers, time)
  @adapter.redis.pipelined do
    treatments_labels_change_numbers.each_slice(IMPRESSIONS_SLICE) do |treatments_labels_change_numbers_slice|
      treatments_labels_change_numbers_slice.each do |split_name, treatment_label_change_number|
        add(split_name,
          'key_name' => key,
          'bucketing_key' => bucketing_key,
          'treatment' => treatment_label_change_number[:treatment],
          'label' => @config.labels_enabled ? treatment_label_change_number[:label] : nil,
          'change_number' => treatment_label_change_number[:change_number],
          '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



40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/cache/repositories/impressions/redis_repository.rb', line 40

def clear
  impressions = impression_keys.each_with_object([]) do |key, memo|
    _, _, ip, = key.split('/')
    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' },
        ip: ip
      }
    end

    @adapter.delete_from_set(key, members)
  end

  impressions
end