MessageTimer.hpp

/*
Copyright (c) 2017-2024,
Battelle Memorial Institute; Lawrence Livermore National Security, LLC; Alliance for Sustainable
Energy, LLC.  See the top-level NOTICE for additional details. All rights reserved.
SPDX-License-Identifier: BSD-3-Clause
*/
#pragma once
 
#include "ActionMessage.hpp"
#include "gmlc/networking/AsioContextManager.h"
 
#include <asio/steady_timer.hpp>
#include <memory>
#include <mutex>
#include <vector>
 
namespace helics {
class MessageTimer: public std::enable_shared_from_this<MessageTimer> {
  public:
    using time_type = decltype(std::chrono::steady_clock::now());
    explicit MessageTimer(std::function<void(ActionMessage&&)> sFunction);
    int32_t addTimerFromNow(std::chrono::nanoseconds time, ActionMessage mess);
    int32_t addTimer(time_type expirationTime, ActionMessage mess);
    void cancelTimer(int32_t index);
    void cancelAll();
    void updateTimer(int32_t timerIndex, time_type expirationTime, ActionMessage mess);
    void updateTimerFromNow(int32_t timerIndex, std::chrono::nanoseconds time, ActionMessage mess);
    bool updateTimer(int32_t timerIndex, time_type expirationTime);
    bool addTimeToTimer(int32_t timerIndex, std::chrono::nanoseconds time);
    void updateMessage(int32_t timerIndex, ActionMessage mess);
    void sendMessage(int32_t timerIndex);
 
  private:
    std::mutex timerLock;  
    std::vector<ActionMessage> buffers;
    std::vector<time_type> expirationTimes;
    const std::function<void(ActionMessage&&)> sendFunction;
    std::vector<std::shared_ptr<asio::steady_timer>> timers;
    std::shared_ptr<gmlc::networking::AsioContextManager> contextPtr;
    decltype(contextPtr->startContextLoop()) loopHandle;
};
}  // namespace helics