D-Net

package eu.dnetlib.msro.workflows.procs;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.concurrent.CopyOnWriteArrayList;

import org.apache.commons.lang3.math.NumberUtils;

import com.fasterxml.jackson.annotation.JsonIgnore;

import eu.dnetlib.msro.workflows.Graph;

import eu.dnetlib.msro.workflows.ProcessStatus;

import eu.dnetlib.msro.workflows.WorkflowProcessInfo;

import eu.dnetlib.msro.workflows.logs.WfLogEntry;

import eu.dnetlib.services.async.AsyncServerCallback;

/**

 * Created by michele on 19/11/15.

 */

public class WorkflowProcess extends WorkflowProcessInfo implements Comparable<WorkflowProcess> {

	@JsonIgnore

	private Graph graph;

	@JsonIgnore

	private AsyncServerCallback callback;

	@JsonIgnore

	private List<Token> tokens = new CopyOnWriteArrayList<>();

	@JsonIgnore

	private Map<String, List<Token>> pausedJoinNodeTokens = new HashMap<String, List<Token>>();

	public void kill() {

		setStatus(ProcessStatus.KILLED);

	}

	public boolean isTerminated() {

		switch (getStatus()) {

		case SUCCESS:

		case FAILURE:

		case KILLED:

			return true;

		default:

			return false;

		}

	}

	@Override

	public int compareTo(final WorkflowProcess wp) {

		return NumberUtils.compare(getPriority(), wp.getPriority());

	}

	public Graph getGraph() {

		return graph;

	}

	public void setGraph(final Graph graph) {

		this.graph = graph;

	}

	public AsyncServerCallback getCallback() {

		return callback;

	}

	public void setCallback(final AsyncServerCallback callback) {

		this.callback = callback;

	}

	public List<Token> getTokens() {

		return tokens;

	}

	public void setTokens(final List<Token> tokens) {

		this.tokens = tokens;

	}

	public Map<String, List<Token>> getPausedJoinNodeTokens() {

		return pausedJoinNodeTokens;

	}

	public void setPausedJoinNodeTokens(final Map<String, List<Token>> pausedJoinNodeTokens) {

		this.pausedJoinNodeTokens = pausedJoinNodeTokens;

	}

	public WfLogEntry prepareWfLog() {

		final WfLogEntry wfLog = new WfLogEntry();

		wfLog.setProcId(getId());

		wfLog.setName(getName());

		wfLog.setFamily(getFamily());

		wfLog.setPriority(getPriority());

		wfLog.setTemplate(isTemplate());

		wfLog.setProfileId(getProfileId());

		wfLog.setParent(getParentProfileId());

		wfLog.setDatasourceId(getDsId());

		wfLog.setDatasourceApi(getDsInterface());

		wfLog.setDatasourceName(getDsName());

		wfLog.setStartDate(getStartDate());

		wfLog.setEndDate(getEndDate());

		wfLog.setSuccess(getStatus() == ProcessStatus.SUCCESS);

		wfLog.setWorkerUrl(getWorkerUrl());

		if (getStatus() == ProcessStatus.FAILURE) {

			wfLog.setError(getError());

			wfLog.setStacktrace(getErrorStacktrace());

		}

		wfLog.setOutputParams(getOutputParams());

		return wfLog;

	}

}