RateLimit zuul.ratelimit.enabled=false報錯

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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章