Fix Cache bug

CodingGay · Mar 21, 2022 · 33628f5 · 33628f5
1 parent b339d75
commit 33628f5
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/core/src/main/java/top/niunaijun/blackreflection/BlackReflection.java b/core/src/main/java/top/niunaijun/blackreflection/BlackReflection.java
@@ -163,15 +163,17 @@ public Object invoke(Object proxy, Method method, Object[] args) throws Throwabl
                 }
             });
 
-            if (caller == null) {
-                sProxyCache.put(clazz, o);
-            } else {
-                Map<Class<?>, Object> callerClassMap = sCallerProxyCache.get(caller);
-                if (callerClassMap == null) {
-                    callerClassMap = new HashMap<>();
-                    sCallerProxyCache.put(caller, callerClassMap);
+            if (CACHE) {
+                if (caller == null) {
+                    sProxyCache.put(clazz, o);
+                } else {
+                    Map<Class<?>, Object> callerClassMap = sCallerProxyCache.get(caller);
+                    if (callerClassMap == null) {
+                        callerClassMap = new HashMap<>();
+                        sCallerProxyCache.put(caller, callerClassMap);
+                    }
+                    callerClassMap.put(clazz, o);
                 }
-                callerClassMap.put(clazz, o);
             }
             return (T) o;
         } catch (ClassNotFoundException e) {