Class: DynamoidAdvancedWhere::QueryMaterializer

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/dynamoid_advanced_where/query_materializer.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(query_builder:) ⇒ QueryMaterializer

Returns a new instance of QueryMaterializer.



16
17
18
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 16

def initialize(query_builder:)
  self.query_builder = query_builder
end

Instance Attribute Details

#query_builder ⇒ Object

Returns the value of attribute query_builder.



8
9
10
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 8

def query_builder
  @query_builder
end

Instance Method Details

#all ⇒ Object 



20
21
22
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 20

def all
  each.to_a
end

#each(&blk) ⇒ Object Also known as: find_each 



24
25
26
27
28
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 24

def each(&blk)
  return enum_for(:each) unless blk

  records.each(&blk)
end

#each_page(&blk) ⇒ Object 



31
32
33
34
35
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 31

def each_page(&blk)
  return enum_for(:each_page) unless blk

  pages.each(&blk)
end

#each_page_via_query ⇒ Object 



49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 49

def each_page_via_query
  query = {
    table_name: table_name
  }.merge(filter_builder.to_query_filter)

  page_start = start_hash

  Enumerator.new do |yielder|
    loop do
      results = client.query(query.merge(exclusive_start_key: page_start))

      items = (results.items || []).each do |item|
        klass.from_database(item.symbolize_keys)
      end

      yielder.yield(items, results)

      (page_start = results.last_evaluated_key) || break
    end
  end.lazy
end

#each_page_via_scan ⇒ Object 



71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 71

def each_page_via_scan
  query = {
    table_name: table_name
  }.merge(filter_builder.to_scan_filter)

  page_start = start_hash

  Enumerator.new do |yielder|
    loop do
      results = client.scan(query.merge(exclusive_start_key: page_start))

      items = (results.items || []).map do |item|
        klass.from_database(item.symbolize_keys)
      end

      yielder.yield(items, results)

      (page_start = results.last_evaluated_key) || break
    end
  end.lazy
end

#filter_builder ⇒ Object 



93
94
95
96
97
98
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 93

def filter_builder
  @filter_builder ||= FilterBuilder.new(
    root_node: query_builder.root_node,
    klass: klass,
  )
end

#pages ⇒ Object 



41
42
43
44
45
46
47
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 41

def pages
  if must_scan?
    each_page_via_scan
  else
    each_page_via_query
  end
end

#records ⇒ Object 



37
38
39
 
# File 'lib/dynamoid_advanced_where/query_materializer.rb', line 37

def records
  pages.flat_map { |i, _| i }
end