D-Net

package eu.dnetlib.msro.workflows.nodes;

import java.util.Map;

import javax.annotation.Resource;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang.StringEscapeUtils;

import org.apache.commons.logging.LogFactory;

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

import com.google.common.collect.Maps;

import com.googlecode.sarasvati.Arc;

import com.googlecode.sarasvati.NodeToken;

import eu.dnetlib.common.logging.DnetLogger;

import eu.dnetlib.common.logging.LogMessage;

import eu.dnetlib.enabling.is.registry.rmi.ISRegistryService;

import eu.dnetlib.enabling.tools.ServiceLocator;

import eu.dnetlib.miscutils.datetime.DateUtils;

import eu.dnetlib.msro.workflows.util.WorkflowsConstants;

/**

 * The success node sets the "isCompletedSuccessfully" env var.

 *

 */

public class SuccessFailureNode extends SimpleJobNode {

	/**

	 * is completed successfully.

	 */

	private boolean success;

	@Resource(name = "registryLocator")

	private ServiceLocator<ISRegistryService> registryLocator;

	@Resource(name = "msroWorkflowLogger")

	private DnetLogger dnetLogger;

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

	@Override

	protected String execute(final NodeToken token) {

		final String profileId = token.getFullEnv().getAttribute(WorkflowsConstants.SYSTEM_WF_PROFILE_ID).trim();

		final String date = DateUtils.calculate_ISO8601(now);

		token.getProcess().getEnv().setAttribute(WorkflowsConstants.SYSTEM_END_HUMAN_DATE, date);

		final Map<String, String> params = mergeEnvAttributes(token);

		final LogMessage logMessage = dnetLogger.newLogMessage().addDetails(params);

		try {

			final String template = IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/msro/workflows/templates/workflow_status.xml.st"));

			final StringTemplate st = new StringTemplate(template);

			st.setAttribute("procId", StringEscapeUtils.escapeXml(params.get(WorkflowsConstants.SYSTEM_WF_PROCESS_ID)));

			st.setAttribute("date", StringEscapeUtils.escapeXml(date));

				st.setAttribute("error", StringEscapeUtils.escapeXml(params.get(WorkflowsConstants.SYSTEM_ERROR)));

			}

			registryLocator.getService().updateProfileNode(profileId, "//STATUS", st.toString());

			token.getProcess().getEnv().setAttribute(WorkflowsConstants.SYSTEM_COMPLETED_SUCCESSFULLY, isSuccess());

			logMessage.addDetail(WorkflowsConstants.SYSTEM_COMPLETED_SUCCESSFULLY, Boolean.toString(isSuccess()));

		} catch (Exception e) {

			log.error("Error updating workflow profile: " + profileId, e);

			token.getProcess().getEnv().setAttribute(WorkflowsConstants.SYSTEM_COMPLETED_SUCCESSFULLY, false);

			logMessage.addDetail(WorkflowsConstants.SYSTEM_COMPLETED_SUCCESSFULLY, Boolean.toString(false));

		}

		logMessage.flush();

		return Arc.DEFAULT_ARC;

	}

			for (String k : map.keySet()) {

				if (!StringUtils.isBlank(k) && (k.startsWith(WorkflowsConstants.DATAPROVIDER_PREFIX) || k.startsWith(WorkflowsConstants.MAIN_LOG_PREFIX))) {

					final String key = StringEscapeUtils.escapeXml(k);

					final String v = map.get(k);

					res.put(key, v != null ? StringEscapeUtils.escapeXml(v) : "null");

				}

		}

		return res;

	}

	private Map<String, String> mergeEnvAttributes(final NodeToken token) {

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

		for (String s : token.getEnv().getAttributeNames()) {

			map.put(s, token.getEnv().getAttribute(s));

		}

		for (String s : token.getFullEnv().getAttributeNames()) {

			map.put(s, token.getFullEnv().getAttribute(s));

		}

		return map;

	}

	public boolean isSuccess() {

		return success;

	}

	@Required

	public void setSuccess(final boolean success) {

		this.success = success;

	}

}