1
0
Fork 0
mirror of https://github.com/fog/fog.git synced 2022-11-09 13:51:43 -05:00
fog--fog/lib/fog/aws/simpledb.rb

199 lines
6.3 KiB
Ruby
Raw Normal View History

require File.expand_path(File.join(File.dirname(__FILE__), '..', 'aws'))
2009-05-19 02:06:49 -04:00
module Fog
module AWS
class SimpleDB < Fog::Service
2009-06-05 19:51:17 -04:00
requires :aws_access_key_id, :aws_secret_access_key
recognizes :host, :nil_string, :path, :port, :scheme, :persistent, :region
2010-06-12 18:31:17 -04:00
request_path 'fog/aws/requests/simpledb'
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
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 = nil
end
def initialize(options={})
2010-03-17 19:07:25 -04:00
@aws_access_key_id = options[:aws_access_key_id]
2011-05-19 18:35:33 -04:00
end
def data
self.class.data[@aws_access_key_id]
end
def reset_data
self.class.data.delete(@aws_access_key_id)
2009-09-08 00:25:50 -04:00
end
2009-05-19 02:06:49 -04:00
end
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={})
2011-02-16 20:25:50 -05:00
require 'fog/core/parser'
@aws_access_key_id = options[:aws_access_key_id]
@aws_secret_access_key = options[:aws_secret_access_key]
@connection_options = options[:connection_options] || {}
2010-06-16 00:04:16 -04:00
@hmac = Fog::HMAC.new('sha256', @aws_secret_access_key)
@nil_string = options[:nil_string]|| 'nil'
options[:region] ||= 'us-east-1'
@host = options[:host] || case options[:region]
when 'ap-northeast-1'
'sdb.ap-northeast-1.amazonaws.com'
when 'ap-southeast-1'
'sdb.ap-southeast-1.amazonaws.com'
when 'eu-west-1'
'sdb.eu-west-1.amazonaws.com'
when 'us-east-1'
2011-03-10 19:34:55 -05:00
'sdb.amazonaws.com'
when 'us-west-1'
'sdb.us-west-1.amazonaws.com'
else
raise ArgumentError, "Unknown region: #{options[:region].inspect}"
end
@path = options[:path] || '/'
@persistent = options[:persistent] || false
@port = options[:port] || 443
@scheme = options[:scheme] || 'https'
@connection = Fog::Connection.new("#{@scheme}://#{@host}:#{@port}#{@path}", @persistent, @connection_options)
end
private
def encode_attributes(attributes, replace_attributes = [], expected_attributes = {})
encoded_attributes = {}
if attributes
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
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
end
2009-05-19 02:06:49 -04:00
end
encoded_attributes
end
2009-05-19 02:06:49 -04:00
def encode_attribute_names(attributes)
Fog::AWS.indexed_param('AttributeName', attributes.map {|attribute| attributes.to_s})
end
2009-05-19 02:06:49 -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
attribute_index = 0
for attribute_key in items[item_key].keys
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
end
item_index += 1
end
end
encoded_attributes
end
2009-05-19 02:06:49 -04:00
def reload
@connection.reset
end
def request(params)
idempotent = params.delete(:idempotent)
parser = params.delete(:parser)
body = Fog::AWS.signed_params(
params,
{
:aws_access_key_id => @aws_access_key_id,
:hmac => @hmac,
:host => @host,
:path => @path,
:port => @port,
:version => '2009-04-15'
}
)
2009-06-05 19:51:17 -04:00
response = @connection.request({
:body => body,
:expects => 200,
:headers => { 'Content-Type' => 'application/x-www-form-urlencoded; charset=utf-8' },
:host => @host,
:idempotent => idempotent,
:method => 'POST',
:parser => parser
})
2009-06-11 13:25:05 -04:00
response
end
2009-05-19 02:06:49 -04:00
def sdb_encode(value)
if value.nil?
@nil_string
else
value.to_s
end
end
end
2009-05-19 02:06:49 -04:00
end
end
end