Examples with LimitAPI com.tencent.polaris.ratelimit.api.core.LimitAPI used on opensource projects

Example 1 with LimitAPI

use of com.tencent.polaris.ratelimit.api.core.LimitAPI in project polaris-java by polarismesh.

the class LocalTest method testSingleThreadLimit.

@Test
public void testSingleThreadLimit() {
    Configuration configuration = TestUtils.configWithEnvAddress();
    try (LimitAPI limitAPI = LimitAPIFactory.createLimitAPIByConfig(configuration)) {
        adjustTime();
        testQuotaAcquire(limitAPI, Consts.createSingleValueMap(new String[] { Consts.LABEL_METHOD }, new String[] { Consts.METHOD_PAY }), Consts.MAX_PAY_COUNT);
        testQuotaAcquire(limitAPI, Consts.createSingleValueMap(new String[] { Consts.LABEL_METHOD }, new String[] { Consts.METHOD_CASH }), Consts.MAX_CASH_COUNT);
        testQuotaAcquire(limitAPI, null, Consts.MAX_SERVICE_COUNT);
        System.out.println("start to wait expired");
        Utils.sleepUninterrupted(10 * 1000);
    }
}

Example 2 with LimitAPI

use of com.tencent.polaris.ratelimit.api.core.LimitAPI in project polaris-java by polarismesh.

the class RateLimitExample method main.

public static void main(String[] args) throws Exception {
    InitResult initResult = LimitExampleUtils.initRateLimitConfiguration(args);
    String namespace = initResult.getNamespace();
    String service = initResult.getService();
    int concurrency = initResult.getConcurrency();
    ScheduledExecutorService executorService = Executors.newScheduledThreadPool(concurrency);
    // 注意：使用本地限流时，限流阈值计数器会存放在LimitAPI实例内部，无法跨实例共享，因此LimitAPI建议通过进程单例模式使用
    try (LimitAPI limitAPI = LimitAPIFactory.createLimitAPI()) {
        Runnable runnable = new Runnable() {

            @Override
            public void run() {
                QuotaRequest quotaRequest = new QuotaRequest();
                quotaRequest.setNamespace(namespace);
                quotaRequest.setService(service);
                quotaRequest.setMethod("echo");
                quotaRequest.setCount(1);
                QuotaResponse quotaResponse = limitAPI.getQuota(quotaRequest);
                System.out.println("quotaResponse is " + quotaResponse.getCode());
            }
        };
        List<ScheduledFuture<?>> futures = new ArrayList<>();
        for (int i = 0; i < concurrency; i++) {
            ScheduledFuture<?> scheduledFuture = executorService.scheduleWithFixedDelay(runnable, 10 + i, 500, TimeUnit.MILLISECONDS);
            futures.add(scheduledFuture);
        }
        Thread.sleep(500000);
        for (ScheduledFuture<?> future : futures) {
            future.cancel(true);
        }
    }
    executorService.shutdown();
}

Example 3 with LimitAPI

use of com.tencent.polaris.ratelimit.api.core.LimitAPI in project spring-cloud-tencent by Tencent.

the class QuotaCheckUtilsTest method setUp.

@Before
public void setUp() {
    limitAPI = mock(LimitAPI.class);
    when(limitAPI.getQuota(any(QuotaRequest.class))).thenAnswer(invocationOnMock -> {
        String serviceName = ((QuotaRequest) invocationOnMock.getArgument(0)).getService();
        if (serviceName.equals("TestApp1")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultOk, 0, "QuotaResultOk"));
        } else if (serviceName.equals("TestApp2")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultOk, 1000, "QuotaResultOk"));
        } else if (serviceName.equals("TestApp3")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultLimited, 0, "QuotaResultLimited"));
        } else {
            throw new RuntimeException("Mock exception.");
        }
    });
}

Example 4 with LimitAPI

use of com.tencent.polaris.ratelimit.api.core.LimitAPI in project spring-cloud-tencent by Tencent.

the class QuotaCheckReactiveFilterTest method setUp.

@Before
public void setUp() {
    MetadataContext.LOCAL_NAMESPACE = "TEST";
    LimitAPI limitAPI = mock(LimitAPI.class);
    when(limitAPI.getQuota(any(QuotaRequest.class))).thenAnswer(invocationOnMock -> {
        String serviceName = ((QuotaRequest) invocationOnMock.getArgument(0)).getService();
        if (serviceName.equals("TestApp1")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultOk, 0, "QuotaResultOk"));
        } else if (serviceName.equals("TestApp2")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultOk, 1000, "QuotaResultOk"));
        } else if (serviceName.equals("TestApp3")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultLimited, 0, "QuotaResultLimited"));
        } else {
            return new QuotaResponse(new QuotaResult(null, 0, null));
        }
    });
    PolarisRateLimitProperties polarisRateLimitProperties = new PolarisRateLimitProperties();
    polarisRateLimitProperties.setRejectRequestTips("RejectRequestTips");
    polarisRateLimitProperties.setRejectHttpCode(419);
    RateLimitRuleLabelResolver rateLimitRuleLabelResolver = mock(RateLimitRuleLabelResolver.class);
    when(rateLimitRuleLabelResolver.getExpressionLabelKeys(anyString(), anyString())).thenReturn(Collections.EMPTY_SET);
    this.quotaCheckReactiveFilter = new QuotaCheckReactiveFilter(limitAPI, labelResolver, polarisRateLimitProperties, rateLimitRuleLabelResolver);
}

Example 5 with LimitAPI

use of com.tencent.polaris.ratelimit.api.core.LimitAPI in project spring-cloud-tencent by Tencent.

the class QuotaCheckServletFilterTest method setUp.

@Before
public void setUp() {
    MetadataContext.LOCAL_NAMESPACE = "TEST";
    LimitAPI limitAPI = mock(LimitAPI.class);
    when(limitAPI.getQuota(any(QuotaRequest.class))).thenAnswer(invocationOnMock -> {
        String serviceName = ((QuotaRequest) invocationOnMock.getArgument(0)).getService();
        if (serviceName.equals("TestApp1")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultOk, 0, "QuotaResultOk"));
        } else if (serviceName.equals("TestApp2")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultOk, 1000, "QuotaResultOk"));
        } else if (serviceName.equals("TestApp3")) {
            return new QuotaResponse(new QuotaResult(QuotaResult.Code.QuotaResultLimited, 0, "QuotaResultLimited"));
        } else {
            return new QuotaResponse(new QuotaResult(null, 0, null));
        }
    });
    PolarisRateLimitProperties polarisRateLimitProperties = new PolarisRateLimitProperties();
    polarisRateLimitProperties.setRejectRequestTips("RejectRequestTips");
    polarisRateLimitProperties.setRejectHttpCode(419);
    RateLimitRuleLabelResolver rateLimitRuleLabelResolver = mock(RateLimitRuleLabelResolver.class);
    when(rateLimitRuleLabelResolver.getExpressionLabelKeys(anyString(), anyString())).thenReturn(Collections.EMPTY_SET);
    this.quotaCheckServletFilter = new QuotaCheckServletFilter(limitAPI, labelResolver, polarisRateLimitProperties, rateLimitRuleLabelResolver);
}