All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.ssssssss.magicapi.cache.DefaultSqlCache Maven / Gradle / Ivy

    package org.ssssssss.magicapi.cache;

import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.concurrent.locks.ReentrantReadWriteLock;

public class DefaultSqlCache extends LinkedHashMap> implements SqlCache {

	private final String separator = ":";

	private final int capacity;

	private final long expire;

	private final ReentrantReadWriteLock lock = new ReentrantReadWriteLock();

	public DefaultSqlCache(int capacity, long expire) {
		super((int) Math.ceil(capacity / 0.75) + 1, 0.75f, true);
		// 容量
		this.capacity = capacity;
		// 固定过期时间
		this.expire = expire;
	}

	@Override
	public void put(String name, String key, Object value) {
		// 封装成过期时间节点
		put(name, key, value, this.expire);
	}

	@Override
	public void put(String name, String key, Object value, long ttl) {
		long expireTime = ttl > 0 ? (System.currentTimeMillis() + ttl) :
				(this.expire > -1 ? System.currentTimeMillis() + this.expire : Long.MAX_VALUE);
		lock.writeLock().lock();
		try {
			// 封装成过期时间节点
			put(name + separator + key, new ExpireNode<>(expireTime, value));
		} finally {
			lock.writeLock().unlock();
		}
	}

	@Override
	public Object get(String name, String key) {
		key = name + separator + key;
		lock.readLock().lock();
		ExpireNode expireNode;
		try {
			expireNode = super.get(key);
		} finally {
			lock.readLock().unlock();
		}
		if (expireNode == null) {
			return null;
		}
		// 惰性删除过期的
//        if (this.expire > -1L && expireNode.expire < System.currentTimeMillis()) {
		if (expireNode.expire < System.currentTimeMillis()) {
			try {
				lock.writeLock().lock();
				super.remove(key);
			} finally {
				lock.writeLock().unlock();
			}
			return null;
		}
		return expireNode.value;
	}

	@Override
	public void delete(String name) {
		try {
			lock.writeLock().lock();
			Iterator>> iterator = super.entrySet().iterator();
			String prefix = name + separator;
			// 清除所有key前缀为name + separator的缓存
			while (iterator.hasNext()) {
				Map.Entry> entry = iterator.next();
				if (entry.getKey().startsWith(prefix)) {
					iterator.remove();
				}
			}
		} finally {
			lock.writeLock().unlock();
		}
	}


	@Override
	protected boolean removeEldestEntry(Map.Entry> eldest) {
		if (this.expire > -1L && size() > capacity) {
			clean();
		}
		// lru淘汰
		return size() > this.capacity;
	}

	/**
	 * 清理已过期的数据
	 */
	private void clean() {
		try {
			lock.writeLock().lock();
			Iterator>> iterator = super.entrySet().iterator();
			long now = System.currentTimeMillis();
			while (iterator.hasNext()) {
				Map.Entry> next = iterator.next();
				// 判断是否过期
				if (next.getValue().expire < now) {
					iterator.remove();
				}
			}
		} finally {
			lock.writeLock().unlock();
		}
	}


	/**
	 * 过期时间节点
	 */
	static class ExpireNode {
		long expire;
		V value;

		ExpireNode(long expire, V value) {
			this.expire = expire;
			this.value = value;
		}
	}
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy