Add latest changes from gitlab-org/gitlab@master

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)