breadcrumb_transport.go

package logger

import (
	"net/http"
	"time"

	"github.com/getsentry/sentry-go"
)

type breadcrumbTransport struct {
	Transport http.RoundTripper

	Level sentry.Level
}

func NewBreadcrumbTransport(level sentry.Level, transport http.RoundTripper) http.RoundTripper {
	if transport == nil {
		transport = http.DefaultTransport
	}
	return &breadcrumbTransport{
		Transport: transport,
		Level:     level,
	}
}

//nolint:contextcheck
func (b breadcrumbTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	span := sentry.StartSpan(req.Context(), req.URL.String(), sentry.ContinueFromRequest(req))
	defer span.Finish()

	//nolint:canonicalheader
	req.Header.Add("sentry-trace", span.ToSentryTrace())

	breadcrumb := sentry.Breadcrumb{
		Data: map[string]interface{}{
			BreadcrumbDataURL:    req.URL.String(),
			BreadcrumbDataMethod: req.Method,
		},
		Level:     b.Level,
		Timestamp: time.Now().UTC(),
		Type:      BreadcrumbTypeHTTP,
	}

	resp, err := b.Transport.RoundTrip(req.WithContext(span.Context()))

	if err == nil {
		span.Status = SpanStatus(resp.StatusCode)
		breadcrumb.Data[BreadcrumbDataStatusCode] = resp.StatusCode
		breadcrumb.Data[BreadcrumbDataReason] = resp.Status
	} else {
		span.Status = sentry.SpanStatusAborted
		breadcrumb.Message = err.Error()
	}

	Hub(span.Context()).AddBreadcrumb(&breadcrumb, nil)

	return resp, err //nolint:wrapcheck
}