java 60 行代码写一个简单可用的并且带过期时间的内存缓存

60 行代码写一个简单可用的并且带过期时间的内存缓存

在一个小项目中需要特别简单的使用一个缓存。如果要引入 ehcache 等其他第三方缓存还需要增加 各种 jar 甚至需要增加配置文件。特别麻烦。

所以就自己写了一个。

思路也很简单。

1、使用一个  static 修饰的 ConcurrentHashMap 对象保存数据。

2、存入数据的时候把存入的实际数据增加一个外包装，顺便加上存入时间，和过期时间。

3、取出数据的时候校验是否过期，如果过期则直接返回 null

4、取数据的时候还可以增加一个数据装载器，如果缓存中没有数据或者已经过期，则调用数据装载器加载最新的数据并且加入缓存，并返回。

基本思路有了下面贴出完整代码。

package com.data;

import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Created by cyy on 2017/9/18.
 */
public class CacheManager {
    private static Map<String,CacheData> CACHE_DATA = new ConcurrentHashMap<>();
    public static <T> T getData(String key,Load<T> load,int expire){
        T data = getData(key);
        if(data == null && load != null){
            data = load.load();
            if(data != null){
                setData(key,data,expire);
            }
        }
        return data;
    }
    public static <T> T getData(String key){
        CacheData<T> data = CACHE_DATA.get(key);
        if(data != null && (data.getExpire() <= 0 || data.getSaveTime() >= new Date().getTime())){
            return data.getData();
        }
        return null;
    }
    public static <T> void setData(String key,T data,int expire){
        CACHE_DATA.put(key,new CacheData(data,expire));
    }
    public static void clear(String key){
        CACHE_DATA.remove(key);
    }
    public static void clearAll(){
        CACHE_DATA.clear();
    }
    public interface Load<T>{
        T load();
    }
    private static class CacheData<T>{
        CacheData(T t,int expire){
            this.data = t;
            this.expire = expire <= 0 ? 0 : expire * 1000;
            this.saveTime = new Date().getTime() + this.expire;
        }
        private T data;
        private long saveTime; // 存活时间
        private long expire;   // 过期时间 小于等于0标识永久存活
        public T getData() {
            return data;
        }
        public long getExpire() {
            return expire;
        }
        public long getSaveTime() {
            return saveTime;
        }
    }
}

使用：

package com.data;

import java.util.Map;

/**
 * Created by Administrator on 2017/9/18.
 */
public class CacheManagerTest {

    public static void main(String [] args)throws Exception{

        String key = "test";

        String value = CacheManager.getData(key, new CacheManager.Load<String>() {
            public String load(){
                return "testValue";
            }
        },2);

        System.out.println("value:"+value);

        Thread.sleep(3 * 1000);

        String value2 = CacheManager.getData(key, new CacheManager.Load<String>() {
            @Override
            public String load() {
                return "什么";
            }
        },3);
        System.out.println("value2:" + value2);
        System.out.println("value3:" + CacheManager.getData(key));

    }


}