// **********************************************************************
//
// Copyright (c) 2001
// MutableRealms, Inc.
// Huntsville, AL, USA
//
// All Rights Reserved
//
// **********************************************************************

#include <Ice/Object.h> // Not included in Ice/Ice.h
#include <Freeze/EvictorI.h>

using namespace std;
using namespace Ice;
using namespace Freeze;

Freeze::EvictorI::EvictorI(const DBPtr& db, EvictorPersistenceMode persistenceMode) :
    _evictorSize(10),
    _deactivated(false),
    _db(db),
    _persistenceMode(persistenceMode),
    _trace(0)
{
    PropertiesPtr properties = _db->getCommunicator()->getProperties();
    string value;

    value = properties->getProperty("Freeze.Trace.Evictor");
    if (!value.empty())
    {
	_trace = atoi(value.c_str());
    }
}

Freeze::EvictorI::~EvictorI()
{
    if (!_deactivated)
    {
	_db->getCommunicator()->getLogger()->warning("evictor has not been deactivated");
    }
}

DBPtr
Freeze::EvictorI::getDB()
{
    JTCSyncT<JTCMutex> sync(*this);

    if (_deactivated)
    {
	throw EvictorDeactivatedException();
    }

    return _db;
}

void
Freeze::EvictorI::setSize(Int evictorSize)
{
    JTCSyncT<JTCMutex> sync(*this);

    if (_deactivated)
    {
	throw EvictorDeactivatedException();
    }

    //
    // Ignore requests to set the evictor size to values smaller than zero.
    //
    if (evictorSize < 0)
    {
	return;
    }

    //
    // Update the evictor size.
    //
    _evictorSize = static_cast<map<string, EvictorElementPtr>::size_type>(evictorSize);

    //
    // Evict as many elements as necessary.
    //
    evict();
}

Int
Freeze::EvictorI::getSize()
{
    JTCSyncT<JTCMutex> sync(*this);

    if (_deactivated)
    {
	throw EvictorDeactivatedException();
    }

    return static_cast<Int>(_evictorSize);
}

void
Freeze::EvictorI::createObject(const string& ident, const ObjectPtr& servant)
{
    JTCSyncT<JTCMutex> sync(*this);

    if (_deactivated)
    {
	throw EvictorDeactivatedException();
    }

    //
    // Save the new Ice Object to the database.
    //
    _db->putServant(ident, servant);
    add(ident, servant);

    if (_trace >= 1)
    {
	ostringstream s;
	s << "created \"" << ident << "\"";
	_db->getCommunicator()->getLogger()->trace("Evictor", s.str());
    }

    //
    // Evict as many elements as necessary.
    //
    evict();
}

void
Freeze::EvictorI::destroyObject(const string& ident)
{
    JTCSyncT<JTCMutex> sync(*this);

    if (_deactivated)
    {
	throw EvictorDeactivatedException();
    }

    //
    // Delete the Ice Object from the database.
    //
    _db->delServant(ident);
    remove(ident);

    if (_trace >= 1)
    {
	ostringstream s;
	s << "destroyed \"" << ident << "\"";
	_db->getCommunicator()->getLogger()->trace("Evictor", s.str());
    }
}

void
Freeze::EvictorI::installServantInitializer(const ServantInitializerPtr& initializer)
{
    JTCSyncT<JTCMutex> sync(*this);

    if (_deactivated)
    {
	throw EvictorDeactivatedException();
    }

    _initializer = initializer;
}

ObjectPtr
Freeze::EvictorI::locate(const ObjectAdapterPtr& adapter, const string& ident, const string&,
			 LocalObjectPtr& cookie)
{
    JTCSyncT<JTCMutex> sync(*this);

    assert(_db);

    //
    // If this operation is called on a deactivated servant locator,
    // it's a bug in Ice.
    //
    assert(!_deactivated);

    EvictorElementPtr element;

    map<string, EvictorElementPtr>::iterator p = _evictorMap.find(ident);
    if (p != _evictorMap.end())
    {
	if (_trace >= 2)
	{
	    ostringstream s;
	    s << "found \"" << ident << "\" in the queue";
	    _db->getCommunicator()->getLogger()->trace("Evictor", s.str());
	}

	//
	// Ice Object found in evictor map. Push it to the front of
	// the evictor list, so that it will be evicted last.
	//
	element = p->second;
	_evictorList.erase(element->position);
	_evictorList.push_front(ident);
	element->position = _evictorList.begin();
    }
    else
    {
	if (_trace >= 2)
	{
	    ostringstream s;
	    s << "couldn't find \"" << ident << "\" in the queue\n"
	      << "loading \"" << ident << "\" from the database";
	    _db->getCommunicator()->getLogger()->trace("Evictor", s.str());
	}

	//
	// Load the Ice Object from database and create and add a
	// Servant for it.
	//
	ObjectPtr servant = _db->getServant(ident);
	if (!servant)
	{
	    //
            // The Ice Object with the given identity does not exist,
            // client will get an ObjectNotExistException.
	    //
	    return 0;
	}
	
	//
	// Add the new Servant to the evictor queue.
	//
	element = add(ident, servant);

	//
	// If an initializer is installed, call it now.
	//
	if (_initializer)
	{
	    _initializer->initialize(adapter, ident, servant);
	}
    }

    //
    // Increase the usage count of the evictor queue element.
    //
    ++element->usageCount;

    //
    // Evict as many elements as necessary.
    //
    evict();

    //
    // Set the cookie and return the servant for the Ice Object.
    //
    cookie = element;
    return element->servant;
}

