Class: Heirloom::AWS::SimpleDB
- Inherits:
-
Object
- Object
- Heirloom::AWS::SimpleDB
- Defined in:
- lib/heirloom/aws/simpledb.rb
Instance Method Summary collapse
- #count(domain) ⇒ Object
- #create_domain(domain) ⇒ Object
- #delete(domain, key) ⇒ Object
- #delete_domain(domain) ⇒ Object
- #domain_empty?(domain) ⇒ Boolean
- #domain_exists?(domain) ⇒ Boolean
- #domains ⇒ Object
-
#initialize(args) ⇒ SimpleDB
constructor
A new instance of SimpleDB.
- #item_count(domain, item) ⇒ Object
- #logger ⇒ Object
- #put_attributes(domain, key, attributes, options = {}) ⇒ Object
- #select(query, opts = {}) ⇒ Object
Constructor Details
#initialize(args) ⇒ SimpleDB
Returns a new instance of SimpleDB.
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# File 'lib/heirloom/aws/simpledb.rb', line 7 def initialize(args) @config = args[:config] @logger = @config.logger fog_args = { :region => @config. } if @config.use_iam_profile fog_args[:use_iam_profile] = true else fog_args[:aws_access_key_id] = @config.access_key fog_args[:aws_secret_access_key] = @config.secret_key end @sdb = Fog::AWS::SimpleDB.new fog_args end |
Instance Method Details
#count(domain) ⇒ Object
86 87 88 89 |
# File 'lib/heirloom/aws/simpledb.rb', line 86 def count(domain) body = @sdb.select("SELECT count(*) FROM `#{domain}`").body body['Items']['Domain']['Count'].first.to_i end |
#create_domain(domain) ⇒ Object
31 32 33 |
# File 'lib/heirloom/aws/simpledb.rb', line 31 def create_domain(domain) @sdb.create_domain(domain) unless domain_exists?(domain) end |
#delete(domain, key) ⇒ Object
82 83 84 |
# File 'lib/heirloom/aws/simpledb.rb', line 82 def delete(domain, key) @sdb.delete_attributes domain, key end |
#delete_domain(domain) ⇒ Object
35 36 37 |
# File 'lib/heirloom/aws/simpledb.rb', line 35 def delete_domain(domain) @sdb.delete_domain(domain) end |
#domain_empty?(domain) ⇒ Boolean
39 40 41 |
# File 'lib/heirloom/aws/simpledb.rb', line 39 def domain_empty?(domain) count(domain).zero? end |
#domain_exists?(domain) ⇒ Boolean
27 28 29 |
# File 'lib/heirloom/aws/simpledb.rb', line 27 def domain_exists?(domain) domains.include? domain end |
#domains ⇒ Object
23 24 25 |
# File 'lib/heirloom/aws/simpledb.rb', line 23 def domains @sdb.list_domains.body['Domains'] end |
#item_count(domain, item) ⇒ Object
91 92 93 94 |
# File 'lib/heirloom/aws/simpledb.rb', line 91 def item_count(domain, item) query = "SELECT count(*) FROM `#{domain}` WHERE itemName() = '#{item}'" @sdb.select(query).body['Items']['Domain']['Count'].first.to_i end |
#logger ⇒ Object
96 97 98 |
# File 'lib/heirloom/aws/simpledb.rb', line 96 def logger @logger end |
#put_attributes(domain, key, attributes, options = {}) ⇒ Object
43 44 45 |
# File 'lib/heirloom/aws/simpledb.rb', line 43 def put_attributes(domain, key, attributes, = {}) @sdb.put_attributes domain, key, attributes, end |
#select(query, opts = {}) ⇒ Object
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
# File 'lib/heirloom/aws/simpledb.rb', line 47 def select(query, opts = {}) has_more = true next_token = nil results = {} logger.debug "Executing simpledb query '#{query}'." if opts[:offset] && opts[:offset] > 0 limit = @sdb.select("#{query} limit #{opts[:offset]}").body if limit['NextToken'] logger.debug "Next token found. Retrieving results." next_token = limit['NextToken'] else logger.debug "No more results. Query complete." has_more = false end end while has_more logger.debug "Retrieving results from next token '#{next_token}'." if next_token more = @sdb.select(query, 'NextToken' => next_token).body more['Items'].each do |k, v| block_given? ? yield(k, v) : results[k] = v end if more['NextToken'] logger.debug "Next token found. Retrieving results." next_token = more['NextToken'] else logger.debug "No more results. Query complete." has_more = false end end results unless block_given? end |