2017-03-25 39 views
1

我想要stackdriver示蹤劑與gRPC一起使用,我需要一些幫助。我一直在尋找這兩個環節的參考,我仍然不能得到它的工作:gRPC - GoLang - Stackdriver tracer

爲簡單起見,我剛剛與hello world gRPC example工作。這裏是我的客戶:

func main() { 

    // Set up a connection to the server. 
    conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure(), grpc.WithUnaryInterceptor(grpc.UnaryClientInterceptor(clientInterceptor))) 
    if err != nil { 
     log.Fatalf("did not connect: %v", err) 
    } 
    defer conn.Close() 
    c := pb.NewGreeterClient(conn) 

    ctx := context.Background() 

    tc, err := trace.NewClient(ctx, "{PROJECT-ID}") 
    if err != nil { 
     log.Fatal(err) 
    } 

    span := tc.NewSpan("/greeter/SayHello") 
    defer span.Finish() 

    ctx = trace.NewContext(ctx, span) 

    r, err := c.SayHello(ctx, &pb.HelloRequest{Name: "world"}) 
    if err != nil { 
     log.Fatalf("could not greet: %v", err) 
    } 

    println("Response:", r.Message) 
} 

func clientInterceptor(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error { 
    // trace current request w/ child span 
    span := trace.FromContext(ctx).NewChild(method) 
    defer span.Finish() 

    // new metadata, or copy of existing 
    md, ok := metadata.FromContext(ctx) 
    if !ok { 
     md = metadata.New(nil) 
    } else { 
     md = md.Copy() 
    } 

    // append trace header to context metadata 
    // header specification: https://cloud.google.com/trace/docs/faq 
    md["X-Cloud-Trace-Context"] = append(
     md["X-Cloud-Trace-Context"], fmt.Sprintf("%s/%d;o=1", span.TraceID(), 0), 
    ) 
    ctx = metadata.NewContext(ctx, md) 

    return invoker(ctx, method, req, reply, cc, opts...) 
} 

..和我GRPC服務器:

// server is used to implement helloworld.GreeterServer. 
type server struct{} 

// SayHello implements helloworld.GreeterServer 
func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) { 
    println("HERE") 
    return &pb.HelloReply{Message: "Hello " + in.Name}, nil 
} 

func main() { 
    lis, err := net.Listen("tcp", ":50051") 
    if err != nil { 
     log.Fatalf("failed to listen: %v", err) 
    } 

    ctx := context.Background() 
    tc, err := trace.NewClient(ctx, "{PROJECT-ID}") 
    if err != nil { 
     log.Fatal(err) 
    } 

    s := grpc.NewServer(EnableGRPCTracingServerOption(tc)) 
    pb.RegisterGreeterServer(s, &server{}) 

    println("listening on :50051") 
    if err := s.Serve(lis); err != nil { 
     log.Fatalf("failed to serve: %v", err) 
    } 
} 

// EnableGRPCTracingServerOption enables parsing google trace header from metadata 
// and adds a new child span to the incoming request context. 
func EnableGRPCTracingServerOption(traceClient *trace.Client) grpc.ServerOption { 
    return grpc.UnaryInterceptor(serverInterceptor(traceClient)) 
} 

func serverInterceptor(traceClient *trace.Client) grpc.UnaryServerInterceptor { 
    return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) { 
     // fetch metadata from request context 
     md, ok := metadata.FromContext(ctx) 
     if !ok { 
      md = metadata.New(nil) 
     } 
     header := strings.Join(md["X-Cloud-Trace-Context"], "") 

     // create new child span from google trace header, add to 
     // current request context 
     span := traceClient.SpanFromHeader(info.FullMethod, header) 
     defer span.Finish() 
     ctx = trace.NewContext(ctx, span) 

     return handler(ctx, req) 
    } 
} 

我,當我運行客戶端發起的痕跡,我得到的錯誤:

rpc error: code = 13 desc = stream terminated by RST_STREAM with error code: 1 

我我感到困惑,因爲我沒有看到有關身份驗證的其他內容。只提供不足以啓動特定項目跟蹤的項目ID。我錯過了什麼?

+0

你看到了什麼,你期望看到什麼?你的問題含糊不清。 –

+0

我期望在項目的stackdriver trace頁面上看到痕跡,並且我沒有看到痕跡。 –

+0

您能否通過以下答案解決問題? –

回答

1

問題同:

defer span.Finish() 

調用不會阻止,因爲我只是做初步測試一個打電話給我的計劃在可以上傳痕跡之前退出。我聯繫了筆者的https://rakyll.org/grpc-trace/她居然更新了她的崗位使用的選項:

defer span.FinishWait() 

該塊,通過允許痕跡,成功上傳程序之前退出固定它。

此外,在長時間運行的web服務器中,這不會是一個問題,因爲該進程不會被終止。

0

我遵循這些相同的教程,並遇到類似的問題。

標題鍵被轉換爲小寫字母。如果你在服務器端用header := strings.Join(md["x-cloud-trace-context"], "")檢索它,你應該很好。

您還可以定義元數據標題:

span := trace.FromContext(ctx).NewChild(method) 
defer span.Finish() 

md := metadata.Pairs(
    "x-cloud-trace-context", fmt.Sprintf("%s/%d;o=1", span.TraceID(), 0), 
) 

ctx = metadata.NewContext(ctx, md)