#include "LoadManager.h"

#include <boost/test/unit_test.hpp>
#include <boost/thread.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>

#include "Log.h"
#include "Config.h"
#include "Application.h"

SETUP_LOG();

static volatile int* uptimePtr = NULL;

int upTime()
{
	static time_t firstCall = time(NULL);
	if (uptimePtr != NULL)
		return *uptimePtr;
	return static_cast<int>(time(NULL) - firstCall);
}

LoadManager::LoadManager(int creditRate, int creditLimit, int debitWarn, int debitLimit) :
		mCreditRate(creditRate), mCreditLimit(creditLimit), mDebitWarn(debitWarn), mDebitLimit(debitLimit),
		mShutdown(false), mArmed(false), mUptime(0), mDeadLock(0), mCosts(LT_MAX)
{
	addLoadCost(LoadCost(LT_InvalidRequest,		-10,		LC_CPU | LC_Network));
	addLoadCost(LoadCost(LT_RequestNoReply,		-1,		LC_CPU | LC_Disk));
	addLoadCost(LoadCost(LT_InvalidSignature,	-100,	LC_CPU));
	addLoadCost(LoadCost(LT_UnwantedData,		-5,		LC_CPU | LC_Network));
	addLoadCost(LoadCost(LT_BadData,			-20,		LC_CPU));

	addLoadCost(LoadCost(LT_NewTrusted,			-10,		0));
	addLoadCost(LoadCost(LT_NewTransaction,		-2,		0));
	addLoadCost(LoadCost(LT_NeededData,			-10,		0));

	addLoadCost(LoadCost(LT_RequestData,		-5,		LC_Disk | LC_Network));
	addLoadCost(LoadCost(LT_CheapQuery,			-1,		LC_CPU));

}

void LoadManager::init()
{
	if (uptimePtr == NULL)
		uptimePtr = static_cast<volatile int *>(&mUptime);
	boost::thread(boost::bind(&LoadManager::threadEntry, this)).detach();
}

LoadManager::~LoadManager()
{
	if (uptimePtr == &mUptime)
		uptimePtr = NULL;
	{
		boost::mutex::scoped_lock sl(mLock);
		mShutdown = true;
	}

	do
	{
		boost::this_thread::sleep(boost::posix_time::milliseconds(100));
		{
			boost::mutex::scoped_lock sl(mLock);
			if (!mShutdown)
				return;
		}
	}
	while (1);
}

void LoadManager::noDeadLock()
{
	boost::mutex::scoped_lock sl(mLock);
	mDeadLock = mUptime;
}

int LoadManager::getCreditRate() const
{
	boost::mutex::scoped_lock sl(mLock);
	return mCreditRate;
}

int LoadManager::getCreditLimit() const
{
	boost::mutex::scoped_lock sl(mLock);
	return mCreditLimit;
}

int LoadManager::getDebitWarn() const
{
	boost::mutex::scoped_lock sl(mLock);
	return mDebitWarn;
}

int LoadManager::getDebitLimit() const
{
	boost::mutex::scoped_lock sl(mLock);
	return mDebitLimit;
}

void LoadManager::setCreditRate(int r)
{
	boost::mutex::scoped_lock sl(mLock);
	mCreditRate = r;
}

void LoadManager::setCreditLimit(int r)
{
	boost::mutex::scoped_lock sl(mLock);
	mCreditLimit = r;
}

void LoadManager::setDebitWarn(int r)
{
	boost::mutex::scoped_lock sl(mLock);
	mDebitWarn = r;
}

void LoadManager::setDebitLimit(int r)
{
	boost::mutex::scoped_lock sl(mLock);
	mDebitLimit = r;
}

void LoadManager::canonicalize(LoadSource& source, int now) const
{
	if (source.mLastUpdate != now)
	{
		if (source.mLastUpdate < now)
		{
			source.mBalance += mCreditRate * (now - source.mLastUpdate);
			if (source.mBalance > mCreditLimit)
			{
				source.mBalance = mCreditLimit;
				source.mLogged = false;
			}
		}
		source.mLastUpdate = now;
	}
}

bool LoadManager::shouldWarn(LoadSource& source) const
{
	{
		boost::mutex::scoped_lock sl(mLock);

		int now = upTime();
		canonicalize(source, now);
		if (source.isPrivileged() || (source.mBalance > mDebitWarn) || (source.mLastWarning == now))
			return false;

		source.mLastWarning = now;
	}
	logWarning(source.getName());
	return true;
}

bool LoadManager::shouldCutoff(LoadSource& source) const
{
	{
		boost::mutex::scoped_lock sl(mLock);
		int now = upTime();
		canonicalize(source, now);
		if (source.isPrivileged() || (source.mBalance > mDebitLimit))
			return false;
		if (source.mLogged)
			return true;
		source.mLogged = true;
	}
	logDisconnect(source.getName());
	return true;
}

