diff options
Diffstat (limited to 'config/initializers/active_record_fix_insert_all.rb')
-rw-r--r-- | config/initializers/active_record_fix_insert_all.rb | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/config/initializers/active_record_fix_insert_all.rb b/config/initializers/active_record_fix_insert_all.rb new file mode 100644 index 00000000000..8ae208dd0e5 --- /dev/null +++ b/config/initializers/active_record_fix_insert_all.rb @@ -0,0 +1,26 @@ +# frozen_string_literal: true + +# This fix is needed to properly support +# columns that perform data mutation to a SQL datatype +# ex. would be `jsonb` and `enum` +# +# This is covered by tests in `BulkInsertSafe` +# that validates handling of different data types + +if Rails.gem_version > Gem::Version.new("6.0.2") + raise Gem::DependencyError, + "Remove patch once the https://github.com/rails/rails/pull/38763 is included" +end + +module ActiveRecordInsertAllBuilderMixin + def extract_types_from_columns_on(table_name, keys:) + columns = connection.schema_cache.columns_hash(table_name) + + unknown_column = (keys - columns.keys).first + raise UnknownAttributeError.new(model.new, unknown_column) if unknown_column + + keys.index_with { |key| model.type_for_attribute(key) } + end +end + +ActiveRecord::InsertAll::Builder.prepend(ActiveRecordInsertAllBuilderMixin) |