简单缓存

hutool手写简单缓存

/**
 * 简单缓存，无超时实现，使用{@link WeakHashMap}实现缓存自动清理
 * @author Looly
 *
 * @param <K> 键类型
 * @param <V> 值类型
 */
public class SimpleCache<K, V> implements Serializable{
	private static final long serialVersionUID = 1L;
	/** 池 */
	private final Map<K, V> cache = new WeakHashMap<>();
	private final ReentrantReadWriteLock cacheLock = new ReentrantReadWriteLock();
	private final ReadLock readLock = cacheLock.readLock();
	private final WriteLock write = cacheLock.writeLock();

	/**
	 * 从缓存池中查找值
	 * @param key 键
	 * @return 值
	 */
	public V get(K key){
		//尝试读取缓存
		readLock.lock();
		V value;
		try {
			value = cache.get(key);
		} finally {
			readLock.unlock();
		}
		return value;
	}
	/**
	 * 从缓存中获得对象，当对象不在缓存中或已经过期返回Func0回调产生的对象
	 * @param key 键
	 * @param supplier 如果不存在回调方法，用于生产值对象
	 * @return 值对象
	 */
	public V get(K key, Funco<V> supplier){
		V v = get(key);
		if(null == v && null != supplier){
			writeLock.lock();
			try{
				//双重检查锁
				v = cache.get(key);
				if(null == v) {
					try {
						v = supplier.call();
					} catch (Exception e){
						throw new RuntimeException(e);
					}
					cache.put(key, v);
				}
			} finally {
				writeLock.unlock();
			}
		}
		return v;
	}

	/**
	 * 放入缓存
	 * @param key 键
	 * @param value 值
	 * @return 值
	 */
	public V put(K key, V value){
		writeLock.lock();
		try {
			cache.put(key, value);
		} finally {
			writeLock.unlock();
		}
		return value;
	}

	/**
	 * 移除缓存
	 * 
	 * @param key 键
	 * @return 移除的值
	 */
	public V remove(K key) {
		writeLock.lock();
		try {
			return cache.remove(key);
		} finally {
			writeLock.unlock();
		}
	}

	/**
	 * 清空缓存池
	 */
	public void clear() {
		writeLock.lock();
		try {
			this.cache.clear();
		} finally {
			writeLock.unlock();
		}
	}
}

/**
 * 无参数的函数对象<br>
 * 接口灵感来自于<a href="http://actframework.org/">ActFramework</a><br>
 * 一个函数接口代表一个一个函数，用于包装一个函数为对象<br>
 * 在JDK8之前，Java的函数并不能作为参数传递，也不能作为返回值存在，此接口用于将一个函数包装成为一个对象，从而传递对象
 * @param <R> 返回值类型
 * @since 4.5.2
 */
public interface Func0<R> {
	/**
	 * 执行函数
	 * 
	 * @return 函数执行结果
	 * @throws Exception 自定义异常
	 */
	R call() throws Exception;
}