cn.acyou.leo.framework.util.CacheUtil Maven / Gradle / Ivy

Go to download
package cn.acyou.leo.framework.util;

import lombok.Data;

import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;
import java.util.function.Function;

/**
 * 本机缓存
 *
 * @author fangyou
 * @version [1.0.0, 2021-08-04 14:07]
 */
public class CacheUtil {
    /**
     * 缓存池
     */
    private static final ConcurrentHashMap CACHE_MAP = new ConcurrentHashMap<>();
    /**
     * 默认缓存 7200 s
     */
    private static final long DEFAULT_CACHE_DURATION = 7200;

    /**
     * 获取缓存值
     *
     * @param key 缓存Key
     * @return 缓存value
     */
    @SuppressWarnings("unchecked")
    public synchronized static  T get(String key) {
        CacheBean cacheBean = CACHE_MAP.get(key);
        if (cacheBean != null) {
            if (cacheBean.getExpiresTime() == null) {
                return (T) cacheBean.getCacheValue();
            }
            if (cacheBean.getExpiresTime() > System.currentTimeMillis()) {
                return (T) cacheBean.getCacheValue();
            } else {
                CACHE_MAP.remove(key);
            }
        }
        return null;
    }

    /**
     * 设置缓存值
     *
     * @param key   缓存Key
     * @param value 缓存value
     */
    public static void put(String key, String value) {
        CacheBean cacheBean = new CacheBean(value, null);
        CACHE_MAP.put(key, cacheBean);
    }


    /**
     * 设置缓存值（带过期时间）
     *
     * @param key      缓存Key
     * @param value    缓存value
     * @param duration duration
     * @param timeUnit 单位
     */
    public static void put(String key, Object value, Long duration, TimeUnit timeUnit) {
        CacheBean cacheBean = new CacheBean(value);
        if (duration != null && duration > 0) {
            long toMillis = timeUnit.toMillis(duration);
            cacheBean.setExpiresTime(System.currentTimeMillis() + toMillis);
        }
        CACHE_MAP.put(key, cacheBean);
    }


    /**
     * 获取和缓存
     *
     * @param key      关键
     * @param function 函数
     * @return {@link String}
     */
    public static  T getAndCache(String key, Function function) {
        return getAndCache(key, DEFAULT_CACHE_DURATION, function);
    }

    /**
     * 获取和缓存
     *
     * @param key      关键
     * @param function 函数
     * @return {@link String}
     */
    @SuppressWarnings("unchecked")
    public static  T getAndCache(String key, Long expire, Function function) {
        Object v = get(key);
        if (v != null) {
            return (T) v;
        }
        Object value = function.apply(key);
        put(key, value, expire, TimeUnit.SECONDS);
        return (T) value;
    }

    /**
     * 清除缓存
     */
    public static void clearAllCache() {
        CACHE_MAP.clear();
    }

    /**
     * 清除缓存
     */
    public static void clearCache(String key) {
        CACHE_MAP.remove(key);
    }

    /**
     * 清除缓存
     */
    public static void clearCache(List keys) {
        keys.forEach(CACHE_MAP::remove);
    }

    @Data
    public static class CacheBean {

        /**
         * 缓存值
         */
        private Object cacheValue;
        /**
         * 过期时间(null 表示长期)
         */
        private Long expiresTime;

        public CacheBean(Object cacheValue) {
            this.cacheValue = cacheValue;
            this.expiresTime = null;
        }

        public CacheBean(Object cacheValue, Long expiresTime) {
            this.cacheValue = cacheValue;
            this.expiresTime = expiresTime;
        }
    }
}