項目中採用了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();
}
};
}
}