forked from cerc-io/ipld-eth-server
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package p2p
|
|
|
|
import (
|
|
logging "github.com/ipfs/go-log"
|
|
p2phost "github.com/libp2p/go-libp2p-host"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
pstore "github.com/libp2p/go-libp2p-peerstore"
|
|
)
|
|
|
|
var log = logging.Logger("p2p-mount")
|
|
|
|
// P2P structure holds information on currently running streams/Listeners
|
|
type P2P struct {
|
|
ListenersLocal *Listeners
|
|
ListenersP2P *Listeners
|
|
Streams *StreamRegistry
|
|
|
|
identity peer.ID
|
|
peerHost p2phost.Host
|
|
peerstore pstore.Peerstore
|
|
}
|
|
|
|
// NewP2P creates new P2P struct
|
|
func NewP2P(identity peer.ID, peerHost p2phost.Host, peerstore pstore.Peerstore) *P2P {
|
|
return &P2P{
|
|
identity: identity,
|
|
peerHost: peerHost,
|
|
peerstore: peerstore,
|
|
|
|
ListenersLocal: newListenersLocal(),
|
|
ListenersP2P: newListenersP2P(peerHost),
|
|
|
|
Streams: &StreamRegistry{
|
|
Streams: map[uint64]*Stream{},
|
|
ConnManager: peerHost.ConnManager(),
|
|
conns: map[peer.ID]int{},
|
|
},
|
|
}
|
|
}
|
|
|
|
// CheckProtoExists checks whether a proto handler is registered to
|
|
// mux handler
|
|
func (p2p *P2P) CheckProtoExists(proto string) bool {
|
|
protos := p2p.peerHost.Mux().Protocols()
|
|
|
|
for _, p := range protos {
|
|
if p != proto {
|
|
continue
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|