D-Net

package eu.dnetlib.data.download;

import java.util.Map;

import org.springframework.beans.BeansException;

import org.springframework.beans.factory.BeanFactory;

import org.springframework.beans.factory.ListableBeanFactory;

import com.google.common.collect.Maps;

import eu.dnetlib.data.download.rmi.DownloadPluginEnumerator;

import eu.dnetlib.data.download.rmi.DownloadServiceException;

 * The Class DownloadPluginEnumeratorImpl.

 */

public class DownloadPluginEnumeratorImpl implements DownloadPluginEnumerator {

	 * bean factory.

	 */

	private ListableBeanFactory beanFactory;

	 * (non-Javadoc)

	 *

		return beanFactory.getBeansOfType(DownloadPlugin.class);

	}

	 * (non-Javadoc)

	 *

		final Map<String, DownloadPlugin> res = Maps.newHashMap();

		for (DownloadPlugin cp : getAll().values()) {

			res.put(cp.getPluginName().toLowerCase(), cp);

		}

		return res;

	}

	 * (non-Javadoc)

	 *

	 */

	public void setBeanFactory(final BeanFactory beanFactory) throws BeansException {

		this.beanFactory = (ListableBeanFactory) beanFactory;

	}

	 * Gets the bean factory.

	 *

	 * @return the bean factory

	 */

		return beanFactory;

	}

	/**

	 * Get given DownloadPlugin or throws exception.

	 *            the protocol

	 *             the download service exception

	public DownloadPlugin get(final String protocol) throws DownloadServiceException {

		final DownloadPlugin plugin = getByProtocols().get(protocol.toLowerCase());

		if (plugin == null) throw new DownloadServiceException("plugin not found for name: " + protocol);

		return plugin;

	}

}