/**
 * $AIST_Release: 4.2.4 $
 * $AIST_Copyright:
 *  Copyright 2003, 2004, 2005, 2006 Grid Technology Research Center,
 *  National Institute of Advanced Industrial Science and Technology
 *  Copyright 2003, 2004, 2005, 2006 National Institute of Informatics
 *  
 *  Licensed under the Apache 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.apache.org/licenses/LICENSE-2.0
 *  
 *  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.
 *  $
 * $RCS_file$ $Revision: 1.7 $ $Date: 2005/10/25 10:48:23 $
 */
//package diag;

import org.apgrid.grpc.ng.NgGrpcFunctionHandle;
import org.gridforum.gridrpc.GrpcClient;
import org.gridforum.gridrpc.GrpcClientFactory;
import org.gridforum.gridrpc.GrpcException;
import org.gridforum.gridrpc.GrpcFunctionHandle;

/**
 * Diagnostic program for Ninf-G
 * This program checks API for session cancel
 */
public class CancelTest implements Runnable {
	private static final String targetExe = "cancel/cancel";
	/* specify time to sleep in second */
	private static final long SLEEP_TIME = 5;

	private GrpcFunctionHandle myHandle;

	public CancelTest (GrpcFunctionHandle myHandle){
		this.myHandle = myHandle;
	}
	
	public static void main (String args[]) {
		System.out.println("=====    CancelTest start     =====");
		GrpcClient client = null;
		GrpcFunctionHandle handle =	null;
		try {
			/* create GrpcClient */
			client = GrpcClientFactory.getClient("org.apgrid.grpc.ng.NgGrpcClient");

			/* check arguments */
			if (args.length < 1) {
				System.err.println("configFileName must be specified in argument");
			}

			/* activate GrpcClient */
			client.activate(args[0]);
			
			/* create GrpcHandle */
			handle = client.getFunctionHandle(targetExe);

			/* call function */
			System.out.print ("Cancel Test : ");
			CancelTest cancelTest = new CancelTest(handle);
			Thread cancelThread = new Thread(cancelTest);
			cancelThread.start();
			
			/* wait for a little */
			try {
				Thread.sleep(SLEEP_TIME * 1000);
			} catch (InterruptedException e) {
				e.printStackTrace();
			}
			
			/* cancel session */
			try {
				handle.cancel();
			} catch (GrpcException e) {
				e.printStackTrace();
			}
			
			/* check if cancel was succeed with grpc_probe */
			NgGrpcFunctionHandle ngFunctionHandle =
				(NgGrpcFunctionHandle) handle;
			if (ngFunctionHandle.isIdle() != true) {
				System.out.println ("FAILED!");
			} else {
				System.out.println ("OK!");
			}
		} catch (GrpcException e) {
			e.printStackTrace();
		} finally {
			/* dispose handle */
			if (handle != null) {
				try {
					handle.dispose();
				} catch (GrpcException e1) {
					e1.printStackTrace();
				}
			}
			try {
				if (client != null) {
					/* deactivate client */
					client.deactivate();
				}
			} catch (GrpcException e) {
				e.printStackTrace();
			}
		}
		System.out.println("===== CancelTest was finished =====");
	}

	/* (non-Javadoc)
	 * @see java.lang.Runnable#run()
	 */
	public void run() {
		try {
			/* start session */
			myHandle.call();
		} catch (GrpcException e) {
			e.printStackTrace();
		}
	}
}