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

Fix querynode stop meet unexpected GetComponentStates (#22590) 

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
pull/22583/head
congqixia 2023-03-07 09:51:50 +08:00 committed by GitHub
parent 29b3690c83
commit f3e2d4a39a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
internal/querycoordv2/mocks/querynode.go
View File

@ -24,6 +24,7 @@ import (
"time" "time"
"github.com/milvus-io/milvus-proto/go-api/commonpb" "github.com/milvus-io/milvus-proto/go-api/commonpb"
"github.com/milvus-io/milvus-proto/go-api/milvuspb"
"github.com/milvus-io/milvus/internal/log" "github.com/milvus-io/milvus/internal/log"
querypb "github.com/milvus-io/milvus/internal/proto/querypb" querypb "github.com/milvus-io/milvus/internal/proto/querypb"
. "github.com/milvus-io/milvus/internal/querycoordv2/params" . "github.com/milvus-io/milvus/internal/querycoordv2/params"
@ -111,6 +112,22 @@ func (node *MockQueryNode) Start() error {
segment.GetSegmentID()) segment.GetSegmentID())
node.segmentVersion[segment.GetSegmentID()] = req.GetVersion() node.segmentVersion[segment.GetSegmentID()] = req.GetVersion()
}).Return(successStatus, nil).Maybe() }).Return(successStatus, nil).Maybe()
node.EXPECT().GetComponentStates(mock.Anything, mock.AnythingOfType("*milvuspb.GetComponentStatesRequest")).
Call.Return(func(context.Context, *milvuspb.GetComponentStatesRequest) *milvuspb.ComponentStates {
select {
case <-node.ctx.Done():
return nil
default:
return &milvuspb.ComponentStates{}
}
}, func(context.Context, *milvuspb.GetComponentStatesRequest) error {
select {
case <-node.ctx.Done():
return grpc.ErrServerStopped
default:
return nil
}
}).Maybe()
// Register // Register
node.session.Init(typeutil.QueryNodeRole, node.addr, false, true) node.session.Init(typeutil.QueryNodeRole, node.addr, false, true)