D-Net

package eu.dnetlib.oai.actions;

import com.google.common.collect.Iterables;

import eu.dnetlib.enabling.resultset.client.ResultSetClient;

import eu.dnetlib.enabling.tools.blackboard.BlackboardJob;

import eu.dnetlib.enabling.tools.blackboard.BlackboardServerHandler;

import eu.dnetlib.miscutils.functional.xml.ApplyXslt;

import eu.dnetlib.oai.sync.OAIStoreSynchronizer;

import eu.dnetlib.rmi.common.ResultSet;

import eu.dnetlib.rmi.provision.MDFInfo;

import eu.dnetlib.utils.MetadataReference;

import eu.dnetlib.utils.MetadataReferenceFactory;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.io.SAXReader;

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

import org.springframework.core.io.ClassPathResource;

public class SyncAction extends AbstractOAIStoreAction {

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

	private final SAXReader saxReader = new SAXReader();

	org.springframework.core.io.Resource dmfXsltResource = new ClassPathResource("/eu/dnetlib/oai/xslt/addDMFBlock.xslt");

	@Autowired

	private OAIStoreSynchronizer synchronizer;

	@Autowired

	private ResultSetClient resultSetClient;

	@Override

	public void execute(final BlackboardServerHandler handler, final BlackboardJob job) throws Exception {

		String epr = job.getParameters().get("oai_syncEpr");

		String recordSource = job.getParameters().get("oai_recordSource");

		boolean alwaysNewRecord = Boolean.valueOf(job.getParameters().get("oai_alwaysNewRecord"));

		final String dbName = job.getParameters().get("oai_dbName");

		final String collectionName = job.getParameters().get("oai_collectionName");

		final MetadataReference mdRef = MetadataReferenceFactory.decode(collectionName, MDREF_SEPARATOR);

		final ResultSet<?> rs = ResultSet.fromJson(epr);

		log.info("Syncronizing content for oai on db " + dbName + "for metadata format: " + mdRef);

		Iterable<String> records = resultSetClient.iter(rs, String.class);

		// in case of DMF there is no "hat element" after oai:metadata: we must add it or the OAI-PMH export will only contain the first

		// element inside oai:metadata.

		if (mdRef.getFormat().equalsIgnoreCase("DMF")) {

			records = addDMFBlock(records);

		}

		this.synchronizer.synchronize(records, mdRef, recordSource, dbName, alwaysNewRecord, () -> {

			handler.done(job);

			return null;

		}, () -> {

			handler.failed(job, new Exception("Error during OAI synchronization on db " + dbName));

			return null;

		});

	}

	protected Iterable<String> addDMFBlock(final Iterable<String> input) {

		final ApplyXslt addDMFBlockXslt = new ApplyXslt(dmfXsltResource);

		return Iterables.transform(input, addDMFBlockXslt::apply);

	}

	public OAIStoreSynchronizer getSynchronizer() {

		return synchronizer;

	}

	public void setSynchronizer(final OAIStoreSynchronizer synchronizer) {

		this.synchronizer = synchronizer;

	}

}