|
|
@@ -5,7 +5,6 @@ import lombok.SneakyThrows;
|
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
import org.springframework.http.server.reactive.ServerHttpRequest;
|
|
|
import org.springframework.stereotype.Component;
|
|
|
-import org.springframework.util.MultiValueMap;
|
|
|
import org.springframework.web.server.ServerWebExchange;
|
|
|
import org.springframework.web.server.WebFilter;
|
|
|
import org.springframework.web.server.WebFilterChain;
|
|
|
@@ -23,7 +22,7 @@ public class TraceIdFilter implements WebFilter {
|
|
|
public Mono<Void> filter(@NotNull ServerWebExchange exchange, @NotNull WebFilterChain chain) {
|
|
|
try {
|
|
|
ServerHttpRequest request = exchange.getRequest();
|
|
|
- String traceId = request.getQueryParams().getFirst("traceId");
|
|
|
+ String traceId = request.getHeaders().getFirst("Trace-Id");
|
|
|
if (traceId == null || traceId.isEmpty()) {
|
|
|
TraceId.set();
|
|
|
} else {
|
wangcheng