mirror of
https://github.com/scroll-tech/scroll.git
synced 2026-01-13 16:08:04 -05:00
95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/scroll-tech/go-ethereum/log"
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"scroll-tech/common/database"
|
|
"scroll-tech/common/observability"
|
|
"scroll-tech/common/utils"
|
|
"scroll-tech/common/version"
|
|
|
|
"scroll-tech/rollup/internal/config"
|
|
"scroll-tech/rollup/internal/controller/blob_uploader"
|
|
)
|
|
|
|
var app *cli.App
|
|
|
|
func init() {
|
|
// Set up blob-uploader app info.
|
|
app = cli.NewApp()
|
|
app.Action = action
|
|
app.Name = "blob-uploader"
|
|
app.Usage = "The Scroll Blob Uploader"
|
|
app.Version = version.Version
|
|
app.Flags = append(app.Flags, utils.CommonFlags...)
|
|
app.Commands = []*cli.Command{}
|
|
app.Before = func(ctx *cli.Context) error {
|
|
return utils.LogSetup(ctx)
|
|
}
|
|
}
|
|
|
|
func action(ctx *cli.Context) error {
|
|
// Load config file.
|
|
cfgFile := ctx.String(utils.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
|
|
db, err := database.InitDB(cfg.DBConfig)
|
|
if err != nil {
|
|
log.Crit("failed to init db connection", "err", err)
|
|
}
|
|
defer func() {
|
|
cancel()
|
|
if err = database.CloseDB(db); err != nil {
|
|
log.Crit("failed to close db connection", "error", err)
|
|
}
|
|
}()
|
|
|
|
registry := prometheus.DefaultRegisterer
|
|
observability.Server(ctx, db)
|
|
|
|
// sanity check config
|
|
if cfg.L2Config.BlobUploaderConfig == nil {
|
|
log.Crit("cfg.L2Config.BlobUploaderConfig must not be nil")
|
|
}
|
|
|
|
blobUploader, err := blob_uploader.NewBlobUploader(ctx.Context, db, cfg.L2Config.BlobUploaderConfig, registry)
|
|
if err != nil {
|
|
log.Crit("failed to create l2 relayer", "config file", cfgFile, "error", err)
|
|
}
|
|
|
|
go utils.Loop(subCtx, 1*time.Second, blobUploader.UploadBlobToS3)
|
|
|
|
// Finish start all blob-uploader functions.
|
|
log.Info("Start blob-uploader successfully", "version", version.Version)
|
|
|
|
// 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 blob uploader cmd instance.
|
|
func Run() {
|
|
if err := app.Run(os.Args); err != nil {
|
|
_, _ = fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|