golang safe heap

heap.go

package main

import (
	"container/heap"
	"sync"
)

// SafeHeap implement a concurrent safe heap.
type SafeHeap struct {
	sync.RWMutex
	h heap.Interface
}

// Push x into heap.
func (h *SafeHeap) Push(x interface{}) {
	h.Lock()
	defer h.Unlock()
	heap.Push(h.h, x)
}

// Pop the minimum or maximum.
func (h *SafeHeap) Pop() interface{} {
	h.Lock()
	defer h.Unlock()
	return heap.Pop(h.h)
}

// Len get length.
func (h *SafeHeap) Len() int {
	h.RLock()
	defer h.RUnlock()
	return h.h.Len()
}

// New create new safe heap.
func New(h heap.Interface) *SafeHeap {
	sh := &SafeHeap{h: h}
	heap.Init(sh.h)
	return sh
}