詳解Spring Cloud Gateway 數(shù)據(jù)庫存儲路由信息的擴(kuò)展方案

 # zuul 的配置形式
 routes:
  pig-auth:
   path: /auth/**
   serviceId: pig-auth
   stripPrefix: true
 # gateway 的配置形式
 routes:
 - id: pigx-auth
   uri: lb://pigx-auth
  predicates:
  - Path=/auth/**
  filters:
  - ValidateCodeGatewayFilter

// 上圖動態(tài)路由的信息保存的默認(rèn)實(shí)現(xiàn)是基于內(nèi)存的實(shí)現(xiàn)
public class InMemoryRouteDefinitionRepository implements RouteDefinitionRepository {
  private final Map<String, RouteDefinition> routes = synchronizedMap(new LinkedHashMap<String, RouteDefinition>());
  @Override
  public Mono<Void> save(Mono<RouteDefinition> route){}
  @Override
  public Mono<Void> delete(Mono<String> routeId){}

  @Override
  public Flux<RouteDefinition> getRouteDefinitions(){}
}

/**
 * @author lengleng
 * @date 2018年11月06日10:27:55
 * <p>
 * 動態(tài)路由處理類
 */
@Slf4j
@AllArgsConstructor
@Service("sysRouteConfService")
public class SysRouteConfServiceImpl extends ServiceImpl<SysRouteConfMapper, SysRouteConf> implements SysRouteConfService {
  private final RedisTemplate redisTemplate;
  private final ApplicationEventPublisher applicationEventPublisher;

  /**
   * 獲取全部路由
   * <p>
   * RedisRouteDefinitionWriter.java
   * PropertiesRouteDefinitionLocator.java
   *
   * @return
   */
  @Override
  public List<SysRouteConf> routes() {
    SysRouteConf condition = new SysRouteConf();
    condition.setDelFlag(CommonConstant.STATUS_NORMAL);
    return baseMapper.selectList(new EntityWrapper<>(condition));
  }

  /**
   * 更新路由信息
   *
   * @param routes 路由信息
   * @return
   */
  @Override
  public Mono<Void> editRoutes(JSONArray routes) {
    // 清空Redis 緩存
    Boolean result = redisTemplate.delete(CommonConstant.ROUTE_KEY);
    log.info("清空網(wǎng)關(guān)路由 {} ", result);

    // 遍歷修改的routes，保存到Redis
    List<RouteDefinitionVo> routeDefinitionVoList = new ArrayList<>();
    routes.forEach(value -> {
      log.info("更新路由 ->{}", value);
      RouteDefinitionVo vo = new RouteDefinitionVo();
      Map<String, Object> map = (Map) value;

      Object id = map.get("routeId");
      if (id != null) {
        vo.setId(String.valueOf(id));
      }

      Object predicates = map.get("predicates");
      if (predicates != null) {
        JSONArray predicatesArray = (JSONArray) predicates;
        List<PredicateDefinition> predicateDefinitionList =
          predicatesArray.toList(PredicateDefinition.class);
        vo.setPredicates(predicateDefinitionList);
      }

      Object filters = map.get("filters");
      if (filters != null) {
        JSONArray filtersArray = (JSONArray) filters;
        List<FilterDefinition> filterDefinitionList
          = filtersArray.toList(FilterDefinition.class);
        vo.setFilters(filterDefinitionList);
      }

      Object uri = map.get("uri");
      if (uri != null) {
        vo.setUri(URI.create(String.valueOf(uri)));
      }

      Object order = map.get("order");
      if (order != null) {
        vo.setOrder(Integer.parseInt(String.valueOf(order)));
      }

      redisTemplate.setHashValueSerializer(new Jackson2JsonRedisSerializer<>(RouteDefinitionVo.class));
      redisTemplate.opsForHash().put(CommonConstant.ROUTE_KEY, vo.getId(), vo);
      routeDefinitionVoList.add(vo);
    });

    // 邏輯刪除全部
    SysRouteConf condition = new SysRouteConf();
    condition.setDelFlag(CommonConstant.STATUS_NORMAL);
    this.delete(new EntityWrapper<>(condition));

    //插入生效路由
    List<SysRouteConf> routeConfList = routeDefinitionVoList.stream().map(vo -> {
      SysRouteConf routeConf = new SysRouteConf();
      routeConf.setRouteId(vo.getId());
      routeConf.setFilters(JSONUtil.toJsonStr(vo.getFilters()));
      routeConf.setPredicates(JSONUtil.toJsonStr(vo.getPredicates()));
      routeConf.setOrder(vo.getOrder());
      routeConf.setUri(vo.getUri().toString());
      return routeConf;
    }).collect(Collectors.toList());
    this.insertBatch(routeConfList);
    log.debug("更新網(wǎng)關(guān)路由結(jié)束 ");

    this.applicationEventPublisher.publishEvent(new RefreshRoutesEvent(this));
    return Mono.empty();
  }
}

 @Slf4j
 @Component
 @AllArgsConstructor
 public class RedisRouteDefinitionWriter implements RouteDefinitionRepository {
   private final RedisTemplate redisTemplate;
 
   @Override
   public Mono<Void> save(Mono<RouteDefinition> route) {
     return route.flatMap(r -> {
       RouteDefinitionVo vo = new RouteDefinitionVo();
       BeanUtils.copyProperties(r, vo);
       log.info("保存路由信息{}", vo);
       redisTemplate.opsForHash().put(CommonConstant.ROUTE_KEY, r.getId(), vo);
       return Mono.empty();
     });
   }
   @Override
   public Mono<Void> delete(Mono<String> routeId) {
     routeId.subscribe(id -> {
       log.info("刪除路由信息{}", id);
       redisTemplate.opsForHash().delete(CommonConstant.ROUTE_KEY, id);
     });
     return Mono.empty();
   }
 
   @Override
   public Flux<RouteDefinition> getRouteDefinitions() {
     redisTemplate.setHashValueSerializer(new Jackson2JsonRedisSerializer<>(RouteDefinitionVo.class));
     List<RouteDefinitionVo> values = redisTemplate.opsForHash().values(CommonConstant.ROUTE_KEY);
     List<RouteDefinition> definitionList = new ArrayList<>();
     values.forEach(vo -> {
       RouteDefinition routeDefinition = new RouteDefinition();
       BeanUtils.copyProperties(vo, routeDefinition);
       definitionList.add(vo);
     });
     log.debug("redis 中路由定義條數(shù)： {}， {}", definitionList.size(), definitionList);
     return Flux.fromIterable(definitionList);
   }
 }