mirror of
https://github.com/XRPLF/rippled.git
synced 2025-12-06 17:27:55 +00:00
96 lines
2.5 KiB
C++
96 lines
2.5 KiB
C++
#ifndef LOADSOURCE__H
|
|
#define LOADSOURCE__H
|
|
|
|
#include <boost/thread/mutex.hpp>
|
|
|
|
#include "types.h"
|
|
|
|
class LoadSource
|
|
{ // a single endpoint that can impose load
|
|
friend class LoadManager;
|
|
|
|
public:
|
|
|
|
// load source flags
|
|
static const int lsfPrivileged = 1;
|
|
static const int lsfOutbound = 2; // outbound connection
|
|
|
|
protected:
|
|
int mBalance;
|
|
int mFlags;
|
|
time_t mLastUpdate;
|
|
time_t mLastWarning;
|
|
|
|
public:
|
|
LoadSource() : mBalance(0), mFlags(0), mLastWarning(0)
|
|
{ mLastUpdate = time(NULL); }
|
|
|
|
bool isPrivileged() const { return (mFlags & lsfPrivileged) != 0; }
|
|
void setPrivileged() { mFlags |= lsfPrivileged; }
|
|
int getBalance() const { return mBalance; }
|
|
|
|
void setOutbound() { mFlags |= lsfOutbound; }
|
|
bool isOutbound() const { return (mFlags & lsfOutbound) != 0; }
|
|
};
|
|
|
|
|
|
class LoadManager
|
|
{ // a collection of load sources
|
|
protected:
|
|
|
|
int mCreditRate; // credits gained/lost per second
|
|
int mCreditLimit; // the most credits a source can have
|
|
int mDebitWarn; // when a source drops below this, we warn
|
|
int mDebitLimit; // when a source drops below this, we cut it off (should be negative)
|
|
|
|
mutable boost::mutex mLock;
|
|
|
|
void canonicalize(LoadSource&, const time_t now) const;
|
|
|
|
public:
|
|
|
|
LoadManager(int creditRate, int creditLimit, int debitWarn, int debitLimit) :
|
|
mCreditRate(creditRate), mCreditLimit(creditLimit), mDebitWarn(debitWarn), mDebitLimit(debitLimit) { ; }
|
|
|
|
int getCreditRate() const;
|
|
int getCreditLimit() const;
|
|
int getDebitWarn() const;
|
|
int getDebitLimit() const;
|
|
void setCreditRate(int);
|
|
void setCreditLimit(int);
|
|
void setDebitWarn(int);
|
|
void setDebitLimit(int);
|
|
|
|
bool shouldWarn(LoadSource&) const;
|
|
bool shouldCutoff(LoadSource&) const;
|
|
bool adjust(LoadSource&, int credits) const; // return value: false=balance okay, true=warn/cutoff
|
|
};
|
|
|
|
class LoadFeeTrack
|
|
{ // structure that tracks our current fee/load schedule
|
|
protected:
|
|
|
|
static const int lftNormalFee = 256; // 256 is the minimum/normal load factor
|
|
static const int lftFeeIncFraction = 16; // increase fee by 1/16
|
|
static const int lftFeeDecFraction = 16; // decrease fee by 1/16
|
|
static const int lftFeeMax = lftNormalFee * 1000000;
|
|
|
|
uint32 mLocalTxnLoadFee; // Scale factor, lftNormalFee = normal fee
|
|
uint32 mRemoteTxnLoadFee; // Scale factor, lftNormalFee = normal fee
|
|
|
|
public:
|
|
|
|
LoadFeeTrack() : mLocalTxnLoadFee(lftNormalFee), mRemoteTxnLoadFee(lftNormalFee) { ; }
|
|
|
|
uint64 scaleFee(uint64 fee);
|
|
|
|
void setRemoteFee(uint32);
|
|
void raiseLocalFee();
|
|
void lowerLocalFee();
|
|
};
|
|
|
|
|
|
#endif
|
|
|
|
// vim:ts=4
|