利用AOP自定义Redis缓存注解

2022-05-05 17:19:03 浏览数 (1)

背景

在查询类开发中我们有使用缓存的场景,一般可以使用Redis作为缓存,来缓解数据库如MySQL的压力。使用缓存的步骤为:

“(1)从Redis缓存中获取数据,如果存在数据,直接返回值。 (2)如果不存在,执行数据库的查询方法 (3)将数据库中的值放入缓存 ”

NO CODE NO BB,代码如下

代码语言:javascript复制
//a.从缓存中获取
String value = redisTemplate.opsForValue().get(key);
if (value != null) {
    log.info("从缓存中读取到值:{}", value);
    return value;
}

//b.从数据库中查询
List<Member> members = memberMapper.listByName(name);

//c.同步value到缓存
value = JSONArray.toJSONString(members);
redisTemplate.opsForValue().set(key, value, expireTimes, TimeUnit.SECONDS);
return value;

如上代码,这里有个问题,我们只是要做个查询而已,也就是只要b行的代码,其他代码不是业务代码,不应该由开发人员去操心。那我们何不用注解的形式代替a和c代码呢。

使用SpringBoot的缓存注解

SpringBoot提供了现成可用的缓存注解@Cacheble

配置类开启缓存注解

代码语言:javascript复制
@Configuration
@EnableCaching
public class RedisConfig extends CachingConfigurerSupport {
    @Bean
    public RedisTemplate<Object,Object> redisTemplate(RedisConnectionFactory factory) {
        RedisTemplate<Object, Object> template = new RedisTemplate<>();
        template.setConnectionFactory(factory);
        ...
     }
     ...
         
    @Bean
    public CacheManager cacheManager(RedisConnectionFactory factory) {
        RedisCacheConfiguration config = RedisCacheConfiguration.defaultCacheConfig()
                .entryTtl(Duration.ofSeconds(60))      //缓存过期时间
                .disableCachingNullValues();

        return RedisCacheManager.builder(factory)
                .cacheDefaults(config)
                .transactionAware()
                .build();
    }
 }

使用@Cacheable注解

代码语言:javascript复制
@Cacheable(value = "member",key = "#name")
public List<Member> listByName(String name) {
    return memberMapper.listByName(name);
}

测试

代码语言:javascript复制
@Test
void listByName() {
    String name = "zhouzhou";
    List<Member> members = memberController.listByName(name);
    log.info("members:{}",members);
}

控制台结果

代码语言:javascript复制
members:[Member(id=1805590839001216, name=zhouzhou, code=109, annotationParam=null)]

上面代码发现使用Spring缓存注解的缓存失效时间还要在配置类中进行配置。于是我在想为什么这个失效时间不做成注解的这一项属性呢,这样自定义失效时间就比较方便了。

自定义缓存注解

注解定义

代码语言:javascript复制
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface CustomizeCache {

    String key();

    String value();

    long expireTimes() default 120L; //默认过期时间120s

    int semaphoreCount() default Integer.MAX_VALUE;  //默认限制线程并发数
}

参数说明如下:

key():缓存的key,一般是一个动态的参数值 vaule():缓存的value,value::key为Redis缓存中拼接的KEY expireTimes():缓存失效时间,默认120s semaphoreCount():共享锁,并发下允许访问的线程数,用于保护数据库。默认为Integer.MAX_VALUE

AOP注解开发

首先创建切面类

代码语言:javascript复制
@Component
@Aspect
@Slf4j
public class CacheAspect {
   ...
}

创建横切面,为注解CustomizeCache添加功能。

代码语言:javascript复制
@Pointcut("@annotation(com.lvshen.demo.redis.cache.CustomizeCache)")
    public void cachePointcut() {
    }

开发缓存功能,定义@Around

代码语言:javascript复制
 @Around("cachePointcut()")
    public Object doCache(ProceedingJoinPoint point) {
    ...
    }

获取方法上注解的内容

代码语言:javascript复制
Method method = point.getTarget().getClass().getMethod(signature.getName(), signature.getMethod().getParameterTypes());
CustomizeCache annotation = method.getAnnotation(CustomizeCache.class);
String keyEl = annotation.key();
String prefix = annotation.value();
long expireTimes = annotation.expireTimes();
int semaphoreCount = annotation.semaphoreCount();

解析SpringEL表达式

代码语言:javascript复制
Object[] args = point.getArgs();
DefaultParameterNameDiscoverer discoverer = new DefaultParameterNameDiscoverer();
String[] parameterNames = discoverer.getParameterNames(method);
for (int i = 0; i < parameterNames.length; i  ) {
    context.setVariable(parameterNames[i], args[i].toString());
}

拼接Redis KEY

代码语言:javascript复制
//解析
String key = prefix   "::"   expression.getValue(context).toString();

判断缓存中是否存在

代码语言:javascript复制
value = redisTemplate.opsForValue().get(key);
if (value != null) {
    log.info("从缓存中读取到值:{}", value);
    return value;
}

自定义组件-创建限流令牌

代码语言:javascript复制
semaphore = new Semaphore(semaphoreCount);
boolean tryAcquire = semaphore.tryAcquire(3000L, TimeUnit.MILLISECONDS);
if (!tryAcquire) {
    //log.info("当前线程【{}】获取令牌失败,等待其他线程释放令牌",   Thread.currentThread().getName());
    throw new RuntimeException(String.format("当前线程【%s】获取令牌失败,等带其他线程释放令牌", Thread.currentThread().getName()));
}

如果缓存没有数据,则执行原本方法。

代码语言:javascript复制
 value = point.proceed();

同步value到缓存

代码语言:javascript复制
redisTemplate.opsForValue().set(key, value, expireTimes, TimeUnit.SECONDS);

最后释放令牌

代码语言:javascript复制
} finally {
    if (semaphore == null) {
     return value;
    } else {
     semaphore.release();
    }
}

调用注解

代码语言:javascript复制
@CustomizeCache(value = "member", key = "#name")
public List<Member> listByNameSelfCache(String name) {
 return memberMapper.listByName(name);
}

测试

代码语言:javascript复制
@Test
void testCache() {
    String name = "lvshen99";
    List<Member> members = memberService.listByNameSelfCache(name);
    log.info("members:{}",members);
}

测试结果

代码语言:javascript复制
members:[Member(id=15, name=lvshen99, code=200, annotationParam=null)]

Redis上显示

我们也可以自定义缓存失效时间,如设置1min失效时间

代码语言:javascript复制
 @CustomizeCache(value = "member", key = "#name",expireTimes = 3600)
 public List<Member> listByNameSelfCache(String name) {
  return memberMapper.listByName(name);
 }

失效时间为,如图,显示3585ms是因为截图的时候过了15ms。

源码地址如下:

代码语言:javascript复制
完整源码Github地址:https://github.com/lvshen9/demo/tree/lvshen-dev/src/main/java/com/lvshen/demo/redis/cache

0 人点赞