No elegant way to get span in Kafka ErrorHandler

I need a way to get span in Kafka ErrorHanlder. I do the next thing in error handler -

    @Override
      public void handle(Exception ex, ConsumerRecord<?, ?> consumerRecord) {
        var curSpan = this.kafkaTracing.nextSpan(consumerRecord).name("on-message").start()
       try (Tracer.SpanInScope ws = tracer.withSpanInScope(curSpan)) {
      log.error(ex);
        } finally {
      span.finish();
    }
  }

But method in KafkaTracing nextSpan() - clear all headers before error handler do its job. And KafkaTracing if final and i can’t inherit from it. I have a question - is it a bug that you don’t give an opportunity not to remove headers from ConsumerRecord?

[jfumard]

Same problem here ! It seems that the ConsumerRecord<?,?> received in the @Override public void handle(Exception ex, ConsumerRecord<?, ?> consumerRecord) method has been stripped from its headers, including the traceId.

[AntonKuzValid]

Probably you can add a ThreadLocal, this will be setted in proxy. example -

@Aspect
@Component
public class KafkaTracingAspect {
  /** Текущий Span потока */
  public static final ThreadLocal<Span> currentSpan = new ThreadLocal<>();

  @Autowired private Tracer tracer;

  @Around("@annotation(org.springframework.kafka.annotation.KafkaListener)")
  public Object kafkaListener(ProceedingJoinPoint joinPoint) throws Throwable {
    currentSpan.set(tracer.currentSpan());
    return joinPoint.proceed();
  }
}