com.cerner.beadledom.client.resteasy.ResteasyCorrelationIdContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of resteasy-client Show documentation

There is a newer version: 4.1.2

package com.cerner.beadledom.client.resteasy;

import com.cerner.beadledom.client.CorrelationIdContext;
import javax.annotation.Nullable;
import javax.ws.rs.core.HttpHeaders;
import org.jboss.resteasy.spi.ResteasyProviderFactory;
import org.slf4j.MDC;

/**
 * A Resteasy implementation of {@link CorrelationIdContext} that retrieves the correlationId from
 * the request headers if present, and falls back to the MDC for compatibility with
 * beadledom-jaxrs-1.0.
 *
 * @author John Leacox
 * @since 1.0
 */
class ResteasyCorrelationIdContext implements CorrelationIdContext {
  private final String headerName;
  private final String mdcName;

  ResteasyCorrelationIdContext(@Nullable String headerName, @Nullable String mdcName) {
    this.headerName = headerName != null ? headerName : CorrelationIdContext.DEFAULT_HEADER_NAME;
    this.mdcName = mdcName != null ? mdcName : CorrelationIdContext.DEFAULT_MDC_NAME;
  }

  @Override
  public String getCorrelationId() {
    HttpHeaders headers = ResteasyProviderFactory.getContextData(HttpHeaders.class);

    String correlationId;
    if (headers != null) {
      correlationId = headers.getHeaderString(headerName);
      if (correlationId != null) {
        return correlationId;
      }
    }

    // Fall back to MDC to support beadledom-jaxrs 1.0. Retrieving from the headers is preferred.
    correlationId = MDC.get(mdcName);

    return correlationId;
  }
}