requestHandler.cpp

 
/* Copyright (c) 2005-2009, Stefan Eilemann <eile@equalizergraphics.com> 
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License version 2.1 as published
 * by the Free Software Foundation.
 *  
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include "requestHandler.h"

#include "lock.h"
#include "scopedMutex.h"
#include "idPool.h"

#include <eq/base/debug.h>

using namespace std;

namespace eq
{
namespace base
{
RequestHandler::RequestHandler( const bool threadSafe )
        : _mutex( threadSafe ? new Lock() : 0 ),
          _requestID( 1 ) 
{}

RequestHandler::~RequestHandler()
{
    while( !_freeRequests.empty( ))
    {
        Request* request = _freeRequests.front();
        _freeRequests.pop_front();
        delete request;
    }

    delete _mutex;
    _mutex = 0;
}

uint32_t RequestHandler::registerRequest( void* data )
{
    ScopedMutex mutex( _mutex );
    if( !_mutex )
        CHECK_THREAD( _thread );

    Request* request;
    if( _freeRequests.empty( ))
        request = new Request;
    else
    {
        request = _freeRequests.front();
        _freeRequests.pop_front();
    }

    request->data = data;
    _requestID = ( _requestID + 1 ) % IDPool::MAX_CAPACITY;
    _requests[_requestID] = request;
    
    return _requestID;
}

void RequestHandler::unregisterRequest( const uint32_t requestID )
{
    ScopedMutex mutex( _mutex );
    if( !_mutex )
        CHECK_THREAD( _thread );

    RequestHash::iterator iter = _requests.find( requestID );
    if( iter == _requests.end( ))
        return;

    Request* request = iter->second;
    _requests.erase( iter );
    _freeRequests.push_front( request );

    if( _mutex )
        _mutex->unset();
}

bool RequestHandler::waitRequest( const uint32_t requestID, void*& rPointer,
                                  const uint32_t timeout )
{
    Request::Result result;
    if( !_waitRequest( requestID, result, timeout ))
        return false;

    rPointer = result.rPointer;
    return true;
}
bool RequestHandler::waitRequest( const uint32_t requestID, uint32_t& rUint32,
                                  const uint32_t timeout )
{
    Request::Result result;
    if( !_waitRequest( requestID, result, timeout ))
        return false;

    rUint32 = result.rUint32;
    return true;
}
bool RequestHandler::waitRequest( const uint32_t requestID, bool& rBool,
                                  const uint32_t timeout )
{
    Request::Result result;
    if( !_waitRequest( requestID, result, timeout ))
        return false;

    rBool = result.rBool;
    return true;
}
bool RequestHandler::waitRequest( const uint32_t requestID )
{
    Request::Result result;
    return _waitRequest( requestID, result, EQ_TIMEOUT_INDEFINITE );
}

bool RequestHandler::_waitRequest( const uint32_t requestID, 
                                   Request::Result& result,
                                   const uint32_t timeout )
{
    if( _mutex ) _mutex->set();
    else         CHECK_THREAD( _thread );

    RequestHash::iterator iter = _requests.find( requestID );
    if( iter == _requests.end( ))
    {
        if( _mutex ) _mutex->unset();
        return false;
    }

    Request*   request       = iter->second;
    if( _mutex ) _mutex->unset();

    const bool requestServed = request->lock.set( timeout );
    if( requestServed )
        result = request->result;

    ScopedMutex mutex( _mutex );
    iter = _requests.find( requestID );
    _requests.erase( iter );
    _freeRequests.push_front( request );
    
    return requestServed;
}

void* RequestHandler::getRequestData( const uint32_t requestID )
{
    ScopedMutex mutex( _mutex );
    RequestHash::const_iterator i = _requests.find( requestID );
    if( i == _requests.end( ))
        return 0;

    return i->second->data;
}

void RequestHandler::serveRequest( const uint32_t requestID, void* result )
{
    if( _mutex ) _mutex->set();
    RequestHash::const_iterator iter = _requests.find( requestID );
    EQASSERTINFO( iter != _requests.end(),
                  "Attempt to serve unregistered request " << requestID );

    Request* request = iter->second;
    if( _mutex ) _mutex->unset();

    request->result.rPointer = result;
    request->lock.unset();
}
void RequestHandler::serveRequest( const uint32_t requestID, uint32_t result )
{
    if( _mutex ) _mutex->set();
    RequestHash::const_iterator iter = _requests.find( requestID );
    EQASSERTINFO( iter != _requests.end(),
                  "Attempt to serve unregistered request " << requestID );

    Request* request = iter->second;
    if( _mutex ) _mutex->unset();

    request->result.rUint32 = result;
    request->lock.unset();
}
void RequestHandler::serveRequest( const uint32_t requestID, bool result )
{
    if( _mutex ) _mutex->set();
    RequestHash::const_iterator iter = _requests.find( requestID );
    EQASSERTINFO( iter != _requests.end(),
                  "Attempt to serve unregistered request " << requestID );

    Request* request = iter->second;
    if( _mutex ) _mutex->unset();

    request->result.rBool = result;
    request->lock.unset();
}
    
bool RequestHandler::isServed( const uint32_t requestID ) const
{
    ScopedMutex mutex( _mutex );
    RequestHash::const_iterator iter = _requests.find( requestID );
    if( iter == _requests.end( ))
        return false;

    Request* request = iter->second;
    if( !request->lock.test( ))
        return true;

    return false;
}

}
}