From 1a7912e300edb09d35759042d7dccfeedfb2e18b Mon Sep 17 00:00:00 2001
From: yah01 <yang.cen@zilliz.com>
Date: Fri, 11 Nov 2022 16:17:04 +0800
Subject: [PATCH] Fix checkFullLoaded always returns all partitions as unloaded
(#20514)
Signed-off-by: yah01 <yang.cen@zilliz.com>
Signed-off-by: yah01 <yang.cen@zilliz.com>
---
internal/proxy/task_statistic.go | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/internal/proxy/task_statistic.go b/internal/proxy/task_statistic.go
index 3cfe1dc413..16cfefde36 100644
--- a/internal/proxy/task_statistic.go
+++ b/internal/proxy/task_statistic.go
@@ -343,8 +343,9 @@ func checkFullLoaded(ctx context.Context, qc types.QueryCoord, collectionName st
for i, percentage := range resp.GetInMemoryPercentages() {
if percentage >= 100 {
loadedPartitionIDs = append(loadedPartitionIDs, resp.GetPartitionIDs()[i])
+ } else {
+ unloadPartitionIDs = append(unloadPartitionIDs, resp.GetPartitionIDs()[i])
}
- unloadPartitionIDs = append(unloadPartitionIDs, resp.GetPartitionIDs()[i])
}
return loadedPartitionIDs, unloadPartitionIDs, nil
}