⚡ (EventLoop): Loop once or multiple times

leka · Dec 19, 2023 · 7a5d8f2 · 7a5d8f2
1 parent 778bf10
commit 7a5d8f2
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 9 deletions.
diff --git a/include/interface/libs/EventLoop.h b/include/interface/libs/EventLoop.h
@@ -16,9 +16,9 @@ class EventLoop
 
 	virtual void registerCallback(const callback_t &) = 0;
 
-	virtual void start() = 0;
-	virtual void stop()	 = 0;
-	virtual void exit()	 = 0;
+	virtual void start(bool loop_once = true) = 0;
+	virtual void stop()						  = 0;
+	virtual void exit()						  = 0;
 
 	virtual void loop() = 0;
 };

diff --git a/libs/EventLoopKit/include/EventLoopKit.h b/libs/EventLoopKit/include/EventLoopKit.h
@@ -18,7 +18,7 @@ class EventLoopKit : public interface::EventLoop
 
 	void registerCallback(const callback_t &callback) final;
 
-	void start() final;
+	void start(bool loop_once = true) final;
 	void stop() final;
 	void exit() final;
 
@@ -29,7 +29,8 @@ class EventLoopKit : public interface::EventLoop
 
 	void loop() final;
 
-	bool must_exit = false;
+	bool _loop_once = true;
+	bool must_exit	= false;
 
 	callback_t _callback {};
 	rtos::Thread _thread {};

diff --git a/libs/EventLoopKit/source/EventLoopKit.cpp b/libs/EventLoopKit/source/EventLoopKit.cpp
@@ -16,11 +16,12 @@ void EventLoopKit::registerCallback(const callback_t &callback)
 	_callback = callback;
 }
 
-void EventLoopKit::start()
+void EventLoopKit::start(bool loop_once)
 {
 	if (_callback == nullptr) {
 		return;
 	}
+	_loop_once = loop_once;
 
 	_flags.set(flag::START);
 }
@@ -43,8 +44,10 @@ void EventLoopKit::loop()
 		return ret;
 	};
 
+	auto should_clear = [&] { return _loop_once; };
+
 	while (keep_running()) {
-		_flags.wait_any(flag::START, true);
+		_flags.wait_any(flag::START, should_clear());
 
 		_callback();
 	}

diff --git a/tests/unit/stubs/stubs/leka/EventLoopKit.h b/tests/unit/stubs/stubs/leka/EventLoopKit.h
@@ -18,7 +18,7 @@ class EventLoopKit : public interface::EventLoop
 
 	void registerCallback(const callback_t &callback) final;
 
-	void start() final;
+	void start(bool loop_once = true) final;
 	void stop() final;
 	void exit() final;
 

diff --git a/tests/unit/stubs/stubs/leka/source/EventLoopKit.cpp b/tests/unit/stubs/stubs/leka/source/EventLoopKit.cpp
@@ -11,7 +11,7 @@ void EventLoopKit::registerCallback(const callback_t &callback)
 	_callback = callback;
 }
 
-void EventLoopKit::start()
+void EventLoopKit::start(bool loop_once)
 {
 	if (_callback == nullptr) {
 		return;