feat: modify http service
This commit is contained in:
3
.vscode/settings.json
vendored
Normal file
3
.vscode/settings.json
vendored
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
{
|
||||||
|
"editor.fontSize": 10
|
||||||
|
}
|
||||||
@@ -1,6 +1,8 @@
|
|||||||
package http
|
package http
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
"go.ipao.vip/atom"
|
"go.ipao.vip/atom"
|
||||||
"go.ipao.vip/atom/container"
|
"go.ipao.vip/atom/container"
|
||||||
"go.ipao.vip/atom/contracts"
|
"go.ipao.vip/atom/contracts"
|
||||||
@@ -46,7 +48,7 @@ func Command() atom.Option {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
type Http struct {
|
type Service struct {
|
||||||
dig.In
|
dig.In
|
||||||
|
|
||||||
App *app.Config
|
App *app.Config
|
||||||
@@ -57,24 +59,24 @@ type Http struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func Serve(cmd *cobra.Command, args []string) error {
|
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{})
|
log.SetFormatter(&log.JSONFormatter{})
|
||||||
|
|
||||||
if http.App.Mode == app.AppModeDevelopment {
|
if svc.App.Mode == app.AppModeDevelopment {
|
||||||
log.SetLevel(log.DebugLevel)
|
log.SetLevel(log.DebugLevel)
|
||||||
|
|
||||||
http.Service.Engine.Get("/swagger/*", swagger.HandlerDefault)
|
svc.Service.Engine.Get("/swagger/*", swagger.HandlerDefault)
|
||||||
}
|
}
|
||||||
http.Service.Engine.Use(errorx.Middleware)
|
svc.Service.Engine.Use(errorx.Middleware)
|
||||||
http.Service.Engine.Use(favicon.New(favicon.Config{
|
svc.Service.Engine.Use(favicon.New(favicon.Config{
|
||||||
Data: []byte{},
|
Data: []byte{},
|
||||||
}))
|
}))
|
||||||
|
|
||||||
group := http.Service.Engine.Group("")
|
group := svc.Service.Engine.Group("")
|
||||||
for _, route := range http.Routes {
|
for _, route := range svc.Routes {
|
||||||
route.Register(group)
|
route.Register(group)
|
||||||
}
|
}
|
||||||
|
|
||||||
return http.Service.Serve()
|
return svc.Service.Serve()
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user