54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package infrastructure
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
fiberLog "github.com/gofiber/fiber/v2/middleware/logger"
|
|
"github.com/gofiber/fiber/v2/middleware/pprof"
|
|
"github.com/wormhole-foundation/wormhole-explorer/parser/internal/sqs"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type Server struct {
|
|
app *fiber.App
|
|
port string
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func NewServer(logger *zap.Logger, port string, pprofEnabled bool, isQueueConsumer bool, consumer *sqs.Consumer, db *mongo.Database) *Server {
|
|
repository := NewRepository(db, logger)
|
|
service := NewService(repository, consumer, isQueueConsumer, logger)
|
|
ctrl := NewController(service)
|
|
app := fiber.New()
|
|
|
|
// config use of middlware.
|
|
if pprofEnabled {
|
|
app.Use(pprof.New())
|
|
}
|
|
app.Use(fiberLog.New(fiberLog.Config{
|
|
Format: "level=info timestamp=${time} method=${method} path=${path} status${status} request_id=${locals:requestid}\n",
|
|
}))
|
|
|
|
api := app.Group("/api")
|
|
api.Get("/health", ctrl.HealthCheck)
|
|
api.Get("/ready", ctrl.ReadyCheck)
|
|
|
|
return &Server{
|
|
app: app,
|
|
port: port,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// Start listen serves HTTP requests from addr.
|
|
func (s *Server) Start() {
|
|
go func() {
|
|
s.app.Listen(":" + s.port)
|
|
}()
|
|
}
|
|
|
|
// Stop gracefull server.
|
|
func (s *Server) Stop() {
|
|
_ = s.app.Shutdown()
|
|
}
|