gateway动态服务名Route配置

需求说明

想做模型部署功能,根据模型的唯一编码注册到nacos并通过gateway调用。因为模型编码是随机生成的,无法提前配置。
如编码为A的模型，在启动时注册到naocs服务名为 predict-A,通过网关对外暴漏路径为/xxx-model/A/

解决方案

自定义PredictFilter,重置Route。

import lombok.extern.slf4j.Slf4j;
import org.springframework.cloud.gateway.filter.GatewayFilter;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.route.Route;
import org.springframework.cloud.gateway.support.ServerWebExchangeUtils;
import org.springframework.core.Ordered;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;
import java.net.URI;
@Slf4j
public class PredictFilter implements GatewayFilter, Ordered {
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {

        String uri = buildUri(exchange.getRequest().getURI());
        log.info("PredictFilter : {}",uri);
        Route route = (Route)exchange.getAttribute(ServerWebExchangeUtils.GATEWAY_ROUTE_ATTR);
        Route newRoute=Route.async()
                .asyncPredicate(route.getPredicate())
                .filters(route.getFilters())
                .id(route.getId())
                .order(route.getOrder())
                .uri(uri).build();
        exchange.getAttributes().put(ServerWebExchangeUtils.GATEWAY_ROUTE_ATTR,newRoute);
        return chain.filter(exchange);
    }


    @Override
    public int getOrder() {
        return 0;
    }

    private String buildUri(URI uri){
        String path = uri.getPath();
        String[] split = path.split("/");
        String serviceName = "predict-" + split[2];
        return "lb://"+serviceName;
    }
}