org.springframework.cloud.gateway.event.RefreshRoutesEvent Java Examples

/**
 * 添加Nacos监听
 */
private void addListener() {
    try {
        nacosConfigProperties.configServiceInstance().addListener(SCG_DATA_ID, SCG_GROUP_ID, new Listener() {
            @Override
            public Executor getExecutor() {
                return null;
            }

            @Override
            public void receiveConfigInfo(String configInfo) {
                publisher.publishEvent(new RefreshRoutesEvent(this));
            }
        });
    } catch (NacosException e) {
        log.error("nacos-addListener-error", e);
    }
}
 

@Test
public void cacheIsRefreshedInTheBackgroundOnEvent() {
	RouteDefinition routeDef1 = routeDef(1);
	RouteDefinition routeDef2 = routeDef(2);

	CachingRouteDefinitionLocator locator = new CachingRouteDefinitionLocator(
			new StubRouteDefinitionLocator(Flux.just(routeDef1),
					Flux.defer(() -> Flux.just(routeDef1, routeDef2))));

	List<RouteDefinition> routes = locator.getRouteDefinitions().collectList()
			.block();
	assertThat(routes).containsExactlyInAnyOrder(routeDef1);

	locator.onApplicationEvent(new RefreshRoutesEvent(this));

	List<RouteDefinition> updatedRoutes = locator.getRouteDefinitions().collectList()
			.block();
	assertThat(updatedRoutes).containsExactlyInAnyOrder(routeDef1, routeDef2);
}
 

/**
 * 刷新路由
 *
 * @return
 */
public Mono<Void> refresh() {
    this.loadRoutes();
    // 触发默认路由刷新事件,刷新缓存路由
    this.publisher.publishEvent(new RefreshRoutesEvent(this));
    return Mono.empty();
}
 

@Test
public void onParentHeartbeatEvent() {
	ApplicationEventPublisher publisher = mock(ApplicationEventPublisher.class);
	RouteRefreshListener listener = new RouteRefreshListener(publisher);

	listener.onApplicationEvent(new ParentHeartbeatEvent(this, 1L));
	listener.onApplicationEvent(new ParentHeartbeatEvent(this, 1L));
	listener.onApplicationEvent(new ParentHeartbeatEvent(this, 2L));

	verify(publisher, times(2)).publishEvent(any(RefreshRoutesEvent.class));
}
 

@Test
public void onHeartbeatEvent() {
	ApplicationEventPublisher publisher = mock(ApplicationEventPublisher.class);
	RouteRefreshListener listener = new RouteRefreshListener(publisher);

	listener.onApplicationEvent(new HeartbeatEvent(this, 1L));
	listener.onApplicationEvent(new HeartbeatEvent(this, 1L));
	listener.onApplicationEvent(new HeartbeatEvent(this, 2L));

	verify(publisher, times(2)).publishEvent(any(RefreshRoutesEvent.class));
}
 

@Override
public boolean supportsEventType(Class<? extends ApplicationEvent> eventType) {
	// from config file
	return PredicateArgsEvent.class.isAssignableFrom(eventType) ||
	// from java dsl
			WeightDefinedEvent.class.isAssignableFrom(eventType) ||
			// force initialization
			RefreshRoutesEvent.class.isAssignableFrom(eventType);
}
 

@Override
public void onApplicationEvent(ApplicationEvent event) {
	if (event instanceof PredicateArgsEvent) {
		handle((PredicateArgsEvent) event);
	}
	else if (event instanceof WeightDefinedEvent) {
		addWeightConfig(((WeightDefinedEvent) event).getWeightConfig());
	}
	else if (event instanceof RefreshRoutesEvent && routeLocator != null) {
		// forces initialization
		routeLocator.ifAvailable(locator -> locator.getRoutes().subscribe());
	}

}
 

@Override
public void onApplicationEvent(RefreshRoutesEvent event) {
	try {
		fetch().collect(Collectors.toList()).subscribe(list -> Flux.fromIterable(list)
				.materialize().collect(Collectors.toList()).subscribe(signals -> {
					applicationEventPublisher
							.publishEvent(new RefreshRoutesResultEvent(this));
					cache.put(CACHE_KEY, signals);
				}, throwable -> handleRefreshError(throwable)));
	}
	catch (Throwable e) {
		handleRefreshError(e);
	}
}
 

