Fix bug with Rails 5.2

See ka8725/migration_data#34 (comment)
westonganger · Dec 27, 2018 · d55bb74 · d55bb74
1 parent 873e520
commit d55bb74
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/active_record/annotate/dumper.rb b/lib/active_record/annotate/dumper.rb
@@ -16,7 +16,11 @@ def dump(table_name, connection = ActiveRecord::Base.connection)
 
       private
         def dumper(connection)
-          ActiveRecord::SchemaDumper.send(:new, connection)
+          if connection.respond_to?(:create_schema_dumper)
+            connection.create_schema_dumper(generate_options(config))
+          else
+            ActiveRecord::SchemaDumper.send(:new, connection)
+          end
         end
 
         def process_annotation(string_io)