很早之前學習了redis，在這次畢業設計中使用redis對熱點資料進行快取，其中使用spring-data-redis還有@Cacheable、@CachePut和@CacheEvict這3個註解。註解的方式避免了在業務方法中參雜太多非業務的邏輯程式碼，比如日誌和事務，都會用AOP配合註解把共用部分抽離出來，同時業務方法也更純粹簡潔。在使用時發現，spring-data-redis雖是強大，但是許多功能用不上，我只需要簡單的快取新增、刪除功能，於是基於自定義註解和AOP知識實現了3個快取註解。

關鍵點

核心程式碼

AOP切面類

package myCacheAnnotation;

import com.alibaba.fastjson.JSON;
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.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.core.LocalVariableTableParameterNameDiscoverer;
import org.springframework.stereotype.Component;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import service.IUserService;

import java.lang.reflect.Method;

/**
 * writer: holien
 * Time: 2018-01-18 11:20
 * Intent: aop配合cache註解實現快取
 */
@Component
@Aspect
public class CacheAspect {

    @Autowired
    private JedisPool jedisPool;

    // 在使用Cache註解的地方切入此切點
    @Around("@annotation(myCacheAnnotation.Cache)")
    private Object handleCache(ProceedingJoinPoint pjp) throws Throwable {
        System.out.println("以下是快取邏輯");
        // 獲取切入的方法物件
        // 這個m是代理物件的，沒有包含註解
        Method m = ((MethodSignature) pjp.getSignature()).getMethod();
        // this()返回代理物件，target()返回目標物件，目標物件反射獲取的method物件才包含註解
        Method methodWithAnnotations = pjp.getTarget().getClass().getDeclaredMethod(pjp.getSignature().getName(), m.getParameterTypes());
        // 根據目標方法物件獲取註解物件
        Cache cacheAnnotation = methodWithAnnotations.getDeclaredAnnotation(myCacheAnnotation.Cache.class);
        // 解析key
        String keyExpr = cacheAnnotation.key();
        Object[] as = pjp.getArgs();
        String key = parseKey(methodWithAnnotations, as, keyExpr);
        // 註解的屬性本質是註解裡的定義的方法
//        Method methodOfAnnotation = a.getClass().getMethod("key");
        // 註解的值本質是註解裡的定義的方法返回值
//        String key = (String) methodOfAnnotation.invoke(a);
        // 到redis中獲取快取
        Jedis jedis = jedisPool.getResource();
        String cache = jedis.get(key);
        if (cache == null) {
            // 若不存在，則到資料庫中去獲取
            Object result = pjp.proceed();
            // 從資料庫獲取後存入redis
            System.out.println("從資料庫獲取的結果以JsonString形式存入redis中");
            jedis.set(key, JSON.toJSONString(result));
            // 若有指定過期時間，則設定
            int expireTime = cacheAnnotation.expire();
            if (expireTime != -1) {
                jedis.expire(key, expireTime);
            }
            return result;
        } else {
            return JSON.parse(cache);
        }
    }

    // 引數2為方法引數值，引數3為註解中某個屬性的值，若含有#則為一個表示式
    // 待解決，解析#user.id的問題
    private String parseKey(Method method, Object[] argValues, String expr) {
        if (expr.contains("#")) {
            String paramName = expr.substring(expr.indexOf('#') + 1);
            // 獲取方法引數名列表
            LocalVariableTableParameterNameDiscoverer discoverer = new LocalVariableTableParameterNameDiscoverer();
            String[] paramNames = discoverer.getParameterNames(method);
            for (int i = 0; i < paramNames.length; i++) {
                if (paramNames[i].equals(paramName)) {
                    return expr.substring(0, expr.indexOf('#')) + argValues[i].toString();
                }
            }
            throw new IllegalArgumentException("解析不了該引數，錯誤引數表示式");
        } else {
            // 不需要解析，直接返回
            return expr;
        }
    }

    // 待解決，解析#user.id的問題
    @Around("@annotation(myCacheAnnotation.CachePut)")
    private Object handleCachePut(ProceedingJoinPoint pjp) throws Throwable {
        return null;
    }

    @Around("@annotation(myCacheAnnotation.CacheEvict)")
    private Object handleCacheEvict(ProceedingJoinPoint pjp) throws Throwable {
        System.out.println("以下是刪除快取邏輯");
        // 獲取切入的方法物件
        // 這個m是代理物件的，沒有包含註解
        Method m = ((MethodSignature) pjp.getSignature()).getMethod();
        // this()返回代理物件，target()返回目標物件，目標物件反射獲取的method物件才包含註解
        Method methodWithAnnotations = pjp.getTarget().getClass().getDeclaredMethod(pjp.getSignature().getName(), m.getParameterTypes());
        // 根據目標方法物件獲取註解物件
        CacheEvict cacheEvictAnnotation = methodWithAnnotations.getDeclaredAnnotation(myCacheAnnotation.CacheEvict.class);
        // 解析key
        String keyExpr = cacheEvictAnnotation.key();
        Object[] as = pjp.getArgs();
        String key = parseKey(methodWithAnnotations, as, keyExpr);
        // 先刪除資料庫中的使用者資訊再刪除快取
        Object result = pjp.proceed();
        Jedis jedis = jedisPool.getResource();
        jedis.del(key);
        System.out.println("刪除快取中的使用者資訊");
        return result;
    }

    public static void main(String[] args) throws Exception {
        ApplicationContext context = new ClassPathXmlApplicationContext("classpath:applicationContext.xml");
        IUserService userService = (IUserService) context.getBean("userService");
//        System.out.println(userService.getUserInfo(10));
        userService.deleteUserInfo(10);
    }

}
 

Cache註解類

package myCacheAnnotation;

import java.lang.annotation.*;

/**
 * writer: holien
 * Time: 2018-01-18 10:49
 * Intent: 自定義快取註解（String、hash）
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface Cache {
    String key();
    int expire() default -1;
}

不足與補充