void
Freeze::EvictorI::finished(const ObjectAdapterPtr&, const string& ident, const string& operation,
			   const ObjectPtr& servant, const LocalObjectPtr& cookie)
{
    JTCSyncT<JTCMutex> sync(*this);

    assert(_db);
    assert(servant);

    //
    // It's possible that the locator has been deactivated already. In
    // this case, _evictorSize is set to zero.
    //
    assert(!_deactivated || _evictorSize);

    //
    // Decrease the usage count of the evictor queue element.
    //
    EvictorElementPtr element = EvictorElementPtr::dynamicCast(cookie);
    assert(element);
    assert(element->usageCount >= 1);
    --element->usageCount;
    
    //
    // If we are in SaveAfterMutatingOperation mode, we must save the
    // Ice Object if this was a mutating call.
    //
    if (_persistenceMode == SaveAfterMutatingOperation)
    {
	if (servant->__isMutating(operation))
	{
	    _db->putServant(ident, servant);
	}
    }

    //
    // Evict as many elements as necessary.
    //
    evict();
}

void
Freeze::EvictorI::deactivate()
{
    JTCSyncT<JTCMutex> sync(*this);

    if (!_deactivated)
    {
	_deactivated = true;
	
	if (_trace >= 1)
	{
	    ostringstream s;
	    s << "deactivating, saving unsaved Ice Objects to the database";
	    _db->getCommunicator()->getLogger()->trace("Evictor", s.str());
	}

	//
	// Set the evictor size to zero, meaning that we will evict
	// everything possible.
	//
	_evictorSize = 0;
	evict();
    }
}

void
Freeze::EvictorI::evict()
{
    list<string>::reverse_iterator p = _evictorList.rbegin();

    //
    // With most STL implementations, _evictorMap.size() is faster
    // than _evictorList.size().
    //
    while (_evictorMap.size() > _evictorSize)
    {
	//
	// Get the last unused element from the evictor queue.
	//
	map<string, EvictorElementPtr>::iterator q;
	while(p != _evictorList.rend())
	{
	    q = _evictorMap.find(*p);
	    assert(q != _evictorMap.end());
	    if (q->second->usageCount == 0)
	    {
		break; // Fine, Servant is not in use.
	    }
	    ++p;
	}
	if(p == _evictorList.rend())
	{
	    //
	    // All Servants are active, can't evict any further.
	    //
	    break;
	}
	string ident = *p;
	EvictorElementPtr element = q->second;

	//
	// If we are in SaveUponEviction mode, we must save the Ice
	// Object that is about to be evicted to persistent store.
	//
	if (_persistenceMode == SaveUponEviction)
	{
	    _db->putServant(ident, element->servant);
	}

	//
	// Remove last unused element from the evictor queue.
	//
	_evictorList.erase(element->position);
	_evictorMap.erase(q);
	++p;

	if (_trace >= 2)
	{
	    ostringstream s;
	    s << "evicted \"" << ident << "\" from the queue\n"
	      << "number of elements in the queue: " << _evictorMap.size();
	    _db->getCommunicator()->getLogger()->trace("Evictor", s.str());
	}
    }

    //
    // If we're deactivated, and if there are no more elements to
    // evict, set _db to zero to break cyclic object
    // dependencies.
    //
    if (_deactivated && _evictorMap.empty())
    {
	assert(_evictorList.empty());
	assert(_evictorSize == 0);
	_db = 0;
    }
}

Freeze::EvictorI::EvictorElementPtr
Freeze::EvictorI::add(const string& ident, const ObjectPtr& servant)
{
    //
    // Ignore the request if the Ice Object is already in the queue.
    //
    map<string, EvictorElementPtr>::const_iterator p = _evictorMap.find(ident);
    if (p != _evictorMap.end())
    {
	return p->second;
    }    

    //
    // Add an Ice Object with its Servant to the evictor queue.
    //
    _evictorList.push_front(ident);
    EvictorElementPtr element = new EvictorElement;
    element->servant = servant;
    element->position = _evictorList.begin();
    element->usageCount = 0;    
    _evictorMap[ident] = element;
    return element;
}

void
Freeze::EvictorI::remove(const string& ident)
{
    //
    // If the Ice Object is currently in the evictor, remove it.
    //
    map<string, EvictorElementPtr>::iterator p = _evictorMap.find(ident);
    if (p != _evictorMap.end())
    {
	_evictorList.erase(p->second->position);
	_evictorMap.erase(p);
    }
}