SpringBoot 接口访问频率限制(二)

package com.stone.study.frequencycontrol; import com.stone.study.common.ErrorCode; import com.stone.study.exception.BusinessException; import org.apache.commons.lang3.ObjectUtils; import javax.annotation.PostConstruct; import java.util.Collections; import java.util.List; import java.util.Map; import java.util.stream.Collectors; public abstract class AbstractFrequencyControlService implements FrequencyControl{ @Override public <T> T executeWithFrequencyControl(FrequencyControlDTO frequencyControl, SupplierThrowWithoutParam<T> supplier) throws Throwable { return executeWithFrequencyControlList(Collections.singletonList(frequencyControl), supplier); } @Override public <T> T executeWithFrequencyControlList(List<FrequencyControlDTO> frequencyControlList, SupplierThrowWithoutParam<T> supplier) throws Throwable { boolean existsFrequencyControlHasNullKey = frequencyControlList.stream().anyMatch(frequencyControl -> ObjectUtils.isEmpty(frequencyControl.getKey())); if(existsFrequencyControlHasNullKey){ throw new BusinessException(ErrorCode.OPERATION_ERROR, "限流策略的Key字段不允许出现空值"); } Map<String, FrequencyControlDTO> frequencyControlDTOMap = frequencyControlList.stream().collect(Collectors.groupingBy(FrequencyControlDTO::getKey, Collectors.collectingAndThen(Collectors.toList(), list -> list.get(0)))); return executeWithFrequencyControlMap((Map<String, FrequencyControlDTO>) frequencyControlDTOMap, supplier); } /** * @param frequencyControlMap 定义的注解频控 Map中的Key-对应redis的单个频控的Key Map中的Value-对应redis的单个频控的Key限制的Value * @param supplier 函数式入参-代表每个频控方法执行的不同的业务逻辑 * @return 业务方法执行的返回值 * @throws Throwable */ private <T> T executeWithFrequencyControlMap(Map<String, FrequencyControlDTO> frequencyControlMap, SupplierThrowWithoutParam<T> supplier) throws Throwable { if (reachRateLimit(frequencyControlMap)) { throw new BusinessException(ErrorCode.OPERATION_ERROR, "操作过于频繁，请稍后再试"); } try { return supplier.get(); } finally { //不管成功还是失败，都增加次数 addFrequencyControlStatisticsCount(frequencyControlMap); } } /** * 是否达到限流阈值 子类实现 每个子类都可以自定义自己的限流逻辑判断 * * @param frequencyControlMap 定义的注解频控 Map中的Key-对应redis的单个频控的Key Map中的Value-对应redis的单个频控的Key限制的Value * @return true-方法被限流 false-方法没有被限流 */ protected abstract boolean reachRateLimit(Map<String, FrequencyControlDTO> frequencyControlMap); /** * 增加限流统计次数 子类实现 每个子类都可以自定义自己的限流统计信息增加的逻辑 * * @param frequencyControlMap 定义的注解频控 Map中的Key-对应redis的单个频控的Key Map中的Value-对应redis的单个频控的Key限制的Value */ protected abstract void addFrequencyControlStatisticsCount(Map<String, FrequencyControlDTO> frequencyControlMap); /** * 注册到工厂中 */ @PostConstruct protected void registerMyselfToFactory() { FrequencyControlStrategyFactory.registerFrequencyController(getStrategyName(), this); } /** * 获取策略名称 * * @return 策略名称 */ protected abstract String getStrategyName(); }