diff --git a/server/notifier.go b/server/notifier.go index 9bd69e7..cccbf0a 100644 --- a/server/notifier.go +++ b/server/notifier.go @@ -57,6 +57,10 @@ func (s *Server) RunNotifier() error { case internal.HeightHash: heightHash, _ := notification.(internal.HeightHash) s.DoNotify(&heightHash) + // Do we need this? + // case peerNotification: + // peer, _ := notification.(peerNotification) + // s.notifyPeerSubs(&Peer{Address: peer.address, Port: peer.port}) default: logrus.Warn("unknown notification type") } diff --git a/server/session.go b/server/session.go index 3d7c931..cd0a8a3 100644 --- a/server/session.go +++ b/server/session.go @@ -47,7 +47,7 @@ type session struct { // headersSub indicates header subscription headersSub bool // peersSub indicates peer subscription - peersBool bool + peersSub bool // headersSubRaw indicates the header subscription mode headersSubRaw bool // client provides the ability to send notifications @@ -103,6 +103,9 @@ func (s *session) doNotify(notification interface{}) { } params = []string{orig, status} case peerNotification: + if !s.peersSub { + return + } note, _ := notification.(peerNotification) method = "server.peers.subscribe" params = []string{note.address, note.port}