expose O(1) len

wantlist/wantlist.go

@@ -8,12 +8,13 @@ import (
 
 type ThreadSafe struct {
 	lk       sync.RWMutex
-	Wantlist
+	Wantlist Wantlist
 }
 
 // not threadsafe
 type Wantlist struct {
 	set map[u.Key]Entry
+	// TODO provide O(1) len accessor if cost becomes an issue
 }
 
 type Entry struct {
@@ -74,6 +75,16 @@ func (w *ThreadSafe) SortedEntries() []Entry {
 	return w.Wantlist.SortedEntries()
 }
 
+func (w *ThreadSafe) Len() int {
+	w.lk.RLock()
+	defer w.lk.RUnlock()
+	return w.Wantlist.Len()
+}
+
+func (w *Wantlist) Len() int {
+	return len(w.set)
+}
+
 func (w *Wantlist) Add(k u.Key, priority int) {
 	if _, ok := w.set[k]; ok {
 		return