first commit

pkg/interceptor/logger/server.go

@@ -0,0 +1,61 @@
+package logger
+
+import (
+	"context"
+	"strings"
+
+	api "gitlab.33.cn/chat/dtalk/pkg/api/logger"
+
+	"gitlab.33.cn/chat/dtalk/pkg/logger"
+
+	"github.com/rs/zerolog"
+	"google.golang.org/grpc"
+)
+
+type Logger interface {
+	Info(msg string, ctx ...interface{})
+	Error(msg string, ctx ...interface{})
+}
+
+type ServerInterceptor struct {
+	log    zerolog.Logger
+	filter []string
+}
+
+func NewServerInterceptor(log zerolog.Logger, filter []string) *ServerInterceptor {
+	return &ServerInterceptor{
+		log:    log,
+		filter: filter,
+	}
+}
+
+func (s *ServerInterceptor) Unary(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
+	//log.Infof("%s req: %v", info.FullMethod, req)
+	for _, method := range s.filter {
+		if strings.HasPrefix(info.FullMethod, method) {
+			return handler(ctx, req)
+		}
+	}
+
+	logt := logger.NewLogWithCtx(ctx, s.log)
+
+	logt.Info().
+		Str("Path", info.FullMethod).
+		Interface("|body", req).
+		Msg("rpc req")
+
+	m, err := handler(ctx, req)
+
+	if err != nil {
+		//log.Error("%s err: %v", info.FullMethod, err)
+		code, msg := api.ParseErr(err)
+		if code != 0 {
+			logt.Error().
+				Int("code", code).
+				Str("msg", msg).
+				Msg("rpc err")
+		}
+	}
+	//log.Info("%s resp: %v", info.FullMethod, m)
+	return m, err
+}

pkg/interceptor/trace/trace.go

@@ -0,0 +1,98 @@
+package trace
+
+import (
+	"context"
+	"gitlab.33.cn/chat/dtalk/pkg/api"
+	"gitlab.33.cn/chat/dtalk/pkg/api/trace"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/metadata"
+)
+
+const (
+	// todo rename, oa那里也要一起改 X-Gateway-*
+	OAOpeId   string = "X-OA-Ope-Id"
+	OATraceId string = "X-OA-Trace-Id"
+)
+
+// NewTraceIdFromMD 从grpc metadata获取traceId
+func NewTraceIdFromMD(md metadata.MD) (string, bool) {
+	if md == nil {
+		return "", false
+	}
+
+	var t string
+
+	if traceId := md.Get(OATraceId); len(traceId) > 0 {
+		t = traceId[0]
+	} else {
+		return "", false
+	}
+
+	return t, true
+}
+
+// NewAddressFromMD 从grpc metadata获取操作者 ID
+func NewAddressFromMD(md metadata.MD) (string, bool) {
+	if md == nil {
+		return "", false
+	}
+
+	var t string
+
+	if opeId := md.Get(OAOpeId); len(opeId) > 0 {
+		t = opeId[0]
+	} else {
+		return "", false
+	}
+
+	return t, true
+}
+
+// ServerUnaryInterceptor 默认令牌服务端一元拦截器
+func ServerUnaryInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
+	return handler(wrapServerContext(ctx), req)
+}
+
+// wrapServerContext 包装服务端上下文
+func wrapServerContext(ctx context.Context) context.Context {
+	md, ok := metadata.FromIncomingContext(ctx)
+	if !ok {
+		return ctx
+	}
+
+	token, ok := NewTraceIdFromMD(md)
+	if !ok {
+		// 如果没有就自己创一个 ID
+		ctx = context.WithValue(ctx, trace.DtalkTraceId, trace.NewTraceIdWithContext(ctx))
+	} else {
+		ctx = context.WithValue(ctx, trace.DtalkTraceId, token)
+	}
+
+	address, ok := NewAddressFromMD(md)
+	if !ok {
+		return ctx
+	}
+	ctx = context.WithValue(ctx, api.Address, address)
+
+	return ctx
+}
+
+// UnaryClientInterceptor 默认令牌客户端一元拦截器
+func UnaryClientInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
+	return invoker(wrapClientContext(ctx), method, req, reply, cc, opts...)
+}
+
+// wrapClientContext 包装客户端上下文
+func wrapClientContext(ctx context.Context) context.Context {
+	t, ok := ctx.Value(trace.DtalkTraceId).(string)
+	if ok {
+		ctx = metadata.AppendToOutgoingContext(ctx, OATraceId, t)
+	}
+
+	token := api.NewAddrWithContext(ctx)
+	if ok {
+		ctx = metadata.AppendToOutgoingContext(ctx, OAOpeId, token)
+	}
+
+	return ctx
+}