Changeset View
Changeset View
Standalone View
Standalone View
src/avalanche/peermanager.h
| Show First 20 Lines • Show All 121 Lines • ▼ Show 20 Lines | using NodeSet = boost::multi_index_container< | ||||
| boost::multi_index::composite_key< | boost::multi_index::composite_key< | ||||
| Node, | Node, | ||||
| boost::multi_index::member<Node, PeerId, &Node::peerid>, | boost::multi_index::member<Node, PeerId, &Node::peerid>, | ||||
| boost::multi_index::member<Node, TimePoint, | boost::multi_index::member<Node, TimePoint, | ||||
| &Node::nextRequestTime>>>>>; | &Node::nextRequestTime>>>>>; | ||||
| NodeSet nodes; | NodeSet nodes; | ||||
| std::unique_ptr<CRollingBloomFilter> rejectedProofs; | |||||
| 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: | ||||
| PeerManager(); | |||||
| /** | /** | ||||
| * Node API. | * Node API. | ||||
| */ | */ | ||||
| bool addNode(NodeId nodeid, const Proof &proof, | bool addNode(NodeId nodeid, const 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; | ||||
| ▲ Show 20 Lines • Show All 41 Lines • ▼ Show 20 Lines | public: | ||||
| // Accssors. | // Accssors. | ||||
| 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; | ||||
| // For testing only | |||||
| void rejectProof(ProofId proofId); | |||||
| private: | private: | ||||
| PeerSet::iterator fetchOrCreatePeer(const Proof &proof); | PeerSet::iterator fetchOrCreatePeer(const 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 | ||||