refactor add newTTLKeyHeap function

shailvipx · Nov 5, 2013 · c05df9e · c05df9e
1 parent b9593b8
commit c05df9e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/store/heap_test.go b/store/heap_test.go
@@ -8,8 +8,7 @@ import (
 )
 
 func TestHeapPushPop(t *testing.T) {
-	h := &TTLKeyHeap{Map: make(map[*Node]int)}
-	heap.Init(h)
+	h := newTTLKeyHeap()
 
 	// add from older expire time to earlier expire time
 	// the path is equal to ttl from now

diff --git a/store/ttl_key_heap.go b/store/ttl_key_heap.go
@@ -10,6 +10,12 @@ type TTLKeyHeap struct {
 	Map   map[*Node]int
 }
 
+func newTTLKeyHeap() *TTLKeyHeap {
+	h := &TTLKeyHeap{Map: make(map[*Node]int)}
+	heap.Init(h)
+	return h
+}
+
 func (h TTLKeyHeap) Len() int {
 	return len(h.Array)
 }