tdlib-fork/td/telegram/DelayDispatcher.h

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2020
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once

#include "td/telegram/net/NetQuery.h"

#include "td/actor/actor.h"

#include "td/utils/Time.h"

#include <queue>

namespace td {

class DelayDispatcher : public Actor {
 public:
  explicit DelayDispatcher(double default_delay) : default_delay_(default_delay) {
  }

  void send_with_callback(NetQueryPtr query, ActorShared<NetQueryCallback> callback);
  void send_with_callback_and_delay(NetQueryPtr query, ActorShared<NetQueryCallback> callback, double delay);

  void close_silent();

 private:
  struct Query {
    NetQueryPtr net_query;
    ActorShared<NetQueryCallback> callback;
    double delay;
  };
  std::queue<Query> queue_;
  Timestamp wakeup_at_;
  double default_delay_;

  void loop() override;
  void tear_down() override;
};

}  // namespace td