refactor: heap reset

keilerkonzept · Sep 26, 2024 · f7bb0a2 · f7bb0a2
1 parent a989134
commit f7bb0a2
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/heap/heap.go b/heap/heap.go
@@ -167,3 +167,11 @@ func (me *Min) Update(item string, fingerprint uint32, count uint32) bool {
 	heap.Fix(me, 0)
 	return true
 }
+
+// Reset resets the heap.
+func (me *Min) Reset() {
+	clear(me.Items)
+	clear(me.Index)
+	me.StoredKeysBytes = 0
+	me.Items = me.Items[:0]
+}
diff --git a/sketch.go b/sketch.go
@@ -210,6 +210,5 @@ func (me *Sketch) SortedSlice() []heap.Item {
 // Reset resets the sketch to an empty state.
 func (me *Sketch) Reset() {
 	clear(me.Buckets)
-	clear(me.Heap.Items)
-	clear(me.Heap.Index)
+	me.Heap.Reset()
 }
diff --git a/sliding/sketch.go b/sliding/sketch.go
@@ -290,6 +290,5 @@ func (me *Sketch) Reset() {
 		clear(me.Buckets[i].Counts)
 	}
 	clear(me.Buckets)
-	clear(me.Heap.Items)
-	clear(me.Heap.Index)
+	me.Heap.Reset()
 }