Spring Cloud - 手写 Gateway 源码，实现自定义局部 FilterFactory

news/2024/10/23 9:31:24/

目录

一、FilterFactory 分析

1.1、前置知识

1.2、分析源码

1.2.1、整体分析

1.2.2、源码分析

1.3、手写源码

1.3.1、基础框架

1.3.2、实现自定义局部过滤器

1.3.3、加参数的自定义局部过滤器器

一、FilterFactory 分析

1.1、前置知识

前面的学习我们知道，GatewayFilter是网关中提供的一种过滤器，可以对进入网关的请求和微服务返回的响应做处理，同时，springcloud 也提供了一些内置的 filter.

比如：StripPrefix，表示给请求的 url 中去表指定的 n 个前缀路由，例如 - StripPrefix=2 那么如果你原本的请求是路由是 /user/list/get ，那么经过 StripPrefix 处理后，就会变成 /get.

如果我们需要自己去实现一个像这样的局部过滤器，该怎么实现呢？

1.2、分析源码

1.2.1、整体分析

例如 StripPrefix，他继承了 AbstractGatewayFilterFactory 这个抽象类.

这里暗含了一层意思：在 application.yml 配置文件中，可以在 filters 配置里写上这个类的前缀 StripPrefix，就表示这个类（后面的 GatewayFilterFactory 是固定写法，就表示他是一个网关过滤器）.

进一步的，如果我们要自定义一个局部过滤器，例如身份认证 Token 过滤器，我们就创建一个类命名为：Token + GatewayFilter，然后继承 AbstractGatewayFilterFactory 抽象类，就表示他是一个局部过滤器.

1.2.2、源码分析

源码如下：

public class StripPrefixGatewayFilterFactory extends AbstractGatewayFilterFactory<Config> {public static final String PARTS_KEY = "parts";public StripPrefixGatewayFilterFactory() {super(Config.class);}public List<String> shortcutFieldOrder() {return Arrays.asList("parts");}public GatewayFilter apply(final Config config) {return new GatewayFilter() {public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {ServerHttpRequest request = exchange.getRequest();ServerWebExchangeUtils.addOriginalRequestUrl(exchange, request.getURI());String path = request.getURI().getRawPath();String[] originalParts = StringUtils.tokenizeToStringArray(path, "/");StringBuilder newPath = new StringBuilder("/");for(int i = 0; i < originalParts.length; ++i) {if (i >= config.getParts()) {if (newPath.length() > 1) {newPath.append('/');}newPath.append(originalParts[i]);}}if (newPath.length() > 1 && path.endsWith("/")) {newPath.append('/');}ServerHttpRequest newRequest = request.mutate().path(newPath.toString()).build();exchange.getAttributes().put(ServerWebExchangeUtils.GATEWAY_REQUEST_URL_ATTR, newRequest.getURI());return chain.filter(exchange.mutate().request(newRequest).build());}public String toString() {return GatewayToStringStyler.filterToStringCreator(StripPrefixGatewayFilterFactory.this).append("parts", config.getParts()).toString();}};}public static class Config {private int parts = 1;public Config() {}public int getParts() {return this.parts;}public void setParts(int parts) {this.parts = parts;}}
}

  • extends AbstractGatewayFilterFactory<Config> ：继承 AbstractGatewayFilterFactory 表示他是一个局部过滤器.  传入一个泛型 Config（是一个静态内部类），是因为在配置 filters 的时候，可能需要给参数指定具体的值，例如 - StripPrefix=2，而 Config 就是来处理这里的 2 这个值的.
  • public static final String PARTS_KEY = "parts"： 这里就是定义一个常量，后面会用上.
  • StripPrefixGatewayFilterFactory() ：构造方法，需要给父类 AbstractGatewayFilterFactory 传递 Config 参数（前面分析过了），将来在 apply 方法中会用上.
  • Config：是一个静态内部类，描述了配置 filters 时，具体要给参数指定的值，并提供了 get 和 set 方法.  这个类就需要传递给父类 AbstractGatewayFilterFactory，最后回传给 apply 方法，在 apply 方法中使用.如果不想给参数指定值，就可以不写 Config 中的内容.
  • shortcutFieldOrder()：这个方法是用来指定 filters 配置中参数值的顺序.  也就是说，如果 Config 个中如果有多个参数，那么你在配置 filters 时，指定参数的多个值，顺序是怎样的？他就是用来指定顺序的.
  • apply(Config config)：局部过滤器的核心类，用来描述过滤规则的.  这里的参数 Config 就是刚刚讲到的 静态内部类，先传递给父类，然后再回传给了 apply 方法，之后我们就可以直接在 apply 方法中去使用 Config 类中的参数.

