foreign_key_column_for has more arguments since rails 7.1

related: rails/rails#47637

lib/ridgepole/ext/schema_dumper.rb

@@ -30,7 +30,7 @@ def foreign_keys(table, stream)
               remove_prefix_and_suffix(foreign_key.to_table).inspect
             ]
 
-            parts << "column: #{foreign_key.column.inspect}" if foreign_key.column != @connection.foreign_key_column_for(foreign_key.to_table)
+            parts << "column: #{foreign_key.column.inspect}" if foreign_key.column != foreign_key_column_for(foreign_key)
 
             parts << "primary_key: #{foreign_key.primary_key.inspect}" if foreign_key.custom_primary_key?
 
@@ -45,6 +45,16 @@ def foreign_keys(table, stream)
           stream.puts add_foreign_key_statements.sort.join("\n")
         end
       end
+
+      private
+
+      def foreign_key_column_for(foreign_key)
+        if ActiveRecord.gem_version < Gem::Version.new('7.1.0')
+          @connection.foreign_key_column_for(foreign_key.to_table)
+        else
+          @connection.foreign_key_column_for(foreign_key.to_table, 'id')
+        end
+      end
     end
   end
 end