pl.piomin.logging.reactive.util.UniqueIDGenerator Maven / Gradle / Ivy

Go to download
package pl.piomin.logging.reactive.util;

import org.slf4j.MDC;
import org.springframework.http.server.reactive.ServerHttpRequest;

import java.util.List;
import java.util.UUID;

public class UniqueIDGenerator {

    private static final String REQUEST_ID_HEADER_NAME = "X-Request-ID";
    private static final String CORRELATION_ID_HEADER_NAME = "X-Correlation-ID";

    public void generateAndSetMDC(ServerHttpRequest request) {
        MDC.clear();

        List requestIds = request.getHeaders().get(REQUEST_ID_HEADER_NAME);
        if (requestIds == null)
            MDC.put(REQUEST_ID_HEADER_NAME, UUID.randomUUID().toString());
        else
            MDC.put(REQUEST_ID_HEADER_NAME, requestIds.get(0));

        List correlationIds = request.getHeaders().get(CORRELATION_ID_HEADER_NAME);
        if (correlationIds == null)
            MDC.put(CORRELATION_ID_HEADER_NAME, UUID.randomUUID().toString());
        else
            MDC.put(CORRELATION_ID_HEADER_NAME, correlationIds.get(0));
    }

}