open-source/MallChat
1
0
Fork 0
mirror of https://github.com/zongzibinbin/MallChat.git synced 2026-03-19 09:53:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

调整项目目录

Browse Source
This commit is contained in:
zengwei-cogiot
2023-09-12 09:44:19 +08:00
parent 0564483102
commit 139cf444ec
30 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

121
mallchat-tools/mallchat-frequency-control/src/main/java/com/abin/frequencycontrol/aspect/FrequencyControlAspect.java Normal file
View File

@@ -0,0 +1,121 @@
package com.abin.frequencycontrol.aspect;
import cn.hutool.core.util.StrUtil;
import com.abin.frequencycontrol.util.RequestHolder;
import com.abin.mallchat.common.FrequencyControlConstant;
import com.abin.mallchat.utils.SpElUtils;
import com.abin.frequencycontrol.annotation.FrequencyControl;
import com.abin.frequencycontrol.domain.dto.FixedWindowDTO;
import com.abin.frequencycontrol.domain.dto.FrequencyControlDTO;
import com.abin.frequencycontrol.domain.dto.SlidingWindowDTO;
import com.abin.frequencycontrol.domain.dto.TokenBucketDTO;
import com.abin.frequencycontrol.service.frequencycontrol.FrequencyControlUtil;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
* 频控实现
*/
@Slf4j
@Aspect
@Component
public class FrequencyControlAspect {
@Around("@annotation(com.abin.frequencycontrol.annotation.FrequencyControl)||@annotation(com.abin.frequencycontrol.annotation.FrequencyControlContainer)")
public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
System.out.println("FrequencyControlAspect start");
Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
FrequencyControl[] annotationsByType = method.getAnnotationsByType(FrequencyControl.class);
Map<String, FrequencyControl> keyMap = new HashMap<>();
String strategy = FrequencyControlConstant.TOTAL_COUNT_WITH_IN_FIX_TIME;
for (int i = 0; i < annotationsByType.length; i++) {
// 获取频控注解
FrequencyControl frequencyControl = annotationsByType[i];
String prefix = StrUtil.isBlank(frequencyControl.prefixKey()) ? /* 默认方法限定名 + 注解排名（可能多个）*/method.toGenericString() + ":index:" + i : frequencyControl.prefixKey();
String key = "";
switch (frequencyControl.target()) {
case EL:
key = SpElUtils.parseSpEl(method, joinPoint.getArgs(), frequencyControl.spEl());
break;
case IP:
key = RequestHolder.get().getIp();
break;
case UID:
key = RequestHolder.get().getUid().toString();
}
keyMap.put(prefix + ":" + key, frequencyControl);
strategy = frequencyControl.strategy();
}
// 将注解的参数转换为编程式调用需要的参数
if (FrequencyControlConstant.TOTAL_COUNT_WITH_IN_FIX_TIME.equals(strategy)) {
// 调用编程式注解 固定窗口
List<FrequencyControlDTO> frequencyControlDTOS = keyMap.entrySet().stream().map(entrySet -> buildFixedWindowDTO(entrySet.getKey(), entrySet.getValue())).collect(Collectors.toList());
return FrequencyControlUtil.executeWithFrequencyControlList(strategy, frequencyControlDTOS, joinPoint::proceed);
} else if (FrequencyControlConstant.TOKEN_BUCKET.equals(strategy)) {
// 调用编程式注解 令牌桶
List<TokenBucketDTO> frequencyControlDTOS = keyMap.entrySet().stream().map(entrySet -> buildTokenBucketDTO(entrySet.getKey(), entrySet.getValue())).collect(Collectors.toList());
return FrequencyControlUtil.executeWithFrequencyControlList(strategy, frequencyControlDTOS, joinPoint::proceed);
} else {
// 调用编程式注解 滑动窗口
List<SlidingWindowDTO> frequencyControlDTOS = keyMap.entrySet().stream().map(entrySet -> buildSlidingWindowFrequencyControlDTO(entrySet.getKey(), entrySet.getValue())).collect(Collectors.toList());
return FrequencyControlUtil.executeWithFrequencyControlList(strategy, frequencyControlDTOS, joinPoint::proceed);
}
}
/**
* 将注解参数转换为编程式调用所需要的参数
*
* @param key 频率控制Key
* @param frequencyControl 注解
* @return 编程式调用所需要的参数-FrequencyControlDTO
*/
private SlidingWindowDTO buildSlidingWindowFrequencyControlDTO(String key, FrequencyControl frequencyControl) {
SlidingWindowDTO frequencyControlDTO = new SlidingWindowDTO();
frequencyControlDTO.setWindowSize(frequencyControl.windowSize());
frequencyControlDTO.setPeriod(frequencyControl.period());
frequencyControlDTO.setCount(frequencyControl.count());
frequencyControlDTO.setUnit(frequencyControl.unit());
frequencyControlDTO.setKey(key);
return frequencyControlDTO;
}
/**
* 将注解参数转换为编程式调用所需要的参数
*
* @param key 频率控制Key
* @param frequencyControl 注解
* @return 编程式调用所需要的参数-FrequencyControlDTO
*/
private TokenBucketDTO buildTokenBucketDTO(String key, FrequencyControl frequencyControl) {
TokenBucketDTO tokenBucketDTO = new TokenBucketDTO(frequencyControl.capacity(), frequencyControl.refillRate());
tokenBucketDTO.setKey(key);
return tokenBucketDTO;
}
/**
* 将注解参数转换为编程式调用所需要的参数
*
* @param key 频率控制Key
* @param frequencyControl 注解
* @return 编程式调用所需要的参数-FrequencyControlDTO
*/
private FixedWindowDTO buildFixedWindowDTO(String key, FrequencyControl frequencyControl) {
FixedWindowDTO fixedWindowDTO = new FixedWindowDTO();
fixedWindowDTO.setCount(frequencyControl.count());
fixedWindowDTO.setTime(frequencyControl.time());
fixedWindowDTO.setUnit(frequencyControl.unit());
fixedWindowDTO.setKey(key);
return fixedWindowDTO;
}
}