tomFlowee
bc47a700a4
As we moved most of the creation of a BufferPool to be via the Streaming::pool() method, which uses a thread-local, it makes sense to start cleaning up the design and make it more modern C++. The above mentioned method would return a reference and you'd see loads of places use `auto &pool =` which is less than ideal. As the number of places where we actually instantiate a BufferPool goes down, the usage of some sort of smart pointer makes more sense. This now makes all APIs use BufferPool be wrapped in a shared_ptr.
93 lines
3.4 KiB
C++
93 lines
3.4 KiB
C++
/*
|
|
* This file is part of the Flowee project
|
|
* Copyright (C) 2018-2021 Tom Zander <tom@flowee.org>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#include "BlockNotificationService.h"
|
|
#include <APIProtocol.h>
|
|
|
|
#include <Logger.h>
|
|
#include <Message.h>
|
|
#include <chain.h>
|
|
#include <primitives/Block.h>
|
|
|
|
#include <streaming/MessageBuilder.h>
|
|
|
|
BlockNotificationService::BlockNotificationService()
|
|
: NetworkService(Api::BlockNotificationService),
|
|
m_pool(std::make_shared<Streaming::BufferPool>())
|
|
{
|
|
ValidationNotifier().addListener(this);
|
|
}
|
|
|
|
BlockNotificationService::~BlockNotificationService()
|
|
{
|
|
ValidationNotifier().removeListener(this);
|
|
}
|
|
|
|
void BlockNotificationService::syncAllTransactionsInBlock(const Block &, CBlockIndex *index)
|
|
{
|
|
const auto list = remotes<RemoteWithBool>(&NetworkService::filterRemoteWithBool);
|
|
if (list.empty())
|
|
return;
|
|
m_pool->reserve(45);
|
|
Streaming::MessageBuilder builder(m_pool);
|
|
builder.add(Api::BlockNotification::BlockHash, index->GetBlockHash());
|
|
builder.add(Api::BlockNotification::BlockHeight, index->nHeight);
|
|
Message message(builder.message(Api::BlockNotificationService, Api::BlockNotification::NewBlockOnChain));
|
|
|
|
for (auto &subinfo : list) {
|
|
subinfo->connection.send(message);
|
|
}
|
|
}
|
|
|
|
void BlockNotificationService::chainReorged(CBlockIndex *oldTip, const std::vector<Block> &revertedBlocks)
|
|
{
|
|
const auto list = remotes<RemoteWithBool>(&NetworkService::filterRemoteWithBool);
|
|
if (list.empty())
|
|
return;
|
|
|
|
/*
|
|
* since the service already sends out which block is the new one in a separate message,
|
|
* all we will do in this one is notify them which blocks have been removed.
|
|
*/
|
|
m_pool->reserve(revertedBlocks.size() * 42);
|
|
CBlockIndex *index = oldTip;
|
|
Streaming::MessageBuilder builder(m_pool);
|
|
for (size_t i = 0; i < revertedBlocks.size(); ++i) {
|
|
assert(index);
|
|
builder.add(Api::BlockNotification::BlockHash, index->GetBlockHash());
|
|
builder.add(Api::BlockNotification::BlockHeight, index->nHeight);
|
|
index = index->pprev;
|
|
}
|
|
Message message(builder.message(Api::BlockNotificationService, Api::BlockNotification::BlocksRemoved));
|
|
|
|
for (auto &subinfo : list) {
|
|
subinfo->connection.send(message);
|
|
}
|
|
}
|
|
|
|
void BlockNotificationService::onIncomingMessage(Remote *remote_, const Message &message, const EndPoint &ep)
|
|
{
|
|
assert(dynamic_cast<RemoteWithBool*>(remote_));
|
|
RemoteWithBool *remote = static_cast<RemoteWithBool*>(remote_);
|
|
if (message.messageId() == Api::BlockNotification::Subscribe) {
|
|
logInfo(Log::BlockNotifactionService) << "Remote" << ep.connectionId << "Wants to hear about blocks";
|
|
remote->enabled = true;
|
|
}
|
|
else if (message.messageId() == Api::BlockNotification::Unsubscribe)
|
|
remote->enabled = false;
|
|
}
|