check length when call announcements.put

fanpei91 · Jan 27, 2019 · 51e915a · 51e915a
1 parent cf4b494
commit 51e915a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dht.go b/dht.go
@@ -51,8 +51,13 @@ func (a *announcements) get() *announcement {
 
 func (a *announcements) put(ac *announcement) {
 	a.mu.Lock()
+	defer a.mu.Unlock()
+
+	if a.ll.Len() >= a.limit {
+		return
+	}
+
 	a.ll.PushBack(ac)
-	a.mu.Unlock()
 
 	select {
 	case a.input <- struct{}{}: