-
Notifications
You must be signed in to change notification settings - Fork 33
/
ScheduledHarvesterTask.cpp
56 lines (49 loc) · 2.25 KB
/
ScheduledHarvesterTask.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
/**
*** Copyright (c) 2016-2019, Jaguar0625, gimre, BloodyRookie, Tech Bureau, Corp.
*** Copyright (c) 2020-present, Jaguar0625, gimre, BloodyRookie.
*** All rights reserved.
***
*** This file is part of Catapult.
***
*** Catapult is free software: you can redistribute it and/or modify
*** it under the terms of the GNU Lesser General Public License as published by
*** the Free Software Foundation, either version 3 of the License, or
*** (at your option) any later version.
***
*** Catapult 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 Lesser General Public License for more details.
***
*** You should have received a copy of the GNU Lesser General Public License
*** along with Catapult. If not, see <http://www.gnu.org/licenses/>.
**/
#include "ScheduledHarvesterTask.h"
#include "catapult/utils/HexFormatter.h"
namespace catapult { namespace harvesting {
ScheduledHarvesterTask::ScheduledHarvesterTask(const ScheduledHarvesterTaskOptions& options, std::unique_ptr<Harvester>&& pHarvester)
: m_harvestingAllowed(options.HarvestingAllowed)
, m_lastBlockElementSupplier(options.LastBlockElementSupplier)
, m_timeSupplier(options.TimeSupplier)
, m_rangeConsumer(options.RangeConsumer)
, m_pHarvester(std::move(pHarvester))
, m_pIsAnyHarvestedBlockPending(std::make_shared<std::atomic_bool>(false))
{}
void ScheduledHarvesterTask::harvest() {
if (*m_pIsAnyHarvestedBlockPending || !m_harvestingAllowed())
return;
auto pLastBlockElement = m_lastBlockElementSupplier();
auto pBlock = m_pHarvester->harvest(*pLastBlockElement, m_timeSupplier());
if (!pBlock)
return;
CATAPULT_LOG(info) << "successfully harvested block at " << pBlock->Height << " with signer " << pBlock->SignerPublicKey;
*m_pIsAnyHarvestedBlockPending = true;
// flag needs to be captured as shared_ptr in order to avoid race condition when harvesting service
// is shutdown before dispatcher service and block completes processing in interim period
m_rangeConsumer(model::BlockRange::FromEntity(std::move(pBlock)), [pIsAnyBlockPending = m_pIsAnyHarvestedBlockPending](
auto,
auto) {
*pIsAnyBlockPending = false;
});
}
}}