enhance:Improve load speed (milvus-io#32898)

fix milvus-io#32897 add memory check when load collection Signed-off-by: xiaofanluan <[email protected]>
congqixia · May 11, 2024 · b044e55 · b044e55
1 parent c0e62e6
commit b044e55
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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) {