D-Net

package eu.dnetlib.services.async;

import java.util.HashMap;

import java.util.Map;

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 Object o, final AsyncClientCallback callback) {

		final String url = remoteBaseUrl + "/async/method/" + remoteMethod;

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

		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;

	}

}