D-Net

package eu.dnetlib.msro.workers.aggregation.collect;

import java.util.List;

import java.util.stream.Collectors;

import org.springframework.beans.BeansException;

import org.springframework.beans.factory.BeanFactory;

import org.springframework.beans.factory.BeanFactoryAware;

import org.springframework.beans.factory.ListableBeanFactory;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import eu.dnetlib.msro.workers.aggregation.collect.plugins.CollectorPlugin;

import eu.dnetlib.msro.workers.aggregation.collect.plugins.DnetCollectorPlugin;

@RestController

@RequestMapping("/collector")

public class CollectorPluginEnumerator implements BeanFactoryAware {

	// private static final Log log = LogFactory.getLog(CollectorPluginEnumerator.class); // NOPMD by marko on 11/24/08 5:02 PM

	/**

	 * bean factory.

	 */

	private ListableBeanFactory beanFactory;

	/**

	 * Get all beans implementing the CollectorPlugin interface.

	 *

	 * @return

	 */

	@RequestMapping("/plugins")

	public List<CollectorPlugin> getAll() {

		return beanFactory.getBeansOfType(CollectorPlugin.class)

				.values()

				.stream()

				.filter(o -> o.getClass().isAnnotationPresent(DnetCollectorPlugin.class))

				.collect(Collectors.toList());

	}

	public ListableBeanFactory getBeanFactory() {

		return beanFactory;

	}

	@Override

	public void setBeanFactory(final BeanFactory beanFactory) throws BeansException {

		this.beanFactory = (ListableBeanFactory) beanFactory;

	}

	/**

	 * Get given CollectorPlugin or throws exception.

	 *

	 * @param protocol

	 * @return

	 * @throws CollectException

	 */

	public CollectorPlugin get(final String protocol) throws CollectException {

		for (final CollectorPlugin cp : getAll()) {

			if (protocol.equalsIgnoreCase(cp.getProtocol())) { return cp; }

		}

		throw new CollectException("plugin not found for protocol: " + protocol);

	}

}