com.tencent.polaris.ratelimit.api.rpc.QuotaResponse Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of polaris-all Show documentation

All in one project for polaris-java

There is a newer version: 2.0.1.0-RC1

/*
 * Tencent is pleased to support the open source community by making polaris-java available.
 *
 * Copyright (C) 2021 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.polaris.ratelimit.api.rpc;

import com.tencent.polaris.api.plugin.ratelimiter.QuotaResult;
import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto;
import com.tencent.polaris.specification.api.v1.traffic.manage.RateLimitProto.Rule;

import java.util.ArrayList;
import java.util.List;

public class QuotaResponse {

    private final QuotaResult quotaResult;

    private RateLimitProto.Rule activeRule;

    private List releaseList;

    public QuotaResponse(QuotaResult quotaResult) {
        this(quotaResult, null);
    }

    public QuotaResponse(QuotaResult quotaResult, List releaseList) {
        this.quotaResult = quotaResult;
        if (releaseList == null) {
            releaseList = new ArrayList<>();
            if (quotaResult.getRelease() != null) {
                releaseList.add(quotaResult.getRelease());
            }
        }
        this.releaseList = releaseList;
    }

    public QuotaResultCode getCode() {
        return QuotaResultCode.values()[quotaResult.getCode().ordinal()];
    }

    public long getWaitMs() {
        return quotaResult.getWaitMs();
    }

    public String getInfo() {
        return quotaResult.getInfo();
    }

    public void setActiveRule(Rule activeRule) {
        this.activeRule = activeRule;
    }

    public String getActiveRuleName() {
        String activeRuleName = this.getActiveRule().getName().getValue();
        return activeRuleName;
    }

    public Rule getActiveRule() {
        return activeRule;
    }

    public void addRelease(Runnable runnable) {
        releaseList.add(runnable);
    }

    public List getReleaseList() {
        return releaseList;
    }

    public void setReleaseList(List releaseList) {
        if (releaseList == null) {
            releaseList = new ArrayList<>();
        }
        this.releaseList = releaseList;
    }
}