SpringCloud组件之 Gateway实战应用

package cn.study.filter; import com.alibaba.nacos.shaded.com.google.gson.Gson; import org.springframework.beans.factory.annotation.Value; import org.springframework.cloud.gateway.filter.GatewayFilterChain; import org.springframework.cloud.gateway.filter.GlobalFilter; import org.springframework.core.Ordered; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.http.HttpStatus; import org.springframework.http.server.reactive.ServerHttpRequest; import org.springframework.http.server.reactive.ServerHttpResponse; import org.springframework.stereotype.Component; import org.springframework.web.server.ServerWebExchange; import reactor.core.publisher.Mono; import javax.annotation.Resource; import java.time.LocalDateTime; import java.time.format.DateTimeFormatter; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Objects; /** * Description: token拦截 * * @author suK * @date 2024/6/5 13:50 * * <pre> * Copyright (c) 2023. All Rights Reserved. * </pre> */ @Component public class TokenGlobalFilter implements GlobalFilter, Ordered { @Value("${login.url}") public String loginUrl; @Resource private RedisTemplate<String, Object> redisTemplate; private static final Gson gson = new Gson(); private static final String TOKEN = "Authorization"; private final static String REDIS_DIR_TOKEN = "token:%s"; @Override public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) { ServerHttpRequest request = exchange.getRequest(); String path = exchange.getRequest().getURI().getPath(); if (loginUrl.contains(path)) { return chain.filter(exchange); } List<String> headerList = request.getHeaders().get(TOKEN); if (null != headerList && headerList.size() > 0) { String token = headerList.get(0); String userStr = (String) redisTemplate.opsForValue().get(String.format(REDIS_DIR_TOKEN, token)); if (Objects.nonNull(userStr)) { try { Map map = gson.fromJson(userStr, Map.class); System.out.printf("%s 用户，于%s,请求了%s 接口%n", map.get("name"), LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy年MM月dd日HH时mm分ss秒")), path); } catch (Exception e) { throw new RuntimeException(e); } return chain.filter(exchange); } } ServerHttpResponse response = exchange.getResponse(); response.getHeaders().set("content-type", "application/json"); Map<String, Object> map = new HashMap<>(6); map.put("code", HttpStatus.UNAUTHORIZED.value()); map.put("msg", "请先登录"); String result = gson.toJson(map); return response.writeWith(Mono.just(response.bufferFactory().wrap((result).getBytes()))); } /** * 过滤器的排序 * * @return */ @Override public int getOrder() { return 0; } }