[hw-3] loms service

loms/internal/infra/grpc/middleware/logging.go

@@ -0,0 +1,24 @@
+package mw
+
+import (
+	"context"
+
+	"github.com/rs/zerolog/log"
+	"google.golang.org/grpc"
+	"google.golang.org/protobuf/encoding/protojson"
+	"google.golang.org/protobuf/proto"
+)
+
+func Logging(ctx context.Context, req any, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp any, err error) {
+	raw, _ := protojson.Marshal((req).(proto.Message))
+	log.Debug().Msgf("request: method: %v, req: %s", info.FullMethod, string(raw))
+
+	if resp, err = handler(ctx, req); err != nil {
+		log.Debug().Msgf("response: method: %v, err: %s", info.FullMethod, err.Error())
+		return
+	}
+	rawResp, _ := protojson.Marshal((resp).(proto.Message))
+	log.Debug().Msgf("response: method: %v, resp: %s", info.FullMethod, string(rawResp))
+
+	return
+}

loms/internal/infra/grpc/middleware/validate.go

@@ -0,0 +1,18 @@
+package mw
+
+import (
+	"context"
+
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+func Validate(ctx context.Context, req any, _ *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (any, error) {
+	if v, ok := req.(interface{ Validate() error }); ok {
+		if err := v.Validate(); err != nil {
+			return nil, status.Error(codes.InvalidArgument, err.Error())
+		}
+	}
+	return handler(ctx, req)
+}