add env var controlling msgs sync epochs

This commit is contained in:
vyzo 2020-09-10 16:22:02 +03:00
parent 0e40a04975
commit dab20efc99

View File

@ -2,6 +2,8 @@ package modules
import ( import (
"context" "context"
"os"
"strconv"
"time" "time"
"github.com/ipfs/go-datastore" "github.com/ipfs/go-datastore"
@ -36,6 +38,19 @@ import (
"github.com/filecoin-project/lotus/node/repo" "github.com/filecoin-project/lotus/node/repo"
) )
var pubsubMsgsSyncEpochs = 10
func init() {
if s := os.Getenv("LOTUS_MSGS_SYNC_EPOCHS"); s != "" {
val, err := strconv.Atoi(s)
if err != nil {
log.Errorf("failed to parse LOTUS_MSGS_SYNC_EPOCHS: %s", err)
return
}
pubsubMsgsSyncEpochs = val
}
}
func RunHello(mctx helpers.MetricsCtx, lc fx.Lifecycle, h host.Host, svc *hello.Service) error { func RunHello(mctx helpers.MetricsCtx, lc fx.Lifecycle, h host.Host, svc *hello.Service) error {
h.SetStreamHandler(hello.ProtocolID, svc.HandleStream) h.SetStreamHandler(hello.ProtocolID, svc.HandleStream)
@ -169,8 +184,8 @@ func HandleIncomingMessages(mctx helpers.MetricsCtx, lc fx.Lifecycle, ps *pubsub
return return
} }
// wait until we are synced within 10 epochs // wait until we are synced within 10 epochs -- env var can override
waitForSync(stmgr, 10, subscribe) waitForSync(stmgr, pubsubMsgsSyncEpochs, subscribe)
} }
func NewLocalDiscovery(lc fx.Lifecycle, ds dtypes.MetadataDS) (*discoveryimpl.Local, error) { func NewLocalDiscovery(lc fx.Lifecycle, ds dtypes.MetadataDS) (*discoveryimpl.Local, error) {