Commit d7bae1a

@@ -76,7 +76,7 @@ module Humble
 
     def apply(row, entity, session)
       puts "#{@attribute} #{@type} #{row} #{entity}"
-      items = session.find_all(@type).to_a
+      items = session.find_all(@type)
       entity.public_send("#{@attribute}=", items)
     end
 

@@ -1,15 +1,17 @@
-class DefaultMapper
-  attr_reader :session, :table
+module Humble
+  class DefaultMapper
+    attr_reader :session, :table
 
-  def initialize(table, session)
-    @table = table
-    @session = session
-  end
+    def initialize(table, session)
+      @table = table
+      @session = session
+    end
 
-  def map_from(row)
-    table.type.new.tap do |entity|
-      table.each do |column|
-        column.apply(row, entity, session)
+    def map_from(row)
+      table.type.new.tap do |entity|
+        table.each do |column|
+          column.apply(row, entity, session)
+        end
       end
     end
   end

@@ -1,14 +1,16 @@
-class IdentityMap
-  def initialize(items = {})
-    @items = items
-  end
+module Humble
+  class IdentityMap
+    def initialize(items = {})
+      @items = items
+    end
 
-  def fetch(key, &block)
-    if @items.key?(key)
-      @items[key]
-    else
-      @items[key] = block.call
-      @items[key]
+    def fetch(key, &block)
+      if @items.key?(key)
+        @items[key]
+      else
+        @items[key] = block.call
+        @items[key]
+      end
     end
   end
 end

@@ -28,7 +28,7 @@ module Humble
     def items
       @items ||= @rows.map do |row|
         @mapper.map_from(row)
-      end.to_a
+      end
     end
   end
 end