Ton

1
/*
2
    This file is part of TON Blockchain Library.
3

4
    TON Blockchain Library is free software: you can redistribute it and/or modify
5
    it under the terms of the GNU Lesser General Public License as published by
6
    the Free Software Foundation, either version 2 of the License, or
7
    (at your option) any later version.
8

9
    TON Blockchain Library is distributed in the hope that it will be useful,
10
    but WITHOUT ANY WARRANTY; without even the implied warranty of
11
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
    GNU Lesser General Public License for more details.
13

14
    You should have received a copy of the GNU Lesser General Public License
15
    along with TON Blockchain Library.  If not, see <http://www.gnu.org/licenses/>.
16

17
    Copyright 2017-2020 Telegram Systems LLP
18
*/
19

20
#include "Pacer.h"
21
namespace ton {
22
namespace rldp2 {
23
Pacer::Pacer(Options options)
24
    : speed_(options.initial_speed)
25
    , capacity_(options.initial_capacity)
26
    , max_capacity_(options.max_capacity)
27
    , time_granularity_(options.time_granularity) {
28
}
29

30
td::Timestamp Pacer::wakeup_at() const {
31
  return wakeup_at_;
32
}
33

34
void Pacer::set_speed(double speed) {
35
  if (speed < 1) {
36
    speed = 1;
37
  }
38
  speed_ = speed;
39
}
40

41
td::optional<td::Timestamp> Pacer::send(double size, td::Timestamp now) {
42
  update_capacity(now);
43

44
  if (size < capacity_) {
45
    capacity_ -= size;
46
    return {};
47
  }
48

49
  size -= capacity_;
50
  capacity_ = 0;
51
  wakeup_at_ = td::Timestamp::in(size / speed_, now);
52
  capacity_at_ = wakeup_at_;
53
  return wakeup_at_;
54
}
55

56
void Pacer::update_capacity(td::Timestamp now) {
57
  if (capacity_at_ && capacity_at_.is_in_past(now)) {
58
    capacity_ += (now.at() - capacity_at_.at()) * speed_;
59
    capacity_ = td::min(capacity_, td::max(max_capacity_, speed_ * time_granularity_));
60
  }
61
  capacity_at_ = now;
62
}
63
}  // namespace rldp2
64
}  // namespace ton
65