implement batching for dropped decisions

honeycombio · Oct 3, 2024 · 178f1ce · 178f1ce
1 parent 60c0892
commit 178f1ce
Show file tree

Hide file tree

Showing 6 changed files with 380 additions and 222 deletions.
diff --git a/app/app_test.go b/app/app_test.go
@@ -35,6 +35,7 @@ import (
 	"github.com/honeycombio/refinery/internal/peer"
 	"github.com/honeycombio/refinery/logger"
 	"github.com/honeycombio/refinery/metrics"
+	"github.com/honeycombio/refinery/pubsub"
 	"github.com/honeycombio/refinery/sample"
 	"github.com/honeycombio/refinery/sharder"
 	"github.com/honeycombio/refinery/transmit"
@@ -198,6 +199,7 @@ func newStartedApp(
 		&inject.Object{Value: shrdr},
 		&inject.Object{Value: noop.NewTracerProvider().Tracer("test"), Name: "tracer"},
 		&inject.Object{Value: collector},
+		&inject.Object{Value: &pubsub.LocalPubSub{}},
 		&inject.Object{Value: metricsr, Name: "metrics"},
 		&inject.Object{Value: metricsr, Name: "genericMetrics"},
 		&inject.Object{Value: metricsr, Name: "upstreamMetrics"},
@@ -778,7 +780,7 @@ func TestPeerRouting_TraceLocalityDisabled(t *testing.T) {
 		Data: map[string]interface{}{
 			"trace_id":                     "2",
 			"meta.refinery.min_span":       true,
-			"meta.annotation_type":         types.SpanAnnotationTypeUnknown,
+			"meta.annotation_type":         types.SpanTypeUnknown,
 			"meta.refinery.root":           false,
 			"meta.refinery.span_data_size": 175,
 		},