import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.http.HttpStatus;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
 
/**
 * 自定义全局过滤器,用于处理请求限流后的响应
 */
public class CustomGlobalFilter implements GlobalFilter {
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        return chain.filter(exchange).then(Mono.defer(() -> {
            ServerHttpResponse response = exchange.getResponse();
            // 如果请求被限流,则返回429状态码
            if (/* 检查请求是否被限流 */) {
                response.setStatusCode(HttpStatus.TOO_MANY_REQUESTS);
                // 清空响应体中的数据
                return DataBufferUtils.join(response.getFlushes())
                        .flatMap(buffer -> {
                            return response.writeWith(Mono.just(buffer));
                        });
            }
            // 请求未被限流,继续正常流程
            return Mono.empty();
        }));
    }
}这段代码定义了一个自定义的全局过滤器,用于在请求被限流后,设置响应状态码为429 TOO_MANY_REQUESTS,并清空响应体中的数据。这样可以避免向客户端返回大量的数据,从而减少潜在的安全风险。