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