2009-05-19 02:06:49 -04:00
|
|
|
module Fog
|
|
|
|
module AWS
|
|
|
|
class SimpleDB
|
2009-06-05 19:51:17 -04:00
|
|
|
|
2009-09-08 00:25:50 -04:00
|
|
|
if Fog.mocking?
|
|
|
|
def self.reset_data
|
|
|
|
@data = { :domains => {} }
|
|
|
|
end
|
|
|
|
def self.data
|
|
|
|
@data
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-08-10 23:43:31 -04:00
|
|
|
def self.reload
|
2009-09-14 00:24:22 -04:00
|
|
|
load "fog/aws/parsers/simpledb/basic.rb"
|
|
|
|
load "fog/aws/parsers/simpledb/domain_metadata.rb"
|
|
|
|
load "fog/aws/parsers/simpledb/get_attributes.rb"
|
|
|
|
load "fog/aws/parsers/simpledb/list_domains.rb"
|
|
|
|
load "fog/aws/parsers/simpledb/select.rb"
|
2009-08-10 23:43:31 -04:00
|
|
|
|
2009-09-14 00:24:22 -04:00
|
|
|
load "fog/aws/requests/simpledb/batch_put_attributes.rb"
|
|
|
|
load "fog/aws/requests/simpledb/create_domain.rb"
|
|
|
|
load "fog/aws/requests/simpledb/delete_attributes.rb"
|
|
|
|
load "fog/aws/requests/simpledb/delete_domain.rb"
|
|
|
|
load "fog/aws/requests/simpledb/domain_metadata.rb"
|
|
|
|
load "fog/aws/requests/simpledb/get_attributes.rb"
|
|
|
|
load "fog/aws/requests/simpledb/list_domains.rb"
|
|
|
|
load "fog/aws/requests/simpledb/put_attributes.rb"
|
|
|
|
load "fog/aws/requests/simpledb/select.rb"
|
2009-09-08 00:25:50 -04:00
|
|
|
|
|
|
|
if Fog.mocking?
|
|
|
|
reset_data
|
|
|
|
end
|
2009-08-10 23:43:31 -04:00
|
|
|
end
|
|
|
|
|
2009-05-19 02:06:49 -04:00
|
|
|
# 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
|
2009-07-07 22:20:54 -04:00
|
|
|
# sdb = SimpleDB.new(
|
|
|
|
# :aws_access_key_id => your_aws_access_key_id,
|
|
|
|
# :aws_secret_access_key => your_aws_secret_access_key
|
|
|
|
# )
|
2009-05-19 02:06:49 -04:00
|
|
|
#
|
|
|
|
# ==== Parameters
|
2009-07-07 22:20:54 -04:00
|
|
|
# * options<~Hash> - config arguments for connection. Defaults to {}.
|
2009-05-19 02:06:49 -04:00
|
|
|
#
|
|
|
|
# ==== Returns
|
2009-07-07 22:20:54 -04:00
|
|
|
# * SimpleDB object with connection to aws.
|
2009-05-19 02:06:49 -04:00
|
|
|
def initialize(options={})
|
|
|
|
@aws_access_key_id = options[:aws_access_key_id]
|
|
|
|
@aws_secret_access_key = options[:aws_secret_access_key]
|
|
|
|
@hmac = HMAC::SHA256.new(@aws_secret_access_key)
|
|
|
|
@host = options[:host] || 'sdb.amazonaws.com'
|
|
|
|
@nil_string = options[:nil_string]|| 'nil'
|
|
|
|
@port = options[:port] || 443
|
|
|
|
@scheme = options[:scheme] || 'https'
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
def encode_attributes(attributes, replace_attributes = [])
|
|
|
|
encoded_attributes = {}
|
2009-06-25 21:02:00 -04:00
|
|
|
if attributes
|
|
|
|
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
|
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
2009-06-25 21:02:00 -04:00
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
encoded_attributes
|
|
|
|
end
|
|
|
|
|
|
|
|
def encode_attribute_names(attributes)
|
2010-01-31 17:07:26 -05:00
|
|
|
AWS.indexed_param('AttributeName', attributes.map {|attribute| attributes.to_s})
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
|
|
|
|
2009-07-11 16:41:21 -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
|
|
|
|
end
|
|
|
|
item_index += 1
|
|
|
|
end
|
|
|
|
end
|
2009-06-25 21:02:00 -04:00
|
|
|
end
|
2009-07-11 16:41:21 -04:00
|
|
|
encoded_attributes
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def request(params, parser)
|
2010-01-27 23:22:22 -05:00
|
|
|
@connection = Fog::Connection.new("#{@scheme}://#{@host}:#{@port}")
|
2009-05-19 02:06:49 -04:00
|
|
|
params.merge!({
|
|
|
|
'AWSAccessKeyId' => @aws_access_key_id,
|
|
|
|
'SignatureMethod' => 'HmacSHA256',
|
|
|
|
'SignatureVersion' => '2',
|
|
|
|
'Timestamp' => Time.now.utc.strftime("%Y-%m-%dT%H:%M:%SZ"),
|
|
|
|
'Version' => '2007-11-07'
|
|
|
|
})
|
|
|
|
|
2009-06-28 01:55:58 -04:00
|
|
|
body = ''
|
2009-06-25 21:02:00 -04:00
|
|
|
for key in params.keys.sort
|
|
|
|
unless (value = params[key]).nil?
|
2009-07-02 15:38:35 -04:00
|
|
|
body << "#{key}=#{CGI.escape(value.to_s).gsub(/\+/, '%20')}&"
|
2009-06-25 21:02:00 -04:00
|
|
|
end
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
|
|
|
|
2009-06-28 01:55:58 -04:00
|
|
|
string_to_sign = "POST\n#{@host}\n/\n" << body.chop
|
2009-05-19 02:06:49 -04:00
|
|
|
hmac = @hmac.update(string_to_sign)
|
2009-06-28 01:55:58 -04:00
|
|
|
body << "Signature=#{CGI.escape(Base64.encode64(hmac.digest).chomp!).gsub(/\+/, '%20')}"
|
2009-06-05 19:51:17 -04:00
|
|
|
|
2009-06-22 01:13:01 -04:00
|
|
|
response = @connection.request({
|
2009-06-28 01:55:58 -04:00
|
|
|
:body => body,
|
2009-07-14 19:04:39 -04:00
|
|
|
:expects => 200,
|
2009-06-28 01:55:58 -04:00
|
|
|
:headers => { 'Content-Type' => 'application/x-www-form-urlencoded' },
|
2009-06-24 23:48:51 -04:00
|
|
|
:host => @host,
|
2009-07-10 18:40:11 -04:00
|
|
|
:method => 'POST',
|
|
|
|
:parser => parser
|
2009-06-22 01:13:01 -04:00
|
|
|
})
|
2009-06-11 13:25:05 -04:00
|
|
|
|
2009-05-20 21:49:20 -04:00
|
|
|
response
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
|
|
|
|
2009-07-11 16:41:21 -04:00
|
|
|
def sdb_encode(value)
|
|
|
|
if value.nil?
|
|
|
|
@nil_string
|
|
|
|
else
|
|
|
|
value.to_s
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-05-19 02:06:49 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2009-08-17 18:11:53 -04:00
|
|
|
|
|
|
|
Fog::AWS::SimpleDB.reload
|