Spring Cloud Gateway 3 和 Spring WebFlux 常用操作

木棉脚

已于 2023-07-13 10:57:07 修改

阅读量273

点赞数

分类专栏： Spring 微服务文章标签： spring java spring boot springcloud gateway

于 2023-07-13 10:04:23 首次发布

本文链接：https://blog.csdn.net/qq_42254218/article/details/131696081

版权

微服务同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

Spring

3 篇文章 0 订阅

订阅专栏

Spring Cloud Gateway 3 和 Spring WebFlux 中拦截器直接响应结果

private Mono<Void> returnNotLogin(ServerWebExchange exchange) {
        ServerHttpResponse response = exchange.getResponse();
        response.getHeaders().setContentType(MediaType.APPLICATION_JSON);
        DataBuffer wrap = response.bufferFactory().wrap(
            JSON.toJSONString(ServerResponseEntity.fail(ResponseEnum.NOT_AUTH), JSONWriter.Feature.WriteMapNullValue)
                .getBytes(StandardCharsets.UTF_8));
        return response.writeWith(Flux.just(wrap));
    }

Spring Cloud Gateway 3 和 Spring WebFlux 中不同过滤器间传递参数、修改gateway转发请求头

@Slf4j
@Component
public class InternalServiceAuthFilter implements GatewayFilter, Ordered {

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        // 不同过滤器间传递参数
        exchange.getAttributes().put("","");
        Object attribute = exchange.getAttribute(Constant.W3_LOGIN_IS_SUCCESS_KEY);
        ServerHttpRequest request = exchange.getRequest();
        
        // 修改gateway转发请求头
        ServerHttpRequest.Builder mutate = request.mutate();
        mutate.header(Constant.OPENX_USER_INFO_HEADER_NAME, JSON.toJSONString(attribute));
        mutate.headers(x -> {
            x.remove(tokenHeader);
            x.remove(wetokenHeader);
        });
        ServerWebExchange newExchange = exchange.mutate().request(request).build();
        return chain.filter(newExchange);
    }

    @Override
    public int getOrder() {
        return 20;
    }
}

Spring Cloud Gateway 3 和 Spring WebFlux 中获取请求ip

@Slf4j
public class IPAddressUtil {

    public static final String LOCALHOST_IPV4 = "127.0.0.1";

    public static final String LOCALHOST_IPV6 = "0:0:0:0:0:0:0:1";

    private static final String UNKNOWN = "unknown";

    public static String getIpAddress(ServerHttpRequest req) {
        String ip = IPAddressUtil.getFirst(req.getHeaders().get("x-forwarded-for"));
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            InetSocketAddress remoteAddress = req.getRemoteAddress();
            if (remoteAddress != null) {
                ip = remoteAddress.getHostString();
            }
        }
        String comma = ",";
        if (ip.contains(comma)) {
            ip = ip.split(",")[0];
        }
        if (LOCALHOST_IPV4.equals(ip) || LOCALHOST_IPV6.equals(ip)) {
            // 获取本机真正的ip地址，内部ip
            try {
                ip = InetAddress.getLocalHost().getHostAddress();
            } catch (UnknownHostException e) {
                log.error(ExceptionUtils.getStackTrace(e));
            }
        }
        return ip;
    }

    public static String getFirst(List<String> list) {
        if (list == null || list.isEmpty()) {
            return "";
        }
        return list.get(0);
    }
}