RateLimitCaller invoke with query param and headers.

pull/952/head
atomzhong 3 years ago
parent 0a19ed1d8a
commit 76735a2d5c

@ -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<String, String> 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;
}
}

Loading…
Cancel
Save