D-Net

package eu.dnetlib.enabling.resultset.registry;

import java.util.UUID;

import org.springframework.beans.factory.annotation.Required;

import eu.dnetlib.enabling.resultset.listener.ResultSetListener;

import eu.dnetlib.rmi.common.ResultSetException;

import net.sf.ehcache.Cache;

import net.sf.ehcache.Element;

public class ResultSetRegistryImpl implements ResultSetRegistry {

	private static final String RS_PREFIX = "rs-";

	/**

	 * rsId -> ResultSet mappings are stored here.

	 */

	private Cache cache;

	@Override

	public String registerResultSet(final ResultSetListener<?> listener) {

		final String newId = RS_PREFIX + UUID.randomUUID();

		this.cache.put(new Element(newId, listener));

		return newId;

	}

	@SuppressWarnings("unchecked")

	@Override

	public <T> ResultSetListener<T> getResultSetById(final String rsId) throws ResultSetException {

		final Element elem = this.cache.get(rsId);

		if (elem != null) {

			final ResultSetListener<T> rs = (ResultSetListener<T>) elem.getObjectValue();

			if (rs != null) { return rs; }

		}

		throw new ResultSetException("Missing resultset with id " + rsId);

	}

	@Override

	public boolean contains(final String rsId) {

		return this.cache.get(rsId) != null;

	}

	public Cache getCache() {

		return this.cache;

	}

	@Required

	public void setCache(final Cache cache) {

		this.cache = cache;

	}

}