From 52b46d6869f994a7958a231430aacceb41f4b7a0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rich=CE=9Brd?= <info@richardramos.me>
Date: Tue, 31 Aug 2021 13:27:42 -0400
Subject: [PATCH] emit ConnStatus always to get an update when peers change
 connectivity (#46)

* emit ConnStatus event always to get an update when peers change
connectivity
* use peers map to determine peer count
---
 waku/v2/node/wakunode2.go | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/waku/v2/node/wakunode2.go b/waku/v2/node/wakunode2.go
index 8134a2a6..07cf0481 100644
--- a/waku/v2/node/wakunode2.go
+++ b/waku/v2/node/wakunode2.go
@@ -163,8 +163,7 @@ func (w *WakuNode) processHostEvent(e interface{}) {
 	newHasHistory := w.HasHistory()
 	log.Info("###ConnStatus isOnline: ", isOnline, "/", newIsOnline, " hasHistory: ",
 		hasHistory, "/", newHasHistory)
-	if w.connStatusChan != nil &&
-		(isOnline != newIsOnline || hasHistory != newHasHistory) {
+	if w.connStatusChan != nil {
 
 		// Creating a copy of the current peers map
 		w.peersMutex.Lock()
@@ -174,8 +173,10 @@ func (w *WakuNode) processHostEvent(e interface{}) {
 		}
 		w.peersMutex.Unlock()
 
-		log.Info("New ConnStatus: ", ConnStatus{IsOnline: newIsOnline, HasHistory: newHasHistory, Peers: p})
-		w.connStatusChan <- ConnStatus{IsOnline: newIsOnline, HasHistory: newHasHistory, Peers: p}
+		connStatus := ConnStatus{IsOnline: newIsOnline, HasHistory: newHasHistory, Peers: p}
+
+		log.Info("New ConnStatus: ", connStatus)
+		w.connStatusChan <- connStatus
 	}
 
 }
@@ -728,7 +729,7 @@ func (w *WakuNode) ClosePeerById(id peer.ID) error {
 }
 
 func (w *WakuNode) PeerCount() int {
-	return len(w.host.Network().Peers())
+	return len(w.peers)
 }
 
 func (w *WakuNode) startKeepAlive(t time.Duration) {