Fix data race between kafkareceiver and batchprocessor (open-telemetr…

…y#2956) (open-telemetry#2957)
punya · Apr 19, 2021 · ec395ec · ec395ec
1 parent 93821c9
commit ec395ec
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kafka_receiver.go b/kafka_receiver.go
@@ -171,8 +171,9 @@ func (c *consumerGroupHandler) ConsumeClaim(session sarama.ConsumerGroupSession,
 			return err
 		}
 
+		spanCount := traces.SpanCount()
 		err = c.nextConsumer.ConsumeTraces(session.Context(), traces)
-		obsreport.EndTraceDataReceiveOp(ctx, c.unmarshaller.Encoding(), traces.SpanCount(), err)
+		obsreport.EndTraceDataReceiveOp(ctx, c.unmarshaller.Encoding(), spanCount, err)
 		if err != nil {
 			return err
 		}