measurable.hpp

Go to the documentation of this file.

 
#ifndef RTCTK_COMPONENTFRAMEWORK_MEASURABLE_HPP
#define RTCTK_COMPONENTFRAMEWORK_MEASURABLE_HPP
 
#include <optional>
 
#include <rtctk/componentFramework/exceptions.hpp>
#include <rtctk/componentFramework/jsonPayload.hpp>
#include <rtctk/componentFramework/measCmdsImpl.hpp>
#include <rtctk/componentFramework/rtcComponent.hpp>
#include <rtctk/componentFramework/stopToken.hpp>
 
namespace rtctk::componentFramework {
namespace detail {
class MeasureContext {
public:
    MeasureContext(rad::cii::Request<std::string, std::string> const& request)
        : m_request(request), m_arg() {
        m_arg = JsonPayload::parse(m_request.GetRequestPayload());
    }
    MeasureContext(MeasureContext&&) noexcept = default;
 
    JsonPayload const& GetArg() const noexcept {
        return m_arg;
    }
 
    void SetResult(JsonPayload const& result) {
        m_result = result.dump();
    }
 
    void SendReply() noexcept {
        try {
            m_request.SetReplyValue(m_result);
        } catch (std::exception const& exception) {
            LOG4CPLUS_ERROR(
                GetLogger(),
                "MeasureContext::SendReply: Failed to send reply: " << exception.what());
        } catch (...) {
            LOG4CPLUS_ERROR(
                GetLogger(),
                "MeasureContext::SendReply: Failed to send reply with unknown exception.");
        }
    }
 
    void SendErrorReply(std::exception_ptr eptr) noexcept {
        try {
            m_request.SetException(
                RequestFailed(NestedExceptionPrinter(std::move(eptr)).Str()));
        } catch (RtctkException const& exception) {
            LOG4CPLUS_ERROR(
                GetLogger(),
                "MeasureContext::SendErrorReply: Failed to send reply: " << exception.what());
        } catch (...) {
            LOG4CPLUS_ERROR(
                GetLogger(),
                "MeasureContext::SendErrorReply: Failed to send reply with unknown exception.");
        }
    }
 
private:
    rad::cii::Request<std::string, std::string> m_request;
    JsonPayload m_arg;
    std::string m_result;
};
}  // namespace detail
 
template <typename Super>
struct Measurable : Super {
    static_assert(std::is_base_of_v<RtcComponent, Super>, "'Measurable' requires 'RtcComponent'");
 
    class BizLogicIf : public Super::BizLogicIf {
    public:
        virtual JsonPayload ActivityMeasuring(StopToken stop_token, JsonPayload const& arg) {
            (void)stop_token;
            return JsonPayload("42");
        }
 
        virtual bool GuardMeasuringAllowed(JsonPayload const& arg) {
            return true;
        }
    };
 
    class InputStage : public Super::InputStage {
    public:
        using Super::InputStage::InputStage;
 
        void Start() override {
            Super::InputStage::Start();
 
            MeasCmdsImpl::Register(this->m_replier, this->m_engine);
        }
    };
 
    class OutputStage : public Super::OutputStage {
    public:
        OutputStage(StateMachineEngine& engine, BizLogicIf& bl) : Super::OutputStage(engine, bl) {
            // Handlers ###################################################################
 
            engine.RegisterRejectHandler<events::Measure, RequestRejected>();
 
            m_measure_success_handler = [this] {
                this->m_engine.PostEvent(std::make_unique<events::MeasureDone>());
            };
 
            m_measure_error_handler = [this](std::exception_ptr eptr) {
                this->m_engine.PostEvent(std::make_unique<events::MeasureError>(eptr));
            };
 
            // No Disable in states ###############################################################
 
            this->m_no_disable_in_states.push_back("On:Operational:MeasureBusy");
 
            // No Update in states ################################################################
 
            this->m_no_update_in_states.push_back("On:Operational:MeasureBusy");
 
            // No Measure in states ###############################################################
 
            this->m_no_measure_in_states.push_back("On:Operational:UpdateBusy");
 
            // Guards #####################################################################
 
            engine.RegisterGuardStatic<events::Measure>(
                "GuardMeasuringAllowed", [this](events::Measure const& ev) {
                    auto act_state = this->m_engine.GetState();
                    for (auto& s : m_no_measure_in_states) {
                        if (act_state.find(s) != std::string::npos) {
                            return false;
                        }
                    }
                    detail::MeasureContext ctx(ev.GetPayload());
                    if (static_cast<BizLogicIf&>(this->m_logic)
                            .GuardMeasuringAllowed(ctx.GetArg())) {
                        // Measure is allowed so context is stored.
                        assert(!m_measure_ctx);
                        m_measure_ctx.emplace(std::move(ctx));
                        return true;
                    }
                    return false;
                });
 
            // Activities #####################################################################
            engine.RegisterActivity(
                "ActivityMeasuring",
                [this](StopToken stop_token) {
                    assert(m_measure_ctx);
                    JsonPayload result =
                        static_cast<BizLogicIf&>(this->m_logic)
                            .ActivityMeasuring(std::move(stop_token), m_measure_ctx->GetArg());
                    // note: SetResult may throw
                    m_measure_ctx->SetResult(result);
                },
                m_measure_success_handler,
                m_measure_error_handler);
 
            // Actions #####################################################################
            engine.RegisterAction("ActionMeasuringDone", [this](auto c) {
                assert(m_measure_ctx);
                m_measure_ctx->SendReply();
                m_measure_ctx.reset();
            });
 
            engine.RegisterActionStatic<events::MeasureError>(
                "ActionMeasuringFailed", [this](events::MeasureError const& ev) {
                    assert(m_measure_ctx);
                    m_measure_ctx->SendErrorReply(ev.GetPayload());
                    m_measure_ctx.reset();
                });
        }
 
    protected:
        std::optional<detail::MeasureContext> m_measure_ctx;
        std::function<void()> m_measure_success_handler;
        std::function<void(std::exception_ptr)> m_measure_error_handler;
        std::list<std::string> m_no_measure_in_states;
    };
 
    class ModelBuilder : public Super::ModelBuilder {
    public:
        ModelBuilder(StateMachineEngine& engine) : Super::ModelBuilder(engine) {
            this->RegisterLayer({"Measurable", {}});
 
            this->mm.AddState(Composite, "On:Operational:RegionMeasure",  "On:Operational");
            this->mm.AddState(Initial,   "On:Operational:MeasureInitial", "On:Operational:RegionMeasure");
            this->mm.AddState(Simple,    "On:Operational:MeasureIdle",    "On:Operational:RegionMeasure");
            this->mm.AddState(Simple,    "On:Operational:MeasureBusy",    "On:Operational:RegionMeasure",   "ActivityMeasuring");
 
            this->mm.AddTrans("On:Operational:MeasureInitial"   , "On:Operational:MeasureIdle" );
            this->mm.AddTrans("On:Operational:MeasureIdle",
                              "On:Operational:MeasureBusy",
                              events::Measure::ID,
                              "GuardMeasuringAllowed");
            this->mm.AddTrans("On:Operational:MeasureBusy",
                              "On:Operational:MeasureIdle",
                              events::MeasureDone::ID,
                              "",
                              "ActionMeasuringDone");
            this->mm.AddTrans("On:Operational:MeasureBusy",
                              "On:Operational:MeasureIdle",
                              events::MeasureError::ID,
                              "",
                              "ActionMeasuringFailed");
        }
    };
};
 
}  // namespace rtctk::componentFramework
 
#endif