ClusterAwareRegionFactory.java

/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later.
 * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
 */
package org.hibernate.test.cache.infinispan.functional.cluster;

import java.util.Hashtable;
import java.util.Properties;

import org.hibernate.boot.spi.SessionFactoryOptions;
import org.hibernate.cache.CacheException;
import org.hibernate.cache.infinispan.InfinispanRegionFactory;
import org.hibernate.cache.infinispan.util.InfinispanMessageLogger;
import org.hibernate.cache.spi.CacheDataDescription;
import org.hibernate.cache.spi.CollectionRegion;
import org.hibernate.cache.spi.EntityRegion;
import org.hibernate.cache.spi.NaturalIdRegion;
import org.hibernate.cache.spi.QueryResultsRegion;
import org.hibernate.cache.spi.RegionFactory;
import org.hibernate.cache.spi.TimestampsRegion;
import org.hibernate.cache.spi.access.AccessType;

import org.hibernate.test.cache.infinispan.util.CacheTestUtil;
import org.infinispan.manager.EmbeddedCacheManager;

/**
 * ClusterAwareRegionFactory.
 * 
 * @author Galder Zamarreño
 * @since 3.5
 */
public class ClusterAwareRegionFactory implements RegionFactory {
	private static final InfinispanMessageLogger log = InfinispanMessageLogger.Provider.getLog(ClusterAwareRegionFactory.class);
	private static final Hashtable<String, EmbeddedCacheManager> cacheManagers = new Hashtable<String, EmbeddedCacheManager>();

	private InfinispanRegionFactory delegate;
	private String cacheManagerName;
	private boolean locallyAdded;

	public ClusterAwareRegionFactory(Properties props) {
		Class<? extends InfinispanRegionFactory> regionFactoryClass =
				(Class<InfinispanRegionFactory>) props.get(DualNodeTest.REGION_FACTORY_DELEGATE);
		delegate = CacheTestUtil.createRegionFactory(regionFactoryClass, props);
	}

	public static EmbeddedCacheManager getCacheManager(String name) {
		return cacheManagers.get(name);
	}

	public static void addCacheManager(String name, EmbeddedCacheManager manager) {
		cacheManagers.put(name, manager);
	}

	public static void clearCacheManagers() {
		for (EmbeddedCacheManager manager : cacheManagers.values()) {
			try {
				manager.stop();
			} catch (Exception e) {
				log.error("Exception cleaning up CacheManager " + manager, e);
			}
		}
		cacheManagers.clear();
	}

	public void start(SessionFactoryOptions settings, Properties properties) throws CacheException {
		cacheManagerName = properties.getProperty(DualNodeTest.NODE_ID_PROP);

		EmbeddedCacheManager existing = getCacheManager(cacheManagerName);
		locallyAdded = (existing == null);

		if (locallyAdded) {
			delegate.start(settings, properties);
			cacheManagers.put(cacheManagerName, delegate.getCacheManager());
		} else {
			delegate.setCacheManager(existing);
		}
	}

	public void stop() {
		if (locallyAdded) cacheManagers.remove(cacheManagerName);
		delegate.stop();
	}

	public CollectionRegion buildCollectionRegion(String regionName, Properties properties,
				CacheDataDescription metadata) throws CacheException {
		return delegate.buildCollectionRegion(regionName, properties, metadata);
	}

	public EntityRegion buildEntityRegion(String regionName, Properties properties,
				CacheDataDescription metadata) throws CacheException {
		return delegate.buildEntityRegion(regionName, properties, metadata);
	}

	public NaturalIdRegion buildNaturalIdRegion(String regionName, Properties properties, CacheDataDescription metadata)
			throws CacheException {
		return delegate.buildNaturalIdRegion( regionName, properties, metadata );
	}

	public QueryResultsRegion buildQueryResultsRegion(String regionName, Properties properties)
				throws CacheException {
		return delegate.buildQueryResultsRegion(regionName, properties);
	}

	public TimestampsRegion buildTimestampsRegion(String regionName, Properties properties)
				throws CacheException {
		return delegate.buildTimestampsRegion(regionName, properties);
	}

	public boolean isMinimalPutsEnabledByDefault() {
		return delegate.isMinimalPutsEnabledByDefault();
	}

	@Override
	public AccessType getDefaultAccessType() {
		return AccessType.TRANSACTIONAL;
	}

	public long nextTimestamp() {
		return delegate.nextTimestamp();
	}
}