mirror of
https://github.com/logos-messaging/logos-messaging-go.git
synced 2026-01-07 08:23:06 +00:00
1991a54d64
Uses libp2p network notifier to determine when a peer connects or disconnects, as well as using the host network peerstore instead of managing out own separate peer map
132 lines
3.2 KiB
Go
132 lines
3.2 KiB
Go
package node
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/libp2p/go-libp2p-core/host"
|
|
"github.com/libp2p/go-libp2p-core/network"
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
"github.com/status-im/go-waku/waku/v2/protocol/filter"
|
|
"github.com/status-im/go-waku/waku/v2/protocol/lightpush"
|
|
"github.com/status-im/go-waku/waku/v2/protocol/store"
|
|
wakurelay "github.com/status-im/go-wakurelay-pubsub"
|
|
)
|
|
|
|
// A map of peer IDs to supported protocols
|
|
type PeerStats map[peer.ID][]string
|
|
|
|
type ConnStatus struct {
|
|
IsOnline bool
|
|
HasHistory bool
|
|
Peers PeerStats
|
|
}
|
|
|
|
type ConnectionNotifier struct {
|
|
h host.Host
|
|
DisconnectChan chan peer.ID
|
|
quit chan struct{}
|
|
}
|
|
|
|
func NewConnectionNotifier(h host.Host) ConnectionNotifier {
|
|
return ConnectionNotifier{
|
|
h: h,
|
|
DisconnectChan: make(chan peer.ID, 100),
|
|
quit: make(chan struct{}),
|
|
}
|
|
}
|
|
|
|
func (c ConnectionNotifier) Listen(n network.Network, m ma.Multiaddr) {
|
|
// called when network starts listening on an addr
|
|
}
|
|
|
|
func (c ConnectionNotifier) ListenClose(n network.Network, m ma.Multiaddr) {
|
|
// called when network stops listening on an addr
|
|
}
|
|
|
|
func (c ConnectionNotifier) Connected(n network.Network, cc network.Conn) {
|
|
// called when a connection opened
|
|
log.Info(fmt.Sprintf("Peer %s connected", cc.RemotePeer()))
|
|
}
|
|
|
|
func (c ConnectionNotifier) Disconnected(n network.Network, cc network.Conn) {
|
|
// called when a connection closed
|
|
log.Info(fmt.Sprintf("Peer %s disconnected", cc.RemotePeer()))
|
|
c.DisconnectChan <- cc.RemotePeer()
|
|
}
|
|
|
|
func (c ConnectionNotifier) OpenedStream(n network.Network, s network.Stream) {
|
|
// called when a stream opened
|
|
}
|
|
|
|
func (c ConnectionNotifier) ClosedStream(n network.Network, s network.Stream) {
|
|
// called when a stream closed
|
|
}
|
|
|
|
func (c ConnectionNotifier) Close() {
|
|
close(c.quit)
|
|
}
|
|
|
|
func (w *WakuNode) sendConnStatus() {
|
|
isOnline, hasHistory := w.Status()
|
|
if w.connStatusChan != nil {
|
|
connStatus := ConnStatus{IsOnline: isOnline, HasHistory: hasHistory, Peers: w.Peers()}
|
|
w.connStatusChan <- connStatus
|
|
}
|
|
|
|
}
|
|
|
|
func (w *WakuNode) connectednessListener() {
|
|
for {
|
|
select {
|
|
case <-w.quit:
|
|
return
|
|
case <-w.protocolEventSub.Out():
|
|
case <-w.identificationEventSub.Out():
|
|
case p := <-w.connectionNotif.DisconnectChan:
|
|
// Notify filter of disconnection
|
|
w.filter.DisconnectChan <- p
|
|
}
|
|
w.sendConnStatus()
|
|
}
|
|
}
|
|
|
|
func (w *WakuNode) Status() (isOnline bool, hasHistory bool) {
|
|
hasRelay := false
|
|
hasLightPush := false
|
|
hasStore := false
|
|
hasFilter := false
|
|
|
|
for _, peer := range w.host.Network().Peers() {
|
|
protocols, err := w.host.Peerstore().GetProtocols(peer)
|
|
if err != nil {
|
|
log.Warn(fmt.Errorf("could not read peer %s protocols", peer))
|
|
}
|
|
|
|
for _, protocol := range protocols {
|
|
if !hasRelay && protocol == string(wakurelay.WakuRelayID_v200) {
|
|
hasRelay = true
|
|
}
|
|
if !hasLightPush && protocol == string(lightpush.LightPushID_v20beta1) {
|
|
hasLightPush = true
|
|
}
|
|
if !hasStore && protocol == string(store.StoreID_v20beta3) {
|
|
hasStore = true
|
|
}
|
|
if !hasFilter && protocol == string(filter.FilterID_v20beta1) {
|
|
hasFilter = true
|
|
}
|
|
}
|
|
}
|
|
|
|
if hasStore {
|
|
hasHistory = true
|
|
}
|
|
|
|
if hasRelay || hasLightPush && (hasStore || hasFilter) {
|
|
isOnline = true
|
|
}
|
|
|
|
return
|
|
}
|