time/test-time-source.cpp

   1 #include "test-time-source.hpp"
   2
   3 #include <stdio.h>
   4
   5 TestTimeSource::TestTimeSource(uint64_t base, TimeUnit unit, bool reliable) :
   6     time(time_to_ns(base, unit)), reliable(reliable)
   7 {
   8 }
   9
  10 uint64_t TestTimeSource::now(TimeUnit unit)
  11 {
  12     return ns_to_time(this->time, unit);
  13 }
  14
  15 #include <stdio.h>
  16
  17 void TestTimeSource::arm(Timer *timer)
  18 {
  19     std::list<Timer *>::iterator i;
  20
  21     for (i = this->events.begin(); i != this->events.end(); ++i) {
  22         if (timer->get_deadline() < (*i)->get_deadline()) {
  23             break;
  24         }
  25     }
  26
  27     this->events.insert(i, timer);
  28 }
  29
  30 void TestTimeSource::cancel(Timer *timer)
  31 {
  32     std::list<Timer *>::iterator i;
  33
  34     for (i = this->events.begin(); i != this->events.end(); ++i) {
  35         if (*i == timer) {
  36             this->events.erase(i);
  37             break;
  38         }
  39     }
  40 }
  41
  42 void TestTimeSource::advance(uint64_t by, TimeUnit unit)
  43 {
  44     std::list<Timer *>::iterator i;
  45     uint64_t target;
  46
  47     target = time_to_ns(ns_to_time(this->time, unit) + by, unit);
  48
  49     if (!this->reliable) {
  50         this->time = target;
  51     }
  52
  53     i = this->events.begin();
  54     while (i != this->events.end() &&
  55            (*i)->get_deadline() <= target) {
  56         Timer *timer = *i;
  57         this->events.pop_front();
  58         if (this->reliable) {
  59             if (this->time < timer->get_deadline()) {
  60                 this->time = timer->get_deadline();
  61             }
  62         }
  63         timer->fire();
  64         i = this->events.begin();
  65     }
  66
  67     this->time = target;
  68 }