2009-05-19 02:06:49 -04:00
|
|
|
module Fog
|
|
|
|
module AWS
|
2010-09-03 04:11:45 -04:00
|
|
|
class SimpleDB < Fog::Service
|
2009-06-05 19:51:17 -04:00
|
|
|
|
2010-12-06 23:00:04 -05:00
|
|
|
requires :aws_access_key_id, :aws_secret_access_key, &inject_parameter_specs
|
|
|
|
recognizes :host, :nil_string, :path, :port, :scheme, :persistent, &inject_parameter_specs
|
2010-11-21 05:56:41 -05:00
|
|
|
|
2010-06-12 18:31:17 -04:00
|
|
|
request_path 'fog/aws/requests/simpledb'
|
2010-09-02 19:01:19 -04:00
|
|
|
request :batch_put_attributes
|
|
|
|
request :create_domain
|
|
|
|
request :delete_attributes
|
|
|
|
request :delete_domain
|
|
|
|
request :domain_metadata
|
|
|
|
request :get_attributes
|
|
|
|
request :list_domains
|
|
|
|
request :put_attributes
|
|
|
|
request :select
|
2010-03-17 19:07:25 -04:00
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
class Mock
|
2009-09-08 00:25:50 -04:00
|
|
|
|
2010-03-17 19:07:25 -04:00
|
|
|
def self.data
|
|
|
|
@data ||= Hash.new do |hash, key|
|
|
|
|
hash[key] = {
|
|
|
|
:domains => {}
|
|
|
|
}
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def self.reset_data(keys=data.keys)
|
|
|
|
for key in [*keys]
|
|
|
|
data.delete(key)
|
|
|
|
end
|
2010-03-14 23:11:43 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def initialize(options={})
|
2010-03-17 19:07:25 -04:00
|
|
|
@aws_access_key_id = options[:aws_access_key_id]
|
|
|
|
@data = self.class.data[@aws_access_key_id]
|
2009-09-08 00:25:50 -04:00
|
|
|
end
|
2009-08-10 23:43:31 -04:00
|
|
|
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
class Real
|
|
|
|
|
|
|
|
# Initialize connection to SimpleDB
|
|
|
|
#
|
|
|
|
# ==== Notes
|
|
|
|
# options parameter must include values for :aws_access_key_id and
|
|
|
|
# :aws_secret_access_key in order to create a connection
|
|
|
|
#
|
|
|
|
# ==== Examples
|
|
|
|
# sdb = SimpleDB.new(
|
|
|
|
# :aws_access_key_id => your_aws_access_key_id,
|
|
|
|
# :aws_secret_access_key => your_aws_secret_access_key
|
|
|
|
# )
|
|
|
|
#
|
|
|
|
# ==== Parameters
|
|
|
|
# * options<~Hash> - config arguments for connection. Defaults to {}.
|
|
|
|
#
|
|
|
|
# ==== Returns
|
|
|
|
# * SimpleDB object with connection to aws.
|
|
|
|
def initialize(options={})
|
|
|
|
@aws_access_key_id = options[:aws_access_key_id]
|
|
|
|
@aws_secret_access_key = options[:aws_secret_access_key]
|
2010-06-16 00:04:16 -04:00
|
|
|
@hmac = Fog::HMAC.new('sha256', @aws_secret_access_key)
|
2010-03-14 23:11:43 -04:00
|
|
|
@host = options[:host] || 'sdb.amazonaws.com'
|
|
|
|
@nil_string = options[:nil_string]|| 'nil'
|
2010-09-29 20:53:32 -04:00
|
|
|
@path = options[:path] || '/'
|
2010-03-14 23:11:43 -04:00
|
|
|
@port = options[:port] || 443
|
|
|
|
@scheme = options[:scheme] || 'https'
|
2010-09-29 20:53:32 -04:00
|
|
|
@connection = Fog::Connection.new("#{@scheme}://#{@host}:#{@port}#{@path}", options[:persistent])
|
2010-03-14 23:11:43 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
2010-06-21 19:56:04 -04:00
|
|
|
def encode_attributes(attributes, replace_attributes = [], expected_attributes = {})
|
2010-03-14 23:11:43 -04:00
|
|
|
encoded_attributes = {}
|
|
|
|
if attributes
|
2010-06-02 17:40:25 -04:00
|
|
|
|
|
|
|
expected_attributes.keys.each_with_index do |exkey, index|
|
|
|
|
for value in Array(expected_attributes[exkey])
|
|
|
|
encoded_attributes["Expected.#{index}.Name"] = exkey.to_s
|
|
|
|
encoded_attributes["Expected.#{index}.Value"] = sdb_encode(value)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
index = 0
|
|
|
|
for key in attributes.keys
|
|
|
|
for value in Array(attributes[key])
|
|
|
|
encoded_attributes["Attribute.#{index}.Name"] = key.to_s
|
|
|
|
if replace_attributes.include?(key)
|
|
|
|
encoded_attributes["Attribute.#{index}.Replace"] = 'true'
|
|
|
|
end
|
|
|
|
encoded_attributes["Attribute.#{index}.Value"] = sdb_encode(value)
|
|
|
|
index += 1
|
2009-06-25 21:02:00 -04:00
|
|
|
end
|
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
2010-03-14 23:11:43 -04:00
|
|
|
encoded_attributes
|
2009-06-25 21:02:00 -04:00
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
def encode_attribute_names(attributes)
|
|
|
|
AWS.indexed_param('AttributeName', attributes.map {|attribute| attributes.to_s})
|
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
def encode_batch_attributes(items, replace_attributes = Hash.new([]))
|
|
|
|
encoded_attributes = {}
|
|
|
|
if items
|
|
|
|
item_index = 0
|
|
|
|
for item_key in items.keys
|
|
|
|
encoded_attributes["Item.#{item_index}.ItemName"] = item_key.to_s
|
|
|
|
for attribute_key in items[item_key].keys
|
|
|
|
attribute_index = 0
|
|
|
|
for value in Array(items[item_key][attribute_key])
|
|
|
|
encoded_attributes["Item.#{item_index}.Attribute.#{attribute_index}.Name"] = attribute_key.to_s
|
|
|
|
if replace_attributes[item_key].include?(attribute_key)
|
|
|
|
encoded_attributes["Item.#{item_index}.Attribute.#{attribute_index}.Replace"] = 'true'
|
|
|
|
end
|
|
|
|
encoded_attributes["Item.#{item_index}.Attribute.#{attribute_index}.Value"] = sdb_encode(value)
|
|
|
|
attribute_index += 1
|
2009-07-11 16:41:21 -04:00
|
|
|
end
|
2010-03-14 23:11:43 -04:00
|
|
|
item_index += 1
|
2009-07-11 16:41:21 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2010-03-14 23:11:43 -04:00
|
|
|
encoded_attributes
|
2009-06-25 21:02:00 -04:00
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
|
2010-06-19 21:56:38 -04:00
|
|
|
def reload
|
|
|
|
@connection.reset
|
|
|
|
end
|
|
|
|
|
2010-03-16 00:58:57 -04:00
|
|
|
def request(params)
|
2010-03-16 01:15:33 -04:00
|
|
|
idempotent = params.delete(:idempotent)
|
2010-03-16 00:58:57 -04:00
|
|
|
parser = params.delete(:parser)
|
|
|
|
|
2010-05-02 16:25:39 -04:00
|
|
|
body = AWS.signed_params(
|
|
|
|
params,
|
|
|
|
{
|
|
|
|
:aws_access_key_id => @aws_access_key_id,
|
|
|
|
:hmac => @hmac,
|
|
|
|
:host => @host,
|
2010-09-29 20:53:32 -04:00
|
|
|
:path => @path,
|
2010-06-02 17:40:25 -04:00
|
|
|
:version => '2009-04-15'
|
2010-05-02 16:25:39 -04:00
|
|
|
}
|
|
|
|
)
|
2009-06-05 19:51:17 -04:00
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
response = @connection.request({
|
2010-03-16 01:15:33 -04:00
|
|
|
:body => body,
|
|
|
|
:expects => 200,
|
|
|
|
:headers => { 'Content-Type' => 'application/x-www-form-urlencoded' },
|
|
|
|
:host => @host,
|
|
|
|
:idempotent => idempotent,
|
|
|
|
:method => 'POST',
|
|
|
|
:parser => parser
|
2010-03-14 23:11:43 -04:00
|
|
|
})
|
2009-06-11 13:25:05 -04:00
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
response
|
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
def sdb_encode(value)
|
|
|
|
if value.nil?
|
|
|
|
@nil_string
|
|
|
|
else
|
|
|
|
value.to_s
|
|
|
|
end
|
2009-07-11 16:41:21 -04:00
|
|
|
end
|
|
|
|
|
2010-03-14 23:11:43 -04:00
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|