feat update pending_task_safety_flag

include/sled/task_queue/pending_task_safety_flag.h

@@ -10,11 +10,11 @@
 
 #include "sled/ref_counted_base.h"
 #include "sled/scoped_refptr.h"
+#include <functional>
 
 namespace sled {
 
-class PendingTaskSafetyFlag final
-    : public sled::RefCountedNonVirtual<PendingTaskSafetyFlag> {
+class PendingTaskSafetyFlag final : public sled::RefCountedNonVirtual<PendingTaskSafetyFlag> {
 public:
     static sled::scoped_refptr<PendingTaskSafetyFlag> Create();
     static sled::scoped_refptr<PendingTaskSafetyFlag> CreateDetached();
@@ -29,11 +29,36 @@ protected:
     explicit PendingTaskSafetyFlag(bool alive) : alive_(alive) {}
 
 private:
-    static sled::scoped_refptr<PendingTaskSafetyFlag>
-    CreateInternal(bool alive);
+    static sled::scoped_refptr<PendingTaskSafetyFlag> CreateInternal(bool alive);
     bool alive_ = true;
 };
 
+class ScopedTaskSafety final {
+public:
+    ScopedTaskSafety() = default;
+
+    explicit ScopedTaskSafety(scoped_refptr<PendingTaskSafetyFlag> flag) : flag_(std::move(flag)) {}
+
+    ~ScopedTaskSafety() { flag_->SetNotAlive(); }
+
+    void reset(scoped_refptr<PendingTaskSafetyFlag> new_flag = PendingTaskSafetyFlag::Create())
+    {
+        flag_->SetNotAlive();
+        flag_ = std::move(new_flag);
+    }
+
+private:
+    scoped_refptr<PendingTaskSafetyFlag> flag_;
+};
+
+inline std::function<void()>
+SafeTask(scoped_refptr<PendingTaskSafetyFlag> flag, std::function<void()> task)
+{
+    return [task, flag]() mutable {
+        if (flag->alive()) { std::move(task)(); }
+    };
+}
+
 }// namespace sled
 
 #endif// SLED_TASK_QUEUE_PENDING_TASK_SAFETY_FLAG_H

include/sled/timer/task_queue_timeout.h

@@ -1,7 +1,9 @@
 #pragma once
+#include "sled/scoped_refptr.h"
 #ifndef SLED_TIMER_QUEUE_TIMEOUT_H
 #define SLED_TIMER_QUEUE_TIMEOUT_H
 
+#include "sled/task_queue/pending_task_safety_flag.h"
 #include "sled/task_queue/task_queue_base.h"
 #include "sled/timer/timeout.h"
 #include <limits>
@@ -12,28 +14,24 @@ typedef uint64_t TimeMs;
 
 class TaskQueueTimeoutFactory {
 public:
-    TaskQueueTimeoutFactory(
-        sled::TaskQueueBase &task_queue,
-        std::function<TimeMs()> get_time,
-        std::function<void(TimeoutID timeout_id)> on_expired)
+    TaskQueueTimeoutFactory(sled::TaskQueueBase &task_queue,
+                            std::function<TimeMs()> get_time,
+                            std::function<void(TimeoutID timeout_id)> on_expired)
         : task_queue_(task_queue),
           get_time_(get_time),
           on_expired_(on_expired)
     {}
 
     std::unique_ptr<Timeout>
-    CreateTimeout(sled::TaskQueueBase::DelayPrecision precision =
-                      sled::TaskQueueBase::DelayPrecision::kHigh)
+    CreateTimeout(sled::TaskQueueBase::DelayPrecision precision = sled::TaskQueueBase::DelayPrecision::kHigh)
     {
-        return std::unique_ptr<TaskQueueTimeout>(
-            new TaskQueueTimeout(*this, precision));
+        return std::unique_ptr<TaskQueueTimeout>(new TaskQueueTimeout(*this, precision));
     }
 
 private:
     class TaskQueueTimeout : public Timeout {
     public:
-        TaskQueueTimeout(TaskQueueTimeoutFactory &parent,
-                         sled::TaskQueueBase::DelayPrecision precision);
+        TaskQueueTimeout(TaskQueueTimeoutFactory &parent, sled::TaskQueueBase::DelayPrecision precision);
         ~TaskQueueTimeout() override;
         void Start(DurationMs duration, TimeoutID timeout_id) override;
         void Stop() override;
@@ -44,6 +42,7 @@ private:
         TimeMs posted_task_expiration_ = std::numeric_limits<TimeMs>::max();
         TimeMs timeout_expiration_ = std::numeric_limits<TimeMs>::max();
         TimeoutID timeout_id_ = TimeoutID(0);
+        scoped_refptr<PendingTaskSafetyFlag> safety_flag_;
     };
 
     sled::TaskQueueBase &task_queue_;

src/task_queue/pending_task_safety_flag.cc

@@ -6,8 +6,7 @@ sled::scoped_refptr<PendingTaskSafetyFlag>
 PendingTaskSafetyFlag::CreateInternal(bool alive)
 {
     // Explicit new, to access private constructor.
-    return sled::scoped_refptr<PendingTaskSafetyFlag>(
-        new PendingTaskSafetyFlag(alive));
+    return sled::scoped_refptr<PendingTaskSafetyFlag>(new PendingTaskSafetyFlag(alive));
 }
 
 sled::scoped_refptr<PendingTaskSafetyFlag>
@@ -31,7 +30,7 @@ PendingTaskSafetyFlag::CreateDetachedInactive()
 void
 PendingTaskSafetyFlag::SetNotAlive()
 {
-    alive_ = true;
+    alive_ = false;
 }
 
 void

src/timer/task_queue_timeout.cc

@@ -1,5 +1,6 @@
 #include "sled/timer/task_queue_timeout.h"
 #include "sled/log/log.h"
+#include "sled/task_queue/pending_task_safety_flag.h"
 #include "sled/units/time_delta.h"
 
 namespace sled {
@@ -23,34 +24,40 @@ TaskQueueTimeoutFactory::TaskQueueTimeout::Start(DurationMs duration_ms, Timeout
         LOGV("timer",
              "New timeout duration is less than scheduled - "
              "ghosting old delayed task");
+        safety_flag_->SetNotAlive();
+        safety_flag_ = PendingTaskSafetyFlag::Create();
     }
 
     posted_task_expiration_ = timeout_expiration_;
+    auto safety_flag = safety_flag_;
     parent_.task_queue_.PostDelayedTaskWithPrecision(
         precision_,
-        [timeout_id, this]() {
-            if (timeout_id != this->timeout_id_) { return; }
-            LOGV("timer", "Timeout expired: {}", timeout_id);
+        SafeTask(safety_flag_,
+                 [timeout_id, this]() {
+                     // if (timeout_id != this->timeout_id_) { return; }
+                     LOGV("timer", "Timeout expired: {}", timeout_id);
+                     ASSERT(posted_task_expiration_ != std::numeric_limits<TimeMs>::max(), "");
+                     posted_task_expiration_ = std::numeric_limits<TimeMs>::max();
 
-            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 {
+                         const TimeMs now = parent_.get_time_();
+                         const DurationMs remaining = timeout_expiration_ - now;
+                         bool is_expired = timeout_expiration_ <= now;
 
-            if (timeout_expiration_ == std::numeric_limits<TimeMs>::max()) {
-                // cancelled timer
-                // do nothing
-            } else {
-                const TimeMs now = parent_.get_time_();
-                if (timeout_expiration_ <= now) {
-                    timeout_expiration_ = std::numeric_limits<TimeMs>::max();
-                    LOGV("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);
-                }
-            }
-        },
+                         timeout_expiration_ = std::numeric_limits<TimeMs>::max();
+
+                         if (!is_expired) {
+                             // continue wait
+                             Start(remaining, timeout_id);
+                         } else {
+                             LOGV("timer", "Timeout Triggered: {}", timeout_id);
+                             parent_.on_expired_(timeout_id_);
+                         }
+                     }
+                 }),
         sled::TimeDelta::Millis(duration_ms));
 }