D-Net

package eu.dnetlib.services.async;

import java.util.HashMap;

import java.util.Map;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.springframework.stereotype.Component;

import org.springframework.web.client.RestTemplate;

@Component

public class AsyncClientUtils {

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

	private static final Map<String, AsyncClientCallback> activeRemoteCalls = new HashMap<>();

	public void invokeRemoteMethod(final String remoteBaseUrl,

			final String remoteMethod,

			final String localBaseUrl,

			final Object o,

			final AsyncClientCallback callback) {

		final String url = remoteBaseUrl + "/async/method/" + remoteMethod + ((StringUtils.isNotBlank(localBaseUrl)) ? "?caller=" + localBaseUrl : "");

		log.info("invoking async method: " + url);

		final String id = (new RestTemplate()).postForObject(url, o, String.class);

		if (callback != null) {

			activeRemoteCalls.put(id, callback);

		}

	}

	public boolean processResponse(final String id, final AsyncResponse response) {

		if (activeRemoteCalls.containsKey(id)) {

			final AsyncClientCallback callback = activeRemoteCalls.get(id);

			callback.updateLastInvocation();

			switch (response.getStatus()) {

			case ASSIGNED:

			case RUNNING:

				callback.onGoing(response);

				break;

			case SUCCESS:

				callback.onDone(response);

				activeRemoteCalls.remove(id);

				break;

			case FAILED:

				callback.onFailed(response);

				activeRemoteCalls.remove(id);

				break;

			default:

				log.warn("Status " + response.getStatus() + " not managed");

				break;

			}

			return true;

		} else {

			log.warn("Unexpected message: " + id);

			return false;

		}

	}

	public void clearRemoteCalls() {

		activeRemoteCalls.clear();

	}

	public void deleteRemoteCall(final String id) {

		activeRemoteCalls.remove(id);

	}

	public static Map<String, AsyncClientCallback> getActiveremotecalls() {

		return activeRemoteCalls;

	}

}