Differential D6828 Diff 22032 src/avalanche/processor.cpp

Changeset View

Standalone View

src/avalanche/processor.cpp

Show First 20 Lines • Show All 227 Lines • ▼ Show 20 Lines
bool AvalancheProcessor::registerVotes(		bool AvalancheProcessor::registerVotes(
NodeId nodeid, const AvalancheResponse &response,		NodeId nodeid, const AvalancheResponse &response,
std::vector<AvalancheBlockUpdate> &updates) {		std::vector<AvalancheBlockUpdate> &updates) {
{		{
// Save the time at which we can query again.		// Save the time at which we can query again.
auto w = peerSet.getWriteView();		auto w = peerSet.getWriteView();
auto it = w->find(nodeid);		auto it = w->find(nodeid);
if (it != w->end()) {		if (it != w->end()) {
w->modify(it, [&response](Peer &p) {		w->modify(it, [&response](Node &n) {
// FIXME: This will override the time even when we received an		// FIXME: This will override the time even when we received an
// old stale message. This should check that the message is		// old stale message. This should check that the message is
// indeed the most up to date one before updating the time.		// indeed the most up to date one before updating the time.
p.nextRequestTime =		n.nextRequestTime =
std::chrono::steady_clock::now() +		std::chrono::steady_clock::now() +
std::chrono::milliseconds(response.getCooldown());		std::chrono::milliseconds(response.getCooldown());
});		});
}		}
}		}

std::vector<CInv> invs;		std::vector<CInv> invs;

▲ Show 20 Lines • Show All 92 Lines • ▼ Show 20 Lines

bool AvalancheProcessor::addPeer(NodeId nodeid, int64_t score, CPubKey pubkey) {		bool AvalancheProcessor::addPeer(NodeId nodeid, int64_t score, CPubKey pubkey) {
return peerSet.getWriteView()		return peerSet.getWriteView()
->insert({nodeid, score, std::chrono::steady_clock::now(),		->insert({nodeid, score, std::chrono::steady_clock::now(),
std::move(pubkey)})		std::move(pubkey)})
.second;		.second;
}		}

CPubKey AvalancheProcessor::getPubKey(NodeId nodeid) const {		bool AvalancheProcessor::forNode(
		NodeId nodeid, std::function<bool(const Node &n)> func) const {
auto r = peerSet.getReadView();		auto r = peerSet.getReadView();
auto it = r->find(nodeid);		auto it = r->find(nodeid);
if (it == r->end()) {		return it != r->end() && func(*it);
return CPubKey();
}

return it->pubkey;
}		}

bool AvalancheProcessor::startEventLoop(CScheduler &scheduler) {		bool AvalancheProcessor::startEventLoop(CScheduler &scheduler) {
return eventLoop.startEventLoop(		return eventLoop.startEventLoop(
scheduler, [this]() { this->runEventLoop(); }, AVALANCHE_TIME_STEP);		scheduler, [this]() { this->runEventLoop(); }, AVALANCHE_TIME_STEP);
}		}

bool AvalancheProcessor::stopEventLoop() {		bool AvalancheProcessor::stopEventLoop() {
▲ Show 20 Lines • Show All 132 Lines • ▼ Show 20 Lines	do {
std::chrono::steady_clock::now() + queryTimeoutDuration;		std::chrono::steady_clock::now() + queryTimeoutDuration;
// Register the query.		// Register the query.
queries.getWriteView()->insert(		queries.getWriteView()->insert(
{pnode->GetId(), current_round, timeout, invs});		{pnode->GetId(), current_round, timeout, invs});
// Set the timeout.		// Set the timeout.
auto w = peerSet.getWriteView();		auto w = peerSet.getWriteView();
auto it = w->find(pnode->GetId());		auto it = w->find(pnode->GetId());
if (it != w->end()) {		if (it != w->end()) {
w->modify(it, [&timeout](Peer &p) {		w->modify(it, [&timeout](Node &n) {
p.nextRequestTime = timeout;		n.nextRequestTime = timeout;
});		});
}		}
}		}

// Send the query to the node.		// Send the query to the node.
connman->PushMessage(		connman->PushMessage(
pnode,		pnode,
CNetMsgMaker(pnode->GetSendVersion())		CNetMsgMaker(pnode->GetSendVersion())
Show All 17 Lines