2016-09-10 03:13:50 +00:00
|
|
|
package floodsub
|
2016-09-10 00:15:39 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
|
2016-10-05 19:47:20 +00:00
|
|
|
inet "github.com/libp2p/go-libp2p-net"
|
|
|
|
ma "github.com/multiformats/go-multiaddr"
|
2016-09-10 00:15:39 +00:00
|
|
|
)
|
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
var _ inet.Notifiee = (*PubSubNotif)(nil)
|
2016-09-10 00:15:39 +00:00
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
type PubSubNotif PubSub
|
2016-09-10 00:15:39 +00:00
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
func (p *PubSubNotif) OpenedStream(n inet.Network, s inet.Stream) {
|
2016-09-10 00:15:39 +00:00
|
|
|
}
|
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
func (p *PubSubNotif) ClosedStream(n inet.Network, s inet.Stream) {
|
2016-09-10 00:15:39 +00:00
|
|
|
}
|
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
func (p *PubSubNotif) Connected(n inet.Network, c inet.Conn) {
|
2016-09-10 00:15:39 +00:00
|
|
|
s, err := p.host.NewStream(context.Background(), c.RemotePeer(), ID)
|
|
|
|
if err != nil {
|
2016-09-13 16:38:07 +00:00
|
|
|
log.Warning("opening new stream to peer: ", err, c.LocalPeer(), c.RemotePeer())
|
2016-09-10 00:15:39 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
select {
|
|
|
|
case p.newPeers <- s:
|
|
|
|
case <-p.ctx.Done():
|
|
|
|
s.Close()
|
|
|
|
}
|
2016-09-10 00:15:39 +00:00
|
|
|
}
|
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
func (p *PubSubNotif) Disconnected(n inet.Network, c inet.Conn) {
|
2016-09-10 00:15:39 +00:00
|
|
|
}
|
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
func (p *PubSubNotif) Listen(n inet.Network, _ ma.Multiaddr) {
|
2016-09-10 00:15:39 +00:00
|
|
|
}
|
|
|
|
|
2016-09-12 20:22:16 +00:00
|
|
|
func (p *PubSubNotif) ListenClose(n inet.Network, _ ma.Multiaddr) {
|
2016-09-10 00:15:39 +00:00
|
|
|
}
|