mirror of
https://github.com/scroll-tech/scroll.git
synced 2026-01-14 16:37:56 -05:00
50040a164e
Co-authored-by: maskpp <maskpp266@gmail.com> Co-authored-by: HAOYUatHZ <37070449+HAOYUatHZ@users.noreply.github.com> Co-authored-by: HAOYUatHZ <HAOYUatHZ@users.noreply.github.com>
120 lines
2.9 KiB
Go
120 lines
2.9 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/scroll-tech/go-ethereum/log"
|
|
"github.com/urfave/cli/v2"
|
|
"gorm.io/gorm"
|
|
|
|
"scroll-tech/common/database"
|
|
"scroll-tech/common/observability"
|
|
"scroll-tech/common/utils"
|
|
"scroll-tech/common/version"
|
|
|
|
"scroll-tech/coordinator/internal/config"
|
|
"scroll-tech/coordinator/internal/controller/api"
|
|
"scroll-tech/coordinator/internal/route"
|
|
)
|
|
|
|
var app *cli.App
|
|
|
|
func init() {
|
|
// Set up coordinator app info.
|
|
app = cli.NewApp()
|
|
app.Action = action
|
|
app.Name = "coordinator"
|
|
app.Usage = "The Scroll L2 Coordinator"
|
|
app.Version = version.Version
|
|
app.Flags = append(app.Flags, utils.CommonFlags...)
|
|
app.Flags = append(app.Flags, apiFlags...)
|
|
app.Before = func(ctx *cli.Context) error {
|
|
return utils.LogSetup(ctx)
|
|
}
|
|
// Register `coordinator-test` app for integration-test.
|
|
utils.RegisterSimulation(app, utils.CoordinatorAPIApp)
|
|
}
|
|
|
|
func action(ctx *cli.Context) error {
|
|
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)
|
|
}
|
|
|
|
db, err := database.InitDB(cfg.DB)
|
|
if err != nil {
|
|
log.Crit("failed to init db connection", "err", err)
|
|
}
|
|
defer func() {
|
|
if err = database.CloseDB(db); err != nil {
|
|
log.Error("can not close db connection", "error", err)
|
|
}
|
|
}()
|
|
|
|
registry := prometheus.DefaultRegisterer
|
|
observability.Server(ctx, db)
|
|
|
|
apiSrv := apiServer(ctx, cfg, db, registry)
|
|
|
|
log.Info(
|
|
"Start coordinator api 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
|
|
log.Info("start shutdown coordinator server ...")
|
|
|
|
closeCtx, cancelExit := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancelExit()
|
|
if err = apiSrv.Shutdown(closeCtx); err != nil {
|
|
log.Warn("shutdown coordinator server failure", "error", err)
|
|
return nil
|
|
}
|
|
|
|
<-closeCtx.Done()
|
|
log.Info("coordinator server exiting success")
|
|
return nil
|
|
}
|
|
|
|
func apiServer(ctx *cli.Context, cfg *config.Config, db *gorm.DB, reg prometheus.Registerer) *http.Server {
|
|
router := gin.New()
|
|
api.InitController(cfg, db, reg)
|
|
route.Route(router, cfg, reg)
|
|
port := ctx.String(httpPortFlag.Name)
|
|
srv := &http.Server{
|
|
Addr: fmt.Sprintf(":%s", port),
|
|
Handler: router,
|
|
ReadHeaderTimeout: time.Minute,
|
|
}
|
|
|
|
go func() {
|
|
if runServerErr := srv.ListenAndServe(); runServerErr != nil && !errors.Is(runServerErr, http.ErrServerClosed) {
|
|
log.Crit("run coordinator http server failure", "error", runServerErr)
|
|
}
|
|
}()
|
|
return srv
|
|
}
|
|
|
|
// Run coordinator.
|
|
func Run() {
|
|
// RunApp the coordinator.
|
|
if err := app.Run(os.Args); err != nil {
|
|
_, _ = fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|