RateLimitCaller invoke with query param and headers.

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

@ -40,6 +40,7 @@ public class CustomLabelResolver implements PolarisRateLimiterLabelServletResolv
private static final Logger LOG = LoggerFactory.getLogger(CustomLabelResolver.class); private static final Logger LOG = LoggerFactory.getLogger(CustomLabelResolver.class);
@Value("${label.key-value:}") @Value("${label.key-value:}")
private String[] keyValues; private String[] keyValues;
@Override @Override
public Map<String, String> resolve(HttpServletRequest request) { public Map<String, String> resolve(HttpServletRequest request) {
// rate limit by some request params. such as query params, headers .. // rate limit by some request params. such as query params, headers ..
@ -51,7 +52,7 @@ public class CustomLabelResolver implements PolarisRateLimiterLabelServletResolv
Map<String, String> labels = new HashMap<>(); Map<String, String> labels = new HashMap<>();
for (String kv : keyValues) { for (String kv : keyValues) {
String key = kv.substring(0, kv.indexOf(":")); String key = kv.substring(0, kv.indexOf(":"));
String value = kv.substring(kv.indexOf(":")+1); String value = kv.substring(kv.indexOf(":") + 1);
labels.put(key, value); labels.put(key, value);
} }

Loading…
Cancel
Save