FauxEventHandler.h

Go to the documentation of this file.

#pragma once

#include <swarmio/Exception.h>
#include <swarmio/data/discovery/Schema.pb.h>
#include <swarmio/services/event/Handler.h>
#include <map>
#include <iostream>

namespace swarmio::simulator 
{
    class FauxEventHandler final : public services::event::Handler
    {
        private:

            std::string _name;

            std::map<std::string, data::discovery::Type> _parameters;

        public:

            FauxEventHandler(const std::string& name)
                : _name(name) { }

            void AddParameter(const std::string& key, data::discovery::Type type)
            {
                _parameters[key] = type;
            }

            virtual void EventWasTriggered(const Node* node, const data::event::Notification& event) override {  }

            virtual data::discovery::Schema DescribeEvent(const std::string& name) override
            {
                if (name == _name)
                {
                    data::discovery::Schema schema;
                    for (auto p : _parameters)
                    {
                        (*schema.mutable_fields())[p.first].set_type(p.second);
                    }
                    return schema;
                }
                else
                {
                    throw Exception("Fake event handler queried for unknown event.");
                }
            }

            const std::string& GetName() const
            {
                return _name;
            }
    };
}