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));

    }


}