65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package http
|
|
|
|
import (
|
|
"atom/container"
|
|
"atom/providers/config"
|
|
"atom/providers/logger"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func init() {
|
|
if err := container.Container.Provide(NewService); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
type Service struct {
|
|
Engine *gin.Engine
|
|
conf *config.Config
|
|
}
|
|
|
|
func (e *Service) Serve() error {
|
|
if e.conf.Http.Https {
|
|
return e.Engine.RunTLS(e.conf.Http.PortString(), e.conf.Http.HttpsCert, e.conf.Http.HttpKey)
|
|
}
|
|
return e.Engine.Run(e.conf.Http.PortString())
|
|
}
|
|
|
|
func NewService(log *logger.Logger, cfg *config.Config) *Service {
|
|
log.Info("init http service with gin...")
|
|
|
|
gin.DefaultWriter = &logger.LevelWriter{Logger: log, Level: zap.InfoLevel}
|
|
gin.DefaultErrorWriter = &logger.LevelWriter{Logger: log, Level: zap.ErrorLevel}
|
|
|
|
if cfg.App.IsDebug() {
|
|
gin.SetMode(gin.DebugMode)
|
|
} else {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
}
|
|
|
|
engine := gin.New()
|
|
|
|
engine.Use(gin.LoggerWithFormatter(func(param gin.LogFormatterParams) string {
|
|
return fmt.Sprintf("%s - [%s] \"%s %s %s %d %s \"%s\" %s\"\n",
|
|
param.ClientIP,
|
|
param.TimeStamp.Format(time.RFC1123),
|
|
param.Method,
|
|
param.Path,
|
|
param.Request.Proto,
|
|
param.StatusCode,
|
|
param.Latency,
|
|
param.Request.UserAgent(),
|
|
param.ErrorMessage,
|
|
)
|
|
}))
|
|
|
|
engine.Use(gin.Recovery())
|
|
|
|
return &Service{Engine: engine, conf: cfg}
|
|
}
|