summaryrefslogtreecommitdiff
path: root/lib/gitlab/import_export/json/streaming_serializer.rb
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.rb15
1 files changed, 12 insertions, 3 deletions
diff --git a/lib/gitlab/import_export/json/streaming_serializer.rb b/lib/gitlab/import_export/json/streaming_serializer.rb
index d1e013a151c..9d28e1abeab 100644
--- a/lib/gitlab/import_export/json/streaming_serializer.rb
+++ b/lib/gitlab/import_export/json/streaming_serializer.rb
@@ -31,10 +31,12 @@ module Gitlab
end
def execute
- serialize_root
+ read_from_replica_if_available do
+ serialize_root
- includes.each do |relation_definition|
- serialize_relation(relation_definition)
+ includes.each do |relation_definition|
+ serialize_relation(relation_definition)
+ end
end
end
@@ -166,6 +168,13 @@ module Gitlab
)
])
end
+
+ def read_from_replica_if_available(&block)
+ return yield unless ::Feature.enabled?(:load_balancing_for_export_workers, type: :development, default_enabled: :yaml)
+ return yield unless ::Gitlab::Database::LoadBalancing.enable?
+
+ ::Gitlab::Database::LoadBalancing::Session.current.use_replicas_for_read_queries(&block)
+ end
end
end
end