bool LoadManager::adjust(LoadSource& source, LoadType t) const
{ // FIXME: Scale by category
	LoadCost cost = mCosts[static_cast<int>(t)];
	return adjust(source, cost.mCost);
}

bool LoadManager::adjust(LoadSource& source, int credits) const
{ // return: true = need to warn/cutoff

	// We do it this way in case we want to add exponential decay later
	int now = upTime();
	canonicalize(source, now);
	source.mBalance += credits;
	if (source.mBalance > mCreditLimit)
		source.mBalance = mCreditLimit;

	if (source.isPrivileged()) // privileged sources never warn/cutoff
		return false;

	if ((source.mBalance >= mDebitLimit) && (source.mLastWarning == now)) // no need to warn
		return false;

	return true;
}

uint64 LoadFeeTrack::mulDiv(uint64 value, uint32 mul, uint64 div)
{ // compute (value)*(mul)/(div) - avoid overflow but keep precision
	static uint64 boundary = (0x00000000FFFFFFFF);
	if (value > boundary)							// Large value, avoid overflow
		return (value / div) * mul;
	else											// Normal value, preserve accuracy
		return (value * mul) / div;
}

uint64 LoadFeeTrack::scaleFeeLoad(uint64 fee, uint64 baseFee, uint32 referenceFeeUnits, bool bAdmin)
{
	static uint64 midrange(0x00000000FFFFFFFF);

	bool big = (fee > midrange);
	if (big)				// big fee, divide first to avoid overflow
		fee /= baseFee;
	else					// normal fee, multiply first for accuracy
		fee *= referenceFeeUnits;

	uint32 feeFactor = std::max(mLocalTxnLoadFee, mRemoteTxnLoadFee);

	// Let admins pay the normal fee until the local load exceeds four times the remote
	if (bAdmin && (feeFactor > mRemoteTxnLoadFee) && (feeFactor < (4 * mRemoteTxnLoadFee)))
		feeFactor = mRemoteTxnLoadFee;

	{
		boost::mutex::scoped_lock sl(mLock);
		fee = mulDiv(fee, feeFactor, lftNormalFee);
	}

	if (big)				// Fee was big to start, must now multiply
		fee *= referenceFeeUnits;
	else					// Fee was small to start, mst now divide
		fee /= baseFee;

	return fee;
}

uint64 LoadFeeTrack::scaleFeeBase(uint64 fee, uint64 baseFee, uint32 referenceFeeUnits)
{
	return mulDiv(fee, referenceFeeUnits, baseFee);
}

uint32 LoadFeeTrack::getRemoteFee()
{
	boost::mutex::scoped_lock sl(mLock);
	return mRemoteTxnLoadFee;
}

uint32 LoadFeeTrack::getLocalFee()
{
	boost::mutex::scoped_lock sl(mLock);
	return mLocalTxnLoadFee;
}

uint32 LoadFeeTrack::getLoadFactor()
{
		boost::mutex::scoped_lock sl(mLock);
		return std::max(mLocalTxnLoadFee, mRemoteTxnLoadFee);
}

void LoadFeeTrack::setRemoteFee(uint32 f)
{
	boost::mutex::scoped_lock sl(mLock);
	mRemoteTxnLoadFee = f;
}

bool LoadFeeTrack::raiseLocalFee()
{
	boost::mutex::scoped_lock sl(mLock);

	if (++raiseCount < 2)
		return false;

	uint32 origFee = mLocalTxnLoadFee;

	if (mLocalTxnLoadFee < mRemoteTxnLoadFee) // make sure this fee takes effect
		mLocalTxnLoadFee = mRemoteTxnLoadFee;

	mLocalTxnLoadFee += (mLocalTxnLoadFee / lftFeeIncFraction); // increment by 1/16th

	if (mLocalTxnLoadFee > lftFeeMax)
		mLocalTxnLoadFee = lftFeeMax;

	if (origFee == mLocalTxnLoadFee)
		return false;
	cLog(lsDEBUG) << "Local load fee raised from " << origFee << " to " << mLocalTxnLoadFee;
	return true;
}

bool LoadFeeTrack::isLoaded()
{
	boost::mutex::scoped_lock sl(mLock);
	return (raiseCount != 0) || (mLocalTxnLoadFee != lftNormalFee);
}

bool LoadFeeTrack::lowerLocalFee()
{
	boost::mutex::scoped_lock sl(mLock);
	uint32 origFee = mLocalTxnLoadFee;
	raiseCount = 0;

	mLocalTxnLoadFee -= (mLocalTxnLoadFee / lftFeeDecFraction ); // reduce by 1/4

	if (mLocalTxnLoadFee < lftNormalFee)
		mLocalTxnLoadFee = lftNormalFee;

	if (origFee == mLocalTxnLoadFee)
		return false;
	cLog(lsDEBUG) << "Local load fee lowered from " << origFee << " to " << mLocalTxnLoadFee;
	return true;
}

