catapult-server-docs/ScheduledHarvesterTask_8h_source.html

#pragma once

#include "Harvester.h"

#include "catapult/chain/ChainFunctions.h"

#include "catapult/disruptor/DisruptorTypes.h"

#include "catapult/model/Elements.h"

#include "catapult/model/EntityInfo.h"

#include "catapult/model/RangeTypes.h"

#include "catapult/functions.h"


namespace catapult { namespace harvesting {


    struct ScheduledHarvesterTaskOptions {

        predicate<> HarvestingAllowed;


        supplier<std::shared_ptr<const model::BlockElement>> LastBlockElementSupplier;


        chain::TimeSupplier TimeSupplier;


        consumer<model::BlockRange&&, const disruptor::ProcessingCompleteFunc&> RangeConsumer;

    };


    class ScheduledHarvesterTask {

    public:

        using TaskOptions = ScheduledHarvesterTaskOptions;


    public:

        explicit 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_isAnyHarvestedBlockPending(false)

        {}


    public:

        void harvest();


    private:

        const decltype(TaskOptions::HarvestingAllowed) m_harvestingAllowed;

        const decltype(TaskOptions::LastBlockElementSupplier) m_lastBlockElementSupplier;

        const decltype(TaskOptions::TimeSupplier) m_timeSupplier;

        const decltype(TaskOptions::RangeConsumer) m_rangeConsumer;

        std::unique_ptr<Harvester> m_pHarvester;


        std::atomic_bool m_isAnyHarvestedBlockPending;

    };

}}