From 76735a2d5c9e5d74b723364a19864d27a4eefdf5 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 | 10 +++++++--- 1 file changed, 7 insertions(+), 3 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 434419408..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 @@ -21,8 +21,11 @@ package com.tencent.cloud.ratelimit.example.service.callee; import java.util.HashMap; import java.util.Map; +import javax.servlet.http.HttpServletRequest; + import com.tencent.cloud.polaris.ratelimit.spi.PolarisRateLimiterLabelServletResolver; -import jakarta.servlet.http.HttpServletRequest; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Component; @@ -34,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 @@ -48,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; } }