问从ServerWebExchange/ServerHttpRequest获取主体作为原始字符串

EN

@Slf4j
public class CachingBodyServerHttpRequestDecorator extends ServerHttpRequestDecorator {

private final StringBuilder cachedBody = new StringBuilder();

public CachingBodyServerHttpRequestDecorator(ServerHttpRequest delegate) {
    super(delegate);
}

@Override
public Flux<DataBuffer> getBody() {
    return super.getBody().doOnNext(this::cache);
}

@SneakyThrows
private void cache(DataBuffer buffer) {
    cachedBody.append(UTF_8.decode(buffer.asByteBuffer()));
}

public String getCachedBody() {
    return cachedBody.toString();
}
}


public final class CachingBodyServerWebExchangeDecorator extends ServerWebExchangeDecorator {

private final ServerHttpRequestDecorator requestDecorator;

public CachingBodyServerWebExchangeDecorator(ServerWebExchange delegate) {
    super(delegate);
    this.requestDecorator = new CachingBodyServerHttpRequestDecorator(delegate.getRequest());
}

@Override
public ServerHttpRequest getRequest() {
    return requestDecorator;
}

}


@Component
@Slf4j
public class SetRequestBodyAttribute implements WebFilter {

    private static ServerHttpRequestPredicate requestsToServe =
            MethodServerHttpRequestPredicate.of(HttpMethod.POST)
                    .and(PathServerHttpRequestPredicate.of("/contacts/import"))
                    .or(MethodServerHttpRequestPredicate.of(HttpMethod.PATCH)
                            .and(PathServerHttpRequestPredicate.of("/contacts/{contact_id}")));

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
        if (requestsToServe.test(exchange.getRequest())) {
             
        return chain.filter(new CachingBodyServerWebExchangeDecorator(exchange));
        }
        return chain.filter(exchange);
    }
}


@UtilityClass
public class ServerHttpRequestUtils {

    public Mono<String> getBodyAsString(ServerHttpRequest request) {
        return Mono.just(((CachingBodyServerHttpRequestDecorator)exchange.getRequest()).getCachedBody())
    }
}


public interface ServerHttpRequestPredicate {

    boolean test(ServerHttpRequest request);

    @NotNull
    default ServerHttpRequestPredicate and(@NotNull ServerHttpRequestPredicate other) {
        Objects.requireNonNull(other);
        return (t) -> test(t) && other.test(t);
    }

    @NotNull
    default ServerHttpRequestPredicate negate() {
        return (t) -> !test(t);
    }

    @NotNull
    default ServerHttpRequestPredicate or(@NotNull ServerHttpRequestPredicate other) {
        Objects.requireNonNull(other);
        return (t) -> test(t) || other.test(t);
    }
}

@Value
public class PathServerHttpRequestPredicate implements ServerHttpRequestPredicate {

    public static PathServerHttpRequestPredicate of(String path) {
        return new PathServerHttpRequestPredicate(path);
    }

    PathPattern pathPattern;

    public PathServerHttpRequestPredicate(String path) {
        Assert.notNull(path, "path should not be null!");
        this.pathPattern = PathPatternParser.defaultInstance.parse(path);
    }

    @Override
    public boolean test(ServerHttpRequest request) {
        PathContainer pathContainer = request.getPath().pathWithinApplication();
        PathPattern.PathMatchInfo info = this.pathPattern.matchAndExtract(pathContainer);
        return info != null;
    }
}

@Value(staticConstructor = "of")
public class MethodServerHttpRequestPredicate implements ServerHttpRequestPredicate{

    HttpMethod httpMethod;

    public MethodServerHttpRequestPredicate(HttpMethod httpMethod) {
        Assert.notNull(httpMethod, "httpMethod should not be null!");
        this.httpMethod = httpMethod;
    }

    @Override
    public boolean test(ServerHttpRequest request) {
        return request.getMethod() == httpMethod;
    }
}

Mono<String> bodyAsStringMono = ServerHttpRequestUtils .getBodyAsString(exchange);