Class: Kiba::Uncommon::Sources::ZOQL

Inherits:
Object
  • Object
show all
Defined in:
lib/kiba/uncommon/sources/zoql.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(zuora_client, zoql_query) ⇒ ZOQL

Returns a new instance of ZOQL.



18
19
20
21
22
23
 
# File 'lib/kiba/uncommon/sources/zoql.rb', line 18

def initialize(zuora_client, zoql_query)
  @zuora_client = zuora_client
  @zoql_query = zoql_query

  Hashie.logger = Logger.new(nil)
end

Instance Attribute Details

#zoql_query ⇒ Object (readonly)

Returns the value of attribute zoql_query.



15
16
17
 
# File 'lib/kiba/uncommon/sources/zoql.rb', line 15

def zoql_query
  @zoql_query
end

#zuora_client ⇒ Object (readonly)

Returns the value of attribute zuora_client.



14
15
16
 
# File 'lib/kiba/uncommon/sources/zoql.rb', line 14

def zuora_client
  @zuora_client
end

Instance Method Details

#each ⇒ Object 



26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/kiba/uncommon/sources/zoql.rb', line 26

def each
  results = query

  loop do
    results.records.each do |result|
      yield result
    end

    break if results['done'] == 'true'
    results = query(results['query_locator'])
  end
end

#query(query_locator = nil) ⇒ Object 



40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/kiba/uncommon/sources/zoql.rb', line 40

def query(query_locator=nil)
  unless query_locator
    response = zuora_client.call! :query, zoql_query
    result = response.to_h['envelope']['body']['query_response']['result']
  else
    response = zuora_client.call! :query_more, query_locator
    result = response.to_h['envelope']['body']['query_more_response']['result']
  end

  # Remove XML fields
  result.records.each do |record|
    record.delete(:ns2)
    record.delete(:type)
    record.delete(:xsi)
  end

  result
end