package org.distributeme.core.qos;

import org.distributeme.core.ClientSideCallContext;
import org.distributeme.core.interceptor.ClientSideRequestInterceptor;
import org.distributeme.core.interceptor.InterceptionContext;
import org.distributeme.core.interceptor.InterceptorResponse;

/**
 * This interceptor inspects slow requests on client side and marks them as failed.
 *
 * @author lrosenberg
 * @since 06.02.15 01:09
 * @version $Id: $Id
 */
public class ClientSideMarkLongRequestAsFailedInterceptor implements ClientSideRequestInterceptor{

	/**
	 * Attribute name for request start.
	 */
	public static final String START_ATTRIBUTE = ClientSideMarkLongRequestAsFailedInterceptor.class.getName()+"-"+"start";
	/**
	 * Timeout property for setting this configuration from outside the VM.
	 */
	public static final String TIMEOUT_PROPERTY = "availabilityTestingFailureTimeout";

	//default 1 minute
	/**
	 * Default setting for the timeout. If property TIMEOUT_PROPERTY is not set, 60 seconds value apply.
	 */
	private long TIMEOUT = 60000L;

	/**
	 * <p>Constructor for ClientSideMarkLongRequestAsFailedInterceptor.</p>
	 */
	public ClientSideMarkLongRequestAsFailedInterceptor(){
		String timeoutProperty = System.getProperty(TIMEOUT_PROPERTY);
		if (timeoutProperty!= null && timeoutProperty.length()>0)
			TIMEOUT = Long.parseLong(timeoutProperty);
	}

	/** {@inheritDoc} */
	@Override
	public InterceptorResponse afterServiceCall(ClientSideCallContext context, InterceptionContext iContext) {
		long start = (Long)iContext.getLocalStore().get(START_ATTRIBUTE);
		long end = System.currentTimeMillis();
		long duration = end - start;
		if (duration > TIMEOUT){
			System.out.println("%%% SLOW CALL "+duration);
			return InterceptorResponse.RETURN_AND_FAIL;
		}
		return InterceptorResponse.CONTINUE;
	}

	/** {@inheritDoc} */
	@Override
	public InterceptorResponse beforeServiceCall(ClientSideCallContext context, InterceptionContext iContext) {
		iContext.getLocalStore().put(START_ATTRIBUTE, System.currentTimeMillis());
		return InterceptorResponse.CONTINUE;
	}
}