From bbe0a2fd03063316e50cf141986bda984599bbda Mon Sep 17 00:00:00 2001
From: 935090232@qq.com <ak473600000>
Date: Tue, 22 Feb 2022 23:41:42 +0800
Subject: [PATCH] Merge branch 'developer'
---
zq-erp/src/main/java/com/matrix/system/common/init/LocalCache.java | 196 ++++++++++++++++++++++++++++++++++++++++++-------
1 files changed, 168 insertions(+), 28 deletions(-)
diff --git a/zq-erp/src/main/java/com/matrix/system/common/init/LocalCache.java b/zq-erp/src/main/java/com/matrix/system/common/init/LocalCache.java
index 26b3098..ed624c2 100644
--- a/zq-erp/src/main/java/com/matrix/system/common/init/LocalCache.java
+++ b/zq-erp/src/main/java/com/matrix/system/common/init/LocalCache.java
@@ -1,73 +1,213 @@
package com.matrix.system.common.init;
+import com.google.common.util.concurrent.ThreadFactoryBuilder;
import com.matrix.core.tools.LogUtil;
+import com.matrix.core.tools.StringUtils;
+import lombok.Data;
-import java.util.Date;
-import java.util.concurrent.ConcurrentHashMap;
-import java.util.concurrent.ConcurrentMap;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.Objects;
+import java.util.Set;
+import java.util.concurrent.*;
+import java.util.concurrent.atomic.AtomicLong;
+import java.util.stream.Collectors;
public class LocalCache {
+ /*
+ * 清理线程运行状态 0 未启动，1 已启动
+ */
+ private static int CLEAR_THREAD_STATUS = 0;
- private static ConcurrentMap<String,Value> localCache=new ConcurrentHashMap(60);
+ private static ConcurrentMap<String, Value> localCache = new ConcurrentHashMap(60);
/**
- * 获取本地缓存
+ * 根据key匹配多个缓存值
+ *
* @param key
* @param <T>
* @return
*/
- public static <T> T get(String key){
- return (T)localCache.get(key);
+ public static <T> Map<String, T> getValues(String key) {
+ return localCache.entrySet().stream()
+ .filter(item -> StringUtils.isMatch(key, item.getKey()))
+ .map(Map.Entry::getValue)
+ .filter(item -> Objects.nonNull(item.value))
+ .collect(Collectors.toMap(Value::getKey, item -> (T) item.value));
+ }
+
+ /**
+ * 获取本地缓存
+ *
+ * @param key
+ * @param <T>
+ * @return
+ */
+ public static <T> T get(String key) {
+ Value value = localCache.get(key);
+ if (Objects.nonNull(value)) {
+ return (T) value.value;
+ }
+ return null;
+ }
+
+ /**
+ * 删除缓存
+ *
+ * @param key
+ * @param <T>
+ * @return
+ */
+ public static <T> T remove(String key) {
+ Value value = localCache.remove(key);
+ if (Objects.nonNull(value)) {
+ return (T) value.value;
+ }
+ return null;
+ }
+
+ /**
+ * 批量删除缓存
+ *
+ * @param key
+ * @return
+ */
+ public static int batchRemove(String key) {
+ int count = 0;
+ Set<Map.Entry<String, Value>> entries = localCache.entrySet();
+ Iterator<Map.Entry<String, Value>> iterator = entries.iterator();
+ while (iterator.hasNext()) {
+ Map.Entry<String, Value> next = iterator.next();
+ if (StringUtils.isMatch(key, next.getKey())) {
+ remove(next.getKey());
+ count++;
+ }
+ }
+ return count;
}
/**
* 保存一个本地缓存
+ *
* @param key
* @param value
*/
- public static void save(String key,Object value){
- if(null!=localCache.put(key,value)){
- LogUtil.debug("覆盖原有缓存{}",key);
+ public static void save(String key, Object value) {
+ if (null != localCache.put(key, buildValue(key, value))) {
+ LogUtil.debug("覆盖原有缓存{}", key);
}
}
- public static void save(String key,Object value,long timeOut){
- if(null!=localCache.put(key,value)){
- LogUtil.debug("覆盖原有缓存{}",key);
+ /**
+ * 设置含过期时间的缓存
+ *
+ * @param key
+ * @param value
+ * @param timeOut 毫秒
+ */
+ public static void save(String key, Object value, long timeOut) {
+ if (null != localCache.put(key, buildValue(key, value, timeOut))) {
+ LogUtil.debug("覆盖原有缓存{}", key);
+ }
+ startClearThread();
+ }
+
+ /**
+ * 重置缓存失效时间
+ * @param key
+ */
+ public static void resetExpire(String key) {
+ Objects.requireNonNull(key);
+ Value value = localCache.get(key);
+ if(Objects.nonNull(value)){
+ value.getCreateTime().set(System.currentTimeMillis());
}
}
+
+ /**
+ * 清理过期对象
+ */
+ private synchronized static void startClearThread() {
+ if (CLEAR_THREAD_STATUS == 0) {
+ ThreadFactory namedThreadFactory = new ThreadFactoryBuilder()
+ .setNameFormat("demo-pool-%d").build();
+ ExecutorService singleThreadPool = new ThreadPoolExecutor(1, 1,
+ 0L, TimeUnit.MILLISECONDS,
+ new LinkedBlockingQueue<Runnable>(1), namedThreadFactory, new ThreadPoolExecutor.AbortPolicy());
+ singleThreadPool.execute(() -> {
+ CLEAR_THREAD_STATUS = 1;
+ while (true) {
+ try {
+ Set<Map.Entry<String, Value>> entries = localCache.entrySet();
+ Iterator<Map.Entry<String, Value>> iterator = entries.iterator();
+ while (iterator.hasNext()) {
+ Map.Entry<String, Value> next = iterator.next();
+
+ if (next.getValue().timeOut == 0) {
+ continue;
+ }
+
+ boolean isTimeOut = (System.currentTimeMillis() - next.getValue().getCreateTime().longValue()) > next.getValue().timeOut;
+ if (isTimeOut) {
+ Value removed = localCache.remove(next.getKey());
+ LogUtil.debug("清除过期对象:{}", removed.value);
+ }
+ }
+ Thread.sleep(1000);
+ } catch (InterruptedException e) {
+ LogUtil.error("清理缓存线程异常停止", e);
+ CLEAR_THREAD_STATUS = 0;
+ }
+ }
+ });
+
+
+ }
+ }
+
+
+ private static Value buildValue(String key, Object value) {
+ return buildValue(key, value, 0);
+ }
+
+
+ private static Value buildValue(String key, Object value, long timeOut) {
+ Value instances = new Value();
+ instances.createTime = new AtomicLong(System.currentTimeMillis());
+ instances.key = key;
+ instances.value = value;
+ instances.timeOut = timeOut;
+ return instances;
+ }
+
+
/**
* 缓存对象
*/
- class Value{
+ @Data
+ static class Value {
/**
- * 过期时间,0 表示不过期
+ * 过期时间,0 表示不过期,单位毫秒
*/
- private long timeOut=0;
+ private long timeOut = 0;
+ /**
+ * 缓存key
+ */
+ private String key;
/**
* 缓存值
*/
- private Object value;
+ private Object value;
/**
* 缓存创建时间
*/
- private long createTime;
+ private AtomicLong createTime;
-
-
-
- public static Value build(String value,long timeOut){
- Value instances=new Value();
- instances.createTime=System.currentTimeMillis();
- instances.value=value;
- instances.timeOut=timeOut;
- return instances;
- }
}
--
Gitblit v1.9.1