Differential D9616 Diff 28715 src/avalanche/peermanager.h

Changeset View

Standalone View

src/avalanche/peermanager.h

Show First 20 Lines • Show All 60 Lines • ▼ Show 20 Lines

struct Peer {		struct Peer {
PeerId peerid;		PeerId peerid;
uint32_t index = -1;		uint32_t index = -1;
uint32_t node_count = 0;		uint32_t node_count = 0;

std::shared_ptr<Proof> proof;		std::shared_ptr<Proof> proof;

Peer(PeerId peerid_, Proof proof_)		Peer(PeerId peerid_, std::shared_ptr<Proof> proof_)
: peerid(peerid_), proof(std::make_shared<Proof>(std::move(proof_))) {}		: peerid(peerid_), proof(std::move(proof_)) {}

const ProofId &getProofId() const { return proof->getId(); }		const ProofId &getProofId() const { return proof->getId(); }
uint32_t getScore() const { return proof->getScore(); }		uint32_t getScore() const { return proof->getScore(); }
};		};

struct proof_index {		struct proof_index {
using result_type = ProofId;		using result_type = ProofId;
result_type operator()(const Peer &p) const { return p.proof->getId(); }		result_type operator()(const Peer &p) const { return p.proof->getId(); }
▲ Show 20 Lines • Show All 44 Lines • ▼ Show 20 Lines	class PeerManager {

static constexpr int SELECT_PEER_MAX_RETRY = 3;		static constexpr int SELECT_PEER_MAX_RETRY = 3;
static constexpr int SELECT_NODE_MAX_RETRY = 3;		static constexpr int SELECT_NODE_MAX_RETRY = 3;

public:		public:
/**		/**
* Node API.		* Node API.
*/		*/
bool addNode(NodeId nodeid, const Proof &proof,		bool addNode(NodeId nodeid, const std::shared_ptr<Proof> &proof,
const Delegation &delegation);		const Delegation &delegation);
bool removeNode(NodeId nodeid);		bool removeNode(NodeId nodeid);

bool forNode(NodeId nodeid, std::function<bool(const Node &n)> func) const;		bool forNode(NodeId nodeid, std::function<bool(const Node &n)> func) const;
bool updateNextRequestTime(NodeId nodeid, TimePoint timeout);		bool updateNextRequestTime(NodeId nodeid, TimePoint timeout);

/**		/**
* Randomly select a node to poll.		* Randomly select a node to poll.
*/		*/
NodeId selectNode();		NodeId selectNode();

/**		/**
* Update the peer set when a new block is connected.		* Update the peer set when a new block is connected.
*/		*/
void updatedBlockTip();		void updatedBlockTip();

/****************************************************		/****************************************************
* Functions which are public for testing purposes. *		* Functions which are public for testing purposes. *
****************************************************/		****************************************************/
/**		/**
* Provide the PeerId associated with the given proof. If the peer does not		* Provide the PeerId associated with the given proof. If the peer does not
* exist, then it is created.		* exist, then it is created.
*/		*/
PeerId getPeerId(const Proof &proof);		PeerId getPeerId(const std::shared_ptr<Proof> &proof);

/**		/**
* Remove an existing peer.		* Remove an existing peer.
*/		*/
bool removePeer(const PeerId peerid);		bool removePeer(const PeerId peerid);

/**		/**
* Randomly select a peer to poll.		* Randomly select a peer to poll.
Show All 14 Lines	public:
// Accessors.		// Accessors.
uint64_t getSlotCount() const { return slotCount; }		uint64_t getSlotCount() const { return slotCount; }
uint64_t getFragmentation() const { return fragmentation; }		uint64_t getFragmentation() const { return fragmentation; }

std::vector<Peer> getPeers() const;		std::vector<Peer> getPeers() const;
std::vector<NodeId> getNodeIdsForPeer(PeerId peerId) const;		std::vector<NodeId> getNodeIdsForPeer(PeerId peerId) const;

private:		private:
PeerSet::iterator fetchOrCreatePeer(const Proof &proof);		PeerSet::iterator fetchOrCreatePeer(const std::shared_ptr<Proof> &proof);
bool addNodeToPeer(const PeerSet::iterator &it);		bool addNodeToPeer(const PeerSet::iterator &it);
bool removeNodeFromPeer(const PeerSet::iterator &it, uint32_t count = 1);		bool removeNodeFromPeer(const PeerSet::iterator &it, uint32_t count = 1);
};		};

/**		/**
* This is an internal method that is exposed for testing purposes.		* This is an internal method that is exposed for testing purposes.
*/		*/
PeerId selectPeerImpl(const std::vector<Slot> &slots, const uint64_t slot,		PeerId selectPeerImpl(const std::vector<Slot> &slots, const uint64_t slot,
const uint64_t max);		const uint64_t max);

} // namespace avalanche		} // namespace avalanche

#endif // BITCOIN_AVALANCHE_PEERMANAGER_H		#endif // BITCOIN_AVALANCHE_PEERMANAGER_H