From 1d00f2f771540c8b90d4cbacd3771fdcf3ecf23c Mon Sep 17 00:00:00 2001 From: Asim Aslam Date: Fri, 24 Jan 2020 22:02:35 +0000 Subject: [PATCH] add trace handler --- service.go | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/service.go b/service.go index abbf23c5..521135e7 100644 --- a/service.go +++ b/service.go @@ -15,6 +15,7 @@ import ( "github.com/micro/go-micro/debug/profile/pprof" "github.com/micro/go-micro/debug/service/handler" "github.com/micro/go-micro/debug/stats" + "github.com/micro/go-micro/debug/trace" "github.com/micro/go-micro/plugin" "github.com/micro/go-micro/server" "github.com/micro/go-micro/util/log" @@ -36,8 +37,15 @@ func newService(opts ...Option) Service { // wrap client to inject From-Service header on any calls options.Client = wrapper.FromService(serviceName, options.Client) + // wrap client to inject From-Service header on any calls + options.Client = wrapper.TraceCall(serviceName, trace.DefaultTrace, options.Client) + // wrap the server to provide handler stats - options.Server.Init(server.WrapHandler(wrapper.HandlerStats(stats.DefaultStats))) + options.Server.Init( + server.WrapHandler(wrapper.HandlerStats(stats.DefaultStats)), + server.WrapHandler(wrapper.TraceHandler(trace.DefaultTrace)), + ) + return &service{ opts: options,