From f9400aff1addf160c0bbb1dc0a60bedebe582a55 Mon Sep 17 00:00:00 2001 From: rogeecn Date: Fri, 28 Feb 2025 09:08:42 +0800 Subject: [PATCH] feat: modify http service --- .vscode/settings.json | 3 +++ .../project/app/service/http/http.go.tpl | 20 ++++++++++--------- 2 files changed, 14 insertions(+), 9 deletions(-) create mode 100644 .vscode/settings.json diff --git a/.vscode/settings.json b/.vscode/settings.json new file mode 100644 index 0000000..2761437 --- /dev/null +++ b/.vscode/settings.json @@ -0,0 +1,3 @@ +{ + "editor.fontSize": 10 +} \ No newline at end of file diff --git a/templates/project/app/service/http/http.go.tpl b/templates/project/app/service/http/http.go.tpl index 05464a4..a717688 100644 --- a/templates/project/app/service/http/http.go.tpl +++ b/templates/project/app/service/http/http.go.tpl @@ -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() }) }