Single matrix memory allocation in DeepCopy (#1390)

Signed-off-by: Artur Melanchyk <[email protected]>
DiceDB · Jan 7, 2025 · c7e9710 · c7e9710
1 parent 9723444
commit c7e9710
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/eval/countminsketch.go b/internal/eval/countminsketch.go
@@ -199,8 +199,9 @@ func (c *CountMinSketch) DeepCopy() *CountMinSketch {
 
 	// Deep copy the matrix
 	matrix := make([][]uint64, c.opts.depth)
+	flatMatrix := make([]uint64, c.opts.depth*c.opts.width) // single memory allocation
 	for row := uint64(0); row < c.opts.depth; row++ {
-		matrix[row] = make([]uint64, c.opts.width)
+		matrix[row] = flatMatrix[row*c.opts.width : (row+1)*c.opts.width : (row+1)*c.opts.width]
 		copy(matrix[row], c.matrix[row])
 	}