lib/ridgepole/ext/schema_dumper.rb in ridgepole-0.6.4.beta5 vs lib/ridgepole/ext/schema_dumper.rb in ridgepole-0.6.4.beta6

@@ -1,32 +1,41 @@
 require 'active_record/schema_dumper'
 
-class ActiveRecord::SchemaDumper
-  def foreign_keys_with_default_name(table, stream)
-    if (foreign_keys = @connection.foreign_keys(table)).any?
-      add_foreign_key_statements = foreign_keys.map do |foreign_key|
-        parts = [
-          "add_foreign_key #{remove_prefix_and_suffix(foreign_key.from_table).inspect}",
-          remove_prefix_and_suffix(foreign_key.to_table).inspect,
-        ]
+module Ridgepole
+  module Ext
+    module SchemaDumper
+      def foreign_keys(table, stream)
+        if (foreign_keys = @connection.foreign_keys(table)).any?
+          add_foreign_key_statements = foreign_keys.map do |foreign_key|
+            parts = [
+              "add_foreign_key #{remove_prefix_and_suffix(foreign_key.from_table).inspect}",
+              remove_prefix_and_suffix(foreign_key.to_table).inspect,
+            ]
 
-        if foreign_key.column != @connection.foreign_key_column_for(foreign_key.to_table)
-          parts << "column: #{foreign_key.column.inspect}"
-        end
+            if foreign_key.column != @connection.foreign_key_column_for(foreign_key.to_table)
+              parts << "column: #{foreign_key.column.inspect}"
+            end
 
-        if foreign_key.custom_primary_key?
-          parts << "primary_key: #{foreign_key.primary_key.inspect}"
-        end
+            if foreign_key.custom_primary_key?
+              parts << "primary_key: #{foreign_key.primary_key.inspect}"
+            end
 
-        parts << "name: #{foreign_key.name.inspect}"
+            parts << "name: #{foreign_key.name.inspect}"
 
-        parts << "on_update: #{foreign_key.on_update.inspect}" if foreign_key.on_update
-        parts << "on_delete: #{foreign_key.on_delete.inspect}" if foreign_key.on_delete
+            parts << "on_update: #{foreign_key.on_update.inspect}" if foreign_key.on_update
+            parts << "on_delete: #{foreign_key.on_delete.inspect}" if foreign_key.on_delete
 
-        "  #{parts.join(', ')}"
-      end
+            "  #{parts.join(', ')}"
+          end
 
-      stream.puts add_foreign_key_statements.sort.join("\n")
+          stream.puts add_foreign_key_statements.sort.join("\n")
+        end
+      end
     end
   end
-  alias_method_chain :foreign_keys, :default_name
+end
+
+module ActiveRecord
+  class SchemaDumper
+    prepend Ridgepole::Ext::SchemaDumper
+  end
 end