60 lines
1.7 KiB
Ruby
60 lines
1.7 KiB
Ruby
require 'pp'
|
|
module Fog
|
|
module VcloudDirector
|
|
module Query
|
|
|
|
def find_by_query(options={})
|
|
type = options.fetch(:type) { self.query_type }
|
|
|
|
results = get_all_results(type, options)
|
|
data = results.map do |query_record|
|
|
model_data = {}
|
|
model_data[:id] = query_record[:href].split('/').last
|
|
model_data[:name] = query_record.fetch(:name) if query_record.key?(:name)
|
|
if self.methods.include?(:populate_model_from_query_record)
|
|
model_data.merge(self.populate_model_from_query_record(query_record))
|
|
else
|
|
model_data
|
|
end
|
|
end
|
|
load(data)
|
|
end
|
|
|
|
private
|
|
|
|
def get_all_results(type, options)
|
|
results = []
|
|
if options.key?(:page)
|
|
page_range = [ Integer(options[:page]) ]
|
|
else
|
|
page_range = (1..get_num_pages(type, options))
|
|
end
|
|
page_range.each do |page|
|
|
results += get_results_page(page, type, options) || []
|
|
end
|
|
results
|
|
end
|
|
|
|
def get_num_pages(type, options)
|
|
body = service.get_execute_query(type, options)
|
|
last_page = body[:lastPage] || 1
|
|
raise "Invalid lastPage (#{last_page}) in query results" unless last_page.is_a? Integer
|
|
last_page.to_i
|
|
end
|
|
|
|
def get_results_page(page, type, options)
|
|
body = service.get_execute_query(type, options.merge({:page=>page})).body
|
|
|
|
record_key = key_of_first_record_or_reference(body)
|
|
body[record_key] = [body[record_key]] if body[record_key].is_a?(Hash)
|
|
body[record_key]
|
|
end
|
|
|
|
def key_of_first_record_or_reference(body)
|
|
body.keys.detect { |key| key.to_s =~ /Record|Reference$/ }
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|