Merge pull request #44 from quan-xie/grpc_trace

add grpc trace

README.md

@@ -18,6 +18,6 @@ go get -u github.com/quan-xie/tuba
 - [x] Traces (支持zipkin、jagger) 
 - [x] RPC (GRPC With K8s headless)
 - [x] Ecode (Customize Error code)
+- [x] Reids (go-redis/v9)
 - [ ] Auth 
-- [ ] Cache  (memcache,redis)
-- [ ] Service Discovery
+

transport/grpc/server.go

@@ -1,13 +1,17 @@
 package grpc
 
 import (
+	"context"
+	"fmt"
 	"net"
 	"sync"
 	"time"
 
 	"github.com/pkg/errors"
 	"github.com/quan-xie/tuba/log"
 	"github.com/quan-xie/tuba/util/xtime"
+	"go.opencensus.io/trace"
+	"google.golang.org/grpc"
 	xgprc "google.golang.org/grpc"
 	"google.golang.org/grpc/health"
 	healthpb "google.golang.org/grpc/health/grpc_health_v1"
@@ -125,3 +129,34 @@ func (s *Server) configuration(c *ServerConfig) (err error) {
 	s.mutex.Unlock()
 	return
 }
+
+// Tracing Interceptor ....
+func TracingInterceptor() grpc.UnaryServerInterceptor {
+	return func(
+		ctx context.Context,
+		req interface{},
+		info *grpc.UnaryServerInfo,
+		handler grpc.UnaryHandler,
+	) (interface{}, error) {
+		// Start a new trace span for the gRPC method.
+		ctx, span := trace.StartSpan(ctx, fmt.Sprintf("gRPC-%s", info.FullMethod))
+		defer span.End()
+
+		// Add gRPC method information to the span attributes.
+		span.AddAttributes(
+			trace.StringAttribute("grpc.method", info.FullMethod),
+		)
+
+		// Call the gRPC handler.
+		resp, err := handler(ctx, req)
+		if err != nil {
+			// Set the error status on the span.
+			span.SetStatus(trace.Status{
+				Code:    trace.StatusCodeInternal,
+				Message: err.Error(),
+			})
+		}
+
+		return resp, err
+	}
+}