demo-Webflux统一返回结果脱敏

上代码

package cn.texous.demo.filter;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.reactivestreams.Publisher;
import org.springframework.core.annotation.Order;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.http.server.reactive.ServerHttpResponseDecorator;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import org.springframework.web.server.WebFilter;
import org.springframework.web.server.WebFilterChain;
import reactor.core.publisher.Mono;

import java.nio.charset.StandardCharsets;

/**
 * 参考来源：<a href="https://mp.weixin.qq.com/s/hPNaES0ycqEmeqm3tTzcZg">文档</a>
 */
@SuppressWarnings("NullableProblems")
@Slf4j
@Order(6)
@Component
@RequiredArgsConstructor
public class DesensitizedFilter implements WebFilter {

    private final DesensitizedService desensitizedService;

    /**
     * 其中有几个地方需要注意
     * ① 修改了返回数据之后，需要重新设置Content-Length头，否则可能造成数据传输不完整的情况
     * ② 获取buffer工厂，使用新的数据重新生成ByteBuffer对象
     * ③ 对于自己创建的ByteBuffer对象，在使用完成之后，需要手动release，否则会造成内存泄露
     *
     * @param exchange the current server exchange
     * @param chain    provides a way to delegate to the next filter
     * @return null
     */
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
        String uri = request.getURI().getPath();
        DesensitizedVo byUrl = desensitizedService.getByUrl(uri, request.getMethodValue());
        if (byUrl != null
                && (!StringUtils.isEmpty(byUrl.getKeys())
                || !StringUtils.isEmpty(byUrl.getDetailKeys()))) {
            ServerHttpResponse original = exchange.getResponse();
            // 创建新的response装饰对象，并传入原始response对象，然后重写writeWith方法
            ServerHttpResponseDecorator newResponse = new ServerHttpResponseDecorator(original) {
                @Override
                public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
                    Mono<DataBuffer> mono = DataBufferUtils.join(body)
                            .map(buffer -> {
                                try {
                                    String data = buffer.toString(StandardCharsets.UTF_8);
                                    String mutatedData = mutate(data, byUrl); // 此处修改返回内容
                                    byte[] bytes = mutatedData.getBytes(StandardCharsets.UTF_8);
                                    getHeaders().setContentLength(bytes.length); // ①
                                    return this.bufferFactory().wrap(bytes); // ②
                                } finally {
                                    DataBufferUtils.release(buffer); // ③
                                }
                            });

                    return super.writeWith(mono);
                }
            };
            return chain.filter(exchange.mutate().response(newResponse).build());
        } else {
            return chain.filter(exchange);
        }
    }

    private String mutate(String data, DesensitizedVo desensitizedVo) {
        try {
            return DesensitizedUtils.desensitizedJsonString(data,
                    desensitizedVo.getKeys(), desensitizedVo.getDetailKeys());
        } catch (Exception e) {
            log.error("mutate data error", e);
        }
        return data;
    }

}