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

fix transfer node meta err (#22420) 

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
pull/22401/head
wei liu 2023-02-27 19:29:47 +08:00 committed by GitHub
parent 167581c08a
commit d433e95ce0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/querycoordv2/meta/resource_manager.go
View File

@ -591,7 +591,7 @@ func (rm *ResourceManager) transferNodeInStore(from string, to string, numNode i
}
toCapacity := rm.groups[to].GetCapacity()
if from != DefaultResourceGroupName {
if to != DefaultResourceGroupName {
// default rg capacity won't be changed
toCapacity = rm.groups[to].GetCapacity() + numNode
}

7
internal/querycoordv2/meta/resource_manager_test.go
View File

@ -196,8 +196,10 @@ func (suite *ResourceManagerSuite) TestRecover() {
err = suite.manager.AddResourceGroup("rg2")
suite.NoError(err)
suite.manager.AssignNode("rg1", 1)
suite.manager.AssignNode("rg2", 2)
suite.manager.AssignNode(DefaultResourceGroupName, 1)
suite.manager.TransferNode(DefaultResourceGroupName, "rg1", 1)
suite.manager.AssignNode(DefaultResourceGroupName, 2)
suite.manager.TransferNode(DefaultResourceGroupName, "rg2", 1)
suite.manager.AssignNode(DefaultResourceGroupName, 3)
suite.manager.AssignNode(DefaultResourceGroupName, 4)
@ -214,7 +216,6 @@ func (suite *ResourceManagerSuite) TestRecover() {
suite.NoError(err)
suite.Equal(1, rg.GetCapacity())
suite.True(suite.manager.ContainsNode("rg1", 1))
print(suite.manager.GetNodes("rg1"))
rg, err = suite.manager.GetResourceGroup("rg2")
suite.NoError(err)