work_loop_handler.hpp

Go to the documentation of this file.

#pragma once
 
#include "commrat/messaging/system/subscription_messages.hpp"
#include "commrat/module/helpers/address_helpers.hpp"
#include <iostream>
#include <type_traits>
 
namespace commrat {
 
template<typename ModuleType>
class WorkLoopHandler {
protected:
    void work_loop() {
        auto& module = static_cast<ModuleType&>(*this);
        
        // Just print a simple message - the address details aren't critical for the log
        std::cout << "[" << module.config_.name << "] work_loop started on WORK mailbox\n" << std::flush;
        
        while (module.running_) {
            // Use receive_any with visitor pattern on work_mailbox
            // BLOCKING receive - waits indefinitely for subscription messages
            std::cout << "[" << module.config_.name << "] work_loop: waiting for message...\n" << std::flush;
            
            auto visitor = [&module](auto&& tims_msg) {
                // tims_msg is TimsMessage<PayloadT>, extract payload
                auto& msg = tims_msg.payload;
                using MsgType = std::decay_t<decltype(msg)>;
                
                // Handle subscription protocol
                if constexpr (std::is_same_v<MsgType, SubscribeRequestType>) {
                    std::cout << "[" << module.config_.name << "] Handling SubscribeRequest\n";
                    module.handle_subscribe_request(msg);
                } else if constexpr (std::is_same_v<MsgType, SubscribeReplyType>) {
                    std::cout << "[" << module.config_.name << "] Handling SubscribeReply\n";
                    module.handle_subscribe_reply(msg);
                } else if constexpr (std::is_same_v<MsgType, UnsubscribeRequestType>) {
                    std::cout << "[" << module.config_.name << "] Handling UnsubscribeRequest\n";
                    module.handle_unsubscribe_request(msg);
                }
            };
            
            // BLOCKING receive on work mailbox (no timeout)
            module.work_mailbox().receive_any(visitor);
        }
        
        std::cout << "[" << module.config_.name << "] work_loop ended\n";
    }
};
 
}  // namespace commrat