Java 网关(Spring Cloud Gateway)深入学习

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) {
        ServerHttpResponse response = exchange.getResponse();

        // 自定义逻辑，例如检查请求头是否包含特定信息
        String headerValue = exchange.getRequest().getHeaders().getFirst("X-My-Header");
        if (headerValue == null || !headerValue.equals("expected-value")) {
            // 如果请求头不匹配，则返回403 Forbidden响应
            response.setStatusCode(HttpStatus.FORBIDDEN);
            return DataBufferUtils.join(response.getBufferFactory())
                    .flatMap(buffer -> {
                        return response.writeWith(Mono.just(buffer));
                    });
        }

        // 如果请求头匹配，则继续执行后续过滤器链
        return chain.filter(exchange);
    }
}