fix timer

src/synchronization/event.cc

@@ -1,7 +1,7 @@
 #include "sled/synchronization/event.h"
 
 namespace sled {
-// constexpr TimeDelta Event::kForever;
+constexpr TimeDelta Event::kForever;
 
 Event::Event() : Event(false, false) {}
 

src/timer/task_queue_timeout.cc

@@ -16,6 +16,7 @@ void
 TaskQueueTimeoutFactory::TaskQueueTimeout::Start(DurationMs duration_ms,
                                                  TimeoutID timeout_id)
 {
+    ASSERT(timeout_expiration_ == std::numeric_limits<TimeMs>::max(), "");
     timeout_expiration_ = parent_.get_time_() + duration_ms;
     timeout_id_ = timeout_id;
 
@@ -30,19 +31,25 @@ TaskQueueTimeoutFactory::TaskQueueTimeout::Start(DurationMs duration_ms,
     parent_.task_queue_.PostDelayedTaskWithPrecision(
         precision_,
         [timeout_id, this]() {
+            LOGV("timer", "Timeout expired: {}", timeout_id);
+            ASSERT(posted_task_expiration_
+                       != std::numeric_limits<TimeMs>::max(),
+                   "");
             posted_task_expiration_ = std::numeric_limits<TimeMs>::max();
+
             if (timeout_expiration_ == std::numeric_limits<TimeMs>::max()) {
                 // cancelled timer
                 // do nothing
             } else {
-                DurationMs remaining =
-                    timeout_expiration_ - parent_.get_time_();
-                timeout_expiration_ = std::numeric_limits<TimeMs>::max();
-                if (remaining > 0) {
-                    Start(remaining, timeout_id);
-                } else {
-                    LOGD("", "Timeout Triggered: {}", timeout_id);
+                const TimeMs now = parent_.get_time_();
+                if (timeout_expiration_ <= now) {
+                    timeout_expiration_ = std::numeric_limits<TimeMs>::max();
+                    LOGD("timer", "Timeout Triggered: {}", timeout_id);
                     parent_.on_expired_(timeout_id_);
+                } else {
+                    const DurationMs remaining = timeout_expiration_ - now;
+                    timeout_expiration_ = std::numeric_limits<TimeMs>::max();
+                    Start(remaining, timeout_id);
                 }
             }
         },

src/timer/timer.cc

@@ -42,6 +42,40 @@ Timer::Start()
     }
 }
 
+void
+Timer::Stop()
+{
+    if (is_running()) {
+        timeout_->Stop();
+        expiration_count_ = 0;
+        is_running_ = false;
+    }
+}
+
+void
+Timer::Trigger(TimerGeneration generation)
+{
+    if (!is_running_ || generation != generation_) { return; }
+    ++expiration_count_;
+    is_running_ = false;
+    // if max_restarts > exppiration_count_ then restart
+    {
+        is_running_ = true;
+        generation_ = TimerGeneration(generation_ + 1);
+        timeout_->Start(duration_, MakeTimeoutId(id_, generation_));
+    }
+
+    sled::optional<DurationMs> new_duration = on_expired_();
+    if (new_duration.has_value() && new_duration != duration_) {
+        duration_ = new_duration.value();
+        if (is_running_) {
+            timeout_->Stop();
+            generation_ = TimerGeneration(generation_ + 1);
+            timeout_->Start(duration_, MakeTimeoutId(id_, generation_));
+        }
+    }
+}
+
 std::unique_ptr<Timer>
 TimerManager::CreateTimer(const std::string &name, Timer::OnExpired on_expired)
 {
@@ -57,4 +91,13 @@ TimerManager::CreateTimer(const std::string &name, Timer::OnExpired on_expired)
     timers_[id] = timer.get();
     return timer;
 }
+
+void
+TimerManager::HandleTimeout(TimeoutID id)
+{
+    TimerID timer_id = id >> 32;
+    TimerGeneration generation = id & 0xffffffff;
+    auto it = timers_.find(timer_id);
+    if (it != timers_.end()) { it->second->Trigger(generation); }
+}
 }// namespace sled