seanyu
3 years ago
parent
2d9afcc4fb
commit
22ece0aba5
@ -0,0 +1,23 @@
|
||||
package com.tencent.cloud.metadata.util;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
|
||||
import static com.tencent.cloud.common.constant.MetadataConstant.DefaultMetadata.DEFAULT_METADATA_SOURCE_SERVICE_NAME;
|
||||
import static com.tencent.cloud.common.constant.MetadataConstant.DefaultMetadata.DEFAULT_METADATA_SOURCE_SERVICE_NAMESPACE;
|
||||
import static com.tencent.cloud.common.metadata.MetadataContext.LOCAL_NAMESPACE;
|
||||
import static com.tencent.cloud.common.metadata.MetadataContext.LOCAL_SERVICE;
|
||||
|
||||
public final class DefaultTransferMedataUtils {
|
||||
|
||||
private DefaultTransferMedataUtils() {
|
||||
}
|
||||
|
||||
public static Map<String, String> getDefaultTransferMedata() {
|
||||
return new HashMap<String, String>() {{
|
||||
put(DEFAULT_METADATA_SOURCE_SERVICE_NAMESPACE, LOCAL_NAMESPACE);
|
||||
put(DEFAULT_METADATA_SOURCE_SERVICE_NAME, LOCAL_SERVICE);
|
||||
}};
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,124 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software distributed
|
||||
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||
* specific language governing permissions and limitations under the License.
|
||||
*
|
||||
*/
|
||||
|
||||
package com.tencent.cloud.polaris.ratelimit.resolver;
|
||||
|
||||
import java.net.InetSocketAddress;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
import java.util.Set;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
|
||||
import com.tencent.cloud.common.metadata.MetadataContextHolder;
|
||||
import com.tencent.cloud.polaris.context.ServiceRuleManager;
|
||||
import com.tencent.cloud.polaris.ratelimit.filter.QuotaCheckServletFilter;
|
||||
import com.tencent.cloud.polaris.ratelimit.spi.PolarisRateLimiterLabelReactiveResolver;
|
||||
import com.tencent.polaris.ratelimit.api.rpc.Argument;
|
||||
import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto;
|
||||
import org.apache.commons.lang.StringUtils;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
import org.springframework.util.CollectionUtils;
|
||||
import org.springframework.web.server.ServerWebExchange;
|
||||
|
||||
import static com.tencent.cloud.common.constant.MetadataConstant.DefaultMetadata.DEFAULT_METADATA_SOURCE_SERVICE_NAME;
|
||||
import static com.tencent.cloud.common.constant.MetadataConstant.DefaultMetadata.DEFAULT_METADATA_SOURCE_SERVICE_NAMESPACE;
|
||||
|
||||
/**
|
||||
* resolve arguments from rate limit rule for Reactive.
|
||||
*
|
||||
* @author seansyyu 2023-03-09
|
||||
*/
|
||||
public class RateLimitRuleArgumentReactiveResolver {
|
||||
|
||||
private static final Logger LOG = LoggerFactory.getLogger(QuotaCheckServletFilter.class);
|
||||
|
||||
private final ServiceRuleManager serviceRuleManager;
|
||||
|
||||
private final PolarisRateLimiterLabelReactiveResolver labelResolver;
|
||||
|
||||
public RateLimitRuleArgumentReactiveResolver(ServiceRuleManager serviceRuleManager, PolarisRateLimiterLabelReactiveResolver labelResolver) {
|
||||
this.serviceRuleManager = serviceRuleManager;
|
||||
this.labelResolver = labelResolver;
|
||||
}
|
||||
|
||||
public Set<Argument> getArguments(ServerWebExchange request, String namespace, String service) {
|
||||
RateLimitProto.RateLimit rateLimitRule = serviceRuleManager.getServiceRateLimitRule(namespace, service);
|
||||
if (rateLimitRule == null) {
|
||||
return Collections.emptySet();
|
||||
}
|
||||
List<RateLimitProto.Rule> rules = rateLimitRule.getRulesList();
|
||||
if (CollectionUtils.isEmpty(rules)) {
|
||||
return Collections.emptySet();
|
||||
}
|
||||
return rules.stream()
|
||||
.flatMap(rule -> rule.getArgumentsList().stream())
|
||||
.map(matchArgument -> {
|
||||
String matchKey = matchArgument.getKey();
|
||||
Argument argument = null;
|
||||
switch (matchArgument.getType()) {
|
||||
case CUSTOM:
|
||||
argument = StringUtils.isBlank(matchKey) ? null :
|
||||
Argument.buildCustom(matchKey, Optional.ofNullable(getCustomResolvedLabels(request).get(matchKey)).orElse(StringUtils.EMPTY));
|
||||
break;
|
||||
case METHOD:
|
||||
argument = Argument.buildMethod(request.getRequest().getMethodValue());
|
||||
break;
|
||||
case HEADER:
|
||||
argument = StringUtils.isBlank(matchKey) ? null :
|
||||
Argument.buildHeader(matchKey, Optional.ofNullable(request.getRequest().getHeaders().getFirst(matchKey)).orElse(StringUtils.EMPTY));
|
||||
break;
|
||||
case QUERY:
|
||||
argument = StringUtils.isBlank(matchKey) ? null :
|
||||
Argument.buildQuery(matchKey, Optional.ofNullable(request.getRequest().getQueryParams().getFirst(matchKey)).orElse(StringUtils.EMPTY));
|
||||
break;
|
||||
case CALLER_SERVICE:
|
||||
String sourceServiceNamespace = MetadataContextHolder.getDisposableMetadata(DEFAULT_METADATA_SOURCE_SERVICE_NAMESPACE, true).orElse(StringUtils.EMPTY);
|
||||
String sourceServiceName = MetadataContextHolder.getDisposableMetadata(DEFAULT_METADATA_SOURCE_SERVICE_NAME, true).orElse(StringUtils.EMPTY);
|
||||
if (!StringUtils.isEmpty(sourceServiceNamespace) && !StringUtils.isEmpty(sourceServiceName)) {
|
||||
argument = Argument.buildCallerService(sourceServiceNamespace, sourceServiceName);
|
||||
}
|
||||
break;
|
||||
case CALLER_IP:
|
||||
InetSocketAddress remoteAddress = request.getRequest().getRemoteAddress();
|
||||
argument = Argument.buildCallerIP(remoteAddress != null ? remoteAddress.getAddress().getHostAddress() : StringUtils.EMPTY);
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
return argument;
|
||||
}).filter(Objects::nonNull).collect(Collectors.toSet());
|
||||
}
|
||||
|
||||
private Map<String, String> getCustomResolvedLabels(ServerWebExchange request) {
|
||||
if (labelResolver != null) {
|
||||
try {
|
||||
return labelResolver.resolve(request);
|
||||
}
|
||||
catch (Throwable e) {
|
||||
LOG.error("resolve custom label failed. resolver = {}", labelResolver.getClass().getName(), e);
|
||||
}
|
||||
}
|
||||
return Collections.emptyMap();
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,123 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software distributed
|
||||
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||
* specific language governing permissions and limitations under the License.
|
||||
*
|
||||
*/
|
||||
|
||||
package com.tencent.cloud.polaris.ratelimit.resolver;
|
||||
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Objects;
|
||||
import java.util.Optional;
|
||||
import java.util.Set;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import javax.servlet.http.HttpServletRequest;
|
||||
|
||||
import com.tencent.cloud.common.metadata.MetadataContextHolder;
|
||||
import com.tencent.cloud.polaris.context.ServiceRuleManager;
|
||||
import com.tencent.cloud.polaris.ratelimit.filter.QuotaCheckServletFilter;
|
||||
import com.tencent.cloud.polaris.ratelimit.spi.PolarisRateLimiterLabelServletResolver;
|
||||
import com.tencent.polaris.ratelimit.api.rpc.Argument;
|
||||
import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto;
|
||||
import org.apache.commons.lang.StringUtils;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
import org.springframework.util.CollectionUtils;
|
||||
|
||||
import static com.tencent.cloud.common.constant.MetadataConstant.DefaultMetadata.DEFAULT_METADATA_SOURCE_SERVICE_NAME;
|
||||
import static com.tencent.cloud.common.constant.MetadataConstant.DefaultMetadata.DEFAULT_METADATA_SOURCE_SERVICE_NAMESPACE;
|
||||
|
||||
/**
|
||||
* resolve arguments from rate limit rule for Servlet.
|
||||
*
|
||||
* @author seansyyu 2023-03-09
|
||||
*/
|
||||
public class RateLimitRuleArgumentServletResolver {
|
||||
|
||||
private static final Logger LOG = LoggerFactory.getLogger(QuotaCheckServletFilter.class);
|
||||
|
||||
private final ServiceRuleManager serviceRuleManager;
|
||||
|
||||
private final PolarisRateLimiterLabelServletResolver labelResolver;
|
||||
|
||||
public RateLimitRuleArgumentServletResolver(ServiceRuleManager serviceRuleManager, PolarisRateLimiterLabelServletResolver labelResolver) {
|
||||
this.serviceRuleManager = serviceRuleManager;
|
||||
this.labelResolver = labelResolver;
|
||||
}
|
||||
|
||||
public Set<Argument> getArguments(HttpServletRequest request, String namespace, String service) {
|
||||
RateLimitProto.RateLimit rateLimitRule = serviceRuleManager.getServiceRateLimitRule(namespace, service);
|
||||
if (rateLimitRule == null) {
|
||||
return Collections.emptySet();
|
||||
}
|
||||
List<RateLimitProto.Rule> rules = rateLimitRule.getRulesList();
|
||||
if (CollectionUtils.isEmpty(rules)) {
|
||||
return Collections.emptySet();
|
||||
}
|
||||
return rules.stream()
|
||||
.flatMap(rule -> rule.getArgumentsList().stream())
|
||||
.map(matchArgument -> {
|
||||
String matchKey = matchArgument.getKey();
|
||||
Argument argument = null;
|
||||
switch (matchArgument.getType()) {
|
||||
case CUSTOM:
|
||||
argument = StringUtils.isBlank(matchKey) ? null :
|
||||
Argument.buildCustom(matchKey, Optional.ofNullable(getCustomResolvedLabels(request).get(matchKey)).orElse(StringUtils.EMPTY));
|
||||
break;
|
||||
case METHOD:
|
||||
argument = Argument.buildMethod(request.getMethod());
|
||||
break;
|
||||
case HEADER:
|
||||
argument = StringUtils.isBlank(matchKey) ? null :
|
||||
Argument.buildHeader(matchKey, Optional.ofNullable(request.getHeader(matchKey)).orElse(StringUtils.EMPTY));
|
||||
break;
|
||||
case QUERY:
|
||||
argument = StringUtils.isBlank(matchKey) ? null :
|
||||
Argument.buildQuery(matchKey, Optional.ofNullable(request.getParameter(matchKey)).orElse(StringUtils.EMPTY));
|
||||
break;
|
||||
case CALLER_SERVICE:
|
||||
String sourceServiceNamespace = MetadataContextHolder.getDisposableMetadata(DEFAULT_METADATA_SOURCE_SERVICE_NAMESPACE, true).orElse(StringUtils.EMPTY);
|
||||
String sourceServiceName = MetadataContextHolder.getDisposableMetadata(DEFAULT_METADATA_SOURCE_SERVICE_NAME, true).orElse(StringUtils.EMPTY);
|
||||
if (!StringUtils.isEmpty(sourceServiceNamespace) && !StringUtils.isEmpty(sourceServiceName)) {
|
||||
argument = Argument.buildCallerService(sourceServiceNamespace, sourceServiceName);
|
||||
}
|
||||
break;
|
||||
case CALLER_IP:
|
||||
argument = Argument.buildCallerIP(Optional.ofNullable(request.getRemoteAddr()).orElse(StringUtils.EMPTY));
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
return argument;
|
||||
}).filter(Objects::nonNull).collect(Collectors.toSet());
|
||||
}
|
||||
|
||||
private Map<String, String> getCustomResolvedLabels(HttpServletRequest request) {
|
||||
if (labelResolver != null) {
|
||||
try {
|
||||
return labelResolver.resolve(request);
|
||||
}
|
||||
catch (Throwable e) {
|
||||
LOG.error("resolve custom label failed. resolver = {}", labelResolver.getClass().getName(), e);
|
||||
}
|
||||
}
|
||||
return Collections.emptyMap();
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,109 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software distributed
|
||||
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||
* specific language governing permissions and limitations under the License.
|
||||
*/
|
||||
|
||||
package com.tencent.cloud.polaris.ratelimit.resolver;
|
||||
|
||||
import java.io.BufferedReader;
|
||||
import java.io.InputStream;
|
||||
import java.io.InputStreamReader;
|
||||
import java.net.InetSocketAddress;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.Collections;
|
||||
import java.util.HashSet;
|
||||
import java.util.Set;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import com.google.protobuf.InvalidProtocolBufferException;
|
||||
import com.google.protobuf.util.JsonFormat;
|
||||
import com.tencent.cloud.common.metadata.MetadataContext;
|
||||
import com.tencent.cloud.polaris.context.ServiceRuleManager;
|
||||
import com.tencent.cloud.polaris.ratelimit.filter.QuotaCheckServletFilterTest;
|
||||
import com.tencent.cloud.polaris.ratelimit.spi.PolarisRateLimiterLabelReactiveResolver;
|
||||
import com.tencent.polaris.ratelimit.api.rpc.Argument;
|
||||
import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.junit.jupiter.api.extension.ExtendWith;
|
||||
|
||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
import org.springframework.mock.http.server.reactive.MockServerHttpRequest;
|
||||
import org.springframework.mock.web.server.MockServerWebExchange;
|
||||
import org.springframework.test.context.junit.jupiter.SpringExtension;
|
||||
import org.springframework.web.server.ServerWebExchange;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.mockito.ArgumentMatchers.anyString;
|
||||
import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@ExtendWith(SpringExtension.class)
|
||||
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT,
|
||||
classes = RateLimitRuleArgumentReactiveResolverTest.TestApplication.class,
|
||||
properties = {
|
||||
"spring.cloud.polaris.namespace=Test", "spring.cloud.polaris.service=TestApp"
|
||||
})
|
||||
public class RateLimitRuleArgumentReactiveResolverTest {
|
||||
|
||||
private final PolarisRateLimiterLabelReactiveResolver labelResolver =
|
||||
exchange -> Collections.singletonMap("xxx", "xxx");
|
||||
|
||||
private RateLimitRuleArgumentReactiveResolver rateLimitRuleArgumentReactiveResolver;
|
||||
|
||||
@BeforeEach
|
||||
void setUp() throws InvalidProtocolBufferException {
|
||||
MetadataContext.LOCAL_NAMESPACE = "TEST";
|
||||
|
||||
ServiceRuleManager serviceRuleManager = mock(ServiceRuleManager.class);
|
||||
|
||||
RateLimitProto.Rule.Builder ratelimitRuleBuilder = RateLimitProto.Rule.newBuilder();
|
||||
InputStream inputStream = QuotaCheckServletFilterTest.class.getClassLoader().getResourceAsStream("ratelimit.json");
|
||||
String json = new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8)).lines().collect(Collectors.joining(""));
|
||||
JsonFormat.parser().ignoringUnknownFields().merge(json, ratelimitRuleBuilder);
|
||||
RateLimitProto.Rule rateLimitRule = ratelimitRuleBuilder.build();
|
||||
RateLimitProto.RateLimit rateLimit = RateLimitProto.RateLimit.newBuilder().addRules(rateLimitRule).build();
|
||||
when(serviceRuleManager.getServiceRateLimitRule(anyString(), anyString())).thenReturn(rateLimit);
|
||||
|
||||
this.rateLimitRuleArgumentReactiveResolver = new RateLimitRuleArgumentReactiveResolver(serviceRuleManager, labelResolver);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetRuleArguments() {
|
||||
// Mock request
|
||||
MetadataContext.LOCAL_SERVICE = "Test";
|
||||
// Mock request
|
||||
MockServerHttpRequest request = MockServerHttpRequest.get("http://127.0.0.1:8080/test")
|
||||
.remoteAddress(new InetSocketAddress("127.0.0.1", 8080))
|
||||
.header("xxx", "xxx")
|
||||
.queryParam("yyy", "yyy")
|
||||
.build();
|
||||
ServerWebExchange exchange = MockServerWebExchange.from(request);
|
||||
Set<Argument> arguments = rateLimitRuleArgumentReactiveResolver.getArguments(exchange, MetadataContext.LOCAL_NAMESPACE, MetadataContext.LOCAL_SERVICE);
|
||||
Set<Argument> exceptRes = new HashSet<>();
|
||||
exceptRes.add(Argument.buildMethod("GET"));
|
||||
exceptRes.add(Argument.buildHeader("xxx", "xxx"));
|
||||
exceptRes.add(Argument.buildQuery("yyy", "yyy"));
|
||||
exceptRes.add(Argument.buildCallerIP("127.0.0.1"));
|
||||
exceptRes.add(Argument.buildCustom("xxx", "xxx"));
|
||||
assertThat(arguments).isEqualTo(exceptRes);
|
||||
}
|
||||
|
||||
@SpringBootApplication
|
||||
protected static class TestApplication {
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,100 @@
|
||||
/*
|
||||
* Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
|
||||
*
|
||||
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
|
||||
*
|
||||
* Licensed under the BSD 3-Clause License (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://opensource.org/licenses/BSD-3-Clause
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software distributed
|
||||
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
||||
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
||||
* specific language governing permissions and limitations under the License.
|
||||
*/
|
||||
|
||||
package com.tencent.cloud.polaris.ratelimit.resolver;
|
||||
|
||||
import java.io.BufferedReader;
|
||||
import java.io.InputStream;
|
||||
import java.io.InputStreamReader;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.Collections;
|
||||
import java.util.HashSet;
|
||||
import java.util.Set;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
import com.google.protobuf.InvalidProtocolBufferException;
|
||||
import com.google.protobuf.util.JsonFormat;
|
||||
import com.tencent.cloud.common.metadata.MetadataContext;
|
||||
import com.tencent.cloud.polaris.context.ServiceRuleManager;
|
||||
import com.tencent.cloud.polaris.ratelimit.filter.QuotaCheckServletFilterTest;
|
||||
import com.tencent.cloud.polaris.ratelimit.spi.PolarisRateLimiterLabelServletResolver;
|
||||
import com.tencent.polaris.ratelimit.api.rpc.Argument;
|
||||
import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.junit.jupiter.api.extension.ExtendWith;
|
||||
|
||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
import org.springframework.mock.web.MockHttpServletRequest;
|
||||
import org.springframework.test.context.junit.jupiter.SpringExtension;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.mockito.ArgumentMatchers.anyString;
|
||||
import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@ExtendWith(SpringExtension.class)
|
||||
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT,
|
||||
classes = RateLimitRuleArgumentServletResolverTest.TestApplication.class,
|
||||
properties = {
|
||||
"spring.cloud.polaris.namespace=Test", "spring.cloud.polaris.service=TestApp"
|
||||
})
|
||||
public class RateLimitRuleArgumentServletResolverTest {
|
||||
|
||||
private final PolarisRateLimiterLabelServletResolver labelResolver =
|
||||
exchange -> Collections.singletonMap("xxx", "xxx");
|
||||
private RateLimitRuleArgumentServletResolver rateLimitRuleArgumentServletResolver;
|
||||
|
||||
@BeforeEach
|
||||
void setUp() throws InvalidProtocolBufferException {
|
||||
MetadataContext.LOCAL_NAMESPACE = "TEST";
|
||||
|
||||
ServiceRuleManager serviceRuleManager = mock(ServiceRuleManager.class);
|
||||
|
||||
RateLimitProto.Rule.Builder ratelimitRuleBuilder = RateLimitProto.Rule.newBuilder();
|
||||
InputStream inputStream = QuotaCheckServletFilterTest.class.getClassLoader().getResourceAsStream("ratelimit.json");
|
||||
String json = new BufferedReader(new InputStreamReader(inputStream, StandardCharsets.UTF_8)).lines().collect(Collectors.joining(""));
|
||||
JsonFormat.parser().ignoringUnknownFields().merge(json, ratelimitRuleBuilder);
|
||||
RateLimitProto.Rule rateLimitRule = ratelimitRuleBuilder.build();
|
||||
RateLimitProto.RateLimit rateLimit = RateLimitProto.RateLimit.newBuilder().addRules(rateLimitRule).build();
|
||||
when(serviceRuleManager.getServiceRateLimitRule(anyString(), anyString())).thenReturn(rateLimit);
|
||||
|
||||
this.rateLimitRuleArgumentServletResolver = new RateLimitRuleArgumentServletResolver(serviceRuleManager, labelResolver);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetRuleArguments() {
|
||||
// Mock request
|
||||
MetadataContext.LOCAL_SERVICE = "Test";
|
||||
MockHttpServletRequest request = new MockHttpServletRequest(null, "GET", "/xxx");
|
||||
request.setParameter("yyy", "yyy");
|
||||
request.addHeader("xxx", "xxx");
|
||||
Set<Argument> arguments = rateLimitRuleArgumentServletResolver.getArguments(request, MetadataContext.LOCAL_NAMESPACE, MetadataContext.LOCAL_SERVICE);
|
||||
Set<Argument> exceptRes = new HashSet<>();
|
||||
exceptRes.add(Argument.buildMethod("GET"));
|
||||
exceptRes.add(Argument.buildHeader("xxx", "xxx"));
|
||||
exceptRes.add(Argument.buildQuery("yyy", "yyy"));
|
||||
exceptRes.add(Argument.buildCallerIP("127.0.0.1"));
|
||||
exceptRes.add(Argument.buildCustom("xxx", "xxx"));
|
||||
assertThat(arguments).isEqualTo(exceptRes);
|
||||
}
|
||||
|
||||
@SpringBootApplication
|
||||
protected static class TestApplication {
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,95 @@
|
||||
{
|
||||
"id": "f7560cce829e4d4c8556a6be63539af5",
|
||||
"service": "xxx",
|
||||
"namespace": "default",
|
||||
"subset": {},
|
||||
"priority": 0,
|
||||
"resource": "QPS",
|
||||
"type": "GLOBAL",
|
||||
"labels": {},
|
||||
"amounts": [
|
||||
{
|
||||
"maxAmount": 1,
|
||||
"validDuration": "1s",
|
||||
"precision": null,
|
||||
"startAmount": null,
|
||||
"minAmount": null
|
||||
}
|
||||
],
|
||||
"action": "REJECT",
|
||||
"disable": false,
|
||||
"report": null,
|
||||
"ctime": "2022-12-11 21:56:59",
|
||||
"mtime": "2023-03-10 15:40:33",
|
||||
"revision": "6eec6f416bee40ecbf664c93add61358",
|
||||
"service_token": null,
|
||||
"adjuster": null,
|
||||
"regex_combine": true,
|
||||
"amountMode": "GLOBAL_TOTAL",
|
||||
"failover": "FAILOVER_LOCAL",
|
||||
"cluster": null,
|
||||
"method": {
|
||||
"type": "EXACT",
|
||||
"value": "/xxx",
|
||||
"value_type": "TEXT"
|
||||
},
|
||||
"arguments": [
|
||||
{
|
||||
"type": "CALLER_SERVICE",
|
||||
"key": "default",
|
||||
"value": {
|
||||
"type": "EXACT",
|
||||
"value": "xxx",
|
||||
"value_type": "TEXT"
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "HEADER",
|
||||
"key": "xxx",
|
||||
"value": {
|
||||
"type": "EXACT",
|
||||
"value": "xxx",
|
||||
"value_type": "TEXT"
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "QUERY",
|
||||
"key": "yyy",
|
||||
"value": {
|
||||
"type": "EXACT",
|
||||
"value": "yyy",
|
||||
"value_type": "TEXT"
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "METHOD",
|
||||
"key": "$method",
|
||||
"value": {
|
||||
"type": "EXACT",
|
||||
"value": "GET",
|
||||
"value_type": "TEXT"
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "CALLER_IP",
|
||||
"key": "$caller_ip",
|
||||
"value": {
|
||||
"type": "EXACT",
|
||||
"value": "127.0.0.1",
|
||||
"value_type": "TEXT"
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "CUSTOM",
|
||||
"key": "xxx",
|
||||
"value": {
|
||||
"type": "EXACT",
|
||||
"value": "xxx",
|
||||
"value_type": "TEXT"
|
||||
}
|
||||
}
|
||||
],
|
||||
"name": "xxx",
|
||||
"etime": "2023-03-10 15:40:33",
|
||||
"max_queue_delay": 30
|
||||
}
|
||||
Loading…
Reference in new issue