Page MenuHomePhabricator
Differential D6793 Diff 21911 src/avalanche/peermanager.h

Changeset View

Standalone View

View Options

src/avalanche/peermanager.h

// Copyright (c) 2020 The Bitcoin developers // Copyright (c) 2020 The Bitcoin developers
// Distributed under the MIT software license, see the accompanying // Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php. // file COPYING or http://www.opensource.org/licenses/mit-license.php.
#ifndef BITCOIN_AVALANCHE_PEERMANAGER_H #ifndef BITCOIN_AVALANCHE_PEERMANAGER_H
#define BITCOIN_AVALANCHE_PEERMANAGER_H #define BITCOIN_AVALANCHE_PEERMANAGER_H
#include <atomic> #include <atomic>
#include <cstdint> #include <cstdint>
#include <vector> #include <vector>
static constexpr size_t NO_PEER = ~size_t(0); static constexpr size_t NO_PEER = ~size_t(0);
struct Slot { struct Slot {
private: private:
uint64_t start; uint64_t start;
uint64_t score; uint32_t score;
public: public:
Slot(uint64_t startIn, uint64_t scoreIn) : start(startIn), score(scoreIn) {} Slot(uint64_t startIn, uint32_t scoreIn) : start(startIn), score(scoreIn) {}
Slot withScore(uint64_t scoreIn) { return Slot(start, scoreIn); } Slot withScore(uint64_t scoreIn) { return Slot(start, scoreIn); }
uint64_t getStart() const { return start; } uint64_t getStart() const { return start; }
uint64_t getStop() const { return start + score; } uint64_t getStop() const { return start + score; }
uint64_t getScore() const { return score; } uint32_t getScore() const { return score; }
bool contains(uint64_t slot) const { bool contains(uint64_t slot) const {
return getStart() <= slot && slot < getStop(); return getStart() <= slot && slot < getStop();
} }
bool precedes(uint64_t slot) const { return slot >= getStop(); } bool precedes(uint64_t slot) const { return slot >= getStop(); }
bool follows(uint64_t slot) const { return getStart() > slot; } bool follows(uint64_t slot) const { return getStart() > slot; }
}; };
class PeerManager { class PeerManager {
std::vector<Slot> slots; std::vector<Slot> slots;
uint64_t slotCount = 0; uint64_t slotCount = 0;
uint64_t fragmentation = 0; uint64_t fragmentation = 0;
public: public:
void addPeer(uint64_t score); void addPeer(uint32_t score);
void rescorePeer(size_t i, uint64_t score); void rescorePeer(size_t i, uint32_t score);
void removePeer(size_t i) { rescorePeer(i, 0); } void removePeer(size_t i) { rescorePeer(i, 0); }
size_t selectPeer() const; size_t selectPeer() const;
// 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; }
}; };
/** /**
* This is an internal method that is exposed for testing purposes. * This is an internal method that is exposed for testing purposes.
*/ */
size_t selectPeerImpl(const std::vector<Slot> &slots, const uint64_t slot, size_t selectPeerImpl(const std::vector<Slot> &slots, const uint64_t slot,
const uint64_t max); const uint64_t max);
#endif // BITCOIN_AVALANCHE_PEERMANAGER_H #endif // BITCOIN_AVALANCHE_PEERMANAGER_H