fix data race

Signed-off-by: Wei Liu <[email protected]>

internal/querycoordv2/balance/score_based_balancer.go

@@ -47,7 +47,7 @@ type ScoreBasedBalancer struct {
 	broker meta.Broker
 
 	// collectionID -> cachedRecordSizeInfo
-	cachedMaxRecordSize map[int64]*cachedRecordSizeInfo
+	cachedMaxRecordSize *typeutil.ConcurrentMap[int64, *cachedRecordSizeInfo]
 }
 
 func NewScoreBasedBalancer(scheduler task.Scheduler,
@@ -60,7 +60,7 @@ func NewScoreBasedBalancer(scheduler task.Scheduler,
 	return &ScoreBasedBalancer{
 		RowCountBasedBalancer: NewRowCountBasedBalancer(scheduler, nodeManager, dist, meta, targetMgr),
 		broker:                broker,
-		cachedMaxRecordSize:   make(map[int64]*cachedRecordSizeInfo),
+		cachedMaxRecordSize:   typeutil.NewConcurrentMap[int64, *cachedRecordSizeInfo](),
 	}
 }
 
@@ -230,7 +230,7 @@ func (b *ScoreBasedBalancer) calculateScore(collectionID, nodeID int64) int {
 }
 
 func (b *ScoreBasedBalancer) estimateRecordSize(collectionID int64) int {
-	cacheInfo, ok := b.cachedMaxRecordSize[collectionID]
+	cacheInfo, ok := b.cachedMaxRecordSize.Get(collectionID)
 	if ok && time.Since(cacheInfo.ts) < 300*time.Second {
 		return cacheInfo.size
 	}
@@ -251,7 +251,7 @@ func (b *ScoreBasedBalancer) estimateRecordSize(collectionID int64) int {
 		// if get schema failed, set record size to 4096 by default
 		recordSize = 4096
 	}
-	b.cachedMaxRecordSize[collectionID] = &cachedRecordSizeInfo{size: recordSize, ts: time.Now()}
+	b.cachedMaxRecordSize.Insert(collectionID, &cachedRecordSizeInfo{size: recordSize, ts: time.Now()})
 	return recordSize
 }