mirror of
https://github.com/scroll-tech/scroll.git
synced 2026-01-22 20:38:13 -05:00
134 lines
3.4 KiB
Go
134 lines
3.4 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/scroll-tech/go-ethereum/ethclient"
|
|
"github.com/scroll-tech/go-ethereum/log"
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"scroll-tech/database"
|
|
|
|
"scroll-tech/common/metrics"
|
|
"scroll-tech/common/version"
|
|
|
|
"scroll-tech/bridge/config"
|
|
"scroll-tech/bridge/relayer"
|
|
"scroll-tech/bridge/utils"
|
|
"scroll-tech/bridge/watcher"
|
|
|
|
cutils "scroll-tech/common/utils"
|
|
)
|
|
|
|
var (
|
|
app *cli.App
|
|
)
|
|
|
|
func init() {
|
|
// Set up rollup-relayer app info.
|
|
app = cli.NewApp()
|
|
|
|
app.Action = action
|
|
app.Name = "rollup-relayer"
|
|
app.Usage = "The Scroll Rollup Relayer"
|
|
app.Version = version.Version
|
|
app.Flags = append(app.Flags, cutils.CommonFlags...)
|
|
app.Commands = []*cli.Command{}
|
|
|
|
app.Before = func(ctx *cli.Context) error {
|
|
return cutils.LogSetup(ctx)
|
|
}
|
|
// Register `rollup-relayer-test` app for integration-test.
|
|
cutils.RegisterSimulation(app, cutils.RollupRelayerApp)
|
|
}
|
|
|
|
func action(ctx *cli.Context) error {
|
|
// Load config file.
|
|
cfgFile := ctx.String(cutils.ConfigFileFlag.Name)
|
|
cfg, err := config.NewConfig(cfgFile)
|
|
if err != nil {
|
|
log.Crit("failed to load config file", "config file", cfgFile, "error", err)
|
|
}
|
|
|
|
subCtx, cancel := context.WithCancel(ctx.Context)
|
|
|
|
// init db connection
|
|
var ormFactory database.OrmFactory
|
|
if ormFactory, err = database.NewOrmFactory(cfg.DBConfig); err != nil {
|
|
log.Crit("failed to init db connection", "err", err)
|
|
}
|
|
defer func() {
|
|
cancel()
|
|
err = ormFactory.Close()
|
|
if err != nil {
|
|
log.Error("can not close ormFactory", "error", err)
|
|
}
|
|
}()
|
|
|
|
// Start metrics server.
|
|
metrics.Serve(subCtx, ctx)
|
|
|
|
// Init l2geth connection
|
|
l2client, err := ethclient.Dial(cfg.L2Config.Endpoint)
|
|
if err != nil {
|
|
log.Error("failed to connect l2 geth", "config file", cfgFile, "error", err)
|
|
return err
|
|
}
|
|
|
|
l2relayer, err := relayer.NewLayer2Relayer(ctx.Context, l2client, ormFactory, cfg.L2Config.RelayerConfig)
|
|
if err != nil {
|
|
log.Error("failed to create l2 relayer", "config file", cfgFile, "error", err)
|
|
return err
|
|
}
|
|
|
|
batchProposer := watcher.NewBatchProposer(subCtx, cfg.L2Config.BatchProposerConfig, l2relayer, ormFactory)
|
|
if err != nil {
|
|
log.Error("failed to create batchProposer", "config file", cfgFile, "error", err)
|
|
return err
|
|
}
|
|
|
|
l2watcher := watcher.NewL2WatcherClient(subCtx, l2client, cfg.L2Config.Confirmations, cfg.L2Config.L2MessengerAddress, cfg.L2Config.L2MessageQueueAddress, cfg.L2Config.WithdrawTrieRootSlot, ormFactory)
|
|
|
|
// Watcher loop to fetch missing blocks
|
|
go cutils.LoopWithContext(subCtx, 2*time.Second, func(ctx context.Context) {
|
|
number, loopErr := utils.GetLatestConfirmedBlockNumber(ctx, l2client, cfg.L2Config.Confirmations)
|
|
if loopErr != nil {
|
|
log.Error("failed to get block number", "err", loopErr)
|
|
return
|
|
}
|
|
l2watcher.TryFetchRunningMissingBlocks(ctx, number)
|
|
})
|
|
|
|
// Batch proposer loop
|
|
go cutils.Loop(subCtx, 2*time.Second, func() {
|
|
batchProposer.TryProposeBatch()
|
|
batchProposer.TryCommitBatches()
|
|
})
|
|
|
|
go cutils.Loop(subCtx, 2*time.Second, l2relayer.ProcessCommittedBatches)
|
|
|
|
// Finish start all rollup relayer functions.
|
|
log.Info("Start rollup-relayer successfully")
|
|
|
|
// Catch CTRL-C to ensure a graceful shutdown.
|
|
interrupt := make(chan os.Signal, 1)
|
|
signal.Notify(interrupt, os.Interrupt)
|
|
|
|
// Wait until the interrupt signal is received from an OS signal.
|
|
<-interrupt
|
|
|
|
return nil
|
|
}
|
|
|
|
// Run rollup relayer cmd instance.
|
|
func Run() {
|
|
if err := app.Run(os.Args); err != nil {
|
|
_, _ = fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|