【OAuth2系列】Spring Cloud Gateway 作为OAuth2 Client接入第三方单点登录代码实践

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.ServerHttpRequest;
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 OAuth2LoginFilter implements GlobalFilter {
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
        ServerHttpResponse response = exchange.getResponse();
 
        // 判断是否需要进行OAuth2认证
        if (需要认证(request)) {
            // 认证未通过，返回401 Unauthorized
            response.setStatusCode(HttpStatus.UNAUTHORIZED);
            return DataBufferUtils.join(response.bufferFactory()
                    .wrap("Authentication Failed".getBytes(StandardCharsets.UTF_8)))
                    .flatMap(buffer -> {
                        return response.writeWith(Mono.just(buffer));
                    });
        }
 
        // 继续执行后续过滤器
        return chain.filter(exchange);
    }
 
    private boolean 需要认证(ServerHttpRequest request) {
        // 逻辑判断是否需要认证，例如检查路径是否需要保护等
        // 返回true表示需要认证，返回false表示不需要认证
        return false;
    }
}

这个代码实例展示了如何在Spring Cloud Gateway中实现一个简单的OAuth2客户端认证过滤器。它检查请求是否需要进行认证，并返回401未授权的状态码和简单的错误信息。如果认证成功，则调用链继续执行后续的过滤器和路由。这个例子仅提供了核心逻辑，实际应用中需要根据具体的认证逻辑进行扩展和完善。