com.huawei.flowcontrol.common.adapte.cse.RuleUtils Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2022-2022 Huawei Technologies Co., Ltd. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * 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.huawei.flowcontrol.common.adapte.cse;

import com.huawei.flowcontrol.common.adapte.cse.match.MatchManager;
import com.huawei.flowcontrol.common.adapte.cse.resolver.AbstractResolver;
import com.huawei.flowcontrol.common.entity.HttpRequestEntity;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Set;

/**
 * 规则获取工具类
 *
 * @author zhouss
 * @since 2022-07-25
 */
public class RuleUtils {
    private RuleUtils() {
    }

    /**
     * 获取规则
     *
     * @param businessName 业务名称
     * @param resolverKey  解析器键
     * @param ruleType 规则类型
     * @param  规则类型
     * @return rule
     */
    public static  T getRule(String businessName, String resolverKey, Class ruleType) {
        final AbstractResolver resolver = ResolverManager.INSTANCE.getResolver(resolverKey);
        return (T)resolver.getRules().get(businessName);
    }

    /**
     * 根据请求体匹配获取规则
     *
     * @param requestEntity 请求体
     * @param resolverKey   解析器键
     * @param ruleType  规则类型
     * @param  规则类型
     * @return List-Rule
     */
    public static  List getRule(HttpRequestEntity requestEntity, String resolverKey, Class ruleType) {
        final Set businessNames = MatchManager.INSTANCE.matchWithCache(requestEntity);
        if (businessNames.isEmpty()) {
            return Collections.emptyList();
        }
        final List result = new ArrayList<>(businessNames.size());
        for (String businessName : businessNames) {
            final T rule = getRule(businessName, resolverKey, ruleType);
            if (rule != null) {
                result.add(rule);
            }
        }
        return result;
    }
}