D-Net

package eu.dnetlib.oai;

import javax.annotation.Resource;

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

import eu.dnetlib.enabling.tools.AbstractBaseService;

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

import eu.dnetlib.oai.mongo.MongoPublisherStore;

import eu.dnetlib.oai.mongo.MongoPublisherStoreDAO;

import eu.dnetlib.oai.sets.OAISetsCounter;

import eu.dnetlib.oai.sync.OAIStoreSynchronizer;

import eu.dnetlib.oai.utils.OAIHelper;

import eu.dnetlib.rmi.common.ResultSet;

import eu.dnetlib.rmi.provision.MDFInfo;

import eu.dnetlib.rmi.provision.OAIStoreService;

import eu.dnetlib.utils.MetadataReference;

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

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

public class OAIStoreServiceImpl extends AbstractBaseService implements OAIStoreService {

	/**

	 * notification handler.

	 */

	private NotificationHandler notificationHandler;

	@Autowired

	private MongoPublisherStoreDAO mongoPublisherStoreDAO;

	@Autowired

	private OAIStoreSynchronizer synchronizer;

	@Resource(name = "oaiSetsCounter")

	private OAISetsCounter setsCounter;

	@Autowired

	private OAIHelper oaiHelper;

	@Autowired

	private ResultSetClient resultSetClient;

	@Override

	public void feed(final ResultSet<String> recordsEpr, final MetadataReference mdfInfo, final String dbName) {

		this.feed(recordsEpr, mdfInfo, "", dbName);

	}

	@Override

	public void feed(final ResultSet<String> recordsEpr, final MetadataReference mdfInfo, final String recordSource, final String dbName) {

		final Iterable<String> iter = this.resultSetClient.iter(recordsEpr, String.class);

		this.synchronizer.synchronize(iter, mdfInfo, recordSource, dbName, false, null, null);

	}

	@Override

	public void updateSetCounts(final String dbName) {

		this.setsCounter.updateCounts("", dbName, null, null);

	}

	@Override

	public void updateSetCounts(final MDFInfo mdfInfo, final String dbName) {

		this.setsCounter.updateCounts(mdfInfo, dbName, null, null);

	}

	@Override

	public void refreshConfiguration(final String dbName) {

		this.oaiHelper.loadConfiguration(dbName);

	}

	@Override

	public void ensureIndexes(final String dbName) {

		this.mongoPublisherStoreDAO.ensureIndex(dbName);

	}

	@Override

	public void ensureIndexes(final MDFInfo mdfInfo, final String dbName) {

		final MongoPublisherStore s = this.mongoPublisherStoreDAO.getStore(mdfInfo.getSourceFormatName(), mdfInfo.getSourceFormatInterpretation(),

				mdfInfo.getSourceFormatLayout(), dbName);

		s.ensureIndices();

	}

	@Override

	public void notify(final String subscriptionId, final String topic, final String isId, final String message) {

		getNotificationHandler().notified(subscriptionId, topic, isId, message);

	}

	public NotificationHandler getNotificationHandler() {

		return this.notificationHandler;

	}

	@Required

	public void setNotificationHandler(final NotificationHandler notificationHandler) {

		this.notificationHandler = notificationHandler;

	}

	public OAIStoreSynchronizer getSynchronizer() {

		return this.synchronizer;

	}

	public void setSynchronizer(final OAIStoreSynchronizer synchronizer) {

		this.synchronizer = synchronizer;

	}

	public OAISetsCounter getSetsCounter() {

		return this.setsCounter;

	}

	public void setSetsCounter(final OAISetsCounter setsCounter) {

		this.setsCounter = setsCounter;

	}

	public OAIHelper getOaiSetHelper() {

		return this.oaiHelper;

	}

	public void setOaiSetHelper(final OAIHelper oaiHelper) {

		this.oaiHelper = oaiHelper;

	}

	public MongoPublisherStoreDAO getMongoPublisherStoreDAO() {

		return this.mongoPublisherStoreDAO;

	}

	public void setMongoPublisherStoreDAO(final MongoPublisherStoreDAO mongoPublisherStoreDAO) {

		this.mongoPublisherStoreDAO = mongoPublisherStoreDAO;

	}

}