Merge cockroachdb#126933

126933: raft: pre-size progress maps r=nvanbenschoten a=nvanbenschoten Drive-by cleanup to pre-size the progress maps when constructing raft `Status` and `SparseStatus` structs. Epic: None Release note: None Co-authored-by: Nathan VanBenschoten <[email protected]>
navsetlur · Jul 10, 2024 · 0639ffb · 0639ffb
2 parents b73328d + 1697b3b
commit 0639ffb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/raft/status.go b/pkg/raft/status.go
@@ -66,7 +66,7 @@ func withProgress(r *raft, visitor func(id pb.PeerID, typ ProgressType, pr track
 }
 
 func getProgressCopy(r *raft) map[pb.PeerID]tracker.Progress {
-	m := make(map[pb.PeerID]tracker.Progress)
+	m := make(map[pb.PeerID]tracker.Progress, len(r.trk.Progress))
 	r.trk.Visit(func(id pb.PeerID, pr *tracker.Progress) {
 		p := *pr
 		p.Inflights = pr.Inflights.Clone()
@@ -107,7 +107,7 @@ func getSparseStatus(r *raft) SparseStatus {
 		BasicStatus: getBasicStatus(r),
 	}
 	if status.RaftState == StateLeader {
-		status.Progress = map[pb.PeerID]tracker.Progress{}
+		status.Progress = make(map[pb.PeerID]tracker.Progress, len(r.trk.Progress))
 		withProgress(r, func(id pb.PeerID, _ ProgressType, pr tracker.Progress) {
 			status.Progress[id] = pr
 		})