1、参考官方文档
我们可以猜测，spring gateway作为网关功能，肯定会提供很多处理请求参数的功能，于是我们查询文档得到如下内容：
image
2、探索GatewayFilterFactory实现规律
- 通过查询spring官方文档可以看到，spring gateway为我们提供了很多
xxxGatewayFilterFactory,而这些factory都有相同点，都是以GatewayFilterFactory结尾的。 - 在类名中，我们可以根据类名进行大胆的猜测，前面的几个单词是描述他的功能的。
- 右侧的
yml配置文件可以看到，filter的配置也是呈现出一定的规律的。
3、从源码获取实现原理
既然我们是要解决自定义请求参数封装问题，那么我们通过上面描述规律，可以很大胆的猜测AddRequestParameteGatewayFilterFactory就是我们要找的目标。那么我们查看一下他的源码，看看他是如何实现的。
首先看看AddRequestParameteGatewayFilterFactory的继承关系，大概了解一下他的组成
image
从继承关系来看，还是比较复杂，有点懵逼，那直接进入代码看看他的实现。
我们只粘贴比较核心的代码进行分析，其他代码暂时不用关心。
@Override
public GatewayFilter apply(NameValueConfig config) {
return new GatewayFilter() {
@Override
public Mono<Void> filter(ServerWebExchange exchange,
GatewayFilterChain chain) {
URI uri = exchange.getRequest().getURI();
StringBuilder query = new StringBuilder();
//获取请求uri的请求参数（GET请求参数通过拼接key=value形式进行传参）
String originalQuery = uri.getRawQuery();
//判断最后一个字符是否是&，如果不是则拼接一个&，以备后续的参数进行连接
if (StringUtils.hasText(originalQuery)) {
query.append(originalQuery);
if (originalQuery.charAt(originalQuery.length() - 1) != "&") {
query.append("&");
}
}
//获取config中的key、value，然后拼接到uri请求参数后面
String value = ServerWebExchangeUtils.expand(exchange, config.getValue());
// TODO urlencode?
query.append(config.getName());
query.append("=");
query.append(value);
//把请求参数重新拼接回去，并放入request中传递到过滤链的下一个请求中去
try {
URI newUri = UriComponentsBuilder.fromUri(uri)
.replaceQuery(query.toString()).build(true).toUri();
ServerHttpRequest request = exchange.getRequest().mutate().uri(newUri)
.build();
return chain.filter(exchange.mutate().request(request).build());
}
catch (RuntimeException ex) {
throw new IllegalStateException(
"Invalid URI query: "" + query.toString() + """);
}
}
此方法是用于封装请求参数的具体实现，代码的具体实现步骤已经通过注释进行说明。
那config中的key：value又是如何传递进来的呢？
4、filter的配置和参数传递
代码实现中出现了config参数的封装，那这个参数是如何获得的呢？这时候我们就要去查看这个filter是如何使用的了。
image
- 从图中可以看到，我们的filter就是在这个配置文件中配置使用的。那为什么配置文件中只有
AddRequestParameter配置，而不是AddRequestParameterGatewayFilterFactory呢？？？ - 其实这是spring的一种约定，实现了
GatewayFilterFactory接口的类在配置使用的时候，需要省略掉后面的GatewayFilterFactory，仅配置前缀即可。 - 同时，配置文件中的red，blue又是做什么的？？？
- 这就是我们需要传递的请求参数了，他以key=red，value=blue的方式进行配置。
- 然后spring会帮我们把这个键值传入到上面的apply方法的config中。我们通过config的源码就能看到究竟：
image
可以看到，config类的实现就是一个name 和value，分别对应了配置文件中的两个参数。
三、基于官方实现进行自定义拓展
我们看到，spring利用同种方式，实现了各种filter。但如果这些都不能满足我们的要求，那该怎么办？？？
1、自定义参数惨景预设
我们现在看到的参数信息都是写死在配置文件中的，无法进行动态参数的传递。我们可以设想一个很普通的场景： 我们需要把用户的登录信息封装到请求参数中，供其他服务使用。那这又该如何实现呢？？？
2、参考官方实现进行自定义
其实如果我们只要模仿官方实现，在uri上拼接我们要传递的动态参数就可以了。具体实现如下：
@Override
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
URI uri = exchange.getRequest().getURI();
StringBuilder query = new StringBuilder();
String originalQuery = uri.getRawQuery();
if (StringUtils.hasText(originalQuery)) {
query.append(originalQuery);
if (originalQuery.charAt(originalQuery.length() - 1) != "&") {
query.append("&");
}
}
String value = ServerWebExchangeUtils.expand(exchange, config.getValue());
// TODO urlencode?
query.append(config.getName());
query.append("=");
query.append(value);
//获取redis中用户的缓存信息，拼接到请求参数后面
String token = exchange.getRequest().getHeaders().getFirst("token");
if (StringUtils.hasText(token)) {
AccountEntity accountEntity = accountAdminApiService.loginAccountAdmin(token);
//通过发射拿到bean的属性和值，以备后面进行传递参数拼接
Map<String, Object> beanMap = beanValue(accountEntity);
if (!CollectionUtils.isEmpty(beanMap)) {
for (String key : beanMap.keySet()) {
query.append("&").append(key).append("=").append(beanMap.get(key));
}
}
}
//以下与官方代码一样，省略...
}
在固定传参参数拼接后面，直接通过token去获取缓存中的用户登陆信息，然后依次拼接对应的属性和值即可。
网友评论