Add parse method for DBI parser

diff --git a/lib/SQL/Translator/Parser/DBI.pm b/lib/SQL/Translator/Parser/DBI.pm
index 42550d0..a93ad3e 100644 (file)
--- a/lib/SQL/Translator/Parser/DBI.pm
+++ b/lib/SQL/Translator/Parser/DBI.pm
@@ -5,6 +5,7 @@ role SQL::Translator::Parser::DBI {
     use DBI::Const::GetInfoReturn;
 
     use MooseX::Types::Moose qw(HashRef Maybe Str);
+    use MooseX::MultiMethods;
 
     use SQL::Translator::Object::Column;
     use SQL::Translator::Object::ForeignKey;
@@ -144,4 +145,10 @@ role SQL::Translator::Parser::DBI {
         $index->add_column($table->get_column($_)) for @index_cols;
         $table->add_index($index);
     }
+
+    multi method parse(Schema $data) { $data }
+
+    multi method parse(Any $) {
+        $self->_add_tables($self->schema);
+    }
 }