1.3、手写源码

1.3.1、基础框架

按照上述分析，不难写出大概模样，例如我们可以模仿源码，创建包 filter.factory ，然后在这个包下定义一个 Token 局部过滤器如下：

@Component // 表示在工厂中创建对象（不能少！）
public class TokenGatewayFilterFactory extends AbstractGatewayFilterFactory<TokenGatewayFilterFactory.Config> {public TokenGatewayFilterFactory() {super(Config.class);}/*** 核心方法: 处理过滤* @param config* @return*/@Overridepublic GatewayFilter apply(Config config) {return new GatewayFilter() {@Overridepublic Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {// 处理过滤逻辑......return chain.filter(exchange);}};}@Overridepublic List<String> shortcutFieldOrder() {return super.shortcutFieldOrder();}public static class Config {}}

那么我们就可以在配置文件中，添加这个自定义的局部过滤器

1.3.2、实现自定义局部过滤器

例如，自定义一个 Token 局部过滤器，那么就可以创建一个类 filter.factory.TokenGatewayFilterFactory 

在 apply 中的过滤逻辑就是，判断前端是否传入 token，如果没有就抛异常，如果有就去 redis 上看看是否存在这个 token，如果存在就放行，不存在就抛异常.

@Slf4j
@Component // 表示在工厂中创建对象
public class TokenGatewayFilterFactory extends AbstractGatewayFilterFactory<TokenGatewayFilterFactory.Config> {@Autowiredprivate StringRedisTemplate redisTemplate;public TokenGatewayFilterFactory() {super(Config.class);}/*** 核心方法: 处理过滤* @param config* @return*/@Overridepublic GatewayFilter apply(Config config) {return new GatewayFilter() {@Overridepublic Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {//1.获取 token 信息//由于 header 中 key 可以重复(包括 parma 也是如此)，因此获取到的是一个 ListList<String> tokens = exchange.getRequest().getHeaders().get(RedisPrefix.TOKEN_KEY);if(tokens == null) {throw new RuntimeException("没有 token 令牌！");}String tokenValue = tokens.get(0);log.info("token: {}", tokenValue);//2.比较 redis 上的 token 数据是否一致(redis 上存储的数据格式为: token前缀 + value)if(!redisTemplate.hasKey(RedisPrefix.TOKEN_KEY + tokenValue)) {throw new RuntimeException("token 令牌不合法！");}return chain.filter(exchange);}};}@Overridepublic List<String> shortcutFieldOrder() {return super.shortcutFieldOrder();}public static class Config {}}

a）例如 redis 存储的数据为

b）执行结果如下：

c）如果没有 token 数据，响应如下：

d）如果有 token，但是 token 值错误，响应如下：

1.3.3、加参数的自定义局部过滤器器

如果在配置 filters 的时候，要指定一些参数，例如 isRequire（boolean类型，表示是否传），name（String 类型）.

那么就可以在 Config 静态内部类中描述，然后在 shortcutFieldOrder() 方法中指定顺序，最后就可以在 apply 中拿到对应的参数，如下：

@Slf4j
@Component // 表示在工厂中创建对象
public class TokenGatewayFilterFactory extends AbstractGatewayFilterFactory<TokenGatewayFilterFactory.Config> {@Autowiredprivate StringRedisTemplate redisTemplate;public TokenGatewayFilterFactory() {super(Config.class);}/*** 核心方法: 处理过滤* @param config* @return*/@Overridepublic GatewayFilter apply(Config config) {return new GatewayFilter() {@Overridepublic Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {log.info("config isRequire: {}", config.isRequire());log.info("config name: {}", config.getName());//1.拿到 Config 中自定义的参数 isRequire，判断是否要进行过滤if(config.isRequire()) {//2.获取 token 信息//由于 header 中 key 可以重复(包括 parma 也是如此)，因此获取到的是一个 ListList<String> tokens = exchange.getRequest().getHeaders().get(RedisPrefix.TOKEN_KEY);if(tokens == null) {throw new RuntimeException("没有 token 令牌！");}String tokenValue = tokens.get(0);log.info("token: {}", tokenValue);//3.比较 redis 上的 token 数据是否一致(redis 上存储的数据格式为: token前缀 + value)if(!redisTemplate.hasKey(RedisPrefix.TOKEN_KEY + tokenValue)) {throw new RuntimeException("token 令牌不合法！");}}return chain.filter(exchange);}};}/*** 指定参数值填写的顺序* @return*/@Overridepublic List<String> shortcutFieldOrder() {return Arrays.asList("require", "name");}/*** 提供 filter 配置中的参数值*/public static class Config {private boolean require;private String name;public boolean isRequire() {return require;}public void setRequire(boolean require) {this.require = require;}public String getName() {return name;}public void setName(String name) {this.name = name;}}}

Ps：这里的不要命名为 isRequire ，会冲突！ 

在配置文件中配置 filters：

执行结果如下：


http://www.ppmy.cn/news/1204086.html

相关文章

两数之和（哈希解法）

两数之和（哈希解法）

题目描述 给定一个整数数组 nums 和一个整数目标值 target&#xff0c;请你在该数组中找出 和为目标值 target 的那 两个 整数&#xff0c;并返回它们的数组下标。 你可以假设每种输入只会对应一个答案。但是&#xff0c;数组中同一个元素在答案里不能重复出现。 你可以按任…
阅读更多...
面包屑实现

面包屑实现

背景&#xff1a;面包屑根据菜单内容显示不同内容。首页永远存在&#xff0c;后面的活动管理及多级菜单的面包屑展示。 实现原理&#xff1a; 通过this.$route.matched获取所有匹配路由&#xff0c;将处理首页外的其他路由设置到一个数组中&#xff0c;再通过数组循环方式显示…
阅读更多...
界面组件Telerik UI for WinForms中文教程 - 如何自定义应用程序文件窗口？

界面组件Telerik UI for WinForms中文教程 - 如何自定义应用程序文件窗口？

Telerik UI for WinForms包含了一个高度可定制的组件&#xff0c;它取代了.NET中默认的OpenFileDialog。在下一个更新版本中&#xff0c;会发布一个向对话框浏览器提那家自定义位置的请求功能&#xff0c;本文演示了这个和另一个自定义功能&#xff0c;它可以帮助用户在浏览文件…
阅读更多...
美容美发行业小程序源码系统+在线预约+会员卡 带完整的搭建教程

美容美发行业小程序源码系统+在线预约+会员卡 带完整的搭建教程

大家好啊&#xff0c;今天源码小编又来给大家分享了&#xff0c;今天要分享的是一款美容美发行业小程序源码系统。系统采用的是PHPMySQL组合开发。现如今&#xff0c;美容美发行业的发展迅速&#xff0c;市场竞争也日趋激烈。为了满足市场需求&#xff0c;提高自身竞争力&#…
阅读更多...
第八章《搞懂算法:逻辑回归是怎么回事》笔记

第八章《搞懂算法:逻辑回归是怎么回事》笔记

8.1 如何理解逻辑回归 逻辑回归根据给定的自变量数据集来估计事件的发生概率&#xff0c;由于结果是一个概率&#xff0c;因此因变量的范围在 0 和 1 之间。 逻辑回归的与线性回归一样&#xff0c;也是以线性函数为基础的&#xff1b;而与线性回归不同的是&#xff0c;逻辑回…
阅读更多...
算法训练 第六周

算法训练 第六周

一、用栈实现队列 本题要求我们使用栈这个数据结构来模拟实现队列的各种操作&#xff0c;我们的具体思路是使用两个栈&#xff0c;将一个栈当作输入栈&#xff0c;用于压入 push传入的数据&#xff1b;另一个栈当作输出栈&#xff0c;用于 pop和 peek 操作。每次 pop 或 peek 时…
阅读更多...
从TCP到Socket，彻底理解网络编程是怎么回事

从TCP到Socket，彻底理解网络编程是怎么回事

进行程序开发的同学&#xff0c;无论Web前端开发、Web后端开发&#xff0c;还是搜索引擎和大数据&#xff0c;几乎所有的开发领域都会涉及到网络编程。比如我们进行Web服务端开发&#xff0c;除了Web协议本身依赖网络外&#xff0c;通常还需要连接数据库&#xff0c;而数据库连…
阅读更多...
Linux中su鉴定故障解决办法

Linux中su鉴定故障解决办法

su鉴定故障问题 刚装完Linux的虚拟机&#xff0c;在终端中输入su命令&#xff0c;输入密码后出现鉴定故障。 原因是在安装linux系统时没有给root用户设置密码&#xff0c;重新设置密码即可。 解决步骤如下 设置root密码 sudo passwd root —>如果没有登录密码 则提示输入…
阅读更多...
最新文章

Copyright @ 2022~2023