spring cloud alibaba 学习（三十一）spring cloud gateway InMemoryRouteDefinitionRepository

---

前言

通过接口注册的路由信息保存在 InMemoryRouteDefinitionRepository 中。

---

一、引入依赖

		<dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-actuator</artifactId>
        </dependency>

添加配置：

management.endpoints.web.exposure.include=*

二、ControllerEndpointHandlerMapping

ControllerEndpointHandlerMapping 用于解析 @RestControllerEndpoint类的方法。
类似于 springmvc 中 RequestMappingInfoHandlerMapping 解析 @Controller 的方法

1、ControllerEndpointHandlerMapping 的创建

在WebFluxEndpointManagementContextConfiguration配置类中，定义了 bean ControllerEndpointHandlerMapping

	@Bean
	@ConditionalOnMissingBean
	public ControllerEndpointHandlerMapping controllerEndpointHandlerMapping(
			ControllerEndpointsSupplier controllerEndpointsSupplier, CorsEndpointProperties corsProperties,
			WebEndpointProperties webEndpointProperties) {
		EndpointMapping endpointMapping = new EndpointMapping(webEndpointProperties.getBasePath());
		return new ControllerEndpointHandlerMapping(endpointMapping, controllerEndpointsSupplier.getEndpoints(),
				corsProperties.toCorsConfiguration());
	}

	public ControllerEndpointHandlerMapping(EndpointMapping endpointMapping,
			Collection<ExposableControllerEndpoint> endpoints, CorsConfiguration corsConfiguration) {
		Assert.notNull(endpointMapping, "EndpointMapping must not be null");
		Assert.notNull(endpoints, "Endpoints must not be null");
		this.endpointMapping = endpointMapping;
		this.handlers = getHandlers(endpoints);
		this.corsConfiguration = corsConfiguration;
		setOrder(-100);
	}

2、在父类AbstractHandlerMethodMapping 的 afterPropertiesSet( ) 方法中，会调用 initHandlerMethods( )

	public void afterPropertiesSet() {
		initHandlerMethods();

		// Total includes detected mappings + explicit registrations via registerMapping
		int total = getHandlerMethods().size();
		if ((logger.isTraceEnabled() && total == 0) || (logger.isDebugEnabled() && total > 0) ) {
			logger.debug(total + " mappings in " + formatMappingName());
		}
	}

（3）initHandlerMethods( )

获取到 @RestControllerEndpoint 标识的类的方法，注册到 ControllerEndpointHandlerMapping 中

三、GatewayControllerEndpoint

类上加了注解 @RestControllerEndpoint(id = “gateway”) , 会被上面的 ControllerEndpointHandlerMapping 类解析。

类中主要的一些方法如下：

1、获取所有路由

访问路径： /actuator/gateway/routes

	@GetMapping("/routes")
	public Flux<Map<String, Object>> routes() {
		return this.routeLocator.getRoutes().map(this::serialize);
	}

2、获取单个路由

访问路径： /actuator/gateway/routes/{id}

	@GetMapping("/routes/{id}")
	public Mono<ResponseEntity<Map<String, Object>>> route(@PathVariable String id) {
		// @formatter:off
		return this.routeLocator.getRoutes()
				.filter(route -> route.getId().equals(id))
				.singleOrEmpty()
				.map(this::serialize)
				.map(ResponseEntity::ok)
				.switchIfEmpty(Mono.just(ResponseEntity.notFound().build()));
		// @formatter:on
	}

3、刷新路由

访问路径： /actuator/gateway/refresh

发布RefreshRoutesEvent事件，会出重新从nacos等地方获取路由

	@PostMapping("/refresh")
	public Mono<Void> refresh() {
		this.publisher.publishEvent(new RefreshRoutesEvent(this));
		return Mono.empty();
	}

4、获取 globalfilters、routefilters、routepredicates

	@GetMapping("/globalfilters")
	public Mono<HashMap<String, Object>> globalfilters() {
		return getNamesToOrders(this.globalFilters);
	}

	@GetMapping("/routefilters")
	public Mono<HashMap<String, Object>> routefilers() {
		return getNamesToOrders(this.GatewayFilters);
	}

	@GetMapping("/routepredicates")
	public Mono<HashMap<String, Object>> routepredicates() {
		return getNamesToOrders(this.routePredicates);
	}

5、新增路由

访问路径： /actuator/gateway/routes/{id}

请求方式是 @PostMapping

会调用 InMemoryRouteDefinitionRepository 的 save( ) 保存路由信息

	@PostMapping("/routes/{id}")
	@SuppressWarnings("unchecked")
	public Mono<ResponseEntity<Object>> save(@PathVariable String id, @RequestBody RouteDefinition route) {

		return Mono.just(route).filter(this::validateRouteDefinition)
				.flatMap(routeDefinition -> this.routeDefinitionWriter.save(Mono.just(routeDefinition).map(r -> {
					r.setId(id);
					log.debug("Saving route: " + route);
					return r;
				})).then(Mono.defer(() -> Mono.just(ResponseEntity.created(URI.create("/routes/" + id)).build()))))
				.switchIfEmpty(Mono.defer(() -> Mono.just(ResponseEntity.badRequest().build())));
	}

6、删除路由

访问路径： /actuator/gateway/routes/{id}

请求方式是 @DeleteMapping

会调用 InMemoryRouteDefinitionRepository 的 delete( ) 删除路由信息

	@DeleteMapping("/routes/{id}")
	public Mono<ResponseEntity<Object>> delete(@PathVariable String id) {
		return this.routeDefinitionWriter.delete(Mono.just(id))
				.then(Mono.defer(() -> Mono.just(ResponseEntity.ok().build())))
				.onErrorResume(t -> t instanceof NotFoundException, t -> Mono.just(ResponseEntity.notFound().build()));
	}

四、InMemoryRouteDefinitionRepository

public class InMemoryRouteDefinitionRepository implements RouteDefinitionRepository {

	//路由信息保存在map中
	private final Map<String, RouteDefinition> routes = synchronizedMap(new LinkedHashMap<String, RouteDefinition>());
	//保存路由
	@Override
	public Mono<Void> save(Mono<RouteDefinition> route) {
		return route.flatMap(r -> {
			if (ObjectUtils.isEmpty(r.getId())) {
				return Mono.error(new IllegalArgumentException("id may not be empty"));
			}
			routes.put(r.getId(), r);
			return Mono.empty();
		});
	}
	//删除路由
	@Override
	public Mono<Void> delete(Mono<String> routeId) {
		return routeId.flatMap(id -> {
			if (routes.containsKey(id)) {
				routes.remove(id);
				return Mono.empty();
			}
			return Mono.defer(() -> Mono.error(new NotFoundException("RouteDefinition not found: " + routeId)));
		});
	}
	//获取路由
	@Override
	public Flux<RouteDefinition> getRouteDefinitions() {
		return Flux.fromIterable(routes.values());
	}

}

总结

1、ControllerEndpointHandlerMapping 解析 @ControllerEndpoint 和 @RestControllerEndpoint，构建路径访问信息；

2、GatewayControllerEndpoint 和 父类 AbstractGatewayControllerEndpoint 提供了对路由的http请求接口；

3、InMemoryRouteDefinitionRepository 用于保存http方式新增的路由。

这种方式新增的路由保存在内存中，在系统重启后就丢失了。