D-Net

package eu.dnetlib.msro.controllers;

import java.util.HashMap;

import java.util.Map;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestParam;

import org.springframework.web.bind.annotation.RestController;

import com.fasterxml.jackson.databind.ObjectMapper;

import eu.dnetlib.clients.msro.Workflow;

import eu.dnetlib.clients.msro.WorkflowTemplate;

import eu.dnetlib.enabling.annotations.DnetService;

import eu.dnetlib.enabling.annotations.DnetServiceType;

import eu.dnetlib.msro.exceptions.MSROException;

import eu.dnetlib.msro.workflows.WorkflowRef;

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

import eu.dnetlib.services.BaseService;

import eu.dnetlib.services.async.AsyncMethodException;

import eu.dnetlib.services.async.AsyncRunnable;

import eu.dnetlib.services.async.AsyncServerCallback;

import eu.dnetlib.services.async.HasAsyncMethods;

@RestController

@RequestMapping("/msro")

@DnetService(DnetServiceType.msro)

public class MsroController extends BaseService implements HasAsyncMethods {

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

	@Autowired

	private WorkflowDispatcher dispatcher;

	@Override

	public AsyncRunnable prepareThread(final String method, final String jsonParams, final AsyncServerCallback callback) throws AsyncMethodException {

		final ObjectMapper mapper = new ObjectMapper();

		return new AsyncRunnable() {

			@Override

			public Map<String, String> prepare() {

				return new HashMap<>();

			}

			@Override

			public void execute() throws AsyncMethodException {

				final WorkflowRef ref;

				try {

					switch (method) {

					case "startWorkflow":

						dispatcher.startWorkflow(mapper.readValue(jsonParams, Workflow.class), getBaseUrl());

						break;

					case "startWorkflowTemplate":

						dispatcher.startWorkflowTemplate(mapper.readValue(jsonParams, WorkflowTemplate.class), getBaseUrl());

						break;

					default:

						log.warn("Invalid method: " + method);

						throw new AsyncMethodException("Invalid method: " + method);

					}

				} catch (final Throwable e) {

					log.warn("Error executing method: " + method, e);

					throw new AsyncMethodException("Error executing method: " + method, e);

				}

			}

		};

	}

	@RequestMapping("startWorkflow")

	public WorkflowRef startWorkflow(

			@RequestParam final String wfId,

			@RequestParam(required = false) final String parent,

			@RequestParam(required = false) final String dsId,

			@RequestParam(required = false) final String ifaceId) throws MSROException {

		return dispatcher.startWorkflow(wfId, parent, dsId, ifaceId, null);

	}

}