path: root/java/demo/IceGrid/allocate/Client.java

// **********************************************************************
//
// Copyright (c) 2003-2007 ZeroC, Inc. All rights reserved.
//
// This copy of Ice is licensed to you under the terms described in the
// ICE_LICENSE file included in this distribution.
//
// **********************************************************************

import Demo.*;

public class Client extends Ice.Application
{
    class ShutdownHook extends Thread
    {
	public void
	run()
	{
	    cleanup();
	    try
	    {
		communicator().destroy();
	    }
	    catch(Ice.LocalException ex)
	    {
		ex.printStackTrace();
	    }
	}
    }

    static private class SessionKeepAliveThread extends Thread
    {
        SessionKeepAliveThread(IceGrid.SessionPrx session, long timeout)
	{
	    _session = session;
	    _timeout = timeout;
	    _terminated = false;
	}

	synchronized public void
	run()
	{
	    while(!_terminated)
	    {
		try
		{
		    wait(_timeout);
		}
		catch(InterruptedException e)
		{
		}
		if(_terminated)
		{
		    break;
		}
		try
		{
		    _session.keepAlive();
		}
		catch(Ice.LocalException ex)
		{
		    break;
		}
	    }
	}

	synchronized private void
	terminate()
	{
	    _terminated = true;
	    notify();
	}

	final private IceGrid.SessionPrx _session;
	final private long _timeout;
	private boolean _terminated;
    }

    private void
    menu()
    {
	System.out.println(
	    "usage:\n" +
	    "t: send greeting\n" +
	    "s: shutdown server\n" +
	    "x: exit\n" +
	    "?: help\n");
    }

    public int
    run(String[] args)
    {
	//
	// Since this is an interactive demo we want to clear the
	// Application installed interrupt callback and install our
	// own shutdown hook.
	//
	setInterruptHook(new ShutdownHook());

	int status = 0;
	IceGrid.RegistryPrx registry = 
	    IceGrid.RegistryPrxHelper.checkedCast(communicator().stringToProxy("DemoIceGrid/Registry"));
	if(registry == null)
	{
	    System.err.println("could not contact registry");
	    return 1;
	}

	java.io.BufferedReader in = new java.io.BufferedReader(new java.io.InputStreamReader(System.in));
	while(true)
	{
	    System.out.println("This demo accepts any user-id / password combination.");

	    try
	    {
		String id;
		System.out.print("user id: ");
		System.out.flush();
		id = in.readLine();
		
		String pw;
		System.out.print("password: ");
		System.out.flush();
		pw = in.readLine();
		
		try
		{
		    synchronized(this)
		    {
			_session = registry.createSession(id, pw);
		    }
		    break;
		}
		catch(IceGrid.PermissionDeniedException ex)
		{
		    System.out.println("permission denied:\n" + ex.reason);
		}
	    }
	    catch(java.io.IOException ex)
	    {
		ex.printStackTrace();
	    }
	}

	synchronized(this)
	{
	    _keepAlive = new SessionKeepAliveThread(_session, registry.getSessionTimeout() / 2);
	    _keepAlive.start();
	}

	try
	{
	    //
	    // First try to retrieve object by identity, which will
	    // work if the application-single.xml descriptor is
	    // used. Otherwise we retrieve object by type, which will
	    // succeed if the application-multiple.xml descriptor is
	    // used.
	    //
	    HelloPrx hello;
	    try
	    {
		hello = HelloPrxHelper.checkedCast(
		    _session.allocateObjectById(communicator().stringToIdentity("hello")));
	    }
	    catch(IceGrid.ObjectNotRegisteredException ex)
	    {
		hello = HelloPrxHelper.checkedCast(_session.allocateObjectByType("::Demo::Hello"));
	    }

	    menu();
	    
	    String line = null;
	    do
	    {
		try
		{
		    System.out.print("==> ");
		    System.out.flush();
		    line = in.readLine();
		    if(line == null)
		    {
			break;
		    }
		    if(line.equals("t"))
		    {
			hello.sayHello();
		    }
		    else if(line.equals("s"))
		    {
			hello.shutdown();
		    }
		    else if(line.equals("x"))
		    {
			// Nothing to do
		    }
		    else if(line.equals("?"))
		    {
			menu();
		    }
		    else
		    {
			System.out.println("unknown command `" + line + "'");
			menu();
		    }
		}
		catch(java.io.IOException ex)
		{
		    ex.printStackTrace();
		}
		catch(Ice.LocalException ex)
		{
		    ex.printStackTrace();
		}
	    }
	    while(!line.equals("x"));
	}
	catch(IceGrid.AllocationException ex)
	{
	    System.err.println("could not allocate object: " + ex.reason);
	    status = 1;
	}
	catch(Exception ex)
	{
	    System.err.println("expected exception: " + ex);
	    status = 1;
	}

	cleanup();

	return status;
    }

    synchronized void
    cleanup()
    {
	//
	// Destroy the keepAlive thread and the sesion object otherwise
	// the session will be kept allocated until the timeout occurs.
	// Destroying the session will release all allocated objects.
	//
	if(_keepAlive != null)
	{
	    _keepAlive.terminate();
	    try
	    {
		_keepAlive.join();
	    }
	    catch(InterruptedException e)
	    {
	    }
	    _keepAlive = null;
	}
	if(_session != null)
	{
	    _session.destroy();
	    _session = null;
	}
    }

    public static void
    main(String[] args)
    {
	Client app = new Client();
	int status = app.main("Client", args, "config.client");
	System.exit(status);
    }

    SessionKeepAliveThread _keepAlive;
    IceGrid.SessionPrx _session;
}