Welcome to mirror list, hosted at ThFree Co, Russian Federation.

gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'lib/gitlab/import_export/json/streaming_serializer.rb')
-rw-r--r--lib/gitlab/import_export/json/streaming_serializer.rb18
1 files changed, 18 insertions, 0 deletions
diff --git a/lib/gitlab/import_export/json/streaming_serializer.rb b/lib/gitlab/import_export/json/streaming_serializer.rb
index ebabf537ce5..59396c6bad2 100644
--- a/lib/gitlab/import_export/json/streaming_serializer.rb
+++ b/lib/gitlab/import_export/json/streaming_serializer.rb
@@ -70,7 +70,11 @@ module Gitlab
batch = batch.preload(key_preloads) if key_preloads
batch.each do |record|
+ before_read_callback(record)
+
items << Raw.new(record.to_json(options))
+
+ after_read_callback(record)
end
end
end
@@ -168,6 +172,20 @@ module Gitlab
def read_from_replica_if_available(&block)
::Gitlab::Database::LoadBalancing::Session.current.use_replicas_for_read_queries(&block)
end
+
+ def before_read_callback(record)
+ remove_cached_external_diff(record)
+ end
+
+ def after_read_callback(record)
+ remove_cached_external_diff(record)
+ end
+
+ def remove_cached_external_diff(record)
+ return unless record.is_a?(MergeRequest)
+
+ record.merge_request_diff&.remove_cached_external_diff
+ end
end
end
end