From da6342779b3228e1796ed4e3c385b52d60821b0f Mon Sep 17 00:00:00 2001
From: Matthew Draper <matthew@trebex.net>
Date: Sat, 30 Jan 2016 05:45:41 +1030
Subject: Intervene on change_table as well as create_table
---
activerecord/lib/active_record/migration/compatibility.rb | 13 +++++++++++++
1 file changed, 13 insertions(+)
(limited to 'activerecord')
diff --git a/activerecord/lib/active_record/migration/compatibility.rb b/activerecord/lib/active_record/migration/compatibility.rb
index 5d742b523b..45e35a4f71 100644
--- a/activerecord/lib/active_record/migration/compatibility.rb
+++ b/activerecord/lib/active_record/migration/compatibility.rb
@@ -30,6 +30,19 @@ module ActiveRecord
end
end
+ def change_table(table_name, options = {})
+ if block_given?
+ super(table_name, options) do |t|
+ class << t
+ prepend TableDefinition
+ end
+ yield t
+ end
+ else
+ super
+ end
+ end
+
def add_reference(*, **options)
options[:index] ||= false
super
--
cgit v1.2.3