From ae7564f0f5397f113b5198e013ea2b6a65e6bc2a Mon Sep 17 00:00:00 2001 From: atomzhong Date: Thu, 6 Apr 2023 16:36:41 +0800 Subject: [PATCH] RateLimitCaller invoke with query param and headers. --- .../example/service/callee/CustomLabelResolver.java | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/spring-cloud-tencent-examples/polaris-ratelimit-example/ratelimit-callee-service/src/main/java/com/tencent/cloud/ratelimit/example/service/callee/CustomLabelResolver.java b/spring-cloud-tencent-examples/polaris-ratelimit-example/ratelimit-callee-service/src/main/java/com/tencent/cloud/ratelimit/example/service/callee/CustomLabelResolver.java index f9f405d4b..9403f7479 100644 --- a/spring-cloud-tencent-examples/polaris-ratelimit-example/ratelimit-callee-service/src/main/java/com/tencent/cloud/ratelimit/example/service/callee/CustomLabelResolver.java +++ b/spring-cloud-tencent-examples/polaris-ratelimit-example/ratelimit-callee-service/src/main/java/com/tencent/cloud/ratelimit/example/service/callee/CustomLabelResolver.java @@ -24,6 +24,8 @@ import java.util.Map; import javax.servlet.http.HttpServletRequest; import com.tencent.cloud.polaris.ratelimit.spi.PolarisRateLimiterLabelServletResolver; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Component; @@ -35,7 +37,7 @@ import org.springframework.stereotype.Component; */ @Component public class CustomLabelResolver implements PolarisRateLimiterLabelServletResolver { - + private static final Logger LOG = LoggerFactory.getLogger(CustomLabelResolver.class); @Value("${label.key-value:}") private String[] keyValues; @Override @@ -49,10 +51,11 @@ public class CustomLabelResolver implements PolarisRateLimiterLabelServletResolv Map labels = new HashMap<>(); for (String kv : keyValues) { String key = kv.substring(0, kv.indexOf(":")); - String value = kv.substring(kv.indexOf(":")); + String value = kv.substring(kv.indexOf(":")+1); labels.put(key, value); } + LOG.info("Current labels:{}", labels); return labels; } }