probers: initiate traces for browser, scripted, and multihttp checks

grafana · Oct 4, 2024 · b386066 · b386066
1 parent c65fdbb
commit b386066
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/internal/prober/browser/browser.go b/internal/prober/browser/browser.go
@@ -6,9 +6,11 @@ import (
 
 	"github.com/grafana/synthetic-monitoring-agent/internal/k6runner"
 	"github.com/grafana/synthetic-monitoring-agent/internal/prober/logger"
+	"github.com/grafana/synthetic-monitoring-agent/internal/tracing"
 	sm "github.com/grafana/synthetic-monitoring-agent/pkg/pb/synthetic_monitoring"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/rs/zerolog"
+	"go.opentelemetry.io/otel/attribute"
 )
 
 const proberName = "browser"
@@ -60,12 +62,18 @@ func (p Prober) Name() string {
 }
 
 func (p Prober) Probe(ctx context.Context, target string, registry *prometheus.Registry, logger logger.Logger) (bool, float64) {
+	ctx, probeSpan := tracing.TracerProvider(ctx).Tracer("").Start(ctx, "browser check")
+	defer probeSpan.End()
+
 	success, err := p.processor.Run(ctx, registry, logger, p.logger)
 	if err != nil {
 		p.logger.Error().Err(err).Msg("running probe")
+		probeSpan.RecordError(err)
 		return false, 0
 	}
 
+	probeSpan.SetAttributes(attribute.Bool("success", success))
+
 	// TODO(mem): implement custom duration extraction.
 	return success, 0
 }
diff --git a/internal/prober/multihttp/multihttp.go b/internal/prober/multihttp/multihttp.go
@@ -8,9 +8,11 @@ import (
 
 	"github.com/grafana/synthetic-monitoring-agent/internal/k6runner"
 	"github.com/grafana/synthetic-monitoring-agent/internal/prober/logger"
+	"github.com/grafana/synthetic-monitoring-agent/internal/tracing"
 	sm "github.com/grafana/synthetic-monitoring-agent/pkg/pb/synthetic_monitoring"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/rs/zerolog"
+	"go.opentelemetry.io/otel/attribute"
 )
 
 const proberName = "multihttp"
@@ -83,12 +85,18 @@ func (p Prober) Name() string {
 }
 
 func (p Prober) Probe(ctx context.Context, target string, registry *prometheus.Registry, logger logger.Logger) (bool, float64) {
+	ctx, probeSpan := tracing.TracerProvider(ctx).Tracer("").Start(ctx, "multihttp check")
+	defer probeSpan.End()
+
 	success, err := p.processor.Run(ctx, registry, logger, p.logger)
 	if err != nil {
 		p.logger.Error().Err(err).Msg("running probe")
+		probeSpan.RecordError(err)
 		return false, 0
 	}
 
+	probeSpan.SetAttributes(attribute.Bool("success", success))
+
 	// TODO(mem): implement custom duration extraction.
 	return success, 0
 }

diff --git a/internal/prober/scripted/scripted.go b/internal/prober/scripted/scripted.go
@@ -6,9 +6,11 @@ import (
 
 	"github.com/grafana/synthetic-monitoring-agent/internal/k6runner"
 	"github.com/grafana/synthetic-monitoring-agent/internal/prober/logger"
+	"github.com/grafana/synthetic-monitoring-agent/internal/tracing"
 	sm "github.com/grafana/synthetic-monitoring-agent/pkg/pb/synthetic_monitoring"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/rs/zerolog"
+	"go.opentelemetry.io/otel/attribute"
 )
 
 const proberName = "scripted"
@@ -60,12 +62,17 @@ func (p Prober) Name() string {
 }
 
 func (p Prober) Probe(ctx context.Context, target string, registry *prometheus.Registry, logger logger.Logger) (bool, float64) {
+	ctx, probeSpan := tracing.TracerProvider(ctx).Tracer("").Start(ctx, "scripted check")
+	defer probeSpan.End()
+
 	success, err := p.processor.Run(ctx, registry, logger, p.logger)
 	if err != nil {
 		p.logger.Error().Err(err).Msg("running probe")
+		probeSpan.RecordError(err)
 		return false, 0
 	}
 
+	probeSpan.SetAttributes(attribute.Bool("success", success))
 	// TODO(mem): implement custom duration extraction.
 	return success, 0
 }