Skip to content

Commit aa8cfb2

Browse files
committed
feat:support rate-limit window expiration.
Signed-off-by: Haotian Zhang <[email protected]>
1 parent 136271d commit aa8cfb2

File tree

1 file changed

+7
-3
lines changed
  • polaris-ratelimit/polaris-ratelimit-client/src/main/java/com/tencent/polaris/ratelimit/client/flow

1 file changed

+7
-3
lines changed

polaris-ratelimit/polaris-ratelimit-client/src/main/java/com/tencent/polaris/ratelimit/client/flow/RateLimitWindow.java

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
package com.tencent.polaris.ratelimit.client.flow;
1919

2020
import com.tencent.polaris.api.config.consumer.LoadBalanceConfig;
21+
import com.tencent.polaris.api.config.consumer.ServiceRouterConfig;
2122
import com.tencent.polaris.api.config.provider.RateLimitConfig;
2223
import com.tencent.polaris.api.plugin.compose.Extensions;
2324
import com.tencent.polaris.api.plugin.ratelimiter.InitCriteria;
@@ -41,6 +42,7 @@
4142
import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto.Rule;
4243
import org.slf4j.Logger;
4344

45+
import java.util.ArrayList;
4446
import java.util.List;
4547
import java.util.Objects;
4648
import java.util.Random;
@@ -146,16 +148,18 @@ public RateLimitWindow(RateLimitWindowSet windowSet, CommonQuotaRequest quotaReq
146148
this.syncParam = quotaRequest.getFlowControlParam();
147149
remoteCluster = getLimiterClusterService(rule.getCluster(), rateLimitConfig);
148150
serviceAddressRepository = buildServiceAddressRepository(rateLimitConfig.getLimiterAddresses(),
149-
uniqueKey, windowSet.getExtensions(), remoteCluster, null, LoadBalanceConfig.LOAD_BALANCE_RING_HASH, "grpc");
151+
uniqueKey, windowSet.getExtensions(), remoteCluster);
150152
allocatingBucket = getQuotaBucket(initCriteria, windowSet.getRateLimitExtension());
151153
lastAccessTimeMs.set(System.currentTimeMillis());
152154
this.rateLimitConfig = rateLimitConfig;
153155
buildRemoteConfigMode();
154156
}
155157

156158
private ServiceAddressRepository buildServiceAddressRepository(List<String> addresses, String hash, Extensions extensions,
157-
ServiceKey remoteCluster, List<String> routers, String lbPolicy, String protocol) {
158-
return new ServiceAddressRepository(addresses, hash, extensions, remoteCluster, routers, lbPolicy, protocol);
159+
ServiceKey remoteCluster) {
160+
List<String> routers = new ArrayList<>();
161+
routers.add(ServiceRouterConfig.DEFAULT_ROUTER_METADATA);
162+
return new ServiceAddressRepository(addresses, hash, extensions, remoteCluster, routers, LoadBalanceConfig.LOAD_BALANCE_RING_HASH, "grpc");
159163
}
160164

161165

0 commit comments

Comments
 (0)