Examples with TokenResult com.alibaba.csp.sentinel.cluster.TokenResult used on opensource projects

Example 21 with TokenResult

use of com.alibaba.csp.sentinel.cluster.TokenResult in project Sentinel by alibaba.

the class ConcurrentClusterFlowChecker method releaseConcurrentToken.

public static TokenResult releaseConcurrentToken(/*@Valid*/
long tokenId) {
    TokenCacheNode node = TokenCacheNodeManager.getTokenCacheNode(tokenId);
    if (node == null) {
        RecordLog.info("[ConcurrentClusterFlowChecker] Token<{}> is already released", tokenId);
        return new TokenResult(TokenResultStatus.ALREADY_RELEASE);
    }
    FlowRule rule = ClusterFlowRuleManager.getFlowRuleById(node.getFlowId());
    if (rule == null) {
        RecordLog.info("[ConcurrentClusterFlowChecker] Fail to get rule by flowId<{}>", node.getFlowId());
        return new TokenResult(TokenResultStatus.NO_RULE_EXISTS);
    }
    if (TokenCacheNodeManager.removeTokenCacheNode(tokenId) == null) {
        RecordLog.info("[ConcurrentClusterFlowChecker] Token<{}> is already released for flowId<{}>", tokenId, node.getFlowId());
        return new TokenResult(TokenResultStatus.ALREADY_RELEASE);
    }
    int acquireCount = node.getAcquireCount();
    AtomicInteger nowCalls = CurrentConcurrencyManager.get(node.getFlowId());
    nowCalls.getAndAdd(-1 * acquireCount);
    ClusterServerStatLogUtil.log("concurrent|release|" + rule.getClusterConfig().getFlowId(), acquireCount);
    return new TokenResult(TokenResultStatus.RELEASE_OK);
}

Example 22 with TokenResult

use of com.alibaba.csp.sentinel.cluster.TokenResult in project Sentinel by alibaba.

the class ParamFlowChecker method passClusterCheck.

private static boolean passClusterCheck(ResourceWrapper resourceWrapper, ParamFlowRule rule, int count, Object value) {
    try {
        Collection<Object> params = toCollection(value);
        TokenService clusterService = pickClusterService();
        if (clusterService == null) {
            // pass in need.
            return fallbackToLocalOrPass(resourceWrapper, rule, count, params);
        }
        TokenResult result = clusterService.requestParamToken(rule.getClusterConfig().getFlowId(), count, params);
        switch(result.getStatus()) {
            case TokenResultStatus.OK:
                return true;
            case TokenResultStatus.BLOCKED:
                return false;
            default:
                return fallbackToLocalOrPass(resourceWrapper, rule, count, params);
        }
    } catch (Throwable ex) {
        RecordLog.warn("[ParamFlowChecker] Request cluster token for parameter unexpected failed", ex);
        return fallbackToLocalOrPass(resourceWrapper, rule, count, value);
    }
}