fix duplicated statistic of load count in method cache.

alibaba · Mar 16, 2018 · d70566c · d70566c
1 parent cb428b5
commit d70566c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/jetcache-anno/src/main/java/com/alicp/jetcache/anno/method/CacheHandler.java b/jetcache-anno/src/main/java/com/alicp/jetcache/anno/method/CacheHandler.java
@@ -146,17 +146,16 @@ private static Object invokeWithCached(CacheInvokeContext context)
             return loadAndCount(context, cache, key);
         }
 
-        CacheLoader loader = (k) -> loadAndCount(context, cache, key);
         try {
             Object result = cache.computeIfAbsent(key, (k) -> {
                 try {
-                    return loader.load(k);
+                    return invokeOrigin(context);
                 } catch (Throwable e) {
                     throw new CacheInvokeException(e.getMessage(), e);
                 }
             });
             if (cache instanceof CacheHandlerRefreshCache) {
-                ((CacheHandlerRefreshCache) cache).addOrUpdateRefreshTask(key, loader);
+                ((CacheHandlerRefreshCache) cache).addOrUpdateRefreshTask(key, (unusedKey) -> invokeOrigin(context));
             }
             return result;
         } catch (CacheInvokeException e) {