Periodic bootstrap
This commit is contained in:
parent
fef544f88e
commit
c0179bd4fa
@ -94,37 +94,16 @@ var DaemonCmd = &cli.Command{
|
|||||||
}
|
}
|
||||||
return lr.SetAPIEndpoint(apima)
|
return lr.SetAPIEndpoint(apima)
|
||||||
}),
|
}),
|
||||||
|
|
||||||
|
node.ApplyIf(func(s *node.Settings) bool { return cctx.Bool("bootstrap") },
|
||||||
|
node.Override(node.BootstrapKey, modules.Bootstrap),
|
||||||
|
),
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
go func() {
|
|
||||||
if !cctx.Bool("bootstrap") {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
err := bootstrap(ctx, api)
|
|
||||||
if err != nil {
|
|
||||||
log.Error("Bootstrap failed: ", err)
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
// TODO: properly parse api endpoint (or make it a URL)
|
// TODO: properly parse api endpoint (or make it a URL)
|
||||||
return serveRPC(api, stop, "127.0.0.1:"+cctx.String("api"))
|
return serveRPC(api, stop, "127.0.0.1:"+cctx.String("api"))
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
func bootstrap(ctx context.Context, api api.FullNode) error {
|
|
||||||
pis, err := build.BuiltinBootstrap()
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, pi := range pis {
|
|
||||||
if err := api.NetConnect(ctx, pi); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
@ -70,6 +70,7 @@ const (
|
|||||||
|
|
||||||
PstoreAddSelfKeysKey = invoke(iota)
|
PstoreAddSelfKeysKey = invoke(iota)
|
||||||
StartListeningKey
|
StartListeningKey
|
||||||
|
BootstrapKey
|
||||||
|
|
||||||
// filecoin
|
// filecoin
|
||||||
SetGenesisKey
|
SetGenesisKey
|
||||||
|
@ -1,15 +1,20 @@
|
|||||||
package modules
|
package modules
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"crypto/rand"
|
"crypto/rand"
|
||||||
"io"
|
"github.com/filecoin-project/go-lotus/build"
|
||||||
"io/ioutil"
|
"github.com/filecoin-project/go-lotus/node/modules/helpers"
|
||||||
|
|
||||||
"github.com/gbrlsnchs/jwt/v3"
|
"github.com/gbrlsnchs/jwt/v3"
|
||||||
logging "github.com/ipfs/go-log"
|
logging "github.com/ipfs/go-log"
|
||||||
|
"github.com/libp2p/go-libp2p-core/host"
|
||||||
"github.com/libp2p/go-libp2p-core/peerstore"
|
"github.com/libp2p/go-libp2p-core/peerstore"
|
||||||
record "github.com/libp2p/go-libp2p-record"
|
record "github.com/libp2p/go-libp2p-record"
|
||||||
|
"go.uber.org/fx"
|
||||||
"golang.org/x/xerrors"
|
"golang.org/x/xerrors"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/filecoin-project/go-lotus/api"
|
"github.com/filecoin-project/go-lotus/api"
|
||||||
"github.com/filecoin-project/go-lotus/chain/types"
|
"github.com/filecoin-project/go-lotus/chain/types"
|
||||||
@ -70,3 +75,46 @@ func APISecret(keystore types.KeyStore, lr repo.LockedRepo) (*dtypes.APIAlg, err
|
|||||||
|
|
||||||
return (*dtypes.APIAlg)(jwt.NewHS256(key.PrivateKey)), nil
|
return (*dtypes.APIAlg)(jwt.NewHS256(key.PrivateKey)), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Bootstrap(mctx helpers.MetricsCtx, lc fx.Lifecycle, host host.Host) {
|
||||||
|
ctx, cancel := context.WithCancel(mctx)
|
||||||
|
|
||||||
|
lc.Append(fx.Hook{
|
||||||
|
OnStart: func(_ context.Context) error {
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
sctx, cancel := context.WithTimeout(ctx, build.BlockDelay*time.Second/2)
|
||||||
|
<-sctx.Done()
|
||||||
|
cancel()
|
||||||
|
|
||||||
|
if ctx.Err() != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(host.Network().Conns()) > 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Warn("No peers connected, performing automatic bootstrap")
|
||||||
|
|
||||||
|
pis, err := build.BuiltinBootstrap()
|
||||||
|
if err != nil {
|
||||||
|
log.Error("Getting bootstrap addrs: ", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, pi := range pis {
|
||||||
|
if err := host.Connect(ctx, pi); err != nil {
|
||||||
|
log.Warn("bootstrap connect failed: ", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
OnStop: func(_ context.Context) error {
|
||||||
|
cancel()
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user