Code to track ledgers we're acquiring from remote node, allow

callbacks to be attached to them, and trigger their state machines.
This commit is contained in:
JoelKatz
2012-01-30 12:56:43 -08:00
parent 7885439bb4
commit e47cba3369

62
LedgerAcquire.h Normal file
View File

@@ -0,0 +1,62 @@
#ifndef __LEDGERACQUIRE__
#define __LEDGERACQUIRE__
#include <vector>
#include "boost/function.hpp"
#include "Ledger.h"
#include "Peer.h"
class LedgerAcquire
{ // A ledger we are trying to acquire
public:
typedef boost::shared_ptr<LedgerAcquire> pointer;
protected:
boost::recursive_mutex mLock;
Ledger::pointer mLedger;
uint256 mHash;
bool mComplete, mFailed, mHaveBase, mHaveState, mHaveTransactions;
std::vector< boost::function<LedgerAcquire::pointer> > mOnComplete;
std::list<boost::weak_ptr<Peer> > mPeers; // peers known to have this ledger
void done(void);
void timerEntry(void);
public:
LedgerAcquire(const uint256& hash);
const uint256& getHash() const { return mHash; }
bool isComplete() const { return mComplete; }
bool isFailed() const { return mFailed; }
bool isBase() const { return mHaveBase; }
bool isAcctStComplete() const { return mHaveState; }
bool isTransComplete() const { return mHaveTransactions; }
Ledger::pointer getLedger() { return mLedger; }
void addTrigger(boost::function<LedgerAcquire::pointer>);
void peerHash(Peer::pointer);
void takeBase(std::vector<unsigned char> data);
void takeTxNode(std::list<uint256> hashes, std::list<std::vector<unsigned char> > data);
void takeAsNode(std::list<uint160> hashes, std::list<std::vector<unsigned char> > data);
void takeTx(std::list<uint256> hashes, std::list<std::vector<unsigned char> > data);
};
class LedgerAcquireMaster
{
protected:
boost::mutex mLock;
std::map<uint256, LedgerAcquire::pointer> mLedgers;
public:
LedgerAcquireMaster() { ; }
LedgerAcquire::pointer findCreate(const uint256& hash);
LedgerAcquire::pointer find(const uint256& hash);
bool hasLedger(const uint256& ledgerHash);
};
#endif