Examples with FlowRule com.alibaba.csp.sentinel.slots.block.flow.FlowRule used on opensource projects

Example 66 with FlowRule

use of com.alibaba.csp.sentinel.slots.block.flow.FlowRule in project Sentinel by alibaba.

the class ConcurrentClusterFlowCheckerTest method setUp.

@Before
public void setUp() {
    FlowRule rule = new FlowRule();
    ClusterFlowConfig config = new ClusterFlowConfig();
    config.setResourceTimeout(500);
    config.setClientOfflineTime(1000);
    config.setFlowId(111L);
    config.setThresholdType(ClusterRuleConstant.FLOW_THRESHOLD_GLOBAL);
    rule.setClusterConfig(config);
    rule.setClusterMode(true);
    rule.setCount(10);
    rule.setResource("test");
    rule.setGrade(RuleConstant.FLOW_GRADE_THREAD);
    ArrayList<FlowRule> rules = new ArrayList<>();
    rules.add(rule);
    ClusterFlowRuleManager.registerPropertyIfAbsent("1-name");
    ClusterFlowRuleManager.loadRules("1-name", rules);
}

Example 67 with FlowRule

use of com.alibaba.csp.sentinel.slots.block.flow.FlowRule in project Sentinel by alibaba.

the class ConcurrentClusterFlowCheckerTest method testEasyAcquireAndRelease.

@Test
public void testEasyAcquireAndRelease() throws InterruptedException {
    setCurrentMillis(System.currentTimeMillis());
    FlowRule rule = ClusterFlowRuleManager.getFlowRuleById(111L);
    ArrayList<Long> list = new ArrayList<>();
    for (int i = 0; i < 10; i++) {
        TokenResult result = ConcurrentClusterFlowChecker.acquireConcurrentToken("127.0.0.1", rule, 1);
        Assert.assertTrue("fail to acquire token", result.getStatus() == TokenResultStatus.OK && result.getTokenId() != 0);
        list.add(result.getTokenId());
    }
    for (int i = 0; i < 10; i++) {
        TokenResult result = ConcurrentClusterFlowChecker.acquireConcurrentToken("127.0.0.1", rule, 1);
        Assert.assertTrue("fail to acquire block token", result.getStatus() == TokenResultStatus.BLOCKED);
    }
    for (int i = 0; i < 10; i++) {
        TokenResult result = ConcurrentClusterFlowChecker.releaseConcurrentToken(list.get(i));
        Assert.assertTrue("fail to release token", result.getStatus() == TokenResultStatus.RELEASE_OK);
    }
    Assert.assertTrue("fail to release token", CurrentConcurrencyManager.get(111L).get() == 0 && TokenCacheNodeManager.getSize() == 0);
}

Example 68 with FlowRule

use of com.alibaba.csp.sentinel.slots.block.flow.FlowRule in project Sentinel by alibaba.

the class TokenCacheNodeManagerTest method setUp.

@Before
public void setUp() {
    FlowRule rule = new FlowRule();
    ClusterFlowConfig config = new ClusterFlowConfig();
    config.setResourceTimeout(500);
    config.setClientOfflineTime(1000);
    config.setFlowId(111L);
    config.setThresholdType(ClusterRuleConstant.FLOW_THRESHOLD_GLOBAL);
    rule.setClusterConfig(config);
    rule.setClusterMode(true);
    rule.setCount(10);
    rule.setResource("test");
    rule.setGrade(RuleConstant.FLOW_GRADE_THREAD);
    ArrayList<FlowRule> rules = new ArrayList<>();
    rules.add(rule);
    ClusterFlowRuleManager.registerPropertyIfAbsent("1-name");
    ClusterFlowRuleManager.loadRules("1-name", rules);
}

Example 69 with FlowRule

use of com.alibaba.csp.sentinel.slots.block.flow.FlowRule in project Sentinel by alibaba.

the class CommonFilterTest method configureRulesFor.

private void configureRulesFor(String resource, int count, String limitApp) {
    FlowRule rule = new FlowRule().setCount(count).setGrade(RuleConstant.FLOW_GRADE_QPS);
    rule.setResource(resource);
    if (StringUtil.isNotBlank(limitApp)) {
        rule.setLimitApp(limitApp);
    }
    FlowRuleManager.loadRules(Collections.singletonList(rule));
}

Example 70 with FlowRule

use of com.alibaba.csp.sentinel.slots.block.flow.FlowRule in project Sentinel by alibaba.

the class CommonFilterMethodTest method configureRulesFor.

private void configureRulesFor(String resource, int count, String limitApp) {
    FlowRule rule = new FlowRule().setCount(count).setGrade(RuleConstant.FLOW_GRADE_QPS);
    rule.setResource(resource);
    if (StringUtil.isNotBlank(limitApp)) {
        rule.setLimitApp(limitApp);
    }
    FlowRuleManager.loadRules(Collections.singletonList(rule));
}