UpdateAwaiter.h

Go to the documentation of this file.

#pragma once

#include <swarmio/services/Awaiter.h>
#include <chrono>

namespace swarmio::services::telemetry
{
    class SWARMIO_API UpdateAwaiter final : public Awaiter<data::telemetry::Update>
    {
        private:

            const Node* _target;

        protected:

            virtual data::telemetry::Update ExtractResponse(const Node* node, const data::Message* message) override
            {
                if (message->content_case() == data::Message::ContentCase::kTmUpdate)
                {
                    return message->tm_update();
                }
                else
                {
                    throw Exception("Last update received is invalid");
                }
            }

            virtual bool IsFinished() override
            {
                return false;
            }

        public:

            UpdateAwaiter(Endpoint* endpoint, uint64_t requestIdentifier, const Node* target)
                : Awaiter(endpoint, requestIdentifier), _target(target)
            {
                FinishConstruction();
            }

            virtual void MailboxWillBeDisconnected() noexcept override 
            {
                // Send unsubscribe message
                data::Message request;
                request.mutable_tm_unsubscribe_request()->set_identifier(GetIdentifier());
                GetEndpoint()->Send(&request, _target);

                // Call base implementation
                Awaiter::MailboxWillBeDisconnected();
            }

            const Node* GetTarget() const
            {
                return _target;
            }
    };
}