mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-01-12 22:37:27 -08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

chore: add tcp events to remote store span

Browse source 
Signed-off-by: Juraj Michalek <juraj.michalek132@gmail.com>
This commit is contained in:
Juraj Michalek 2024-10-25 16:46:00 +02:00
parent f131cdd4c5
commit 2e7c739d44
1 changed files with 32 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
storage/remote/client.go
View file

@ -18,8 +18,10 @@ import (
"context"
"errors"
"fmt"
"go.opentelemetry.io/otel/attribute"
"io"
"net/http"
"net/http/httptrace"
"strconv"
"strings"
"time"
@ -279,7 +281,36 @@ func (c *Client) Store(ctx context.Context, req []byte, attempt int) (WriteRespo
ctx, span := otel.Tracer("").Start(ctx, "Remote Store", trace.WithSpanKind(trace.SpanKindClient))
defer span.End()
httpResp, err := c.Client.Do(httpReq.WithContext(ctx))
httpReqTrace := &httptrace.ClientTrace{
GetConn: func(hostPort string) {
span.AddEvent("GetConn", trace.WithAttributes(attribute.String("host", hostPort)))
},
GotConn: func(info httptrace.GotConnInfo) {
span.AddEvent("GotConn", trace.WithAttributes(
attribute.Bool("reused", info.Reused),
attribute.Bool("wasIdle", info.WasIdle),
//attribute.Duration("idleTime", info.IdleTime),
))
},
DNSStart: func(info httptrace.DNSStartInfo) {
span.AddEvent("DNSStart", trace.WithAttributes(attribute.String("host", info.Host)))
},
DNSDone: func(info httptrace.DNSDoneInfo) {
span.AddEvent("DNSDone", trace.WithAttributes(attribute.Bool("coalesced", info.Coalesced)))
},
ConnectStart: func(network, addr string) {
span.AddEvent("ConnectStart", trace.WithAttributes(attribute.String("network", network), attribute.String("addr", addr)))
},
ConnectDone: func(network, addr string, err error) {
attrs := []attribute.KeyValue{attribute.String("network", network), attribute.String("addr", addr)}
if err != nil {
attrs = append(attrs, attribute.String("error", err.Error()))
}
span.AddEvent("ConnectDone", trace.WithAttributes(attrs...))
},
}
httpResp, err := c.Client.Do(httpReq.WithContext(httptrace.WithClientTrace(ctx, httpReqTrace)))
if err != nil {
// Errors from Client.Do are from (for example) network errors, so are
// recoverable.