From b044e5503e07db59dc0dadb9b99b8bffd1eac194 Mon Sep 17 00:00:00 2001 From: Xiaofan <83447078+xiaofan-luan@users.noreply.github.com> Date: Fri, 10 May 2024 19:29:31 -0700 Subject: [PATCH] enhance:Improve load speed (#32898) fix #32897 add memory check when load collection Signed-off-by: xiaofanluan --- internal/querycoordv2/observers/collection_observer.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/internal/querycoordv2/observers/collection_observer.go b/internal/querycoordv2/observers/collection_observer.go index 2a4b3b55a9..64fe02febc 100644 --- a/internal/querycoordv2/observers/collection_observer.go +++ b/internal/querycoordv2/observers/collection_observer.go @@ -128,6 +128,7 @@ func (ob *CollectionObserver) LoadCollection(ctx context.Context, collectionID i } ob.loadTasks.Insert(key, LoadTask{LoadType: querypb.LoadType_LoadCollection, CollectionID: collectionID}) + ob.checkerController.Check() } func (ob *CollectionObserver) LoadPartitions(ctx context.Context, collectionID int64, partitionIDs []int64) { @@ -140,6 +141,7 @@ func (ob *CollectionObserver) LoadPartitions(ctx context.Context, collectionID i } ob.loadTasks.Insert(key, LoadTask{LoadType: querypb.LoadType_LoadPartition, CollectionID: collectionID, PartitionIDs: partitionIDs}) + ob.checkerController.Check() } func (ob *CollectionObserver) Observe(ctx context.Context) {