spring 服务网关Gateway 自定义filter 程序写法

最新推荐文章于 2024-09-14 01:02:52 发布

背着代码去游走

最新推荐文章于 2024-09-14 01:02:52 发布

阅读量1.4k

点赞数

分类专栏：分布式文章标签： gateway spring java

本文链接：https://blog.csdn.net/fdqzq/article/details/105268882

版权

分布式专栏收录该内容

23 篇文章 2 订阅

订阅专栏

Gateway 服务网关

快速开始

@Configuration
public class GatewayRoutes {
   @Autowired
   private DefaultInit defaultInit;
   @Autowired
   private MyGatewayFilter myGatewayFilter;
   @Bean
   public RouteLocator routeLocator(RouteLocatorBuilder builder) {

       return builder.routes()
               .route("api",p -> p.path("/api/**").and().asyncPredicate(initMyRoutePredicateFactory().applyAsync(config -> config.setHeaderName("Authorization"))).filters(f -> f.filter(myGatewayFilter).hystrix(config -> config
.setName("some")
.setFallbackUri("forward:/fallback"))).uri("https://www.baidu.com/")).build();
   }
   @Bean
   public MyRoutePredicateFactory initMyRoutePredicateFactory(){
       return new MyRoutePredicateFactory(MyRoutePredicateFactory.Config.class);
   }

}

可通过自定义filter控制跳转路径或者权限控制也可以通过自定义断言PredicateFactory控制自定义权限

filter 这里实现了自定义跳转路径及执行时间打印

public class MyGatewayFilter implements GatewayFilter ,Ordered{
   private static final Logger logger = LoggerFactory.getLogger(BaseController.class);
   private static final String TIME = "TIME";


   @Override
   public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {

       exchange.getAttributes().put(TIME, System.currentTimeMillis());
       Route route = exchange.getAttribute(GATEWAY_ROUTE_ATTR);
       String rawPath = exchange.getRequest().getURI().getRawPath();
       rawPath = rawPath.substring(rawPath.indexOf(route.getUri().getPath())+route.getUri().getPath().length());
       //重定义路由
       Route newRoute = Route.async().asyncPredicate(route.getPredicate()).filters(route.getFilters()).id(route.getId()).order(route.getOrder())
               .uri(route.getUri().toString()+rawPath).build();
       exchange.getAttributes().put(GATEWAY_ROUTE_ATTR, newRoute);
       return chain.filter(exchange).then(Mono.fromRunnable(() -> {
           Long start = exchange.getAttribute(TIME);
           if (start != null) {
               logger.info("请求路径:{}, 响应时长:{} ms",exchange.getRequest().getURI(),(System.currentTimeMillis() - start) );
           }
       }));
   }

   @Override
   public int getOrder() {

       return 0;
   }

}

MyRoutePredicateFactory 这里实现简单token判断

public class MyRoutePredicateFactory extends AbstractRoutePredicateFactory<MyRoutePredicateFactory.Config> {

   /**
   * @param configClass
   */
   public MyRoutePredicateFactory(Class<Config> configClass) {
       super(configClass);
   }

   @Override
   public Predicate<ServerWebExchange> apply(Config config) {

       return exchange -> {
//判断header里有放token
HttpHeaders headers = exchange.getRequest().getHeaders();
List<String> header = headers.get(config.getHeaderName());
//TODO token逻辑判断
return header!=null&&header.size() > 0;
};
   }


   public static class Config {

private String headerName;

public String getHeaderName() {
return headerName;
}

public void setHeaderName(String headerName) {
this.headerName = headerName;
}
}
}

fallback

@RestController
public class FallbackController {
   private static final Logger logger = LoggerFactory.getLogger(FallbackController .class);

   @RequestMapping("/fallback")
public Mono<String> fallback(ServerWebExchange exchange) {

return Mono.just("该服务暂时不可用");
}
}