org.springframework.cloud.gateway.filter.ratelimit.KeyResolver Java Examples

@Bean(name = UriKeyResolver.BEAN_NAME)
public KeyResolver getUriKeyResolver() {
    return new UriKeyResolver();
}
 

@Bean
KeyResolver userKeyResolver() {
    return exchange -> Mono.just("1");
}
 

@Bean
KeyResolver resolver2() {
	return exchange -> Mono.just("notallowedkey");
}
 

@Bean
@Primary
KeyResolver resolver1() {
	return exchange -> Mono.just("allowedkey");
}
 

private void assertFilterFactory(KeyResolver keyResolver, String key, boolean allowed,
		HttpStatus expectedStatus) {
	assertFilterFactory(keyResolver, key, allowed, expectedStatus, null);
}
 

public Config setKeyResolver(KeyResolver keyResolver) {
	this.keyResolver = keyResolver;
	return this;
}
 

public KeyResolver getKeyResolver() {
	return keyResolver;
}
 

@SuppressWarnings("unchecked")
@Override
public GatewayFilter apply(Config config) {
	KeyResolver resolver = getOrDefault(config.keyResolver, defaultKeyResolver);
	RateLimiter<Object> limiter = getOrDefault(config.rateLimiter,
			defaultRateLimiter);
	boolean denyEmpty = getOrDefault(config.denyEmptyKey, this.denyEmptyKey);
	HttpStatusHolder emptyKeyStatus = HttpStatusHolder
			.parse(getOrDefault(config.emptyKeyStatus, this.emptyKeyStatusCode));

	return (exchange, chain) -> resolver.resolve(exchange).defaultIfEmpty(EMPTY_KEY)
			.flatMap(key -> {
				if (EMPTY_KEY.equals(key)) {
					if (denyEmpty) {
						setResponseStatus(exchange, emptyKeyStatus);
						return exchange.getResponse().setComplete();
					}
					return chain.filter(exchange);
				}
				String routeId = config.getRouteId();
				if (routeId == null) {
					Route route = exchange
							.getAttribute(ServerWebExchangeUtils.GATEWAY_ROUTE_ATTR);
					routeId = route.getId();
				}
				return limiter.isAllowed(routeId, key).flatMap(response -> {

					for (Map.Entry<String, String> header : response.getHeaders()
							.entrySet()) {
						exchange.getResponse().getHeaders().add(header.getKey(),
								header.getValue());
					}

					if (response.isAllowed()) {
						return chain.filter(exchange);
					}

					setResponseStatus(exchange, config.getStatusCode());
					return exchange.getResponse().setComplete();
				});
			});
}
 

public KeyResolver getDefaultKeyResolver() {
	return defaultKeyResolver;
}
 

public RequestRateLimiterGatewayFilterFactory(RateLimiter defaultRateLimiter,
		KeyResolver defaultKeyResolver) {
	super(Config.class);
	this.defaultRateLimiter = defaultRateLimiter;
	this.defaultKeyResolver = defaultKeyResolver;
}
 

@Bean
@ConditionalOnBean({ RateLimiter.class, KeyResolver.class })
public RequestRateLimiterGatewayFilterFactory requestRateLimiterGatewayFilterFactory(
		RateLimiter rateLimiter, KeyResolver resolver) {
	return new RequestRateLimiterGatewayFilterFactory(rateLimiter, resolver);
}
 

@Bean(name = PrincipalNameKeyResolver.BEAN_NAME)
@ConditionalOnBean(RateLimiter.class)
@ConditionalOnMissingBean(KeyResolver.class)
public PrincipalNameKeyResolver principalNameKeyResolver() {
	return new PrincipalNameKeyResolver();
}
 

/**
 * ip地址限流
 *
 * @return 限流key
 */
@Bean
@Primary
public KeyResolver remoteAddressKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
}
 

@Bean
public KeyResolver userKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getQueryParams().getFirst("user"));
}
 

@Bean(name = "apiKeyResolver")
@Primary
public KeyResolver apiKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getPath().value());
}
 

@Bean(value = "remoteAddrKeyResolver")
public KeyResolver remoteAddrKeyResolver() {
	return exchange -> Mono.just(exchange.getRequest().getRemoteAddress().getAddress().getHostAddress());
}
 

@Bean
public KeyResolver pathKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getPath().value());
}
 

@Bean
KeyResolver userKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getQueryParams().getFirst("fwcloud"));
}
 

/**
 * IP限流 (通过exchange对象可以获取到请求信息，这边用了HostName)
 */
@Bean
@Primary
public KeyResolver ipKeyResolver() {
	return exchange -> Mono.just(exchange.getRequest().getRemoteAddress().getAddress().getHostAddress());
}
 

/**
 * 用户限流 (通过exchange对象可以获取到请求信息，获取当前请求的用户 TOKEN)
 */
@Bean
public KeyResolver userKeyResolver() {
	//使用这种方式限流，请求Header中必须携带X-Access-Token参数
	return exchange -> Mono.just(exchange.getRequest().getHeaders().getFirst(GlobalAccessTokenFilter.X_ACCESS_TOKEN));
}
 

/**
* 接口限流 (获取请求地址的uri作为限流key)
*/
@Bean
public KeyResolver apiKeyResolver() {
	return exchange -> Mono.just(exchange.getRequest().getPath().value());
}
 

@Bean(name = "userKeyResolver")
public KeyResolver userKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getQueryParams().getFirst("userId"));
}
 

@Bean(name = "ipKeyResolver")
public KeyResolver ipKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
}
 

/**
 * 按客户端IP限流
 * @return
 */
@Bean
public KeyResolver ipKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
}
 

/**
 * 请求路径限流
 *
 * @return 限流key
 */
@Bean
public KeyResolver apiKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getPath().value());
}
 

/**
 * 接口限流
 *
 * @return
 */
@Bean
public KeyResolver apiKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getPath().value());
}
 

/**
 * IP限流
 *
 * @return
 */
@Bean
public KeyResolver ipKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
}
 

/**
 * username限流
 *
 * @return 限流key
 */
@Bean
public KeyResolver userKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getQueryParams().getFirst("username"));
}
 

/**
 * 请求路径限流
 *
 * @return 限流key
 */
@Bean
public KeyResolver apiKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getPath().value());
}
 

/**
 * ip地址限流
 *
 * @return 限流key
 */
@Bean
public KeyResolver remoteAddressKeyResolver() {
    return exchange -> Mono.just(exchange.getRequest().getRemoteAddress().getHostName());
}