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

chore(httpc): extend httpc reqs with tracing

pull/18377/head
Johnny Steenbergen 2020-06-04 14:53:35 -07:00 committed by Johnny Steenbergen
parent bc9b69b96f
commit dccae684fd
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/httpc/req.go
View File

@ -12,6 +12,7 @@ import (
"strings"
"github.com/influxdata/influxdb/v2"
"github.com/influxdata/influxdb/v2/kit/tracing"
)
const (
@ -144,6 +145,19 @@ func (r *Req) Do(ctx context.Context) error {
}
func (r *Req) do(ctx context.Context) error {
span, ctx := tracing.StartSpanFromContextWithOperationName(ctx, r.req.URL.String())
defer span.Finish()
u := r.req.URL
span.LogKV(
"scheme", u.Scheme,
"host", u.Host,
"path", u.Path,
"query_params", u.Query().Encode(),
)
tracing.InjectToHTTPRequest(span, r.req)
resp, err := r.client.Do(r.req.WithContext(ctx))
if err != nil {
return err
@ -153,6 +167,11 @@ func (r *Req) do(ctx context.Context) error {
resp.Body.Close()
}()
span.LogKV(
"response_code", resp.StatusCode,
"response_byte", resp.ContentLength,
)
if r.respFn != nil {
if err := r.respFn(resp); err != nil {
return err