Class: Curator::Riak::DataStore

Inherits:
Object
  • Object
show all
Defined in:
lib/curator/riak/data_store.rb

Direct Known Subclasses

Curator::ResettableRiak::DataStore

Instance Method Summary collapse

Instance Method Details

#_bucket(name) ⇒ Object 



68
69
70
 
# File 'lib/curator/riak/data_store.rb', line 68

def _bucket(name)
  client.bucket(_bucket_name(name))
end

#_bucket_name(name) ⇒ Object 



72
73
74
 
# File 'lib/curator/riak/data_store.rb', line 72

def _bucket_name(name)
  bucket_prefix + ":" + name
end

#_deserialize(data) ⇒ Object 



76
77
78
79
80
81
 
# File 'lib/curator/riak/data_store.rb', line 76

def _deserialize(data)
  deserialized_data = data.dup
  deserialized_data["created_at"] = Time.parse(data["created_at"]) if data["created_at"]
  deserialized_data["updated_at"] = Time.parse(data["updated_at"]) if data["updated_at"]
  deserialized_data
end

#_find_key_by_index(bucket, index_name, query) ⇒ Object 



83
84
85
 
# File 'lib/curator/riak/data_store.rb', line 83

def _find_key_by_index(bucket, index_name, query)
  bucket.get_index("#{index_name}_bin", query)
end

#_normalized_index_data(index_data) ⇒ Object 



87
88
89
90
91
92
93
 
# File 'lib/curator/riak/data_store.rb', line 87

def _normalized_index_data(index_data)
  if index_data.is_a?(Array)
    index_data.join(", ")
  else
    index_data
  end
end

#bucket_prefix ⇒ Object 



15
16
17
 
# File 'lib/curator/riak/data_store.rb', line 15

def bucket_prefix
  "#{Curator.config.bucket_prefix}:#{Curator.config.environment}"
end

#client ⇒ Object 



7
8
9
10
11
12
13
 
# File 'lib/curator/riak/data_store.rb', line 7

def client
  return @client if @client
  return @client = Curator.config.client if Curator.config.client

  yml_config = YAML.load(File.read(Curator.config.riak_config_file))[Curator.config.environment]
  @client = ::Riak::Client.new(yml_config)
end

#delete(bucket_name, key) ⇒ Object 



19
20
21
22
23
 
# File 'lib/curator/riak/data_store.rb', line 19

def delete(bucket_name, key)
  bucket = _bucket(bucket_name)
  object = bucket.get(key)
  object.delete
end

#find_all(bucket_name) ⇒ Object 



41
42
43
44
 
# File 'lib/curator/riak/data_store.rb', line 41

def find_all(bucket_name)
  bucket = _bucket(bucket_name)
  bucket.keys.map { |key| find_by_key(bucket_name, key) }
end

#find_by_attribute(bucket_name, index_name, query) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/curator/riak/data_store.rb', line 56

def find_by_attribute(bucket_name, index_name, query)
  return [] if query.nil?

  bucket = _bucket(bucket_name)
  begin
    keys = _find_key_by_index(bucket, index_name.to_s, query)
    keys.map { |key| find_by_key(bucket_name, key) }
  rescue ::Riak::HTTPFailedRequest => failed_request
    raise failed_request unless failed_request.not_found?
  end
end

#find_by_key(bucket_name, key) ⇒ Object 



46
47
48
49
50
51
52
53
54
 
# File 'lib/curator/riak/data_store.rb', line 46

def find_by_key(bucket_name, key)
  bucket = _bucket(bucket_name)
  begin
    object = bucket.get(key)
    { :key => object.key, :data => _deserialize(object.data) } unless object.data.empty?
  rescue ::Riak::HTTPFailedRequest => failed_request
    raise failed_request unless failed_request.not_found?
  end
end

#ping ⇒ Object 



25
26
27
 
# File 'lib/curator/riak/data_store.rb', line 25

def ping
  client.ping
end

#save(options) ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/curator/riak/data_store.rb', line 29

def save(options)
  bucket = _bucket(options[:collection_name])
  object = ::Riak::RObject.new(bucket, options[:key])
  object.content_type = options.fetch(:content_type, "application/json")
  object.data = options[:value]
  options.fetch(:index, {}).each do |index_name, index_data|
    object.indexes["#{index_name}_bin"] << _normalized_index_data(index_data)
  end
  result = object.store
  result.key
end