package com.matrix.system.common.init; import com.google.common.util.concurrent.ThreadFactoryBuilder; import com.matrix.core.tools.LogUtil; import java.util.Iterator; import java.util.Map; import java.util.Set; import java.util.concurrent.*; public class LocalCache { /* * 清理线程运行状态 0 未启动,1 已启动 */ private static int CLEAR_THREAD_STATUS = 0; private static ConcurrentMap localCache = new ConcurrentHashMap(60); /** * 获取本地缓存 * * @param key * @param * @return */ public static T get(String key) { return (T) localCache.get(key); } /** * 保存一个本地缓存 * * @param key * @param value */ public static void save(String key, Object value) { if (null != localCache.put(key, buildValue(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(value, timeOut))) { LogUtil.debug("覆盖原有缓存{}", key); } startClearThread(); } /** * 清理过期对象 */ 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(1), namedThreadFactory, new ThreadPoolExecutor.AbortPolicy()); singleThreadPool.execute(() -> { CLEAR_THREAD_STATUS = 1; while (true) { try { Set> entries = localCache.entrySet(); Iterator> iterator = entries.iterator(); while (iterator.hasNext()) { Map.Entry next = iterator.next(); if (next.getValue().timeOut == 0) { continue; } boolean isTimeOut = (System.currentTimeMillis() - next.getValue().createTime) > 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(Object value) { return buildValue(value, 0); } private static Value buildValue(Object value, long timeOut) { Value instances = new Value(); instances.createTime = System.currentTimeMillis(); instances.value = value; instances.timeOut = timeOut; return instances; } /** * 缓存对象 */ static class Value { /** * 过期时间,0 表示不过期,单位毫秒 */ private long timeOut = 0; /** * 缓存值 */ private Object value; /** * 缓存创建时间 */ private long createTime; } }