com.tencent.polaris.plugins.circuitbreaker.composite.CircuitBreakerRuleDictionary Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of circuitbreaker-composite Show documentation
Polaris Plugins Circuitbreaker With Rule JAR
There is a newer version: 2.0.0.0-RC1
Show newest version
/*
 * Tencent is pleased to support the open source community by making Polaris available.
 *
 * Copyright (C) 2019 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.plugins.circuitbreaker.composite;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.regex.Pattern;

import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import com.tencent.polaris.api.plugin.circuitbreaker.entity.Resource;
import com.tencent.polaris.api.pojo.ServiceKey;
import com.tencent.polaris.api.pojo.ServiceRule;
import com.tencent.polaris.api.utils.CollectionUtils;
import com.tencent.polaris.plugins.circuitbreaker.composite.utils.CircuitBreakerUtils;
import com.tencent.polaris.plugins.circuitbreaker.composite.utils.MatchUtils;
import com.tencent.polaris.specification.api.v1.fault.tolerance.CircuitBreakerProto;

import static com.tencent.polaris.plugins.circuitbreaker.composite.utils.MatchUtils.matchMethod;

public class CircuitBreakerRuleDictionary {

	private final Map>> allRules = new HashMap<>();

	private final Function regexToPattern;

	private final Object updateLock = new Object();

	public CircuitBreakerRuleDictionary(Function regexToPattern) {
		this.regexToPattern = regexToPattern;
		allRules.put(CircuitBreakerProto.Level.SERVICE, CacheBuilder.newBuilder().build());
		allRules.put(CircuitBreakerProto.Level.METHOD, CacheBuilder.newBuilder().build());
		allRules.put(CircuitBreakerProto.Level.INSTANCE, CacheBuilder.newBuilder().build());
	}

	public CircuitBreakerProto.CircuitBreakerRule lookupCircuitBreakerRule(Resource resource) {
		synchronized (updateLock) {
			Cache> serviceKeyListCache = allRules.get(resource.getLevel());
			if (null == serviceKeyListCache) {
				return null;
			}
			return selectRule(resource, serviceKeyListCache.getIfPresent(resource.getService()), regexToPattern);
		}
	}

	private static CircuitBreakerProto.CircuitBreakerRule selectRule(Resource resource,
			List sortedRules, Function regexToPattern) {
		if (CollectionUtils.isEmpty(sortedRules)) {
			return null;
		}
		for (CircuitBreakerProto.CircuitBreakerRule cbRule : sortedRules) {
			CircuitBreakerProto.RuleMatcher ruleMatcher = cbRule.getRuleMatcher();
			CircuitBreakerProto.RuleMatcher.DestinationService destination = ruleMatcher.getDestination();
			if (!MatchUtils.matchService(resource.getService(), destination.getNamespace(), destination.getService())) {
				continue;
			}
			CircuitBreakerProto.RuleMatcher.SourceService source = ruleMatcher.getSource();
			if (!MatchUtils.matchService(resource.getCallerService(), source.getNamespace(), source.getService())) {
				continue;
			}
			boolean methodMatched = matchMethod(resource, destination.getMethod(), regexToPattern);
			if (methodMatched) {
				return cbRule;
			}
		}
		return null;
	}

	/**
	 * rule on server has been changed, clear all caches to make it pull again
	 * @param serviceKey target service
	 */
	public void onServiceChanged(ServiceKey serviceKey) {
		synchronized (updateLock) {
			clearRules(serviceKey);
		}
	}

	private void clearRules(ServiceKey serviceKey) {
		allRules.get(CircuitBreakerProto.Level.SERVICE).invalidate(serviceKey);
		allRules.get(CircuitBreakerProto.Level.METHOD).invalidate(serviceKey);
		allRules.get(CircuitBreakerProto.Level.INSTANCE).invalidate(serviceKey);
	}

	public void putServiceRule(ServiceKey serviceKey, ServiceRule serviceRule) {
		synchronized (updateLock) {
			if (null == serviceRule || null == serviceRule.getRule()) {
				clearRules(serviceKey);
				return;
			}
			CircuitBreakerProto.CircuitBreaker circuitBreaker = (CircuitBreakerProto.CircuitBreaker) serviceRule.getRule();
			List rules = circuitBreaker.getRulesList();
			List subServiceRules = new ArrayList<>();
			List subMethodRules = new ArrayList<>();
			List subInstanceRules = new ArrayList<>();
			for (CircuitBreakerProto.CircuitBreakerRule rule : rules) {
				if (!rule.getEnable() || !CircuitBreakerUtils.checkRule(rule)) {
					continue;
				}
				CircuitBreakerProto.Level level = rule.getLevel();
				switch (level) {
				case SERVICE:
					subServiceRules.add(rule);
					break;
				case INSTANCE:
					subInstanceRules.add(rule);
					break;
				case METHOD:
					subMethodRules.add(rule);
					break;
				}
			}
			subServiceRules = sortCircuitBreakerRules(subServiceRules);
			subMethodRules = sortCircuitBreakerRules(subMethodRules);
			subInstanceRules = sortCircuitBreakerRules(subInstanceRules);
			allRules.get(CircuitBreakerProto.Level.SERVICE).put(serviceKey, subServiceRules);
			allRules.get(CircuitBreakerProto.Level.METHOD).put(serviceKey, subMethodRules);
			allRules.get(CircuitBreakerProto.Level.INSTANCE).put(serviceKey, subInstanceRules);
		}
	}

	private static List sortCircuitBreakerRules(List rules) {
		if (CollectionUtils.isEmpty(rules)) {
			return rules;
		}
		List outRules = new ArrayList<>(rules);
		outRules.sort(new Comparator() {
			@Override
			public int compare(CircuitBreakerProto.CircuitBreakerRule rule1, CircuitBreakerProto.CircuitBreakerRule rule2) {
				// 1. compare destination service
				CircuitBreakerProto.RuleMatcher ruleMatcher1 = rule1.getRuleMatcher();
				String destNamespace1 = ruleMatcher1.getDestination().getNamespace();
				String destService1 = ruleMatcher1.getDestination().getService();
				String destMethod1 = ruleMatcher1.getDestination().getMethod().getValue().getValue();

				CircuitBreakerProto.RuleMatcher ruleMatcher2 = rule2.getRuleMatcher();
				String destNamespace2 = ruleMatcher2.getDestination().getNamespace();
				String destService2 = ruleMatcher2.getDestination().getService();
				String destMethod2 = ruleMatcher2.getDestination().getMethod().getValue().getValue();

				int svcResult = MatchUtils.compareService(destNamespace1, destService1, destNamespace2, destService2);
				if (svcResult != 0) {
					return svcResult;
				}
				if (rule1.getLevel() == CircuitBreakerProto.Level.METHOD && rule1.getLevel() == rule2.getLevel()) {
					int methodResult = MatchUtils.compareSingleValue(destMethod1, destMethod2);
					if (methodResult != 0) {
						return methodResult;
					}
				}
				// 2. compare source service
				String srcNamespace1 = ruleMatcher1.getSource().getNamespace();
				String srcService1 = ruleMatcher1.getSource().getService();
				String srcNamespace2 = ruleMatcher2.getSource().getNamespace();
				String srcService2 = ruleMatcher2.getSource().getService();
				return MatchUtils.compareService(srcNamespace1, srcService1, srcNamespace2, srcService2);
			}
		});
		return outRules;
	}

}