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

// **********************************************************************
//
// Copyright (c) 2003-2006 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.
//
// **********************************************************************

#include <Ice/Ice.h>
#include <IceGrid/Registry.h>
#include <Hello.h>

using namespace std;
using namespace Demo;

class SessionKeepAliveThread : public IceUtil::Thread, public IceUtil::Monitor<IceUtil::Mutex>
{
public:

    SessionKeepAliveThread(const IceGrid::SessionPrx& session, long timeout) :
	_session(session),
        _timeout(IceUtil::Time::seconds(timeout)),
        _destroy(false)
    {
    }

    virtual void
    run()
    {
        Lock sync(*this);
        while(!_destroy)
        {
            timedWait(_timeout);
            if(_destroy)
            {
	        break;
	    }
            try
            {
                _session->keepAlive();
            }
            catch(const Ice::Exception&)
            {
		break;
            }
        }
    }

    void
    destroy()
    {
        Lock sync(*this);
        _destroy = true;
        notify();
    }

private:

    const IceGrid::SessionPrx _session;
    const IceUtil::Time _timeout;
    bool _destroy;
};

typedef IceUtil::Handle<SessionKeepAliveThread> SessionKeepAliveThreadPtr;

class HelloClient : public Ice::Application
{
public:

    virtual int run(int, char*[]);
    virtual void interruptCallback(int);

private:

    void cleanup();
    void menu();
    string trim(const string&);

    IceUtil::Mutex _mutex;
    IceGrid::SessionPrx _session;
    SessionKeepAliveThreadPtr _keepAlive;
};

int
main(int argc, char* argv[])
{
    HelloClient app;
    return app.main(argc, argv, "config.client");
}

int
HelloClient::run(int argc, char* argv[])
{
    int status = EXIT_SUCCESS;

    //
    // Since this is an interactive demo we want the custom interrupt
    // callback to be called when the process is interrupted.
    //
    callbackOnInterrupt();

    IceGrid::RegistryPrx registry = IceGrid::RegistryPrx::checkedCast(
	communicator()->stringToProxy("DemoIceGrid/Registry"));
    if(!registry)
    {
        cerr << argv[0] << ": could not contact registry" << endl;
        return EXIT_FAILURE;
    }

    while(true)
    {
	cout << "This demo accepts any user-id / password combination.\n";

	string id;
	cout << "user id: " << flush;
	getline(cin, id);
	id = trim(id);

	string password;
	cout << "password: " << flush;
	getline(cin, password);
	password = trim(password);

	try
	{
	    IceUtil::Mutex::Lock sync(_mutex);
	    _session = registry->createSession(id, password);
 	    break;
	}
	catch(const IceGrid::PermissionDeniedException& ex)
	{
	    cout << "permission denied:\n" << ex.reason << endl;
	}
    }

    {
	IceUtil::Mutex::Lock sync(_mutex);
	_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 = HelloPrx::checkedCast(_session->allocateObjectById(communicator()->stringToIdentity("hello")));
	}
	catch(const IceGrid::ObjectNotRegisteredException&)
	{
	    hello = HelloPrx::checkedCast(_session->allocateObjectByType("::Demo::Hello"));
	}
	
	menu();
	
	char c;
	do
	{
	    try
	    {
		cout << "==> ";
		cin >> c;
		if(c == 't')
		{
		    hello->sayHello();
		}
		else if(c == 's')
		{
		    hello->shutdown();
		}
		else if(c == 'x')
		{
		    // Nothing to do
		}
		else if(c == '?')
		{
		    menu();
		}
		else
		{
		    cout << "unknown command `" << c << "'" << endl;
		    menu();
		}
	    }
	    catch(const Ice::Exception& ex)
	    {
		cerr << ex << endl;
	    }
	}
	while(cin.good() && c != 'x');
    }
    catch(const IceGrid::AllocationException& ex)
    {
	cerr << argv[0] << ": could not allocate object: " << ex.reason << endl;
	status = EXIT_FAILURE;
    }
    catch(...)
    {
	cerr << "unexpected exception" << endl;
	status = EXIT_FAILURE;
    }

    cleanup();
    return status;
}

void
HelloClient::interruptCallback(int)
{
    cleanup();

    try
    {
	communicator()->destroy();
    }
    catch(const IceUtil::Exception& ex)
    {
	cerr << appName() << ": " << ex << endl;
    }
    catch(...)
    {
	cerr << appName() << ": unknown exception" << endl;
    }
    exit(EXIT_SUCCESS);
}

void
HelloClient::cleanup()
{
    IceUtil::Mutex::Lock sync(_mutex);
    //
    // 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)
    {
	_keepAlive->destroy();
	_keepAlive->getThreadControl().join();
	_keepAlive = 0;
    }
    if(_session)
    {
	_session->destroy();
	_session = 0;
    }
}

void
HelloClient::menu()
{
    cout <<
	"usage:\n"
	"t: send greeting\n"
	"s: shutdown server\n"
	"x: exit\n"
	"?: help\n";
}

string
HelloClient::trim(const string& s)
{
    static const string delims = "\t\r\n ";
    string::size_type last = s.find_last_not_of(delims);
    if(last != string::npos)
    {
        return s.substr(s.find_first_not_of(delims), last+1);
    }
    return s;
}