Examples with StreamCounterSet com.tencent.polaris.ratelimit.client.flow.StreamCounterSet used on opensource projects

Example 1 with StreamCounterSet

use of com.tencent.polaris.ratelimit.client.flow.StreamCounterSet 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 StreamCounterSet

use of com.tencent.polaris.ratelimit.client.flow.StreamCounterSet in project polaris-java by polarismesh.

the class RemoteSyncTask method doRemoteInit.

/**
 * 发送初始化请求
 */
private void doRemoteInit() {
    StreamCounterSet streamCounterSet = asyncRateLimitConnector.getStreamCounterSet(window.getWindowSet().getRateLimitExtension().getExtensions(), window.getRemoteCluster(), window.getUniqueKey(), serviceIdentifier);
    // 拿不到限流集群的实例的时候
    if (streamCounterSet == null) {
        LOG.error("[doRemoteInit] failed, stream counter is null. remote cluster:{},", window.getRemoteCluster());
        return;
    }
    // 调整时间
    if (!adjustTime(streamCounterSet)) {
        LOG.error("[doRemoteInit] adjustTime failed.remote cluster:{},svcKey:{}", window.getRemoteCluster(), window.getSvcKey());
        return;
    }
    StreamObserver<RateLimitRequest> streamClient = streamCounterSet.preCheckAsync(serviceIdentifier, window);
    if (streamClient == null) {
        LOG.error("[doRemoteInit] failed, stream client is null. remote cluster:{},svcKey:{}", window.getRemoteCluster(), window.getSvcKey());
        return;
    }
    // clientId
    Builder initRequest = RateLimitInitRequest.newBuilder();
    initRequest.setClientId(window.getWindowSet().getClientId());
    // target
    LimitTarget.Builder target = LimitTarget.newBuilder();
    target.setNamespace(window.getSvcKey().getNamespace());
    target.setService(window.getSvcKey().getService());
    target.setLabels(window.getLabels());
    initRequest.setTarget(target);
    // QuotaTotal
    QuotaMode quotaMode = QuotaMode.forNumber(window.getRule().getAmountModeValue());
    QuotaBucket allocatingBucket = window.getAllocatingBucket();
    Map<Integer, AmountInfo> amountInfos = allocatingBucket.getAmountInfo();
    if (MapUtils.isNotEmpty(amountInfos)) {
        for (Map.Entry<Integer, AmountInfo> entry : amountInfos.entrySet()) {
            QuotaTotal.Builder total = QuotaTotal.newBuilder();
            total.setDuration(entry.getKey());
            total.setMode(quotaMode);
            total.setMaxAmount((int) entry.getValue().getMaxAmount());
            initRequest.addTotals(total.build());
        }
    }
    RateLimitRequest rateLimitInitRequest = RateLimitRequest.newBuilder().setCmd(RateLimitCmd.INIT).setRateLimitInitRequest(initRequest).build();
    streamClient.onNext(rateLimitInitRequest);
}