Spring Cloud Gateway编码实现任意地址跳转，Alibaba技术官内部私藏MySQL笔记分享

com.bolingcavalry

common

${project.version}

org.springframework.cloud

spring-cloud-starter-gateway

org.springframework.boot

spring-boot-maven-plugin

com.bolingcavalry.gateway.GatewayDynamicRouteApplication

repackage

• 启动类是普通的SpringBoot启动类：

package com.bolingcavalry.gateway;

import org.springframework.boot.SpringApplication;

import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication

public class GatewayDynamicRouteApplication {

public static void main(String[] args) {

SpringApplication.run(GatewayDynamicRouteApplication.class,args);

}

}

• 接下来是本篇的核心，自定义过滤器类，代码中已经添加了详细的注释，有几处要注意的地方稍后会提到：

package com.bolingcavalry.gateway.filter;

import lombok.Data;

import lombok.ToString;

import lombok.extern.slf4j.Slf4j;

import org.springframework.cloud.gateway.filter.GatewayFilter;

import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;

import org.springframework.cloud.gateway.route.Route;

import org.springframework.http.HttpHeaders;

import org.springframework.http.HttpMethod;

import org.springframework.http.server.reactive.ServerHttpRequest;

import org.springframework.stereotype.Component;

import org.springframework.util.MultiValueMap;

import org.springframework.web.util.UriComponentsBuilder;

import java.net.URI;

import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.GATEWAY_ROUTE_ATTR;

@Component

@Slf4j

public class BizLogicRouteGatewayFilterFactory extends AbstractGatewayFilterFactory<BizLogicRouteGatewayFilterFactory.BizLogicRouteConfig> {

private static final String TAG_TEST_USER = “tag-test-user”;

public BizLogicRouteGatewayFilterFactory() {

super(BizLogicRouteConfig.class);

}

@Override

public GatewayFilter apply(BizLogicRouteConfig config) {

return (exchange, chain) -> {

// 本次的请求对象

ServerHttpRequest request = exchange.getRequest();

// 调用方请求时的path

String rawPath = request.getURI().getRawPath();

log.info(“rawPath [{}]”, rawPath);

// 请求头

HttpHeaders headers = request.getHeaders();

// 请求方法

HttpMethod httpMethod = request.getMethod();

// 请求参数

MultiValueMap<String, String> queryParams = request.getQueryParams();

// 这就是定制的业务逻辑，isTestUser等于ture代表当前请求来自测试用户，需要被转发到测试环境

boolean isTestUser = false;

// 如果header中有tag-test-user这个key，并且值等于true(不区分大小写)，

// 就认为当前请求是测试用户发来的

if (headers.containsKey(TAG_TEST_USER)) {

String tageTestUser = headers.get(TAG_TEST_USER).get(0);

if (“true”.equalsIgnoreCase(tageTestUser)) {

isTestUser = true;

}

}

URI uri;

if (isTestUser) {

log.info(“这是测试用户的请求”);

// 从配置文件中得到测试环境的uri

uri = UriComponentsBuilder.fromHttpUrl(config.getTestEnvUri() + rawPath).queryParams(queryParams).build().toUri();

} else {

log.info(“这是普通用户的请求”);

// 从配置文件中得到正式环境的uri

uri = UriComponentsBuilder.fromHttpUrl(config.getProdEnvUri() + rawPath).queryParams(queryParams).build().toUri();

}

// 生成新的Request对象，该对象放弃了常规路由配置中的spring.cloud.gateway.routes.uri字段

ServerHttpRequest serverHttpRequest = request.mutate().uri(uri).method(httpMethod).headers(httpHeaders -> httpHeaders = httpHeaders).build();

// 取出当前的route对象

Route route = exchange.getAttribute(GATEWAY_ROUTE_ATTR);

//从新设置Route地址

Route newRoute =

Route.async().asyncPredicate(route.getPredicate()).filters(route.getFilters()).id(route.getId())

.order(route.getOrder()).uri(uri).build();

// 放回exchange中

exchange.getAttributes().put(GATEWAY_ROUTE_ATTR,newRoute);

// 链式处理，交给下一个过滤器

return chain.filter(exchange.mutate().request(serverHttpRequest).build());

};

}

/**

• 这是过滤器的配置类，配置信息会保存在此处

*/

@Data

@ToString

public static class BizLogicRouteConfig {

// 生产环境的服务地址

private String prodEnvUri;

// 测试环境的服务地址

private String testEnvUri;

}

}

• 上述代码中要注意的地方如下：

1. BizLogicRouteConfig是过滤器的配置类，可以在使用过滤器时在配置文件中配置prodEnvUri和testEnvUri的值，在代码中可以通过这两个字段取得配置值

2. 过滤器的工厂类名为BizLogicRouteGatewayFilterFactory，按照规则，过滤器的名字是BizLogicRoute

3. 在apply方法中，重新创建ServerHttpRequest和Route对象，它们的参数可以按照业务需求随意设置，然后再将这两个对象设置给SpringCloud gateway的处理链中，接下来，处理链上的其他过滤拿到的就是新的ServerHttpRequest和Route对象了

配置

小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数初中级Java工程师，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年最新Java开发全套学习资料》送给大家，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频

如果你觉得这些内容对你有帮助，可以添加下面V无偿领取！（备注Java）

份《2024年最新Java开发全套学习资料》送给大家，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。**
[外链图片转存中…(img-2s6yuhqH-1710891245570)]
[外链图片转存中…(img-7NTZtVFX-1710891245571)]
[外链图片转存中…(img-kUwzR38I-1710891245571)]

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频

如果你觉得这些内容对你有帮助，可以添加下面V无偿领取！（备注Java）
[外链图片转存中…(img-sicc9NiY-1710891245571)]