From 4411a325222dd52a438e5e47d92b87f81b661fb7 Mon Sep 17 00:00:00 2001
From: redkale <redkale@qq.com>
Date: Thu, 13 Jun 2024 15:23:12 +0800
Subject: [PATCH] CachedActionFunc

---
 docs/cached.md                                | 22 +++++++++++
 .../org/redkale/cached/CachedManager.java     | 19 +++++++++
 .../org/redkale/cached/spi/CachedAction.java  | 29 ++++++++++++--
 .../redkale/cached/spi/CachedActionFunc.java  | 17 ++++++++
 .../cached/spi/CachedAsmMethodBoost.java      |  6 +--
 .../cached/spi/CachedManagerService.java      | 39 +++++++++++++++++--
 .../redkale/net/http/RestDeleteMapping.java   |  2 +-
 .../org/redkale/net/http/RestGetMapping.java  |  2 +-
 .../redkale/net/http/RestPatchMapping.java    |  2 +-
 .../org/redkale/net/http/RestPostMapping.java |  2 +-
 .../org/redkale/net/http/RestPutMapping.java  |  2 +-
 11 files changed, 127 insertions(+), 15 deletions(-)
 create mode 100644 src/main/java/org/redkale/cached/spi/CachedActionFunc.java

diff --git a/docs/cached.md b/docs/cached.md
index 7c24a646b..882ad87c8 100644
--- a/docs/cached.md
+++ b/docs/cached.md
@@ -51,6 +51,28 @@
     }
 ```
 
+## 动态调整缓存时长
+&emsp;&emsp;使用```@Resource```注入多个```CachedManager```
+```java
+
+    @Resource
+    private CachedManager cachedManager;
+
+    @Cached(key = "#{code}_#{map.id}", remoteExpire = "60", timeUnit = TimeUnit.MILLISECONDS)
+    public String getName(String code, Map<String, Long> map) {
+        return code + "-" + map;
+    }
+
+    public void updateExpire() {
+        Duration expire = Duration.ofMillis(600);
+        cachedManager.acceptCachedAction("#{code}_#{map.id}", action -> {
+            //将缓存时长改成600毫秒，并开启本地缓存
+            action.setLocalExpire(expire);
+            action.setRemoteExpire(expire);
+        });
+    }
+```
+
 ## 缓存配置
 ```xml
     <!--
diff --git a/src/main/java/org/redkale/cached/CachedManager.java b/src/main/java/org/redkale/cached/CachedManager.java
index 5e573b65b..9df3358f9 100644
--- a/src/main/java/org/redkale/cached/CachedManager.java
+++ b/src/main/java/org/redkale/cached/CachedManager.java
@@ -5,7 +5,10 @@ package org.redkale.cached;
 
 import java.lang.reflect.Type;
 import java.time.Duration;
+import java.util.List;
 import java.util.concurrent.CompletableFuture;
+import java.util.function.Consumer;
+import org.redkale.cached.spi.CachedAction;
 import org.redkale.inject.Resourcable;
 import org.redkale.util.ThrowSupplier;
 
@@ -58,6 +61,22 @@ public interface CachedManager extends Resourcable {
         }
     }
 
