最近的项目没有用到redis,就在网上借鉴了下以下代码,一开始以为key不会自动过期,要调到checkExpiry这个方法才会,但发现key确实会到时间自动过期,但我没明白原理,还是说确实是调了但是我没发现,求大佬们解惑。
import java.util.*;
/**
* @Title: ExpiryMap 可以设置过期时间的Map
* @description ExpiryMap继承至HashMap 重写了所有对外的方法,对每个key值都设置了有效期
*/
public class ExpiryMap<K, V> extends HashMap<K, V> {
private static final long serialVersionUID = 1L;
/**
* default expiry time 2s
*/
private long EXPIRY = 1000 * 2;
private HashMap<K, Long> expiryMap = new HashMap<>();
/** 缓存实例对象 */
private volatile static ExpiryMap<String, String> SameUrlMap;
/**
* 采用单例模式获取实例
* @return
*/
public static ExpiryMap getInstance() {
//第一次判空,提高效率
if (null == SameUrlMap) {
//保证线程安全
synchronized (ExpiryMap.class) {
//第二次判空,保证单例对象的唯一性,防止第一次有多个线程进入第一个if判断
if (null == SameUrlMap) {
SameUrlMap = new ExpiryMap<>();
}
}
}
return SameUrlMap;
}
public ExpiryMap(){
super();
}
public ExpiryMap(long defaultExpiryTime){
this(1 << 4, defaultExpiryTime);
}
public ExpiryMap(int initialCapacity, long defaultExpiryTime){
super(initialCapacity);
this.EXPIRY = defaultExpiryTime;
}
@Override
public V put(K key, V value) {
expiryMap.put(key, System.currentTimeMillis() + EXPIRY);
return super.put(key, value);
}
@Override
public boolean containsKey(Object key) {
return !checkExpiry(key, true) && super.containsKey(key);
}
/**
* @param key
* @param value
* @param expiryTime 键值对有效期 毫秒
* @return
*/
public V put(K key, V value, long expiryTime) {
expiryMap.put(key, System.currentTimeMillis() + expiryTime);
return super.put(key, value);
}
@Override
public int size() {
return entrySet().size();
}
@Override
public boolean isEmpty() {
return entrySet().size() == 0;
}
@Override
public boolean containsValue(Object value) {
if (value == null) {
return Boolean.FALSE;
}
Set<Entry<K, V>> set = super.entrySet();
Iterator<Entry<K, V>> iterator = set.iterator();
while (iterator.hasNext()) {
java.util.Map.Entry<K, V> entry = iterator.next();
if(value.equals(entry.getValue())){
if(checkExpiry(entry.getKey(), false)) {
iterator.remove();
return Boolean.FALSE;
}else {
return Boolean.TRUE;
}
}
}
return Boolean.FALSE;
}
@Override
public Collection<V> values() {
Collection<V> values = super.values();
if(values == null || values.size() < 1) {
return values;
}
Iterator<V> iterator = values.iterator();
while (iterator.hasNext()) {
V next = iterator.next();
if(!containsValue(next)) {
iterator.remove();
}
}
return values;
}
@Override
public V get(Object key) {
if (key == null) {
return null;
}
if(checkExpiry(key, true)) {
return null;
}
return super.get(key);
}
/**
*
* @Description: 是否过期
* @param key
* @return null:不存在或key为null -1:过期 存在且没过期返回value 因为过期的不是实时删除,所以稍微有点作用
*/
public Object isInvalid(Object key) {
if (key == null) {
return null;
}
if(!expiryMap.containsKey(key)){
return null;
}
long expiryTime = expiryMap.get(key);
boolean flag = System.currentTimeMillis() > expiryTime;
if(flag){
super.remove(key);
expiryMap.remove(key);
return -1;
}
return super.get(key);
}
@Override
public void putAll(Map<? extends K, ? extends V> m) {
for (Map.Entry<? extends K, ? extends V> e : m.entrySet()) {
expiryMap.put(e.getKey(), System.currentTimeMillis() + EXPIRY);
}
super.putAll(m);
}
@Override
public Set<Map.Entry<K,V>> entrySet() {
Set<java.util.Map.Entry<K, V>> set = super.entrySet();
Iterator<java.util.Map.Entry<K, V>> iterator = set.iterator();
while (iterator.hasNext()) {
java.util.Map.Entry<K, V> entry = iterator.next();
if(checkExpiry(entry.getKey(), false)) {
iterator.remove();
}
}
return set;
}
/**
*
* @Description: 是否过期
* @param isRemoveSuper true super删除
* @return
*/
private boolean checkExpiry(Object key, boolean isRemoveSuper){
if(!expiryMap.containsKey(key)){
return Boolean.FALSE;
}
long expiryTime = expiryMap.get(key);
boolean flag = System.currentTimeMillis() > expiryTime;
if(flag){
if(isRemoveSuper) {
super.remove(key);
}
expiryMap.remove(key);
}
return flag;
}
public static void main(String[] args) throws InterruptedException {
ExpiryMap<String, String> map = new ExpiryMap<>();
map.put("test", "xxx");
map.put("test2", "ankang", 5000);
System.out.println("test==" + map.get("test"));
Thread.sleep(3000);
System.out.println("test==" + map.get("test"));
System.out.println("test2==" + map.get("test2"));
Thread.sleep(3000);
System.out.println("test2==" + map.get("test2"));
}
}
这个代码中,每个key都有一个对应的过期时间,存储在expiryMap中,这个过期时间是在put方法中设置的,等于当前时间加上有效期。然后,在每次调用get、containsKey、containsValue等方法时,都会先检查key是否过期,如果过期了,就会从HashMap和expiryMap中删除key,并返回null或false。这样就实现了一个可以设置过期时间的Map。