@Test
public void onInstanceRegisteredEvent() {
	ApplicationEventPublisher publisher = mock(ApplicationEventPublisher.class);
	RouteRefreshListener listener = new RouteRefreshListener(publisher);

	listener.onApplicationEvent(new InstanceRegisteredEvent<>(this, new Object()));

	verify(publisher).publishEvent(any(RefreshRoutesEvent.class));
}
 

@Test
public void cacheIsNotClearedOnEvent() {
	RouteDefinition routeDef1 = routeDef(1);
	RouteDefinition routeDef2 = routeDef(2);

	CountDownLatch latch = new CountDownLatch(1);
	CachingRouteDefinitionLocator locator = new CachingRouteDefinitionLocator(
			new StubRouteDefinitionLocator(Flux.just(routeDef1), Flux.defer(() -> {
				try {
					latch.await();
				}
				catch (InterruptedException e) {
					Thread.currentThread().interrupt();
					throw new IllegalStateException(e);
				}
				return Flux.just(routeDef1, routeDef2);
			}).subscribeOn(Schedulers.single())));

	List<RouteDefinition> routes = locator.getRouteDefinitions().collectList()
			.block();
	assertThat(routes).containsExactlyInAnyOrder(routeDef1);

	locator.onApplicationEvent(new RefreshRoutesEvent(this));

	routes = locator.getRouteDefinitions().collectList().block();
	assertThat(routes).containsExactlyInAnyOrder(routeDef1);

	latch.countDown();
}
 

private void waitUntilRefreshFinished(CachingRouteLocator locator,
		List<RefreshRoutesResultEvent> resultEvents) throws InterruptedException {
	CountDownLatch cdl = new CountDownLatch(1);
	locator.setApplicationEventPublisher(o -> {
		resultEvents.add((RefreshRoutesResultEvent) o);
		cdl.countDown();
	});
	locator.onApplicationEvent(new RefreshRoutesEvent(this));

	assertThat(cdl.await(5, TimeUnit.SECONDS)).isTrue();
}
 

private void reset() {
	this.publisher.publishEvent(new RefreshRoutesEvent(this));
}
 

public void refresh() {
    publisher.publishEvent(new RefreshRoutesEvent(this));
}
 

@Override
public void onApplicationEvent(final RefreshRoutesEvent event) {
    CACHE.clear();
}
 

@Override
public void onApplicationEvent(final RefreshRoutesEvent event) {
    CACHE.clear();
}
 

@Override
public void onApplicationEvent(final RefreshRoutesEvent event) {
    Optional.ofNullable(CACHE.get(getCacheName())).ifPresent(Map::clear);
}
 

@Override
public void onApplicationEvent(RefreshRoutesEvent event) {
	fetch().materialize().collect(Collectors.toList())
			.doOnNext(routes -> cache.put(CACHE_KEY, routes)).subscribe();
}
 

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

private void refreshGatewayRouteDefinition() {
    logger.info("Refreshing Gateway RouteDefinition!");
    this.publisher.publishEvent(new RefreshRoutesEvent(this));
    logger.info("Gateway RouteDefinition refreshed!");
}
 

/**
 * 增加路由
 * @param definition
 * @return
 */
public String add(RouteDefinition definition) {
    routeDefinitionWriter.save(Mono.just(definition)).subscribe();
    this.publisher.publishEvent(new RefreshRoutesEvent(this));
    return "success";
}
 

private void notifyChanged() {
    this.publisher.publishEvent(new RefreshRoutesEvent(this));
}
 

private void notifyChanged() {
    this.publisher.publishEvent(new RefreshRoutesEvent(this));
}
 

/**
 * 增加路由
 *
 * @param definition definition
 * @return String
 */
public String add(RouteDefinition definition) {
    routeDefinitionWriter.save(Mono.just(definition)).subscribe();
    this.applicationEventPublisher.publishEvent(new RefreshRoutesEvent(this));
    return "success";
}