D-Net

package eu.dnetlib.openaire.exporter.datasource.clients;

import java.io.IOException;

import java.nio.charset.Charset;

import java.util.Queue;

import eu.dnetlib.OpenaireExporterConfig;

import eu.dnetlib.enabling.datasources.common.DatasourceManagerException;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;

import eu.dnetlib.openaire.exporter.datasource.clients.utils.IndexDsInfo;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

import org.springframework.cache.annotation.CacheEvict;

import org.springframework.cache.annotation.Cacheable;

import org.springframework.core.io.ClassPathResource;

import org.springframework.http.HttpStatus;

import org.springframework.stereotype.Component;

/**

 * Created by claudio on 20/10/2016.

 */

@Component

public class ISLookupClient {

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

	@Autowired

	private OpenaireExporterConfig config;

	@Autowired

	private ISLookUpService isLookUpService;

	@Cacheable("datasources-is-cache")

	public IndexDsInfo calculateCurrentIndexDsInfo(final Queue<Throwable> errors) throws DatasourceManagerException {

		log.warn("calculateCurrentIndexDsInfo(): not using cache");

		final String[] arr = _isLookUp(_getQuery(config.getFindIndexDsInfo()), errors).split("@@@");

		return new IndexDsInfo(

				_isLookUp(_getQuery(config.getFindSolrIndexUrl()), errors),

				arr[0].trim(), arr[1].trim(), arr[2].trim());

	}

	@Cacheable("datasources-is-cache")

	public String getObjectStoreId(final String dsId, final Queue<Throwable> errors) throws DatasourceManagerException {

		log.warn(String.format("getObjectStoreId(%s): not using cache", dsId));

		final String xqueryTemplate = _getQuery(config.getFindObjectStore());

		return _isLookUp(String.format(xqueryTemplate, dsId), errors);

	}

	private String _getQuery(final ClassPathResource resource) throws DatasourceManagerException {

		try {

			return IOUtils.toString(resource.getInputStream(), Charset.defaultCharset());

		} catch (IOException e) {

			throw new DatasourceManagerException(HttpStatus.INTERNAL_SERVER_ERROR.value(), "Error accessing classpath resource (xquery)", e);

		}

	}

	private String _isLookUp(final String xquery, final Queue<Throwable> errors) {

		try {

			log.debug(String.format("running xquery: %s", xquery));

			final String res = isLookUpService.getResourceProfileByQuery(xquery);

			log.debug(String.format("query result: %s", res));

			return res;

		} catch (Throwable e) {

			errors.add(e);

			return "";

		}

	}

	@CacheEvict(cacheNames = "datasources-is-cache", allEntries = true)

	public void dropCache() {

		log.info("dropped dsManager IS cache");

	}

}