Periodic bootstrap

This commit is contained in:
Łukasz Magiera 2019-10-11 02:31:06 +02:00
parent fef544f88e
commit c0179bd4fa
3 changed files with 56 additions and 28 deletions

View File

@ -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
}

View File

@ -70,6 +70,7 @@ const (
PstoreAddSelfKeysKey = invoke(iota) PstoreAddSelfKeysKey = invoke(iota)
StartListeningKey StartListeningKey
BootstrapKey
// filecoin // filecoin
SetGenesisKey SetGenesisKey

View File

@ -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
},
})
}