admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix mock querynode server session not revoked (#17229)

Browse Source 
Revoke mock querynode server session when it's stopped
This PR reduces the running time of TestLoadBalanceIndexedSegmentsAfterNodeDown from 60+ seconds to less than 1+ seconds
Also related to #17212 #17215

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
This commit is contained in:
congqixia 2022-06-18 18:22:12 +08:00 committed by GitHub
parent 144f7a2d80
commit 50cecc65ed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internal/querycoord/mock_querynode_server_test.go
View File

@ -22,6 +22,7 @@ import (
"net"
"strconv"
"sync"
"time"
"go.uber.org/zap"
"google.golang.org/grpc"
@ -172,6 +173,9 @@ func (qs *queryNodeServerMock) start() error {
func (qs *queryNodeServerMock) stop() error {
qs.cancel()
if qs.session != nil {
qs.session.Revoke(time.Second)
}
if qs.grpcServer != nil {
qs.grpcServer.GracefulStop()
}