129 lines
3.1 KiB
Go
129 lines
3.1 KiB
Go
package ipfscluster
|
|
|
|
import (
|
|
"github.com/ipfs-cluster/ipfs-cluster/api"
|
|
|
|
peer "github.com/libp2p/go-libp2p/core/peer"
|
|
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
// ConnectGraph returns a description of which cluster peers and ipfs
|
|
// daemons are connected to each other.
|
|
func (c *Cluster) ConnectGraph() (api.ConnectGraph, error) {
|
|
ctx, span := trace.StartSpan(c.ctx, "cluster/ConnectGraph")
|
|
defer span.End()
|
|
|
|
cg := api.ConnectGraph{
|
|
ClusterID: c.host.ID(),
|
|
IDtoPeername: make(map[string]string),
|
|
IPFSLinks: make(map[string][]peer.ID),
|
|
ClusterLinks: make(map[string][]peer.ID),
|
|
ClusterTrustLinks: make(map[string]bool),
|
|
ClustertoIPFS: make(map[string]peer.ID),
|
|
}
|
|
members, err := c.consensus.Peers(ctx)
|
|
if err != nil {
|
|
return cg, err
|
|
}
|
|
|
|
for _, member := range members {
|
|
// one of the entries is for itself, but that shouldn't hurt
|
|
cg.ClusterTrustLinks[member.String()] = c.consensus.IsTrustedPeer(ctx, member)
|
|
}
|
|
|
|
peers := make([][]api.ID, len(members))
|
|
errs := make([]error, len(members))
|
|
|
|
for i, member := range members {
|
|
in := make(chan struct{})
|
|
close(in)
|
|
out := make(chan api.ID, 1024)
|
|
errCh := make(chan error, 1)
|
|
go func(i int) {
|
|
defer close(errCh)
|
|
|
|
errCh <- c.rpcClient.Stream(
|
|
ctx,
|
|
member,
|
|
"Cluster",
|
|
"Peers",
|
|
in,
|
|
out,
|
|
)
|
|
}(i)
|
|
var ids []api.ID
|
|
for id := range out {
|
|
ids = append(ids, id)
|
|
}
|
|
peers[i] = ids
|
|
errs[i] = <-errCh
|
|
}
|
|
|
|
for i, err := range errs {
|
|
p := members[i].String()
|
|
cg.ClusterLinks[p] = make([]peer.ID, 0)
|
|
if err != nil { // Only setting cluster connections when no error occurs
|
|
logger.Debugf("RPC error reaching cluster peer %s: %s", p, err.Error())
|
|
continue
|
|
}
|
|
|
|
selfConnection, pID := c.recordClusterLinks(&cg, p, peers[i])
|
|
cg.IDtoPeername[p] = pID.Peername
|
|
// IPFS connections
|
|
if !selfConnection {
|
|
logger.Warnf("cluster peer %s not its own peer. No ipfs info ", p)
|
|
continue
|
|
}
|
|
c.recordIPFSLinks(&cg, pID)
|
|
}
|
|
|
|
return cg, nil
|
|
}
|
|
|
|
func (c *Cluster) recordClusterLinks(cg *api.ConnectGraph, p string, peers []api.ID) (bool, api.ID) {
|
|
selfConnection := false
|
|
var pID api.ID
|
|
for _, id := range peers {
|
|
if id.Error != "" {
|
|
logger.Debugf("Peer %s errored connecting to its peer %s", p, id.ID.Pretty())
|
|
continue
|
|
}
|
|
if id.ID.String() == p {
|
|
selfConnection = true
|
|
pID = id
|
|
} else {
|
|
cg.ClusterLinks[p] = append(cg.ClusterLinks[p], id.ID)
|
|
}
|
|
}
|
|
return selfConnection, pID
|
|
}
|
|
|
|
func (c *Cluster) recordIPFSLinks(cg *api.ConnectGraph, pID api.ID) {
|
|
ipfsID := pID.IPFS.ID
|
|
if pID.IPFS.Error != "" { // Only setting ipfs connections when no error occurs
|
|
logger.Warnf("ipfs id: %s has error: %s. Skipping swarm connections", ipfsID.Pretty(), pID.IPFS.Error)
|
|
return
|
|
}
|
|
|
|
pid := pID.ID.String()
|
|
ipfsPid := ipfsID.String()
|
|
|
|
if _, ok := cg.IPFSLinks[pid]; ok {
|
|
logger.Warnf("ipfs id: %s already recorded, one ipfs daemon in use by multiple cluster peers", ipfsID.Pretty())
|
|
}
|
|
cg.ClustertoIPFS[pid] = ipfsID
|
|
cg.IPFSLinks[ipfsPid] = make([]peer.ID, 0)
|
|
var swarmPeers []peer.ID
|
|
err := c.rpcClient.Call(
|
|
pID.ID,
|
|
"IPFSConnector",
|
|
"SwarmPeers",
|
|
struct{}{},
|
|
&swarmPeers,
|
|
)
|
|
if err != nil {
|
|
return
|
|
}
|
|
cg.IPFSLinks[ipfsPid] = swarmPeers
|
|
}
|