Influxdata
/
influxdb
mirror of https://github.com/influxdata/influxdb.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix client tracing propagation from doubly recording (#14204) 

Closes #14203

* chore(vendor): update to opentracing 1.1.0

* fix(http): finish client span

* fix(http): remove extra span injection
pull/14205/head
Chris Goller 2019-06-26 00:48:15 -05:00 committed by GitHub
parent 530ceb9cbd
commit 14e7daa6e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
10 changed files with 7 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
go.mod
View File

@ -61,7 +61,7 @@ require (
github.com/nats-io/nuid v1.0.0 // indirect
github.com/onsi/ginkgo v1.7.0 // indirect
github.com/onsi/gomega v1.4.3 // indirect
github.com/opentracing/opentracing-go v1.0.2
github.com/opentracing/opentracing-go v1.1.0
github.com/philhofer/fwd v1.0.0 // indirect
github.com/pkg/errors v0.8.1
github.com/prometheus/client_golang v0.9.0

2
go.sum
View File

@ -294,6 +294,8 @@ github.com/opencontainers/image-spec v1.0.1 h1:JMemWkRwHx4Zj+fVxWoMCFm/8sYGGrUVo
github.com/opencontainers/image-spec v1.0.1/go.mod h1:BtxoFyWECRxE4U/7sNtV5W15zMzWCbyJoFRP3s7yZA0=
github.com/opentracing/opentracing-go v1.0.2 h1:3jA2P6O1F9UOrWVpwrIo17pu01KWvNWg4X946/Y5Zwg=
github.com/opentracing/opentracing-go v1.0.2/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=
github.com/opentracing/opentracing-go v1.1.0 h1:pWlfV3Bxv7k65HYwkikxat0+s3pV4bsqf19k25Ur8rU=
github.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=
github.com/pascaldekloe/goe v0.1.0 h1:cBOtyMzM9HTpWjXfbbunk26uA6nG3a8n06Wieeh0MwY=
github.com/pascaldekloe/goe v0.1.0/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc=
github.com/pelletier/go-buffruneio v0.2.0/go.mod h1:JkE26KsDizTr40EUHkXVtNPvgGtbSNq5BcowyYOWdKo=

3
http/bucket_service.go
View File

@ -605,7 +605,6 @@ func (s *BucketService) FindBucketByID(ctx context.Context, id influxdb.ID) (*in
return nil, err
}
SetToken(s.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, s.InsecureSkipVerify)
resp, err := hc.Do(req)
@ -692,7 +691,6 @@ func (s *BucketService) FindBuckets(ctx context.Context, filter influxdb.BucketF
req.URL.RawQuery = query.Encode()
SetToken(s.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, s.InsecureSkipVerify)
resp, err := hc.Do(req)
@ -745,7 +743,6 @@ func (s *BucketService) CreateBucket(ctx context.Context, b *influxdb.Bucket) er
req.Header.Set("Content-Type", "application/json")
SetToken(s.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, s.InsecureSkipVerify)

3
http/client.go
View File

@ -54,6 +54,7 @@ func NewService(addr, token string) *Service {
}
}
// NewURL concats addr and path.
func NewURL(addr, path string) (*url.URL, error) {
u, err := url.Parse(addr)
if err != nil {
@ -63,6 +64,7 @@ func NewURL(addr, path string) (*url.URL, error) {
return u, nil
}
// NewClient returns an http.Client that pools connections and injects a span.
func NewClient(scheme string, insecure bool) *traceClient {
hc := &traceClient{
Client: http.Client{
@ -84,6 +86,7 @@ type traceClient struct {
// Do injects the trace and then performs the request.
func (c *traceClient) Do(r *http.Request) (*http.Response, error) {
span, _ := tracing.StartSpanFromContext(r.Context())
defer span.Finish()
tracing.InjectToHTTPRequest(span, r)
return c.Client.Do(r)
}

1
http/influxdb/client.go
View File

@ -80,6 +80,7 @@ type traceClient struct {
// Do injects the trace and then performs the request.
func (c *traceClient) Do(r *http.Request) (*http.Response, error) {
span, _ := tracing.StartSpanFromContext(r.Context())
defer span.Finish()
tracing.InjectToHTTPRequest(span, r)
return c.Client.Do(r)
}

2
http/influxdb/source_proxy_query_service.go
View File

@ -92,7 +92,6 @@ func (s *SourceProxyQueryService) fluxQuery(ctx context.Context, w io.Writer, re
hreq.Header.Set("Authorization", s.Token)
hreq.Header.Set("Content-Type", "application/json")
hreq = hreq.WithContext(ctx)
tracing.InjectToHTTPRequest(span, hreq)
hc := newTraceClient(u.Scheme, s.InsecureSkipVerify)
resp, err := hc.Do(hreq)
@ -140,7 +139,6 @@ func (s *SourceProxyQueryService) influxQuery(ctx context.Context, w io.Writer,
params.Set("rp", compiler.RP)
hreq.URL.RawQuery = params.Encode()
tracing.InjectToHTTPRequest(span, hreq)
hc := newTraceClient(u.Scheme, s.InsecureSkipVerify)
resp, err := hc.Do(hreq)

4
http/org_service.go
View File

@ -656,7 +656,6 @@ func (s *OrganizationService) FindOrganizations(ctx context.Context, filter infl
if err != nil {
return nil, 0, tracing.LogError(span, err)
}
tracing.InjectToHTTPRequest(span, req)
SetToken(s.Token, req)
hc := NewClient(url.Scheme, s.InsecureSkipVerify)
@ -707,7 +706,6 @@ func (s *OrganizationService) CreateOrganization(ctx context.Context, o *influxd
if err != nil {
return tracing.LogError(span, err)
}
tracing.InjectToHTTPRequest(span, req)
req.Header.Set("Content-Type", "application/json")
SetToken(s.Token, req)
@ -754,7 +752,6 @@ func (s *OrganizationService) UpdateOrganization(ctx context.Context, id influxd
if err != nil {
return nil, tracing.LogError(span, err)
}
tracing.InjectToHTTPRequest(span, req)
req.Header.Set("Content-Type", "application/json")
SetToken(s.Token, req)
@ -793,7 +790,6 @@ func (s *OrganizationService) DeleteOrganization(ctx context.Context, id influxd
if err != nil {
return tracing.LogError(span, err)
}
tracing.InjectToHTTPRequest(span, req)
SetToken(s.Token, req)

2
http/query_handler.go
View File

@ -370,7 +370,6 @@ func (s *FluxService) Query(ctx context.Context, w io.Writer, r *query.ProxyRequ
hreq.Header.Set("Content-Type", "application/json")
hreq.Header.Set("Accept", "text/csv")
hreq = hreq.WithContext(ctx)
tracing.InjectToHTTPRequest(span, hreq)
hc := NewClient(u.Scheme, s.InsecureSkipVerify)
resp, err := hc.Do(hreq)
@ -438,7 +437,6 @@ func (s *FluxQueryService) Query(ctx context.Context, r *query.Request) (flux.Re
hreq.Header.Set("Content-Type", "application/json")
hreq.Header.Set("Accept", "text/csv")
hreq = hreq.WithContext(ctx)
tracing.InjectToHTTPRequest(span, hreq)
hc := NewClient(u.Scheme, s.InsecureSkipVerify)
resp, err := hc.Do(hreq)

1
http/source_proxy_service.go
View File

@ -54,7 +54,6 @@ func (s *SourceProxyQueryService) queryFlux(ctx context.Context, w io.Writer, re
hreq.Header.Set("Authorization", fmt.Sprintf("Token %s", s.Token))
hreq.Header.Set("Content-Type", "application/json")
hreq = hreq.WithContext(ctx)
tracing.InjectToHTTPRequest(span, hreq)
hc := NewClient(u.Scheme, s.InsecureSkipVerify)
resp, err := hc.Do(hreq)

11
http/task_service.go
View File

@ -1373,7 +1373,6 @@ func (t TaskService) FindTaskByID(ctx context.Context, id platform.ID) (*platfor
return nil, err
}
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
resp, err := hc.Do(req)
@ -1435,7 +1434,6 @@ func (t TaskService) FindTasks(ctx context.Context, filter platform.TaskFilter)
return nil, 0, err
}
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
resp, err := hc.Do(req)
@ -1482,7 +1480,6 @@ func (t TaskService) CreateTask(ctx context.Context, tc platform.TaskCreate) (*p
req.Header.Set("Content-Type", "application/json")
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
@ -1525,7 +1522,6 @@ func (t TaskService) UpdateTask(ctx context.Context, id platform.ID, upd platfor
req.Header.Set("Content-Type", "application/json")
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
@ -1564,7 +1560,6 @@ func (t TaskService) DeleteTask(ctx context.Context, id platform.ID) error {
req.Header.Set("Content-Type", "application/json")
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
@ -1603,7 +1598,6 @@ func (t TaskService) FindLogs(ctx context.Context, filter platform.LogFilter) ([
return nil, 0, err
}
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
@ -1657,7 +1651,6 @@ func (t TaskService) FindRuns(ctx context.Context, filter platform.RunFilter) ([
req.Header.Set("Content-Type", "application/json")
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
@ -1700,7 +1693,6 @@ func (t TaskService) FindRunByID(ctx context.Context, taskID, runID platform.ID)
}
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
@ -1744,7 +1736,6 @@ func (t TaskService) RetryRun(ctx context.Context, taskID, runID platform.ID) (*
}
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
@ -1792,7 +1783,6 @@ func (t TaskService) ForceRun(ctx context.Context, taskID platform.ID, scheduled
}
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)
@ -1843,7 +1833,6 @@ func (t TaskService) CancelRun(ctx context.Context, taskID, runID platform.ID) e
}
SetToken(t.Token, req)
tracing.InjectToHTTPRequest(span, req)
hc := NewClient(u.Scheme, t.InsecureSkipVerify)