unistack-org/micro
4
0
Fork 2
Code Issues 18 Pull Requests Actions Packages Projects Releases 324 Wiki Activity

Refactor debug/trace ready for Jaeger

Browse Source
This commit is contained in:
Micro
2020-01-29 15:45:11 +00:00
parent 1108cc5e91
commit 62c067adcd
10 changed files with 130 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
util/wrapper/wrapper.go
View File

@@ -19,7 +19,7 @@ type traceWrapper struct {
client.Client
name string
trace trace.Trace
trace trace.Tracer
}
var (
@@ -97,7 +97,7 @@ func HandlerStats(stats stats.Stats) server.HandlerWrapper {
}
// TraceCall is a call tracing wrapper
func TraceCall(name string, t trace.Trace, c client.Client) client.Client {
func TraceCall(name string, t trace.Tracer, c client.Client) client.Client {
return &traceWrapper{
name: name,
trace: t,
@@ -106,7 +106,7 @@ func TraceCall(name string, t trace.Trace, c client.Client) client.Client {
}
// TraceHandler wraps a server handler to perform tracing
func TraceHandler(t trace.Trace) server.HandlerWrapper {
func TraceHandler(t trace.Tracer) server.HandlerWrapper {
// return a handler wrapper
return func(h server.HandlerFunc) server.HandlerFunc {
// return a function that returns a function