From b8fccf89e0a13c895217d38b32c7f704d3380c99 Mon Sep 17 00:00:00 2001
From: rkumar555 <rkumar@gitlab.com>
Date: Mon, 22 Jul 2024 14:15:43 +0200
Subject: [PATCH] Set reserved_storage_bytes in the node_assignment
Set the value of reserved_storage_bytes to the space required in the
node.
Changelog: added
MR: https://gitlab.com/gitlab-org/gitlab/-/merge_requests/159736
EE: true
---
ee/app/services/search/zoekt/scheduling_service.rb | 3 ++-
ee/spec/services/search/zoekt/scheduling_service_spec.rb | 1 +
2 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/ee/app/services/search/zoekt/scheduling_service.rb b/ee/app/services/search/zoekt/scheduling_service.rb
index 915272990afd5..9b88656e7e905 100644
--- a/ee/app/services/search/zoekt/scheduling_service.rb
+++ b/ee/app/services/search/zoekt/scheduling_service.rb
@@ -227,7 +227,8 @@ def node_assignment
namespace_id: zoekt_enabled_namespace.root_namespace_id,
zoekt_node_id: node.id,
zoekt_enabled_namespace: zoekt_enabled_namespace,
- replica: Replica.for_enabled_namespace!(zoekt_enabled_namespace)
+ replica: Replica.for_enabled_namespace!(zoekt_enabled_namespace),
+ reserved_storage_bytes: space_required
)
zoekt_index.state = :ready if Feature.disabled?(:zoekt_initial_indexing_task)
zoekt_indices << zoekt_index
diff --git a/ee/spec/services/search/zoekt/scheduling_service_spec.rb b/ee/spec/services/search/zoekt/scheduling_service_spec.rb
index ca5ecf0acc546..eaa95e0f0c470 100644
--- a/ee/spec/services/search/zoekt/scheduling_service_spec.rb
+++ b/ee/spec/services/search/zoekt/scheduling_service_spec.rb
@@ -336,6 +336,7 @@
index = zkt_enabled_namespace2.indices.last
expect(index).not_to be_nil
expect(index.namespace_id).to eq zkt_enabled_namespace2.root_namespace_id
+ expect(index.reserved_storage_bytes).not_to be_nil
expect(index).to be_pending
end
--
GitLab