Fix data race in pipeline/manager (#24127)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
This commit is contained in:
congqixia 2023-05-16 15:17:23 +08:00 committed by GitHub
parent 74bba2320a
commit a960cc2a1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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{},
}
}