一、注解
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.util.concurrent.TimeUnit;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface RedisCache {
String key();
String prefix();
int expireTime() default 36000;
TimeUnit timeunit() default TimeUnit.SECONDS;
}
二、注解切面类
import com.dewav.tms.framework.aspectj.lang.annotation.RedisCache;
import com.dewav.tms.framework.redis.RedisService;
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.context.expression.MethodBasedEvaluationContext;
import org.springframework.core.DefaultParameterNameDiscoverer;
import org.springframework.expression.EvaluationContext;
import org.springframework.expression.Expression;
import org.springframework.expression.spel.standard.SpelExpressionParser;
import org.springframework.stereotype.Component;
import org.springframework.util.Assert;
import javax.annotation.Resource;
import java.lang.reflect.Method;
import java.util.concurrent.TimeUnit;
@Component
@Aspect
@Slf4j
public class RedisCacheAspect {
private final SpelExpressionParser spelExpressionParser = new SpelExpressionParser();
private final DefaultParameterNameDiscoverer parameterNameDiscoverer = new DefaultParameterNameDiscoverer();
@Resource
private RedisService redisService;
@Around("@annotation(redisCache)")
public Object queryByRedisCache(ProceedingJoinPoint pjp, RedisCache redisCache) {
int expireTime = redisCache.expireTime();
String key = redisCache.key();
String prefix = redisCache.prefix();
Assert.hasText(key, "@RedisCache key不能为空!");
Assert.hasText(prefix, "@RedisCache prefix不能为空!");
String evaluateExpression = evaluateExpression(key, pjp);
String redisKey = prefix + evaluateExpression;
Object obj = null;
try {
if (null != redisKey){
if (redisService.hasKey(redisKey)){
obj = redisService.getCacheObject(redisKey);
return obj;
}
}
} catch (Exception e) {
log.error("从Redis获取"+redisKey+"失败:"+e.getMessage());
}
try{
obj = pjp.proceed();
}catch(Throwable e){
log.error("执行异常", e);
throw new RuntimeException(e.getMessage());
}
if (null != redisKey){
redisService.setCacheObject(redisKey, obj, expireTime, TimeUnit.SECONDS);
}
return obj;
}
private String evaluateExpression(String expression, ProceedingJoinPoint point) {
Object target = point.getTarget();
Object[] args = point.getArgs();
MethodSignature methodSignature = (MethodSignature) point.getSignature();
Method method = methodSignature.getMethod();
EvaluationContext context = new MethodBasedEvaluationContext(target, method, args, parameterNameDiscoverer);
Expression exp = spelExpressionParser.parseExpression(expression);
return exp.getValue(context, String.class);
}
}
三、使用样例
@Override
@RedisCache(key = "{#customerId + ':' + #dictType}", prefix = CacheConstants.SYS_DICT_KEY)
public List<SysDictData> selectDictDataByType(Long customerId, String dictType) {
return dictDataMapper.selectDictDataByType(customerId, dictType);
}
@Override
@RedisCache(key = "#apiKey", prefix = CacheConstants.OPEN_KEY + "apiInfo:apiKey:")
public EdiApiInfo getByApiKey(String apiKey) {
return baseMapper.selectOne(new QueryWrapper<EdiApiInfo>()
.eq("api_key", apiKey)
.eq("status", true)
.eq("del_flag", false));
}