|
@@ -1,7 +1,7 @@
|
|
|
package cn.iocoder.yudao.framework.redis.core;
|
|
|
|
|
|
+import cn.hutool.core.util.NumberUtil;
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
-import org.springframework.boot.convert.DurationStyle;
|
|
|
import org.springframework.cache.annotation.Cacheable;
|
|
|
import org.springframework.data.redis.cache.RedisCache;
|
|
|
import org.springframework.data.redis.cache.RedisCacheConfiguration;
|
|
@@ -9,12 +9,12 @@ import org.springframework.data.redis.cache.RedisCacheManager;
|
|
|
import org.springframework.data.redis.cache.RedisCacheWriter;
|
|
|
|
|
|
import java.time.Duration;
|
|
|
-import java.time.temporal.ChronoUnit;
|
|
|
|
|
|
/**
|
|
|
* 支持自定义过期时间的 {@link RedisCacheManager} 实现类
|
|
|
*
|
|
|
- * 在 {@link Cacheable#cacheNames()} 格式为 "key#ttl" 时,# 后面的 ttl 为过期时间,单位为秒
|
|
|
+ * 在 {@link Cacheable#cacheNames()} 格式为 "key#ttl" 时,# 后面的 ttl 为过期时间。
|
|
|
+ * 单位为最后一个字母(支持的单位有:d 天,h 小时,m 分钟,s 秒),默认单位为 s 秒
|
|
|
*
|
|
|
* @author 芋道源码
|
|
|
*/
|
|
@@ -42,10 +42,42 @@ public class TimeoutRedisCacheManager extends RedisCacheManager {
|
|
|
// 移除 # 后面的 : 以及后面的内容,避免影响解析
|
|
|
names[1] = StrUtil.subBefore(names[1], StrUtil.COLON, false);
|
|
|
// 解析时间
|
|
|
- Duration duration = DurationStyle.detectAndParse(names[1], ChronoUnit.SECONDS);
|
|
|
+ Duration duration = parseDuration(names[1]);
|
|
|
cacheConfig = cacheConfig.entryTtl(duration);
|
|
|
}
|
|
|
return super.createRedisCache(names[0], cacheConfig);
|
|
|
}
|
|
|
|
|
|
+ /**
|
|
|
+ * 解析过期时间 Duration
|
|
|
+ *
|
|
|
+ * @param ttlStr 过期时间字符串
|
|
|
+ * @return 过期时间 Duration
|
|
|
+ */
|
|
|
+ private Duration parseDuration(String ttlStr) {
|
|
|
+ String timeUnit = StrUtil.subSuf(ttlStr, -1);
|
|
|
+ switch (timeUnit) {
|
|
|
+ case "d":
|
|
|
+ return Duration.ofDays(removeDurationSuffix(ttlStr));
|
|
|
+ case "h":
|
|
|
+ return Duration.ofHours(removeDurationSuffix(ttlStr));
|
|
|
+ case "m":
|
|
|
+ return Duration.ofMinutes(removeDurationSuffix(ttlStr));
|
|
|
+ case "s":
|
|
|
+ return Duration.ofSeconds(removeDurationSuffix(ttlStr));
|
|
|
+ default:
|
|
|
+ return Duration.ofSeconds(Long.parseLong(ttlStr));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 移除多余的后缀,返回具体的时间
|
|
|
+ *
|
|
|
+ * @param ttlStr 过期时间字符串
|
|
|
+ * @return 时间
|
|
|
+ */
|
|
|
+ private Long removeDurationSuffix(String ttlStr) {
|
|
|
+ return NumberUtil.parseLong(StrUtil.sub(ttlStr, 0, ttlStr.length() - 1));
|
|
|
+ }
|
|
|
+
|
|
|
}
|