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

}

}