S4U_Daemon.h

#ifndef WRENCH_SIM4U_DAEMON_H
#define WRENCH_SIM4U_DAEMON_H

#include <string>

#include <simgrid/s4u.hpp>
#include <iostream>

//#define ACTOR_TRACKING_OUTPUT yes


namespace wrench {

    /***********************/
    /***********************/

    class Simulation;

    class S4U_Daemon {

        class LifeSaver {
        public:
            explicit LifeSaver(std::shared_ptr<S4U_Daemon> &reference) : reference(reference) {}

            std::shared_ptr<S4U_Daemon> reference;
        };

    public:
        std::string process_name;
        std::string mailbox_name;
        std::string hostname;

        S4U_Daemon(std::string hostname, std::string process_name_prefix, std::string mailbox_prefix);

        // Daemon without a mailbox (not needed?)
//        S4U_Daemon(std::string hostname, std::string process_name_prefix);

        virtual ~S4U_Daemon();

        void startDaemon(bool daemonized, bool auto_restart);

        void createLifeSaver(std::shared_ptr<S4U_Daemon> reference);

        virtual void cleanup();

        virtual int main() = 0;

        void setCleanlyTerminated();
        bool hasCleanlyTerminated();
        bool isSetToAutoRestart();

        bool join();

        void suspend();

        void resume();

        std::string getName();

        LifeSaver *life_saver = nullptr;

        Simulation *simulation;
    protected:

        friend class S4U_DaemonActor;
        void runMainMethod();

        void killActor();

        void acquireDaemonLock();

        void releaseDaemonLock();

        unsigned int num_starts = 0;
    private:
        // Lock use typically to prevent kill() from killing the actor
        // while it's in the middle of doing something critical
        simgrid::s4u::MutexPtr daemon_lock;
        
        simgrid::s4u::ActorPtr s4u_actor;
        
        bool cleanly_terminated; // Set to true after main returns
        bool auto_restart; // Set to true if daemon is supposed to auto-restart

#ifdef ACTOR_TRACKING_OUTPUT
        std::string process_name_prefix;
#endif

    };

    /***********************/
    /***********************/
};


#endif //WRENCH_SIM4U_DAEMONWITHMAILBOX_H