diff --git a/src/main/java/com/github/hcsp/annotation/Cache.java b/src/main/java/com/github/hcsp/annotation/Cache.java index b8651b5..63febca 100644 --- a/src/main/java/com/github/hcsp/annotation/Cache.java +++ b/src/main/java/com/github/hcsp/annotation/Cache.java @@ -1,5 +1,9 @@ package com.github.hcsp.annotation; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; + +@Retention(RetentionPolicy.RUNTIME) public @interface Cache { // 标记缓存的时长(秒),默认60s int cacheSeconds() default 60; diff --git a/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java b/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java index fb5d531..28c9087 100644 --- a/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java +++ b/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java @@ -1,5 +1,17 @@ package com.github.hcsp.annotation; +import net.bytebuddy.ByteBuddy; +import net.bytebuddy.implementation.MethodDelegation; +import net.bytebuddy.implementation.bind.annotation.*; + +import net.bytebuddy.matcher.ElementMatchers; + +import java.lang.reflect.Method; +import java.util.Arrays; +import java.util.Objects; +import java.util.concurrent.Callable; +import java.util.concurrent.ConcurrentHashMap; + public class CacheClassDecorator { // 将传入的服务类Class进行增强 // 使得返回一个具有如下功能的Class: @@ -7,10 +19,53 @@ public class CacheClassDecorator { // 这意味着,在短时间内调用同一个服务的同一个@Cache方法两次 // 它实际上只被调用一次,第二次的结果直接从缓存中获取 // 注意,缓存的实现需要是线程安全的 - public static Class decorate(Class klass) { - return klass; + public static Class decorate(Class klass) { + return new ByteBuddy() + .subclass(klass) + .method(ElementMatchers.isAnnotatedWith(Cache.class)) + .intercept(MethodDelegation.to(CacheAdvisor.class)) + .make() + .load(klass.getClassLoader()) + .getLoaded(); + } + + public static class CacheAdvisor { + private static ConcurrentHashMap cache = new ConcurrentHashMap<>(); + + @RuntimeType + public static Object cache( + @SuperCall Callable superCall, + @Origin Method method, + @This Object thisObject, + @AllArguments Object[] arguments + ) throws Exception { + CacheKey cacheKey = new CacheKey(thisObject, method.getName(), arguments); + CacheValue getValueFromCache = cache.get(cacheKey); + if (getValueFromCache != null) { + if(isExpiredForCache(getValueFromCache,method)){ + return putInCache(superCall, cacheKey); + } + return getValueFromCache.value; + } else { + return putInCache(superCall, cacheKey); + } + + } + + private static Object putInCache(Callable superCall, CacheKey cacheKey) throws Exception { + long time = System.currentTimeMillis(); + Object result = superCall.call(); + cache.put(cacheKey, new CacheValue(result, time)); + return result; + } + private static boolean isExpiredForCache(CacheValue cacheValue, Method method) { + int cacheSeconds = method.getAnnotation(Cache.class).cacheSeconds(); + return System.currentTimeMillis() - cacheValue.time >= cacheSeconds*1000; + } } + + public static void main(String[] args) throws Exception { DataService dataService = decorate(DataService.class).getConstructor().newInstance(); @@ -18,10 +73,51 @@ public static void main(String[] args) throws Exception { System.out.println(dataService.queryData(1)); Thread.sleep(1 * 1000); System.out.println(dataService.queryData(1)); + Thread.sleep(3 * 1000); + System.out.println(dataService.queryData(1)); // 无缓存的查询:两次都执行了真正的查询操作 System.out.println(dataService.queryDataWithoutCache(1)); Thread.sleep(1 * 1000); System.out.println(dataService.queryDataWithoutCache(1)); } + + private static class CacheKey { + private final Object thisObject; + private final String methodName; + private final Object[] arguments; + + public CacheKey(Object thisObject, String methodName, Object[] arguments) { + this.thisObject = thisObject; + this.methodName = methodName; + this.arguments = arguments; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + CacheKey cacheKey = (CacheKey) o; + return Objects.equals(thisObject, cacheKey.thisObject) && + Objects.equals(methodName, cacheKey.methodName) && + Arrays.equals(arguments, cacheKey.arguments); + } + + @Override + public int hashCode() { + int result = Objects.hash(thisObject, methodName); + result = 31 * result + Arrays.hashCode(arguments); + return result; + } + } + + private static class CacheValue { + private final Object value; + private final long time; + + public CacheValue(Object value, long time) { + this.value = value; + this.time = time; + } + } } diff --git a/src/main/java/com/github/hcsp/annotation/DataService.java b/src/main/java/com/github/hcsp/annotation/DataService.java index fefb56c..8765de3 100644 --- a/src/main/java/com/github/hcsp/annotation/DataService.java +++ b/src/main/java/com/github/hcsp/annotation/DataService.java @@ -12,7 +12,7 @@ public class DataService { * @param id 数据ID * @return 查询到的数据列表 */ - @Cache + @Cache(cacheSeconds=2) public List queryData(int id) { // 模拟一个查询操作 Random random = new Random();