Json::Value LoadFeeTrack::getJson(uint64 baseFee, uint32 referenceFeeUnits)
{
	Json::Value j(Json::objectValue);

	{
		boost::mutex::scoped_lock sl(mLock);

		// base_fee = The cost to send a "reference" transaction under no load, in millionths of a Ripple
		j["base_fee"] = Json::Value::UInt(baseFee);

		// load_fee = The cost to send a "reference" transaction now, in millionths of a Ripple
		j["load_fee"] = Json::Value::UInt(
			mulDiv(baseFee, std::max(mLocalTxnLoadFee, mRemoteTxnLoadFee), lftNormalFee));
	}

	return j;
}

int LoadManager::getUptime()
{
	boost::mutex::scoped_lock sl(mLock);
	return mUptime;
}

static void LogDeadLock(int dlTime)
{
	cLog(lsWARNING) << "Server stalled for " << dlTime << " seconds.";
}

void LoadManager::threadEntry()
{
	NameThread("loadmgr");
	boost::posix_time::ptime t = boost::posix_time::microsec_clock::universal_time();
	while (1)
	{
		{
			boost::mutex::scoped_lock sl(mLock);
			if (mShutdown)
			{
				mShutdown = false;
				return;
			}
			++mUptime;

			int dlTime = mUptime - mDeadLock;
			if (mArmed && (dlTime >= 10))
			{
				if ((dlTime % 10) == 0)
				{
					boost::thread(BIND_TYPE(&LogDeadLock, dlTime)).detach();
				}

				assert (dlTime < 500);
			}

		}

		bool change;
		if (theApp->getJobQueue().isOverloaded())
		{
			cLog(lsINFO) << theApp->getJobQueue().getJson(0);
			change = theApp->getFeeTrack().raiseLocalFee();
		}
		else
			change = theApp->getFeeTrack().lowerLocalFee();
		if (change)
			theApp->getOPs().reportFeeChange();

		t += boost::posix_time::seconds(1);
		boost::posix_time::time_duration when = t - boost::posix_time::microsec_clock::universal_time();

		if ((when.is_negative()) || (when.total_seconds() > 1))
		{
			cLog(lsWARNING) << "time jump";
			t = boost::posix_time::microsec_clock::universal_time();
		}
		else
			boost::this_thread::sleep(when);
	}
}

void LoadManager::logWarning(const std::string& source) const
{
	if (source.empty())
		cLog(lsDEBUG) << "Load warning from empty source";
	else
		cLog(lsINFO) << "Load warning: " << source;
}

void LoadManager::logDisconnect(const std::string& source) const
{
	if (source.empty())
		cLog(lsINFO) << "Disconnect for empty source";
	else
		cLog(lsWARNING) << "Disconnect for: " << source;
}

BOOST_AUTO_TEST_SUITE(LoadManager_test)

BOOST_AUTO_TEST_CASE(LoadFeeTrack_test)
{
	cLog(lsDEBUG) << "Running load fee track test";

	Config d; // get a default configuration object
	LoadFeeTrack l;

	BOOST_REQUIRE_EQUAL(l.scaleFeeBase(10000, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE), 10000);
	BOOST_REQUIRE_EQUAL(l.scaleFeeLoad(10000, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE, false), 10000);
	BOOST_REQUIRE_EQUAL(l.scaleFeeBase(1, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE), 1);
	BOOST_REQUIRE_EQUAL(l.scaleFeeLoad(1, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE, false), 1);

	// Check new default fee values give same fees as old defaults
	BOOST_REQUIRE_EQUAL(l.scaleFeeBase(d.FEE_DEFAULT, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE), 10);
	BOOST_REQUIRE_EQUAL(l.scaleFeeBase(d.FEE_ACCOUNT_RESERVE, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE), 200 * SYSTEM_CURRENCY_PARTS);
	BOOST_REQUIRE_EQUAL(l.scaleFeeBase(d.FEE_OWNER_RESERVE, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE), 50 * SYSTEM_CURRENCY_PARTS);
	BOOST_REQUIRE_EQUAL(l.scaleFeeBase(d.FEE_NICKNAME_CREATE, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE), 1000);
	BOOST_REQUIRE_EQUAL(l.scaleFeeBase(d.FEE_OFFER, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE), 10);
	BOOST_REQUIRE_EQUAL(l.scaleFeeBase(d.FEE_CONTRACT_OPERATION, d.FEE_DEFAULT, d.TRANSACTION_FEE_BASE), 1);

}

BOOST_AUTO_TEST_SUITE_END()

// vim:ts=4