2017-01-30 12:12:25 +00:00
|
|
|
package ipfscluster
|
|
|
|
|
|
|
|
import (
|
2017-11-08 19:04:04 +00:00
|
|
|
"fmt"
|
|
|
|
host "github.com/libp2p/go-libp2p-host"
|
2017-01-30 12:12:25 +00:00
|
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
|
|
peerstore "github.com/libp2p/go-libp2p-peerstore"
|
|
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
|
|
)
|
|
|
|
|
2017-11-08 19:04:04 +00:00
|
|
|
// peerManager provides wrappers peerset control
|
2017-01-30 12:12:25 +00:00
|
|
|
type peerManager struct {
|
2017-11-08 19:04:04 +00:00
|
|
|
host host.Host
|
2017-01-30 12:12:25 +00:00
|
|
|
}
|
|
|
|
|
2017-11-08 19:04:04 +00:00
|
|
|
func newPeerManager(h host.Host) *peerManager {
|
|
|
|
return &peerManager{h}
|
2017-01-30 12:12:25 +00:00
|
|
|
}
|
|
|
|
|
2017-11-08 19:04:04 +00:00
|
|
|
func (pm *peerManager) addPeer(addr ma.Multiaddr) error {
|
|
|
|
logger.Debugf("adding peer address %s", addr)
|
2017-02-02 22:52:06 +00:00
|
|
|
pid, decapAddr, err := multiaddrSplit(addr)
|
2017-01-30 12:12:25 +00:00
|
|
|
if err != nil {
|
2017-02-02 22:52:06 +00:00
|
|
|
return err
|
2017-01-30 12:12:25 +00:00
|
|
|
}
|
2017-11-08 19:04:04 +00:00
|
|
|
pm.host.Peerstore().AddAddr(pid, decapAddr, peerstore.PermanentAddrTTL)
|
2017-02-02 22:52:06 +00:00
|
|
|
return nil
|
2017-01-30 12:12:25 +00:00
|
|
|
}
|
|
|
|
|
2017-11-08 19:04:04 +00:00
|
|
|
func (pm *peerManager) rmPeer(pid peer.ID) error {
|
|
|
|
logger.Debugf("forgetting peer %s", pid.Pretty())
|
|
|
|
pm.host.Peerstore().ClearAddrs(pid)
|
2017-01-30 12:12:25 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-11-08 19:04:04 +00:00
|
|
|
// func (pm *peerManager) savePeers() {
|
|
|
|
// peers := pm.peersAddrs()
|
|
|
|
// logger.Debugf("saving peers: %s", peers)
|
|
|
|
// pm.cluster.config.Peers = peers
|
|
|
|
// pm.cluster.config.NotifySave()
|
|
|
|
// }
|
2017-01-30 12:12:25 +00:00
|
|
|
|
2017-11-08 19:04:04 +00:00
|
|
|
// func (pm *peerManager) isPeer(p peer.ID) bool {
|
|
|
|
// if p == pm.cluster.id {
|
|
|
|
// return true
|
|
|
|
// }
|
|
|
|
|
|
|
|
// peers := pm.cluster.consensus.Peers()
|
|
|
|
// for _, pid := range peers {
|
|
|
|
// if p == pid {
|
|
|
|
// return true
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
// return false
|
|
|
|
// }
|
2017-01-30 12:12:25 +00:00
|
|
|
|
2017-02-02 22:52:06 +00:00
|
|
|
// cluster peer addresses (NOT including ourselves)
|
2017-11-08 19:04:04 +00:00
|
|
|
func (pm *peerManager) addresses(peers []peer.ID) []ma.Multiaddr {
|
2017-10-27 20:11:14 +00:00
|
|
|
addrs := []ma.Multiaddr{}
|
2017-11-08 19:04:04 +00:00
|
|
|
if peers == nil {
|
|
|
|
return addrs
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, p := range peers {
|
|
|
|
if p == pm.host.ID() {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
peerAddr, _ := ma.NewMultiaddr(fmt.Sprintf("/ipfs/%s", peer.IDB58Encode(p)))
|
|
|
|
for _, a := range pm.host.Peerstore().Addrs(p) {
|
|
|
|
addrs = append(addrs, a.Encapsulate(peerAddr))
|
2017-02-02 22:52:06 +00:00
|
|
|
}
|
2017-01-30 12:12:25 +00:00
|
|
|
}
|
2017-02-02 22:52:06 +00:00
|
|
|
return addrs
|
|
|
|
}
|
2017-01-30 12:12:25 +00:00
|
|
|
|
2017-11-08 19:04:04 +00:00
|
|
|
func (pm *peerManager) importAddresses(addrs []ma.Multiaddr) error {
|
|
|
|
for _, a := range addrs {
|
|
|
|
pm.addPeer(a)
|
2017-10-31 10:20:14 +00:00
|
|
|
}
|
2017-01-30 12:12:25 +00:00
|
|
|
return nil
|
|
|
|
}
|