Sfoglia il codice sorgente

fix: 修复 clientinterceptors/tracinginterceptor.go 显示接受消息字节为0 (#2003)

Atlan 2 anni fa
parent
commit
af6d37c33d
1 ha cambiato i file con 4 aggiunte e 2 eliminazioni
  1. 4 2
      zrpc/internal/clientinterceptors/tracinginterceptor.go

+ 4 - 2
zrpc/internal/clientinterceptors/tracinginterceptor.go

@@ -26,9 +26,11 @@ func UnaryTracingInterceptor(ctx context.Context, method string, req, reply inte
 	defer span.End()
 
 	ztrace.MessageSent.Event(ctx, 1, req)
-	ztrace.MessageReceived.Event(ctx, 1, reply)
 
-	if err := invoker(ctx, method, req, reply, cc, opts...); err != nil {
+	err := invoker(ctx, method, req, reply, cc, opts...)
+	// fix: https://github.com/zeromicro/go-zero/issues/1954
+	ztrace.MessageReceived.Event(ctx, 1, reply)
+	if err != nil {
 		s, ok := status.FromError(err)
 		if ok {
 			span.SetStatus(codes.Error, s.Message())