lib/rom/boot.rb in rom-0.4.0 vs lib/rom/boot.rb in rom-0.4.1

@@ -37,10 +37,13 @@
     #     end
     #   end
     #
     # @api public
     def schema(&block)
-      @schema = DSL.new(self).schema(&block)
+      base_relations = DSL.new(self).schema(&block)
+      base_relations.each do |repo, relations|
+        (@schema[repo] ||= []).concat(relations)
+      end
       self
     end
 
     # Relation definition DSL
     #