Examples with LocalQuotaInfo com.tencent.polaris.api.plugin.ratelimiter.LocalQuotaInfo used on opensource projects

Example 1 with LocalQuotaInfo

use of com.tencent.polaris.api.plugin.ratelimiter.LocalQuotaInfo in project polaris-java by polarismesh.

the class RemoteSyncTask method doRemoteAcquire.

/**
 * 上报
 */
private void doRemoteAcquire() {
    StreamCounterSet streamCounterSet = asyncRateLimitConnector.getStreamCounterSet(window.getWindowSet().getRateLimitExtension().getExtensions(), window.getRemoteCluster(), window.getUniqueKey(), serviceIdentifier);
    if (streamCounterSet == null) {
        LOG.error("[doRemoteAcquire] failed, stream counter is null. remote cluster:{},", window.getRemoteCluster());
        return;
    }
    if (!streamCounterSet.hasInit(serviceIdentifier)) {
        LOG.warn("[doRemoteAcquire] has not inited. serviceKey:{}", window.getSvcKey());
        doRemoteInit();
        return;
    }
    // 调整时间
    if (!adjustTime(streamCounterSet)) {
        LOG.error("[doRemoteAcquire] adjustTime failed.remote cluster:{},svcKey:{}", window.getRemoteCluster(), window.getSvcKey());
        return;
    }
    StreamObserver<RateLimitRequest> streamClient = streamCounterSet.preCheckAsync(serviceIdentifier, window);
    if (streamClient == null) {
        LOG.error("[doRemoteAcquire] failed, stream client is null. remote cluster:{}", window.getRemoteCluster());
        return;
    }
    RateLimitReportRequest.Builder rateLimitReportRequest = RateLimitReportRequest.newBuilder();
    // clientKey
    rateLimitReportRequest.setClientKey(streamCounterSet.getClientKey());
    // timestamp
    long curTimeMilli = System.currentTimeMillis();
    long serverTimeMilli = curTimeMilli + asyncRateLimitConnector.getTimeDiff().get();
    rateLimitReportRequest.setTimestamp(serverTimeMilli);
    // quotaUses
    Map<Integer, LocalQuotaInfo> localQuotaInfos = window.getAllocatingBucket().fetchLocalUsage(serverTimeMilli);
    for (Map.Entry<Integer, LocalQuotaInfo> entry : localQuotaInfos.entrySet()) {
        QuotaSum.Builder quotaSum = QuotaSum.newBuilder();
        quotaSum.setUsed((int) entry.getValue().getQuotaUsed());
        quotaSum.setLimited((int) entry.getValue().getQuotaLimited());
        quotaSum.setCounterKey(streamCounterSet.getInitRecord().get(serviceIdentifier).getDurationRecord().get(entry.getKey()));
        rateLimitReportRequest.addQuotaUses(quotaSum.build());
    }
    RateLimitRequest rateLimitRequest = RateLimitRequest.newBuilder().setCmd(RateLimitCmd.ACQUIRE).setRateLimitReportRequest(rateLimitReportRequest).build();
    streamClient.onNext(rateLimitRequest);
}

Example 2 with LocalQuotaInfo

use of com.tencent.polaris.api.plugin.ratelimiter.LocalQuotaInfo in project polaris-java by polarismesh.

the class RemoteAwareBucket method fetchLocalUsage.

@Override
public Map<Integer, LocalQuotaInfo> fetchLocalUsage(long curTimeMs) {
    Map<Integer, LocalQuotaInfo> localInfos = new HashMap<>();
    for (Map.Entry<Long, TokenBucket> entry : tokenBucketMap.entrySet()) {
        TokenBucket tokenBucket = entry.getValue();
        Result result = tokenBucket.getSlidingWindow().acquireCurrentValues(curTimeMs);
        LocalQuotaInfo localQuotaInfo = new LocalQuotaInfo(result.getPassed(), result.getLimited());
        localInfos.put(tokenBucket.getValidDurationSecond(), localQuotaInfo);
    }
    return localInfos;
}