You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.2 KiB
55 lines
1.2 KiB
package logging
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/go-kratos/kratos/v2/errors"
|
|
"github.com/go-kratos/kratos/v2/log"
|
|
"github.com/go-kratos/kratos/v2/transport/http"
|
|
)
|
|
|
|
// httpServerLog is a middleware server handler when transport is KindHTTP
|
|
func httpServerLog(logger log.Logger, ctx context.Context, args string, err error) {
|
|
info, ok := http.FromServerContext(ctx)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
level := log.LevelInfo
|
|
if err != nil {
|
|
level = log.LevelError
|
|
}
|
|
|
|
log.WithContext(ctx, logger).Log(level,
|
|
"kind", "server",
|
|
"component", "http",
|
|
"http.target", info.Request.RequestURI,
|
|
"http.method", info.Request.Method,
|
|
"http.args", args,
|
|
"http.code", errors.Code(err),
|
|
"http.error", extractError(err),
|
|
)
|
|
}
|
|
|
|
// httpClientLog is a middleware client handler when transport is KindHTTP
|
|
func httpClientLog(logger log.Logger, ctx context.Context, args string, err error) {
|
|
info, ok := http.FromClientContext(ctx)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
level := log.LevelInfo
|
|
if err != nil {
|
|
level = log.LevelError
|
|
}
|
|
|
|
log.WithContext(ctx, logger).Log(level,
|
|
"kind", "client",
|
|
"component", "http",
|
|
"http.target", info.Request.RequestURI,
|
|
"http.method", info.Request.Method,
|
|
"http.args", args,
|
|
"http.code", errors.Code(err),
|
|
"http.error", extractError(err),
|
|
)
|
|
}
|
|
|