Class: Curator::ResettableRiak::DataStore
Instance Method Summary
collapse
#_bucket, #_bucket_name, #_deserialize, #_find_key_by_index, #_normalized_index_data, #client, #delete, #find_by_attribute, #find_by_key, #ping
Instance Method Details
#bucket_prefix ⇒ Object
7
8
9
10
|
# File 'lib/curator/resettable_riak/data_store.rb', line 7
def bucket_prefix
job = "#{ENV['JOB_NAME'].gsub(/[^[:alnum:]]/, '_')}" if ENV['JOB_NAME'].present?
[Curator.config.bucket_prefix, job, Curator.config.environment].compact.join(':')
end
|
#exclude_from_reset(&block) ⇒ Object
12
13
14
15
16
17
|
# File 'lib/curator/resettable_riak/data_store.rb', line 12
def exclude_from_reset(&block)
@exclude_from_reset = true
yield
ensure
@exclude_from_reset = false
end
|
#remove_all_keys ⇒ Object
19
20
21
22
23
24
25
26
27
28
|
# File 'lib/curator/resettable_riak/data_store.rb', line 19
def remove_all_keys
silence_warnings do
buckets = client.buckets.select { |bucket| bucket.name.start_with?(bucket_prefix) }
buckets.each do |bucket|
bucket.keys do |keys|
keys.each { |key| bucket.delete(key) }
end
end
end
end
|
#reset! ⇒ Object
30
31
32
33
34
35
36
37
|
# File 'lib/curator/resettable_riak/data_store.rb', line 30
def reset!
@bucket_names ||= {}
deletable_buckets = @bucket_names.each do |bucket_name, keys|
bucket = _bucket(bucket_name)
keys.each {|key| bucket.delete(key)}
end
@bucket_names = {}
end
|
#save(options) ⇒ Object
39
40
41
42
43
44
45
46
47
48
49
|
# File 'lib/curator/resettable_riak/data_store.rb', line 39
def save(options)
key = super
unless @exclude_from_reset
@bucket_names ||= {}
@bucket_names[options[:collection_name]] ||= []
@bucket_names[options[:collection_name]] << key
end
key
end
|