D-Net

package eu.dnetlib.openaire.exporter.community;

import java.io.IOException;

import java.nio.charset.Charset;

import java.util.List;

import java.util.Map;

import java.util.Queue;

import java.util.concurrent.LinkedBlockingQueue;

import java.util.function.Function;

import java.util.stream.Collectors;

import com.google.common.collect.Lists;

import com.google.common.escape.Escaper;

import com.google.common.xml.XmlEscapers;

import eu.dnetlib.OpenaireExporterConfig;

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

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

import eu.dnetlib.openaire.exporter.context.Context;

import eu.dnetlib.openaire.exporter.context.ContextMappingUtils;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

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.stereotype.Component;

/**

 * Created by claudio on 20/10/2016.

 */

@Component

public class ISClientImpl implements ISClient {

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

	@Autowired

	private OpenaireExporterConfig config;

	@Autowired

	private ISLookUpService isLookUpService;

	@Override

	@Cacheable("context-cache")

	public Map<String, Context> getFunderContextMap() throws IOException {

		return _processContext(

				new LinkedBlockingQueue<>(),

				_getQuery(config.getFindFunderContexts()));

	}

	@Override

	@Cacheable("context-cache")

	public Map<String, Context> getCommunityContextMap() throws IOException {

		return _processContext(

				new LinkedBlockingQueue<>(),

				_getQuery(config.getFindCommunityContexts()));

	}

	@Override

	@CacheEvict(value = "context-cache", allEntries = true)

	public void updateContextParam(final String id, final String name, final String value) {

		final Queue<Throwable> errors = Lists.newLinkedList();

		_quickSeachProfile(getXQuery(id, name, value), errors);

		if (!errors.isEmpty()) {

			errors.forEach(e -> log.error(e));

		}

	}

	@Override

	@CacheEvict(value = "context-cache", allEntries = true)

	public void updateContextAttribute(final String id, final String name, final String value) {

		final Queue<Throwable> errors = Lists.newLinkedList();

		final Escaper esc = XmlEscapers.xmlAttributeEscaper();

		_quickSeachProfile(String.format(

				"update value collection('/db/DRIVER/ContextDSResources/ContextDSResourceType')" +

						"/RESOURCE_PROFILE/BODY/CONFIGURATION/context[./@id = '%s']/@%s with '%s'", id, name, escape(esc, value)),

				errors);

		if (!errors.isEmpty()) {

			errors.forEach(e -> log.error(e));

		}

	}

	@Override

	@CacheEvict(value = "context-cache", allEntries = true)

	public void addConcept(final String id, final String categoryId, final String data) {

		final Queue<Throwable> errors = Lists.newLinkedList();

		_quickSeachProfile(String.format(

				"update insert %s into collection('/db/DRIVER/ContextDSResources/ContextDSResourceType')" +

				"/RESOURCE_PROFILE/BODY/CONFIGURATION/context[./@id = '%s']/category[./@id = '%s']", data, id, categoryId), errors);

		if (!errors.isEmpty()) {

			errors.forEach(e -> log.error(e));

		}

	}

	@Override

	@CacheEvict(value = "context-cache", allEntries = true)

	public void removeConcept(final String id, final String categoryId, final String conceptId) {

		final Queue<Throwable> errors = Lists.newLinkedList();

		_quickSeachProfile(String.format(

				"for $concept in collection('/db/DRIVER/ContextDSResources/ContextDSResourceType')" +

						"/RESOURCE_PROFILE/BODY/CONFIGURATION/context[./@id = '%s']" +

						"/category[./@id = '%s']/concept[./@id = '%s']" +

						"return update delete $concept", id, categoryId, conceptId),

				errors);

		if (!errors.isEmpty()) {

			errors.forEach(e -> log.error(e.getMessage()));

		}

	}

	/// HELPERS

	private String getXQuery(final String id, final String name, final String value) {

		final Escaper esc = XmlEscapers.xmlContentEscaper();

		if (StringUtils.isNotBlank(value)) {

			return String.format(

					"update replace collection('/db/DRIVER/ContextDSResources/ContextDSResourceType')" +

							"/RESOURCE_PROFILE/BODY/CONFIGURATION/context[./@id = '%s']/param[./@name = '%s'] with <param name='%s'>%s</param>", id, name, name,

					escape(esc, value));

		} else {

			return String.format(

					"update replace collection('/db/DRIVER/ContextDSResources/ContextDSResourceType')" +

							"/RESOURCE_PROFILE/BODY/CONFIGURATION/context[./@id = '%s']/param[./@name = '%s'] with <param name='%s'/>", id, name, name);

		}

	}

	public static String escape(final Escaper esc, final String value) {

		return StringUtils.isNotBlank(value) ? esc.escape(value) : "";

	}

	private Map<String, Context> _processContext(final Queue<Throwable> errors, final String xquery) throws IOException {

		try {

			return getContextProfiles(errors, xquery).stream()

					.filter(StringUtils::isNotBlank)

					.map(s -> ContextMappingUtils.parseContext(s, errors))

					.collect(Collectors.toMap(

							Context::getId,

							Function.identity()));

		} finally {

			if (!errors.isEmpty()) {

				log.error(errors);

				errors.forEach(e -> e.printStackTrace());

			}

		}

	}

	private List<String> getContextProfiles(final Queue<Throwable> errors, final String xquery) throws IOException {

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

		return _quickSeachProfile(xquery, errors);

	}

	private String _getQuery(final ClassPathResource resource) throws IOException {

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

	}

	private String _isLookUp(final String xquery) throws ISLookUpException {

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

		final String res = isLookUpService.getResourceProfileByQuery(xquery);

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

		return res;

	}

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

		final List<String> res = Lists.newArrayList();

		try {

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

			final List<String> list = isLookUpService.quickSearchProfile(xquery);

			if (list != null) {

				res.addAll(list);

			}

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

		} catch (Throwable e) {

			errors.add(e);

			return Lists.newArrayList();

		} finally {

			return res;

		}

	}

	@CacheEvict(cacheNames = { "context-cache" }, allEntries = true)

	public void dropCache() {

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

	}

}