feat: modify http service

This commit is contained in:
rogeecn
2025-02-28 09:08:42 +08:00
parent 65bfc96e5b
commit f9400aff1a
2 changed files with 14 additions and 9 deletions

3
.vscode/settings.json vendored Normal file
View File

@@ -0,0 +1,3 @@
{
"editor.fontSize": 10
}

View File

@@ -1,6 +1,8 @@
package http
import (
"context"
"go.ipao.vip/atom"
"go.ipao.vip/atom/container"
"go.ipao.vip/atom/contracts"
@@ -46,7 +48,7 @@ func Command() atom.Option {
)
}
type Http struct {
type Service struct {
dig.In
App *app.Config
@@ -57,24 +59,24 @@ type Http struct {
}
func Serve(cmd *cobra.Command, args []string) error {
return container.Container.Invoke(func(http Http) error {
return container.Container.Invoke(func(ctx context.Context, svc Service) error {
log.SetFormatter(&log.JSONFormatter{})
if http.App.Mode == app.AppModeDevelopment {
if svc.App.Mode == app.AppModeDevelopment {
log.SetLevel(log.DebugLevel)
http.Service.Engine.Get("/swagger/*", swagger.HandlerDefault)
svc.Service.Engine.Get("/swagger/*", swagger.HandlerDefault)
}
http.Service.Engine.Use(errorx.Middleware)
http.Service.Engine.Use(favicon.New(favicon.Config{
svc.Service.Engine.Use(errorx.Middleware)
svc.Service.Engine.Use(favicon.New(favicon.Config{
Data: []byte{},
}))
group := http.Service.Engine.Group("")
for _, route := range http.Routes {
group := svc.Service.Engine.Group("")
for _, route := range svc.Routes {
route.Register(group)
}
return http.Service.Serve()
return svc.Service.Serve()
})
}