lib/rom/sql/schema/dsl.rb in rom-sql-0.9.1 vs lib/rom/sql/schema/dsl.rb in rom-sql-1.0.0.beta1

@@ -1,19 +1,22 @@
+require 'rom/sql/type'
 require 'rom/sql/schema/inferrer'
 require 'rom/sql/schema/associations_dsl'
 
 module ROM
   module SQL
     class Schema < ROM::Schema
       class DSL < ROM::Schema::DSL
         attr_reader :associations_dsl
 
         def associations(&block)
-          @associations_dsl = AssociationsDSL.new(name, &block)
+          @associations_dsl = AssociationsDSL.new(relation, &block)
         end
 
         def call
-          SQL::Schema.new(name, attributes, opts)
+          SQL::Schema.define(
+            relation, opts.merge(attributes: attributes.values, type_class: SQL::Type)
+          )
         end
 
         def opts
           opts = { inferrer: inferrer }