From a960cc2a1bfe421fdd7f5b8337865a2a1d514e85 Mon Sep 17 00:00:00 2001
From: congqixia <congqi.xia@zilliz.com>
Date: Tue, 16 May 2023 15:17:23 +0800
Subject: [PATCH] Fix data race in pipeline/manager (#24127)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
---
 internal/querynodev2/pipeline/manager.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/internal/querynodev2/pipeline/manager.go b/internal/querynodev2/pipeline/manager.go
index ba67753dd..a4a3aa6ab 100644
--- a/internal/querynodev2/pipeline/manager.go
+++ b/internal/querynodev2/pipeline/manager.go
@@ -49,10 +49,12 @@ type manager struct {
 
 	tSafeManager TSafeManager
 	dispatcher   msgdispatcher.Client
-	mu           sync.Mutex
+	mu           sync.RWMutex
 }
 
 func (m *manager) Num() int {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
 	return len(m.channel2Pipeline)
 }
 
@@ -162,6 +164,5 @@ func NewManager(dataManager *DataManager,
 		delegators:       delegators,
 		tSafeManager:     tSafeManager,
 		dispatcher:       dispatcher,
-		mu:               sync.Mutex{},
 	}
 }
-- 
GitLab