D-Net

package eu.dnetlib.data.mdstore.utils;

import java.io.IOException;

import java.io.StringWriter;

import java.io.Writer;

import java.util.Map;

import com.google.common.collect.Maps;

import eu.dnetlib.clients.is.InformationServiceClient;

import eu.dnetlib.miscutils.datetime.DateUtils;

import eu.dnetlib.msro.controllers.MsroWorkerController;

import freemarker.template.Configuration;

import freemarker.template.Template;

import freemarker.template.TemplateException;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

import org.springframework.stereotype.Component;

/**

 * Created by claudio on 24/03/2017.

 */

@Component

public class MDStoreProfileManager {

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

	@Autowired

	private InformationServiceClient isClient;

	@Autowired

	private MsroWorkerController msroWorkerController;

	@Autowired

	private Configuration freemakerCfg;

	public String registerProfile(String format, String layout, String interpretation) throws MDStoreException {

		final Map<String, Object> input = Maps.newHashMap();

		input.put("serviceId", msroWorkerController.getProfileId());

		input.put("format", format);

		input.put("layout", layout);

		input.put("interpretation", interpretation);

		log.debug(String.format("registering new mdstore profile %s-%s-%s", format, layout, interpretation));

		return isClient.register(processTemplate(input));

	}

	public void deleteProfile(final String mdId) throws MDStoreException {

		isClient.deleteProfile(mdId);

	}

	public void updateSize(final String mdId, final long size) {

		final String now = DateUtils.now_ISO8601();

		isClient.xupdate("for $x in //RESOURCE_PROFILE[.//RESOURCE_IDENTIFIER/@value = '" + mdId + "']"

				+ "return update value $x//LAST_STORAGE_DATE with '" + now + "'");

		isClient.xupdate("for $x in //RESOURCE_PROFILE[.//RESOURCE_IDENTIFIER/@value = '" + mdId + "']"

				+ "return update value $x//NUMBER_OF_RECORDS with '" + size + "'");

	}

	// HELPERS

	private String processTemplate(final Map<String, Object> input) throws MDStoreException {

		try(Writer writer = new StringWriter()) {

			final Template template = freemakerCfg.getTemplate("mdstore.ftl");

			template.process(input, writer);

			return writer.toString();

		} catch (IOException | TemplateException e) {

			throw new MDStoreException(e);

		}

	}

}