com.jeesuite.cache.local.EhCacheLevel1CacheProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jeesuite-cache Show documentation

There is a newer version: 1.4.0

/**
 * 
 */
package com.jeesuite.cache.local;

import java.io.IOException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import net.sf.ehcache.Cache;
import net.sf.ehcache.CacheManager;
import net.sf.ehcache.Element;

/**
 * @description 

 * @author vakin
 * @date 2016年12月4日
 */
public class EhCacheLevel1CacheProvider implements Level1CacheProvider {

	private CacheManager manager;
	private Map caches = new ConcurrentHashMap();
	
	private String ehcacheName;
	
	public void setEhcacheName(String ehcacheName) {
		this.ehcacheName = ehcacheName;
	}

	@Override
	public void start() {
		if (ehcacheName != null)
			manager = CacheManager.getCacheManager(ehcacheName);
		if (manager == null) {
		    manager = new CacheManager();
		}
	}
	
	@Override
	public void close() throws IOException {
		manager.shutdown();
	}

	@Override
	public boolean set(String cacheName, String key, Object value) {
		getCacheHolder(cacheName).put(new Element( key, value ));
		return true;
	}

	@Override
	public  T get(String cacheName, String key) {
		Element element = getCacheHolder(cacheName).get(key);
		if(element == null)return null;
		return (T) element.getObjectValue();
	}

	@Override
	public void remove(String cacheName, String key) {
		getCacheHolder(cacheName).remove(key);
	}

	@Override
	public void remove(String cacheName) {
		getCacheHolder(cacheName).removeAll();
	}

	@Override
	public void clearAll() {
		for (Cache cache : caches.values()) {
			cache.removeAll();
		}
	}
	
	private Cache getCacheHolder(String cacheName){
		return getAndNotexistsCreateCache(cacheName);
	}
	
	private Cache getAndNotexistsCreateCache(String cacheName){
		Cache cache = caches.get(cacheName);
		if(cache != null)return cache;
		synchronized (caches) {
			if((cache = caches.get(cacheName)) != null)return cache;
			manager.addCache(cacheName);
			cache = manager.getCache(cacheName);
			caches.put(cacheName, cache);
		}
		return cache;	
	}

}