70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package distribyted
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
peerstore "github.com/libp2p/go-libp2p-peerstore"
|
|
|
|
coreiface "github.com/ipfs/interface-go-ipfs-core"
|
|
path "github.com/ipfs/interface-go-ipfs-core/path"
|
|
)
|
|
|
|
type PstoreMngr struct {
|
|
ctx context.Context
|
|
dht coreiface.DhtAPI
|
|
swarm coreiface.SwarmAPI
|
|
store peerstore.Peerstore
|
|
rootLinks []path.Path
|
|
}
|
|
|
|
func NewPstoreMngr(
|
|
ctx context.Context,
|
|
rootLinks []path.Path,
|
|
dht coreiface.DhtAPI,
|
|
swarm coreiface.SwarmAPI,
|
|
store peerstore.Peerstore,
|
|
) *PstoreMngr {
|
|
return &PstoreMngr{ctx, dht, swarm, store, rootLinks}
|
|
}
|
|
|
|
func (p *PstoreMngr) Start() {
|
|
p.loadKnownRootPeers()
|
|
p.startProvidersFinder()
|
|
}
|
|
|
|
func (p *PstoreMngr) loadKnownRootPeers() {
|
|
for _, peer := range p.store.Peers() {
|
|
pp := peer
|
|
go func() {
|
|
fmt.Println("loading known peer", pp)
|
|
addrs := p.store.Addrs(pp)
|
|
|
|
if err := p.swarm.Connect(p.ctx, peerstore.PeerInfo{
|
|
ID: pp,
|
|
Addrs: addrs,
|
|
}); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
|
|
func (p *PstoreMngr) startProvidersFinder() {
|
|
for _, link := range p.rootLinks {
|
|
fmt.Println("loading new root link", link)
|
|
ll := link
|
|
go func() {
|
|
pInfo, err := p.dht.FindProviders(p.ctx, ll)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
for i := range pInfo {
|
|
fmt.Println("Adding new address from peer", i.ID, "LINK", ll)
|
|
p.store.SetAddrs(i.ID, i.Addrs, 24*30*time.Hour)
|
|
}
|
|
}()
|
|
}
|
|
}
|