cerc-io/lotus
15
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

protect full node connection

Browse Source
This commit is contained in:
Aarsh Shah 2022-02-04 13:04:07 +04:00
parent e155eb9fa3
commit ca46a79144
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
markets/idxprov/mesh.go
View File

@ -4,6 +4,10 @@ import (
"context"
"fmt"
"github.com/libp2p/go-libp2p-core/network"
"github.com/libp2p/go-libp2p-core/peer"
"github.com/filecoin-project/lotus/api/v1api"
logging "github.com/ipfs/go-log/v2"
@ -21,16 +25,25 @@ type Libp2pMeshCreator struct {
}
func (mc Libp2pMeshCreator) Connect(ctx context.Context) error {
addrs, err := mc.fullnodeApi.NetAddrsListen(ctx)
faddrs, err := mc.fullnodeApi.NetAddrsListen(ctx)
if err != nil {
return err
}
if err := mc.idxProvHost.Connect(ctx, addrs); err != nil {
// if we already have a connection to the full node, there's nothing to do here
if mc.idxProvHost.Network().Connectedness(faddrs.ID) == network.Connected {
return nil
}
// otherwise, connect to the full node, ask it to protect the connection and protect the connection on our end too
if err := mc.idxProvHost.Connect(ctx, faddrs); err != nil {
return fmt.Errorf("failed to connect index provider host with the full node: %w", err)
}
mc.idxProvHost.ConnManager().Protect(addrs.ID, "markets")
log.Debugw("successfully connected to full node", "fullNodeInfo", addrs.String())
mc.idxProvHost.ConnManager().Protect(faddrs.ID, "markets")
mc.fullnodeApi.NetProtectAdd(ctx, []peer.ID{mc.idxProvHost.ID()})
log.Debugw("successfully connected to full node and asked it protect indexer provider peer conn", "fullNodeInfo", faddrs.String(),
"idxProviderPeerId", mc.idxProvHost.ID())
return nil
}