StrandOwnerLifetimeExtender.h

Go to the documentation of this file.

 
#pragma once
#include "catapult/utils/WrappedWithOwnerDecorator.h"
#include <boost/asio.hpp>
 
namespace catapult { namespace thread {
 
    template<typename TOwner>
    class StrandOwnerLifetimeExtender {
    public:
        explicit StrandOwnerLifetimeExtender(boost::asio::io_context::strand& strand) : m_strand(strand)
        {}
 
    public:
        template<typename THandler>
        auto wrap(const std::shared_ptr<TOwner>& pOwner, THandler handler) {
            // when wrap is called the returned callback needs to extend the lifetime of pOwner
            utils::WrappedWithOwnerDecorator<THandler> wrappedHandler(pOwner, std::move(handler));
            return m_strand.wrap(wrappedHandler);
        }
 
    public:
        template<typename THandler>
        void post(const std::shared_ptr<TOwner>& pOwner, THandler handler) {
            // ensure all handlers extend the lifetime of pOwner and post to a strand
            boost::asio::post(m_strand, [pOwner, handler]() {
                handler(pOwner);
            });
        }
 
    private:
        boost::asio::io_context::strand& m_strand;
    };
}}