From 47c9e77155a8023c64d25cf1ceedde18ced9c706 Mon Sep 17 00:00:00 2001 From: Jiquan Long Date: Fri, 1 Jul 2022 10:58:18 +0800 Subject: [PATCH] Fix querycoord ut data race (#17980) Signed-off-by: longjiquan --- internal/querycoord/mock_querynode_server_test.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/internal/querycoord/mock_querynode_server_test.go b/internal/querycoord/mock_querynode_server_test.go index a5026f6844..be5fc406f1 100644 --- a/internal/querycoord/mock_querynode_server_test.go +++ b/internal/querycoord/mock_querynode_server_test.go @@ -280,11 +280,13 @@ func (qs *queryNodeServerMock) GetMetrics(ctx context.Context, req *milvuspb.Get } totalMemUsage := uint64(0) + globalSegInfosMutex.RLock() for _, info := range qs.segmentInfos { if nodeIncluded(qs.queryNodeID, info.NodeIds) { totalMemUsage += uint64(info.MemSize) } } + globalSegInfosMutex.RUnlock() nodeInfos := metricsinfo.QueryNodeInfos{ BaseComponentInfos: metricsinfo.BaseComponentInfos{ Name: metricsinfo.ConstructComponentName(typeutil.QueryNodeRole, qs.queryNodeID),