From bd8925bf9552734c5012eab1e6fb888eb809082e Mon Sep 17 00:00:00 2001 From: concise <15521608275@163.com> Date: Sun, 30 Aug 2020 16:13:22 +0800 Subject: [PATCH 1/3] =?UTF-8?q?=E5=9F=BA=E4=BA=8E=E6=B3=A8=E8=A7=A3?= =?UTF-8?q?=E7=9A=84=E7=BC=93=E5=AD=98=E8=A3=85=E9=A5=B0=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pom.xml | 12 ++ .../com/github/hcsp/annotation/Cache.java | 4 + .../hcsp/annotation/CacheClassDecorator.java | 114 +++++++++++++++++- .../github/hcsp/annotation/DataService.java | 2 +- 4 files changed, 130 insertions(+), 2 deletions(-) diff --git a/pom.xml b/pom.xml index cf436d9..e321ca2 100644 --- a/pom.xml +++ b/pom.xml @@ -76,6 +76,18 @@ 5.6.0 test + + + net.bytebuddy + byte-buddy + 1.10.14 + + + + net.bytebuddy + byte-buddy + 1.10.14 + 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..bf1c81d 100644 --- a/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java +++ b/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java @@ -1,5 +1,20 @@ package com.github.hcsp.annotation; +import net.bytebuddy.ByteBuddy; +import net.bytebuddy.implementation.MethodDelegation; +import net.bytebuddy.implementation.bind.annotation.AllArguments; +import net.bytebuddy.implementation.bind.annotation.Origin; +import net.bytebuddy.implementation.bind.annotation.RuntimeType; +import net.bytebuddy.implementation.bind.annotation.SuperCall; +import net.bytebuddy.implementation.bind.annotation.This; +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 +22,105 @@ public class CacheClassDecorator { // 这意味着，在短时间内调用同一个服务的同一个@Cache方法两次 // 它实际上只被调用一次，第二次的结果直接从缓存中获取 // 注意，缓存的实现需要是线程安全的 + @SuppressWarnings("unchecked") public static Class decorate(Class klass) { - return klass; + return (Class) new ByteBuddy() + // 对带有cache缓存的方法进行增强 + .subclass(klass) + .method(ElementMatchers.isAnnotatedWith(Cache.class)) + .intercept(MethodDelegation.to(CacheAdvisor.class)) + .make() + .load(klass.getClassLoader()) + .getLoaded(); + + } + + private static class CacheKey { + private Object thisObject; + private String methodName; + private Object[] arguments; + + 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 Object value; + private long time; + + CacheValue(Object value, long time) { + this.value = value; + this.time = time; + } + } + + + public static class CacheAdvisor { + // 第一次执行真正的查询，第二次从换从中获取，那就要先写个map来存储第一次的缓存 + 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); + final CacheValue resultExistInCache = cache.get(cacheKey); + if (resultExistInCache != null) { + // 缓存的结果是什么时候生成的？ + if (cacheExpires(resultExistInCache, method)) { + // 缓存过期 + return invokeRealMethodAndPutIntoCache(superCall, cacheKey); + } else { + return resultExistInCache.value; + } + + } else { + // super.queryData(1) + return invokeRealMethodAndPutIntoCache(superCall, cacheKey); + } + } + + private static Object invokeRealMethodAndPutIntoCache(@SuperCall Callable superCall, CacheKey cacheKey) throws Exception { + Object realMethodInvocationResult = superCall.call(); + cache.put(cacheKey, new CacheValue(realMethodInvocationResult, System.currentTimeMillis())); + return realMethodInvocationResult; + } + + private static boolean cacheExpires(CacheValue cacheValue, Method method) { + long time = cacheValue.time; + int cacheSeconds = method.getAnnotation(Cache.class).cacheSeconds(); + return System.currentTimeMillis() - time > cacheSeconds * 1000; + } } + public static void main(String[] args) throws Exception { DataService dataService = decorate(DataService.class).getConstructor().newInstance(); @@ -18,6 +128,8 @@ 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)); diff --git a/src/main/java/com/github/hcsp/annotation/DataService.java b/src/main/java/com/github/hcsp/annotation/DataService.java index fefb56c..d01da0d 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(); From 424698f6277c3de3293b9c3315530faa0a4b8e87 Mon Sep 17 00:00:00 2001 From: concise <15521608275@163.com> Date: Sun, 30 Aug 2020 16:18:36 +0800 Subject: [PATCH 2/3] =?UTF-8?q?=E5=9F=BA=E4=BA=8E=E6=B3=A8=E8=A7=A3?= =?UTF-8?q?=E7=9A=84=E7=BC=93=E5=AD=98=E8=A3=85=E9=A5=B0=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/github/hcsp/annotation/CacheClassDecorator.java | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java b/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java index bf1c81d..971210b 100644 --- a/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java +++ b/src/main/java/com/github/hcsp/annotation/CacheClassDecorator.java @@ -72,7 +72,7 @@ private static class CacheValue { private Object value; private long time; - CacheValue(Object value, long time) { + CacheValue(Object value, long time) { this.value = value; this.time = time; } @@ -80,29 +80,24 @@ private static class CacheValue { public static class CacheAdvisor { - // 第一次执行真正的查询，第二次从换从中获取，那就要先写个map来存储第一次的缓存 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); final CacheValue resultExistInCache = cache.get(cacheKey); if (resultExistInCache != null) { - // 缓存的结果是什么时候生成的？ if (cacheExpires(resultExistInCache, method)) { - // 缓存过期 return invokeRealMethodAndPutIntoCache(superCall, cacheKey); } else { return resultExistInCache.value; } } else { - // super.queryData(1) return invokeRealMethodAndPutIntoCache(superCall, cacheKey); } } From 485c50acff172dfacb0cd872eb0ab1216bc4523f Mon Sep 17 00:00:00 2001 From: concise <15521608275@163.com> Date: Sun, 30 Aug 2020 16:35:29 +0800 Subject: [PATCH 3/3] =?UTF-8?q?=E5=9F=BA=E4=BA=8E=E6=B3=A8=E8=A7=A3?= =?UTF-8?q?=E7=9A=84=E7=BC=93=E5=AD=98=E8=A3=85=E9=A5=B0=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pom.xml | 64 +++++++++++++++++++++++---------------------------------- 1 file changed, 26 insertions(+), 38 deletions(-) diff --git a/pom.xml b/pom.xml index e321ca2..26b6ec5 100644 --- a/pom.xml +++ b/pom.xml @@ -76,18 +76,6 @@ 5.6.0 test - - - net.bytebuddy - byte-buddy - 1.10.14 - - - - net.bytebuddy - byte-buddy - 1.10.14 - @@ -99,31 +87,31 @@ -Dfile.encoding=UTF-8 - - maven-checkstyle-plugin - 3.1.0 - - ${basedir}/.circleci/checkstyle.xml - true - false - - - - compile - compile - - check - - - - - - com.puppycrawl.tools - checkstyle - 8.29 - - - + + maven-checkstyle-plugin + 3.1.0 + + ${basedir}/.circleci/checkstyle.xml + true + false + + + + compile + compile + + check + + + + + + com.puppycrawl.tools + checkstyle + 8.29 + + + - + \ No newline at end of file