restructure

providers/http/gin/engine.go

@@ -0,0 +1,54 @@
+package gin
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/rogeecn/atom/container"
+	"github.com/rogeecn/atom/providers/http"
+	"github.com/rogeecn/atom/providers/log"
+	"go.uber.org/dig"
+
+	"github.com/gin-gonic/gin"
+)
+
+type Service struct {
+	conf   *http.Config
+	Engine *gin.Engine
+}
+
+func (e *Service) Use(middleware ...gin.HandlerFunc) gin.IRoutes {
+	return e.Engine.Use(middleware...)
+}
+
+func (e *Service) Serve() error {
+	if e.conf.Https {
+		return e.Engine.RunTLS(e.conf.PortString(), e.conf.HttpsCert, e.conf.HttpKey)
+	}
+	return e.Engine.Run(e.conf.PortString())
+}
+
+func Provide(cfg *http.Config, opts ...dig.ProvideOption) error {
+	return container.Container.Provide(func() (http.Service, error) {
+		gin.DefaultWriter = log.LevelWriter{Level: log.InfoLevel}
+		gin.DefaultErrorWriter = log.LevelWriter{Level: log.ErrorLevel}
+
+		engine := gin.New()
+		engine.Use(gin.Recovery())
+		engine.Use(gin.LoggerWithFormatter(func(param gin.LogFormatterParams) string {
+			return fmt.Sprintf(`%s - [%s] "%s %s %s %d %s '%q' %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,
+			)
+		}))
+
+		return &Service{Engine: engine, conf: cfg}, nil
+	}, opts...)
+}