feat: complete template

backend/providers/http/contracts.go

@@ -1,8 +0,0 @@
-package http
-
-type Route interface{}
-
-type Service interface {
-	Serve() error
-	GetEngine() interface{}
-}

backend/providers/http/engine.go

@@ -1,11 +1,11 @@
-package fiber
+package http
 
 import (
 	"errors"
+	"fmt"
+	"runtime/debug"
 	"time"
 
-	"backend/providers/http"
-
 	"git.ipao.vip/rogeecn/atom/container"
 	"git.ipao.vip/rogeecn/atom/utils/opt"
 	log "github.com/sirupsen/logrus"
@@ -19,31 +19,17 @@ func DefaultProvider() container.ProviderContainer {
 	return container.ProviderContainer{
 		Provider: Provide,
 		Options: []opt.Option{
-			opt.Prefix(http.DefaultPrefix),
+			opt.Prefix(DefaultPrefix),
 		},
 	}
 }
 
 type Service struct {
-	conf   *http.Config
+	conf   *Config
 	Engine *fiber.App
 }
 
-func (e *Service) Use(middleware ...interface{}) fiber.Router {
-	return e.Engine.Use(middleware...)
-}
-
-func (e *Service) GetEngine() interface{} {
-	return e.Engine
-}
-
-func (e *Service) Close() {
-	if err := e.Engine.Shutdown(); err != nil {
-		log.Error("http server shutdown error: ", err)
-	}
-}
-
-func (e *Service) Serve() error {
+func (svc *Service) Serve() error {
 	listenConfig := fiber.ListenConfig{
 		EnablePrintRoutes: true,
 		OnShutdownSuccess: func() {
@@ -52,32 +38,40 @@ func (e *Service) Serve() error {
 		OnShutdownError: func(err error) {
 			log.Error("http server shutdown error: ", err)
 		},
-		DisableStartupMessage: true,
+		// DisableStartupMessage: true,
 	}
 
-	if e.conf.Tls != nil {
-		if e.conf.Tls.Cert == "" || e.conf.Tls.Key == "" {
+	if svc.conf.Tls != nil {
+		if svc.conf.Tls.Cert == "" || svc.conf.Tls.Key == "" {
 			return errors.New("tls cert or key is empty")
 		}
-		listenConfig.CertFile = e.conf.Tls.Cert
-		listenConfig.CertKeyFile = e.conf.Tls.Key
+		listenConfig.CertFile = svc.conf.Tls.Cert
+		listenConfig.CertKeyFile = svc.conf.Tls.Key
 	}
+	container.AddCloseAble(func() {
+		svc.Engine.Shutdown()
+	})
 
-	return e.Engine.Listen(e.conf.PortString(), listenConfig)
+	return svc.Engine.Listen(svc.conf.PortString(), listenConfig)
 }
 
 func Provide(opts ...opt.Option) error {
 	o := opt.New(opts...)
-	var config http.Config
+	var config Config
 	if err := o.UnmarshalConfig(&config); err != nil {
 		return err
 	}
 
-	return container.Container.Provide(func(l *log.Logger) (http.Service, error) {
+	return container.Container.Provide(func() (*Service, error) {
 		engine := fiber.New(fiber.Config{
 			StrictRouting: true,
 		})
-		engine.Use(recover.New())
+		engine.Use(recover.New(recover.Config{
+			EnableStackTrace: true,
+			StackTraceHandler: func(c fiber.Ctx, e any) {
+				log.WithError(e.(error)).Error(fmt.Sprintf("panic: %v\n%s\n", e, debug.Stack()))
+			},
+		}))
 
 		if config.StaticRoute != nil && config.StaticPath != nil {
 			engine.Use(config.StaticRoute, config.StaticPath)
@@ -89,8 +83,6 @@ func Provide(opts ...opt.Option) error {
 			TimeZone:   "Asia/Shanghai",
 		}))
 
-		svc := &Service{Engine: engine, conf: &config}
-		container.AddCloseAble(svc.Close)
-		return svc, nil
+		return &Service{Engine: engine, conf: &config}, nil
 	}, o.DiOptions()...)
 }