milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix load timeout (#21156) 

Signed-off-by: lixinguo <xinguo.li@zilliz.com>

Signed-off-by: lixinguo <xinguo.li@zilliz.com>
Co-authored-by: lixinguo <xinguo.li@zilliz.com>
pull/21104/head
smellthemoon 2022-12-12 18:43:23 +08:00 committed by GitHub
parent b0afdb2180
commit b04eb17088
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internal/querycoordv2/observers/collection_observer.go
View File

@ -138,7 +138,6 @@ func (ob *CollectionObserver) observeLoadStatus() {
collections := ob.meta.CollectionManager.GetAllCollections()
for _, collection := range collections {
if collection.LoadPercentage == 100 {
delete(ob.collectionLoadedCount, collection.GetCollectionID())
continue
}
ob.observeCollectionLoadStatus(collection)
@ -150,7 +149,6 @@ func (ob *CollectionObserver) observeLoadStatus() {
}
for _, partition := range partitions {
if partition.LoadPercentage == 100 {
delete(ob.partitionLoadedCount, partition.GetPartitionID())
continue
}
ob.observePartitionLoadStatus(partition)
@ -204,6 +202,7 @@ func (ob *CollectionObserver) observeCollectionLoadStatus(collection *meta.Colle
}
ob.collectionLoadedCount[collection.GetCollectionID()] = loadedCount
if updated.LoadPercentage == 100 {
delete(ob.collectionLoadedCount, collection.GetCollectionID())
ob.targetMgr.UpdateCollectionCurrentTarget(updated.CollectionID)
updated.Status = querypb.LoadStatus_Loaded
ob.meta.CollectionManager.UpdateCollection(updated)
@ -267,6 +266,7 @@ func (ob *CollectionObserver) observePartitionLoadStatus(partition *meta.Partiti
}
ob.partitionLoadedCount[partition.GetPartitionID()] = loadedCount
if updated.LoadPercentage == 100 {
delete(ob.partitionLoadedCount, partition.GetPartitionID())
ob.targetMgr.UpdateCollectionCurrentTarget(partition.GetCollectionID(), partition.GetPartitionID())
updated.Status = querypb.LoadStatus_Loaded
ob.meta.CollectionManager.PutPartition(updated)