Spring Boot中的API网关实现

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 someHeaderValue = exchange.getRequest().getHeaders().getFirst("Some-Header");
        if (someHeaderValue == null) {
            // 如果没有这个请求头，返回403 Forbidden
            response.setStatusCode(HttpStatus.FORBIDDEN);
            return DataBufferUtils.join(response.getBufferFactory())
                    .flatMap(buffer -> {
                        return response.writeWith(Mono.just(buffer));
                    });
        }
 
        // 如果请求头存在，继续请求处理
        return chain.filter(exchange);
    }
}

这段代码定义了一个全局过滤器，用于检查进入API网关的请求是否包含特定的请求头。如果请求头不存在，它将返回403 Forbidden响应。如果请求头存在，请求将被继续处理。这是一个简单的权限控制示例，实际应用中可以根据需要进行更复杂的逻辑判断。