debug 日志等级下启用日志中间件

This commit is contained in:
xmdhs 2023-10-08 15:46:53 +08:00
parent fadd6f3390
commit 04a6cc96c1
No known key found for this signature in database
GPG Key ID: E809D6D43DEFCC95
2 changed files with 5 additions and 2 deletions

View File

@ -97,7 +97,7 @@ func (y *Yggdrasil) Auth(handle http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
ctx := r.Context() ctx := r.Context()
a, err := utils.DeCodeBody[yggdrasil.ValidateToken](r.Body, y.validate) a, err := utils.DeCodeBody[yggdrasil.ValidateToken](r.Body, y.validate)
if err != nil { if err != nil || a.AccessToken == "" {
token := y.getTokenbyAuthorization(ctx, w, r) token := y.getTokenbyAuthorization(ctx, w, r)
if token == "" { if token == "" {
return return

View File

@ -1,6 +1,7 @@
package route package route
import ( import (
"context"
"log/slog" "log/slog"
"net/http" "net/http"
@ -15,7 +16,9 @@ import (
func NewRoute(handelY *yggdrasil.Yggdrasil, handel *handle.Handel, c config.Config, sl slog.Handler) http.Handler { func NewRoute(handelY *yggdrasil.Yggdrasil, handel *handle.Handel, c config.Config, sl slog.Handler) http.Handler {
r := chi.NewRouter() r := chi.NewRouter()
r.Use(middleware.RequestID) r.Use(middleware.RequestID)
if sl.Enabled(context.Background(), slog.LevelDebug) {
r.Use(NewStructuredLogger(sl)) r.Use(NewStructuredLogger(sl))
}
r.Use(middleware.Recoverer) r.Use(middleware.Recoverer)
r.Use(cors.AllowAll().Handler) r.Use(cors.AllowAll().Handler)
if c.RaelIP { if c.RaelIP {