项目中采用了spring-cloud-zuul-ratelimit,并发测试的时候想禁用掉限流zuul.ratelimit.enabled=false,结果报以下错误。
Description:
Parameter 0 of method ratelimitKeyGenerator in com.uecent.zuul.server.config.RateLimitConfig required a bean of type 'com.marcosbarbero.cloud.autoconfigure.zuul.ratelimit.config.properties.RateLimitProperties' that could not be found.
Action:
Consider defining a bean of type 'com.marcosbarbero.cloud.autoconfigure.zuul.ratelimit.config.properties.RateLimitProperties' in your configuration.
报错的地方是自定义了limit的key生成方式:
@Bean
public RateLimitKeyGenerator ratelimitKeyGenerator(RateLimitProperties properties, RateLimitUtils rateLimitUtils) {
return new DefaultRateLimitKeyGenerator(properties, rateLimitUtils) {
@Override
public String key(HttpServletRequest request, Route route, RateLimitProperties.Policy policy) {
final StringJoiner joiner = new StringJoiner(":");
joiner.add(properties.getKeyPrefix());
if (route != null) {
joiner.add(route.getId());
}
// url
joiner.add(Optional.ofNullable(route).map(Route::getPath).orElse(StringUtils.EMPTY));
// user
joiner.add(Optional.ofNullable(request.getHeader(GlobalConsts.TOKEN_HEADER)).orElse("anonymous"));
// ip
joiner.add(Optional.ofNullable(request.getHeader("X-Real-IP")).orElse(request.getRemoteAddr()));
return joiner.toString();
}
};
}
出错的原因是第一个参数没有正常注入,找不到这个bean了。
查看RateLimit的源代码,发现spring boot 自启动配置的地方,默认了zuul.ratelimit.enabled=true才会加载配置文件。于是在我自定义生成limitkey的地方也加上这个限制就可以了。
完整代码:
RateLimitConfig.java
package com.uecent.zuul.server.config;
import com.marcosbarbero.cloud.autoconfigure.zuul.ratelimit.config.RateLimitKeyGenerator;
import com.marcosbarbero.cloud.autoconfigure.zuul.ratelimit.config.RateLimitUtils;
import com.marcosbarbero.cloud.autoconfigure.zuul.ratelimit.config.properties.RateLimitProperties;
import com.marcosbarbero.cloud.autoconfigure.zuul.ratelimit.support.DefaultRateLimitKeyGenerator;
import com.uecent.common.core.constant.GlobalConsts;
import org.apache.commons.lang3.StringUtils;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.cloud.netflix.zuul.filters.Route;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import javax.servlet.http.HttpServletRequest;
import java.util.Optional;
import java.util.StringJoiner;
/**
* 自定义RateLimit的key生成方式
* @author :wangjf
* @date :2020/5/18 17:05
* @description:provider-com
* @version: v1.1.0
*/
@Configuration
// 关键就是下面这句
@ConditionalOnProperty(prefix = RateLimitProperties.PREFIX, name="enabled", havingValue = "true")
public class RateLimitConfig {
@Bean
public RateLimitKeyGenerator ratelimitKeyGenerator(RateLimitProperties properties, RateLimitUtils rateLimitUtils) {
return new DefaultRateLimitKeyGenerator(properties, rateLimitUtils) {
@Override
public String key(HttpServletRequest request, Route route, RateLimitProperties.Policy policy) {
final StringJoiner joiner = new StringJoiner(":");
joiner.add(properties.getKeyPrefix());
if (route != null) {
joiner.add(route.getId());
}
// url
joiner.add(Optional.ofNullable(route).map(Route::getPath).orElse(StringUtils.EMPTY));
// user
joiner.add(Optional.ofNullable(request.getHeader(GlobalConsts.TOKEN_HEADER)).orElse("anonymous"));
// ip
joiner.add(Optional.ofNullable(request.getHeader("X-Real-IP")).orElse(request.getRemoteAddr()));
return joiner.toString();
}
};
}
}