2024年Java最全Spring Cloud Gateway自定义过滤器实战(观测断路器状态变化)，总结拼多多，美团JAVA面试经验

import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
 
import java.nio.charset.StandardCharsets;
 
public class CustomGlobalFilter implements GlobalFilter, Ordered {
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpResponse response = exchange.getResponse();
 
        // 注意：这里的response.getBody()可能已经有值了，需要确保不会产生副作用
        // 使用DataBufferUtils.join()可以确保不会有副作用
        Mono<Void> modifiedResponseBody = response.getBody()
            .map(dataBuffer -> {
                byte[] contents = new byte[dataBuffer.readableByteCount()];
                dataBuffer.read(contents);
                // 这里可以添加逻辑来观测断路器的状态变化
                // 例如，可以根据返回的数据内容来判断服务健康状况
                // 断路器状态变化的观测逻辑放在这里
                return buffer(contents);
            });
 
        // 重写返回的response的body
        return chain.filter(exchange).then(Mono.defer(() -> {
            // 重新设置返回内容
            response.getHeaders().setContentType(org.springframework.http.MediaType.APPLICATION_JSON);
            return modifiedResponseBody;
        }));
    }
 
    private DataBuffer buffer(byte[] data) {
        // 创建一个数据缓冲区并写入数据
        return ...; // 这里需要创建一个数据缓冲区并写入修改后的数据
    }
 
    @Override
    public int getOrder() {
        // 设置全局过滤器的执行顺序
        return ...; // 返回合适的顺序值
    }
}

在这个代码实例中，我们创建了一个自定义的全局过滤器CustomGlobalFilter，实现了GlobalFilter和Ordered接口。在filter方法中，我们首先获取了响应体，然后观测并可能修改响应体的内容，最后重写了响应体，并设置了正确的内容类型。我们需要注意，在实际的应用中，应该避免在获取响应体后再次修改响应体导致的副作用。在这个代码实例中，我们假设了buffer方法来创建一个新的数据缓冲区并写入修改后的数据。实际应用中，你需要使用合适的方法来创建DataBuffer。