From 58032b964fa5dee3924847943548093af6dcb098 Mon Sep 17 00:00:00 2001 From: 935090232@qq.com <ak473600000> Date: Thu, 28 Oct 2021 10:32:24 +0800 Subject: [PATCH] Merge branch 'score_shop' into api_score_meger --- zq-erp/src/main/java/com/matrix/system/common/init/LocalCache.java | 215 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 215 insertions(+), 0 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 new file mode 100644 index 0000000..ed624c2 --- /dev/null +++ b/zq-erp/src/main/java/com/matrix/system/common/init/LocalCache.java @@ -0,0 +1,215 @@ +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.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); + + /** + * 根据key匹配多个缓存值 + * + * @param key + * @param <T> + * @return + */ + 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, buildValue(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; + } + + + + + /** + * 缓存对象 + */ + @Data + static class Value { + + /** + * 过期时间,0 表示不过期,单位毫秒 + */ + private long timeOut = 0; + /** + * 缓存key + */ + private String key; + /** + * 缓存值 + */ + private Object value; + + /** + * 缓存创建时间 + */ + private AtomicLong createTime; + + + } + + +} -- Gitblit v1.9.1