+    /**
+     * 获取{@link org.redkale.cached.spi.CachedAction}集合
+     *
+     * @return CachedAction集合
+     */
+    public List<CachedAction> getCachedActions();
+
+    /**
+     * 处理指定缓存key的{@link org.redkale.cached.spi.CachedAction}<br>
+     * 可用于动态调整缓存时长
+     *
+     * @param templetKey 缓存key
+     * @param consumer 处理函数
+     */
+    public void acceptCachedAction(String templetKey, Consumer<CachedAction> consumer);
+
     // -------------------------------------- 本地缓存 --------------------------------------
     /**
      * 本地获取缓存数据, 过期返回null
diff --git a/src/main/java/org/redkale/cached/spi/CachedAction.java b/src/main/java/org/redkale/cached/spi/CachedAction.java
index 02e816b1d..7c68a3d9f 100644
--- a/src/main/java/org/redkale/cached/spi/CachedAction.java
+++ b/src/main/java/org/redkale/cached/spi/CachedAction.java
@@ -63,7 +63,7 @@ public class CachedAction {
     private final String[] paramNames;
 
     // 模板key
-    String templetKey;
+    final String templetKey;
 
     // 缓存key生成器
     private CachedKeyGenerator keyGenerator;
@@ -91,10 +91,9 @@ public class CachedAction {
         this.resultType = this.async ? ((ParameterizedType) returnType).getActualTypeArguments()[0] : returnType;
     }
 
-    void init(ResourceFactory resourceFactory, Object service) {
+    String init(ResourceFactory resourceFactory, Object service) {
         this.manager = resourceFactory.load(cached.getManager(), CachedManager.class);
-        String key = environment.getPropertyValue(cached.getKey());
-        this.templetKey = key;
+        final String key = environment.getPropertyValue(cached.getKey());
         if (key.startsWith("@")) { // 动态加载缓存key生成器
             String generatorName = key.substring(1);
             this.keyGenerator = resourceFactory.findChild(generatorName, CachedKeyGenerator.class);
@@ -104,6 +103,8 @@ public class CachedAction {
         }
         this.localExpire = createDuration(cached.getLocalExpire());
         this.remoteExpire = createDuration(cached.getRemoteExpire());
+        ((CachedActionFunc) this.manager).addAction(this);
+        return key;
     }
 
     @ClassDepends
@@ -160,6 +161,26 @@ public class CachedAction {
         return method;
     }
 
+    public String getTempletKey() {
+        return templetKey;
+    }
+
+    public Duration getLocalExpire() {
+        return localExpire;
+    }
+
+    public void setLocalExpire(Duration localExpire) {
+        this.localExpire = localExpire;
+    }
+
+    public Duration getRemoteExpire() {
+        return remoteExpire;
+    }
+
+    public void setRemoteExpire(Duration remoteExpire) {
+        this.remoteExpire = remoteExpire;
+    }
+
     @Override
     public String toString() {
         return "{"
diff --git a/src/main/java/org/redkale/cached/spi/CachedActionFunc.java b/src/main/java/org/redkale/cached/spi/CachedActionFunc.java
new file mode 100644
index 000000000..e0e70f31a
--- /dev/null
+++ b/src/main/java/org/redkale/cached/spi/CachedActionFunc.java
@@ -0,0 +1,17 @@
+/*
+
+*/
+
+package org.redkale.cached.spi;
+
+/**
+ * 增加 {@link org.redkale.cached.spi.CachedAction}， {@link org.redkale.cached.CachedManager}的实现类也必须实现本接口
+ *
+ * <p>详情见: https://redkale.org
+ *
+ * @author zhangjx
+ * @since 2.8.0
+ */
+public interface CachedActionFunc {
+    void addAction(CachedAction action);
+}
diff --git a/src/main/java/org/redkale/cached/spi/CachedAsmMethodBoost.java b/src/main/java/org/redkale/cached/spi/CachedAsmMethodBoost.java
index 94dc93479..e3ad958fa 100644
--- a/src/main/java/org/redkale/cached/spi/CachedAsmMethodBoost.java
+++ b/src/main/java/org/redkale/cached/spi/CachedAsmMethodBoost.java
@@ -237,9 +237,9 @@ public class CachedAsmMethodBoost extends AsmMethodBoost {
         actionMap.forEach((field, action) -> {
             try {
                 resourceFactory.inject(action);
-                action.init(resourceFactory, service);
-                if (action.templetKey.indexOf('@') < 0
-                        && action.templetKey.indexOf('{') < 0
+                final String tkey = action.init(resourceFactory, service);
+                if (tkey.indexOf('@') < 0
+                        && tkey.indexOf('{') < 0
                         && action.getMethod().getParameterCount() > 0) {
                     // 一般有参数的方法，Cached.key应该是动态的
                     logger.log(
diff --git a/src/main/java/org/redkale/cached/spi/CachedManagerService.java b/src/main/java/org/redkale/cached/spi/CachedManagerService.java
index 47b2552a9..a18fdfa9f 100644
--- a/src/main/java/org/redkale/cached/spi/CachedManagerService.java
+++ b/src/main/java/org/redkale/cached/spi/CachedManagerService.java
@@ -5,9 +5,13 @@ package org.redkale.cached.spi;
 
 import java.lang.reflect.Type;
 import java.time.Duration;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.Objects;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.CopyOnWriteArrayList;
+import java.util.function.Consumer;
 import java.util.function.Function;
 import java.util.logging.Level;
 import java.util.logging.Logger;
@@ -40,7 +44,7 @@ import org.redkale.util.TypeToken;
 @Component
 @AutoLoad(false)
 @ResourceType(CachedManager.class)
-public class CachedManagerService implements CachedManager, Service {
+public class CachedManagerService implements CachedManager, CachedActionFunc, Service {
 
     public static final String CACHED_CHANNEL_TOPIC_PREFIX = "cached-update-channel";
 
@@ -75,6 +79,8 @@ public class CachedManagerService implements CachedManager, Service {
     // 缓存无效时使用的异步锁
     private final ConcurrentHashMap<String, CachedAsyncLock> asyncLockMap = new ConcurrentHashMap<>();
 
+    protected final List<CachedAction> actions = new CopyOnWriteArrayList<>();
+
     @Resource(required = false)
     protected Application application;
 
@@ -173,10 +179,37 @@ public class CachedManagerService implements CachedManager, Service {
         }
     }
 
+    @Override
+    public void addAction(CachedAction action) {
+        actions.add(action);
+    }
+
+    /**
+     * 获取{@link org.redkale.cached.spi.CachedAction}集合
+     *
+     * @return CachedAction集合
+     */
+    @Override
+    public List<CachedAction> getCachedActions() {
+        return new ArrayList<>(actions);
+    }
+
+    /**
+     * 处理指定缓存key的{@link org.redkale.cached.spi.CachedAction}
+     *
+     * @param templetKey 缓存key
+     * @param consumer 处理函数
+     */
+    @Override
+    public void acceptCachedAction(String templetKey, Consumer<CachedAction> consumer) {
+        actions.stream()
+                .filter(v -> Objects.equals(v.getTempletKey(), templetKey))
+                .forEach(consumer);
+    }
+
     @Override
     public String toString() {
-        return getClass().getSimpleName() + "_" + Objects.hash(this) + "{name='" + name + "', schema='" + schema
-                + "'}";
+        return getClass().getSimpleName() + "_" + Objects.hash(this) + "{name='" + name + "', schema='" + schema + "'}";
     }
 
     // -------------------------------------- 本地缓存 --------------------------------------
diff --git a/src/main/java/org/redkale/net/http/RestDeleteMapping.java b/src/main/java/org/redkale/net/http/RestDeleteMapping.java
index e6a25dc16..0eb5f6371 100644
--- a/src/main/java/org/redkale/net/http/RestDeleteMapping.java
+++ b/src/main/java/org/redkale/net/http/RestDeleteMapping.java
@@ -17,7 +17,7 @@ import java.lang.annotation.Target;
  * <p>详情见: https://redkale.org
  *
  * @since 2.8.0
- * @org.redkale.net.http.RestMapping
+ * @see org.redkale.net.http.RestMapping
  * @see org.redkale.net.http.RestService
  * @author zhangjx
  */
diff --git a/src/main/java/org/redkale/net/http/RestGetMapping.java b/src/main/java/org/redkale/net/http/RestGetMapping.java
index 288335a48..3ca1cc3e5 100644
--- a/src/main/java/org/redkale/net/http/RestGetMapping.java
+++ b/src/main/java/org/redkale/net/http/RestGetMapping.java
@@ -17,7 +17,7 @@ import java.lang.annotation.Target;
  * <p>详情见: https://redkale.org
  *
  * @since 2.8.0
- * @org.redkale.net.http.RestMapping
+ * @see org.redkale.net.http.RestMapping
  * @see org.redkale.net.http.RestService
  * @author zhangjx
  */
diff --git a/src/main/java/org/redkale/net/http/RestPatchMapping.java b/src/main/java/org/redkale/net/http/RestPatchMapping.java
index 7be4a08e5..afcd1cbbc 100644
--- a/src/main/java/org/redkale/net/http/RestPatchMapping.java
+++ b/src/main/java/org/redkale/net/http/RestPatchMapping.java
@@ -17,7 +17,7 @@ import java.lang.annotation.Target;
  * <p>详情见: https://redkale.org
  *
  * @since 2.8.0
- * @org.redkale.net.http.RestMapping
+ * @see org.redkale.net.http.RestMapping
  * @see org.redkale.net.http.RestService
  * @author zhangjx
  */
diff --git a/src/main/java/org/redkale/net/http/RestPostMapping.java b/src/main/java/org/redkale/net/http/RestPostMapping.java
index 9f18c60be..de1c4e492 100644
--- a/src/main/java/org/redkale/net/http/RestPostMapping.java
+++ b/src/main/java/org/redkale/net/http/RestPostMapping.java
@@ -17,7 +17,7 @@ import java.lang.annotation.Target;
  * <p>详情见: https://redkale.org
  *
  * @since 2.8.0
- * @org.redkale.net.http.RestMapping
+ * @see org.redkale.net.http.RestMapping
  * @see org.redkale.net.http.RestService
  * @author zhangjx
  */
diff --git a/src/main/java/org/redkale/net/http/RestPutMapping.java b/src/main/java/org/redkale/net/http/RestPutMapping.java
index 747d4868f..dbe5ba3a7 100644
--- a/src/main/java/org/redkale/net/http/RestPutMapping.java
+++ b/src/main/java/org/redkale/net/http/RestPutMapping.java
@@ -17,7 +17,7 @@ import java.lang.annotation.Target;
  * <p>详情见: https://redkale.org
  *
  * @since 2.8.0
- * @org.redkale.net.http.RestMapping
+ * @see org.redkale.net.http.RestMapping
  * @see org.redkale.net.http.RestService
  * @author zhangjx
  */