[TASKS] Create TaskWithState

aleemont1 · Nov 24, 2023 · da14226 · da14226
1 parent a853916
commit da14226
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/smart_bridge/src/kernel/TaskWithState.h b/smart_bridge/src/kernel/TaskWithState.h
@@ -0,0 +1,28 @@
+#ifndef __TASK_WITH_STATE__
+#define __TASK_WITH_STATE__
+
+#include "Task.h"
+#include <Arduino.h>
+
+class TaskWithState : public Task
+{
+public:
+    TaskWithState() : Task(){}; // Default constructor
+
+    void setState(const int state)
+    {
+        this->state = state;
+        this->stateTimestamp = millis();
+    }
+
+    long timeInState()
+    {
+        return millis() - stateTimestamp;
+    }
+
+private:
+    int state;
+    long stateTimestamp;
+};
+
+#endif