如何使用网关构建网关服务并实施动态路由选择？
最编程
2024-03-25 08:56:07
...
@Component
public class AuthFilter implements GlobalFilter, Ordered {
@Autowired
private RedisTemplate<String, String> redisTemplate;
private static final String TOKEN_HEADER_KEY = "auth_token";
@Override
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
// 1.获取请求对象
ServerHttpRequest request = exchange.getRequest();
// 2.获取token
String token = getToken(request);
ServerHttpResponse response = exchange.getResponse();
if (StringUtils.isBlank(token)) {
// 3.token为空 返回401
response.setStatusCode(HttpStatus.UNAUTHORIZED);
return response.setComplete();
}
// 4.验证token是否有效
String userId = getUserIdByToken(token);
if (StringUtils.isBlank(userId)) {
// 5.token无效 返回401
response.setStatusCode(HttpStatus.UNAUTHORIZED);
return response.setComplete();
}
// token有效，后续业务处理
// 从写请求头，方便业务系统从请求头获取用户id进行权限相关处理
ServerHttpRequest.Builder builder = exchange.getRequest().mutate();
request = builder.header("user_id", userId).build();
// 延长缓存过期时间-token缓存用户如果一直在操作就会一直重置过期
// 这样避免用户操作过程中突然过期影响业务操作及体验，只有用户操作间隔时间大于缓存过期时间才会过期
resetTokenExpirationTime(token, userId);
// 完成验证
return chain.filter(exchange);
}
@Override
public int getOrder() {
// 优先级 越小越优先
return 0;
}
/**
* 从redis中获取用户id
* 在登录操作时候 登陆成功会生成一个token, redis得key为auth_token:token 值为用户id
*
* @param token
* @return
*/
private String getUserIdByToken(String token) {
String redisKey = String.join(":", "auth_token", token);
return redisTemplate.opsForValue().get(redisKey);
}
/**
* 重置token过期时间
*
* @param token
* @param userId
*/
private void resetTokenExpirationTime(String token, String userId) {
String redisKey = String.join(":", "auth_token", token);
redisTemplate.opsForValue().set(redisKey, userId, 2, TimeUnit.HOURS);
}
/**
* 获取token
*
* @param request
* @return
*/
private static String getToken(ServerHttpRequest request) {
HttpHeaders headers = request.getHeaders();
// 从请求头获取token
String token = headers.getFirst(TOKEN_HEADER_KEY);
if (StringUtils.isBlank(token)) {
// 请求头无token则从url获取token
token = request.getQueryParams().getFirst(TOKEN_HEADER_KEY);
}
if (StringUtils.isBlank(token)) {
// 请求头和url都没有token则从cookies获取
HttpCookie cookie = request.getCookies().getFirst(TOKEN_HEADER_KEY);
if (cookie != null) {
token = cookie.getValue();
}
}
return token;
}
}
上一篇： OSPF 详解 (II)
推荐阅读
-
谈API网关和应用网关--从技术选型谈起：API网关的性能是第一指标，一般会选择Kong、Apisix等基于OpenResty+Lua的高性能网关（得益于Ngnix基于C++的高性能无阻塞网络IO模型），应用网关一般是结合自身业务的技术栈来选择，比如SpringCloud Gateway、Zuul等。当然，这也不是绝对的，如果你对 Kong 非常熟悉，用它来做应用网关也不是不可能。 一些开源网关项目的例子： Kong Apisix 特使 Traefik SpringCloud 网关 Zuul / Zuul2 接下来，我们将重点介绍应用网关。在网格中，应用网关侧重于以下功能（与 API 网关不同） 动态路由 服务发现 服务聚合/协调 可观察性 如果您使用的是 Sping 技术栈，使用 SpringCloud Gateway 和 Zuul 可以轻松重用现有类库，如集成您的注册表，使用 Hystrix、resilience4j 完成熔断和限流功能等，快速完成一个生产级可用应用网关，如果引入新的复杂技术栈 成本将直线上升。根据使用场景的不同，性能有时并不是第一指标，但通常我们很容易陷入性能误区。
-
如何使用网关构建网关服务并实施动态路由选择？