6447ea51d2
This takes advantange of the latest features in go-cid, peer.ID and go-multiaddr and makes the Go types serializable by default. This means we no longer need to copy between Pin <-> PinSerial, or ID <-> IDSerial etc. We can now efficiently binary-encode these types using short field keys and without parsing/stringifying (in many cases it just a cast). We still get the same json output as before (with minor modifications for Cids). This should greatly improve Cluster performance and memory usage when dealing with large collections of items. License: MIT Signed-off-by: Hector Sanjuan <hector@protocol.ai>
71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
|
|
multiaddr "github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
// PeersToStrings IDB58Encodes a list of peers.
|
|
func PeersToStrings(peers []peer.ID) []string {
|
|
strs := make([]string, len(peers))
|
|
for i, p := range peers {
|
|
if p != "" {
|
|
strs[i] = peer.IDB58Encode(p)
|
|
}
|
|
}
|
|
return strs
|
|
}
|
|
|
|
// StringsToPeers decodes peer.IDs from strings.
|
|
func StringsToPeers(strs []string) []peer.ID {
|
|
peers := make([]peer.ID, len(strs))
|
|
for i, p := range strs {
|
|
var err error
|
|
peers[i], err = peer.IDB58Decode(p)
|
|
if err != nil {
|
|
logger.Debugf("'%s': %s", p, err)
|
|
}
|
|
}
|
|
return peers
|
|
}
|
|
|
|
// Libp2pMultiaddrSplit takes a LibP2P multiaddress (/<multiaddr>/ipfs/<peerID>)
|
|
// and decapsulates it, parsing the peer ID. Returns an error if there is
|
|
// any problem (for example, the provided address not being a Libp2p one).
|
|
func Libp2pMultiaddrSplit(addr ma.Multiaddr) (peer.ID, ma.Multiaddr, error) {
|
|
pid, err := addr.ValueForProtocol(ma.P_IPFS)
|
|
if err != nil {
|
|
err = fmt.Errorf("invalid peer multiaddress: %s: %s", addr, err)
|
|
logger.Error(err)
|
|
return "", nil, err
|
|
}
|
|
|
|
ipfs, _ := ma.NewMultiaddr("/ipfs/" + pid)
|
|
decapAddr := addr.Decapsulate(ipfs)
|
|
|
|
peerID, err := peer.IDB58Decode(pid)
|
|
if err != nil {
|
|
err = fmt.Errorf("invalid peer ID in multiaddress: %s: %s", pid, err)
|
|
logger.Error(err)
|
|
return "", nil, err
|
|
}
|
|
return peerID, decapAddr, nil
|
|
}
|
|
|
|
// MustLibp2pMultiaddrJoin takes a LibP2P multiaddress and a peer ID and
|
|
// encapsulates a new /ipfs/<peerID> address. It will panic if the given
|
|
// peer ID is bad.
|
|
func MustLibp2pMultiaddrJoin(addr Multiaddr, p peer.ID) Multiaddr {
|
|
v := addr.Value()
|
|
pidAddr, err := multiaddr.NewMultiaddr("/ipfs/" + peer.IDB58Encode(p))
|
|
// let this break badly
|
|
if err != nil {
|
|
panic("called MustLibp2pMultiaddrJoin with bad peer!")
|
|
}
|
|
return Multiaddr{Multiaddr: v.Encapsulate(pidAddr)}
|
|
}
|