fix issue with order being wrong

[dbsrgits/SQL-Translator-2.0-ish.git] / lib / SQL / Translator / Object / Schema.pm

diff --git a/lib/SQL/Translator/Object/Schema.pm b/lib/SQL/Translator/Object/Schema.pm
index 446c553..80f268d 100644 (file)
--- a/lib/SQL/Translator/Object/Schema.pm
+++ b/lib/SQL/Translator/Object/Schema.pm
@@ -88,7 +88,11 @@ class SQL::Translator::Object::Schema extends SQL::Translator::Object {
         $self->$orig($view->name, $view)
     }
 
-    around add_procedure(Procedure $procedure does coerce) { $self->$orig($procedure->name, $procedure) }
+    around add_procedure(Procedure $procedure does coerce) {
+        $procedure->schema($self);
+        $self->$orig($procedure->name, $procedure) 
+    }
+
     around add_trigger(Trigger $trigger does coerce) {
         $trigger->schema($self);
         $self->$orig($trigger->name, $trigger);;
@@ -108,6 +112,17 @@ class SQL::Translator::Object::Schema extends SQL::Translator::Object {
         $self->$orig($name);
     }
 
+    around remove_trigger(Trigger|Str $trigger) {
+        my $name = is_Trigger($trigger) ? $trigger->name : $trigger;
+####        die "Can't drop non-existant trigger " . $name unless $self->exists_trigger($name);
+        $self->$orig($name);
+    }
+
+    around remove_procedure(Procedure|Str $procedure) {
+        my $name = is_Procedure($procedure) ? $procedure->name : $procedure;
+        $self->$orig($name);
+    }
+
     method order { }
     method perform_action_when { }
     method database_events { }