42 lines
858 B
Go
42 lines
858 B
Go
package floodsub
|
|
|
|
import (
|
|
"context"
|
|
|
|
inet "github.com/libp2p/go-libp2p-net"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
var _ inet.Notifiee = (*PubSubNotif)(nil)
|
|
|
|
type PubSubNotif PubSub
|
|
|
|
func (p *PubSubNotif) OpenedStream(n inet.Network, s inet.Stream) {
|
|
}
|
|
|
|
func (p *PubSubNotif) ClosedStream(n inet.Network, s inet.Stream) {
|
|
}
|
|
|
|
func (p *PubSubNotif) Connected(n inet.Network, c inet.Conn) {
|
|
s, err := p.host.NewStream(context.Background(), c.RemotePeer(), ID)
|
|
if err != nil {
|
|
log.Warning("opening new stream to peer: ", err, c.LocalPeer(), c.RemotePeer())
|
|
return
|
|
}
|
|
|
|
select {
|
|
case p.newPeers <- s:
|
|
case <-p.ctx.Done():
|
|
s.Close()
|
|
}
|
|
}
|
|
|
|
func (p *PubSubNotif) Disconnected(n inet.Network, c inet.Conn) {
|
|
}
|
|
|
|
func (p *PubSubNotif) Listen(n inet.Network, _ ma.Multiaddr) {
|
|
}
|
|
|
|
func (p *PubSubNotif) ListenClose(n inet.Network, _ ma.Multiaddr) {
|
|
}
|