/**
 * Copyright 2005-2011 The Kuali Foundation
 *
 * Licensed under the Educational Community License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.opensource.org/licenses/ecl2.php
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.kuali.rice.ksb.messaging;

import org.kuali.rice.ksb.api.messaging.AsynchronousCall;
import org.kuali.rice.ksb.api.messaging.AsynchronousCallback;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.xml.namespace.QName;

public class TestCallback implements AsynchronousCallback {
	/**
	 * 
	 */
	private static final long serialVersionUID = 8558495197148582373L;

	public static boolean callbackCalled;

	public static List<Long> CURRENT_MILLIS_WHEN_CALLED;

	public static int NUMBER_CALL_BACKS = 0;

	public static Map<QName, Integer> SERVICE_CALL_COUNT_TRACKED = new HashMap<QName, Integer>();
	public static Map<QName, List<AsynchronousCall>> SERVICES_CALLS_TRACKED = new HashMap<QName, List<AsynchronousCall>>();

	static {
		CURRENT_MILLIS_WHEN_CALLED = new ArrayList<Long>();
	}

	private int numberCallbacks = 0;

	public static boolean isCallbackCalled() {
		return callbackCalled;
	}

	public static void setCallbackCalled(boolean callbackCalled) {
		TestCallback.callbackCalled = callbackCalled;
	}

	public synchronized void callback(Serializable returnObject, AsynchronousCall methodCall) {
		CURRENT_MILLIS_WHEN_CALLED.add(System.currentTimeMillis());
		NUMBER_CALL_BACKS++;
		setCallbackCalled(true);
		this.numberCallbacks++;
		QName serviceName = methodCall.getServiceConfiguration().getServiceName();
		Integer callCount = SERVICE_CALL_COUNT_TRACKED.get(serviceName);
		if (callCount == null) {
			SERVICE_CALL_COUNT_TRACKED.put(methodCall.getServiceConfiguration().getServiceName(), 1);
		} else {
			SERVICE_CALL_COUNT_TRACKED.put(methodCall.getServiceConfiguration().getServiceName(), callCount + 1);
		}
		
		List<AsynchronousCall> serviceCallsTracked = SERVICES_CALLS_TRACKED.get(serviceName);
		if (serviceCallsTracked == null) {
			serviceCallsTracked = new ArrayList<AsynchronousCall>();
			SERVICES_CALLS_TRACKED.put(serviceName, serviceCallsTracked);
		}
		serviceCallsTracked.add(methodCall);
		
		System.out.println("!!!Callback called number callbacks " + this.numberCallbacks);
	}

	public static void clearCallbacks() {
		NUMBER_CALL_BACKS = 0;
		SERVICE_CALL_COUNT_TRACKED = new HashMap<QName, Integer>();
	}

	/**
	 * sometimes it's more convenient to use a non static counter above when
	 * doing in memory queueing. Other times when doing persistent async
	 * messaging a static counter is needed. Everything could be converted to
	 * this method if the tests using the above method clear the static count
	 * before putting testing against callbacks.
	 * 
	 * @param callbacks
	 * @param millisDelay
	 */
	public void pauseUntilNumberCallbacksUsingStaticCounter(int callbacks, QName serviceName) {
		int numPauses = 0;
		while (true) {
			synchronized (this.getClass()) {
				if (serviceName == null) {
					if (NUMBER_CALL_BACKS >= callbacks) {
						System.out.println("!!!Returning number callback met");
						return;
					}
				} else {
					if (SERVICE_CALL_COUNT_TRACKED.get(serviceName) != null && SERVICE_CALL_COUNT_TRACKED.get(serviceName) >= callbacks) {
						System.out.println("!!!Returning number callback met for service " + serviceName);
						return;
					}
						// attributes will not be an exact match.
					for (Map.Entry<QName, Integer> serviceCall : SERVICE_CALL_COUNT_TRACKED.entrySet()) {
						if (serviceCall.getKey().getLocalPart().lastIndexOf(serviceName.getLocalPart()) > -1 && serviceCall.getKey().getNamespaceURI().equals(serviceName.getNamespaceURI()) && serviceCall.getValue() >= callbacks) {
							System.out.println("!!!Returning number callback met for service " + serviceName);
							return;
						}
					}
				}
			}
			if (numPauses > 60 * 1) {
				return;
			}
			try {
				numPauses++;
				System.out.println("!!!Test callback pausing for 1 second");
				Thread.sleep(1000);
			} catch (InterruptedException e) {
			    // nothing to do
			}
		}
	}
}
