class Curator::ResettableRiak::DataStore

Public Instance Methods

bucket_prefix() click to toggle source
# 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() { || ... } click to toggle source
# 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() click to toggle source
# 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!() click to toggle source
# 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) click to toggle source
Calls superclass method
# 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