D-Net

package eu.dnetlib.msro.workflows.nodes;

import eu.dnetlib.dli.resolver.model.DLIResolvedObject;

import eu.dnetlib.dli.resolver.model.serializer.ResolverSerializer;

import eu.dnetlib.enabling.resultset.factory.ResultSetFactory;

import eu.dnetlib.enabling.tools.DnetStreamSupport;

import eu.dnetlib.msro.workflows.graph.Arc;

import eu.dnetlib.msro.workflows.procs.Env;

import eu.dnetlib.pid.resolver.model.ObjectType;

import eu.dnetlib.pid.resolver.store.ResolverStore;

import eu.dnetlib.rmi.common.ResultSet;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

/**

 * Created by sandro on 10/11/16.

 */

public class FilterResolvedRecordJobNode extends SimpleJobNode {

	private static final Log log = LogFactory.getLog(FilterResolvedRecordJobNode.class);

	private final static String recordTemplate =

			"<root xmlns:oaf=\"http://namespace.dnet.eu/oaf\" xmlns:dri=\"http://www.driver-repository.eu/namespace/dri\" xmlns:dc=\"http://purl.org/dc/elements/1.1/\">%s</root>";

	private String type;

	private String eprParam;

	@Autowired

	private ResolverStore resolverStore;

	@Autowired

	private ResultSetFactory resultSetFactory;

	@Autowired

	private ResolverSerializer resolverSerializer;

	@Override

	protected String execute(final Env env) throws Exception {

		log.info("HERE");

		final Iterable<String> resultIterator = () -> DnetStreamSupport.generateStreamFromIterator(resolverStore.getAllRecords().iterator())

				.map(record -> {

					final ObjectType type = record.getType();

                    if (type == null || !type.toString().equals(this.type)) {

                        return "<JUNK/>";

                    }

                    final String resultXML = String.format(recordTemplate, resolverSerializer.serializeToXML((DLIResolvedObject) record));

                    return resultXML;

				}).iterator();

		final ResultSet<String> resultSet = resultSetFactory.createResultSet(resultIterator);

		env.setAttribute(eprParam, resultSet);

		return Arc.DEFAULT_ARC;

	}

	public String getType() {

		return type;

	}

	public void setType(final String type) {

		this.type = type;

	}

	public String getEprParam() {

		return eprParam;

	}

	public void setEprParam(final String eprParam) {

		this.eprParam = eprParam;

	}

}