新增日志中间件

http/middleware/logging/logging.go

@@ -0,0 +1,125 @@
+package logging
+
+import (
+	"context"
+	"fmt"
+
+	"git.ikuban.com/server/kratos-utils/http/param"
+
+	context2 "git.ikuban.com/server/kratos-utils/http/context"
+
+	"github.com/go-kratos/kratos/v2/errors"
+	"github.com/go-kratos/kratos/v2/log"
+	"github.com/go-kratos/kratos/v2/middleware"
+	"github.com/go-kratos/kratos/v2/transport/grpc"
+	"github.com/go-kratos/kratos/v2/transport/http"
+)
+
+// Option is HTTP logging option.
+type Option func(*options)
+
+type options struct {
+	logger log.Logger
+}
+
+// WithLogger with middleware logger.
+func WithLogger(logger log.Logger) Option {
+	return func(o *options) {
+		o.logger = logger
+	}
+}
+
+// Server is an server logging middleware.
+func Server(opts ...Option) middleware.Middleware {
+	options := options{
+		logger: log.DefaultLogger,
+	}
+	for _, o := range opts {
+		o(&options)
+	}
+	log := log.NewHelper("middleware/logging", options.logger)
+	return func(handler middleware.Handler) middleware.Handler {
+		return func(ctx context.Context, req interface{}) (interface{}, error) {
+			var (
+				path      string
+				method    string
+				params    string
+				component string
+			)
+			if info, ok := http.FromServerContext(ctx); ok {
+				component = "HTTP"
+				path = info.Request.RequestURI
+				method = info.Request.Method
+				params = info.Request.Form.Encode()
+			} else if info, ok := grpc.FromServerContext(ctx); ok {
+				component = "gRPC"
+				path = info.FullMethod
+				method = "POST"
+				params = req.(fmt.Stringer).String()
+			}
+			fromPath := param.GetFromPath(ctx)
+			ctx = context2.AppendToContext(ctx, "from_path", path)
+			reply, err := handler(ctx, req)
+			if err != nil {
+				log.Errorw(
+					"kind", "server",
+					"component", component,
+					"path", path,
+					"from", fromPath,
+					"method", method,
+					"params", params,
+					"code", errors.Code(err),
+					"error", err.Error(),
+				)
+				return nil, err
+			}
+			return reply, nil
+		}
+	}
+}
+
+// Client is an client logging middleware.
+func Client(opts ...Option) middleware.Middleware {
+	options := options{
+		logger: log.DefaultLogger,
+	}
+	for _, o := range opts {
+		o(&options)
+	}
+	log := log.NewHelper("middleware/logging", options.logger)
+	return func(handler middleware.Handler) middleware.Handler {
+		return func(ctx context.Context, req interface{}) (interface{}, error) {
+			var (
+				path      string
+				method    string
+				params    string
+				component string
+			)
+			if info, ok := http.FromClientContext(ctx); ok {
+				component = "HTTP"
+				path = info.Request.RequestURI
+				method = info.Request.Method
+				params = info.Request.Form.Encode()
+			} else if info, ok := grpc.FromClientContext(ctx); ok {
+				path = info.FullMethod
+				method = "POST"
+				component = "gRPC"
+				params = req.(fmt.Stringer).String()
+			}
+			reply, err := handler(ctx, req)
+			if err != nil {
+				log.Errorw(
+					"kind", "client",
+					"component", component,
+					"path", path,
+					"method", method,
+					"params", params,
+					"code", errors.Code(err),
+					"error", err.Error(),
+				)
+				return nil, err
+			}
+			return reply, nil
+		}
+	}
+}

http/param/base.go

@@ -12,3 +12,11 @@ func GetQuery(c context.Context) url.Values {
 	}
 	return url.Values{}
 }
+
+func GetFromPath(c context.Context) string {
+	path := c.Value("from_path")
+	if _, ok := path.(string); ok {
+		return path.(string)
+	}
+	return ""
+}