X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Finternaltask.h;h=608ea9662609bb0e5912f65a247f193f3a292504;hb=d1eb133ab529cdae131be7b150209f03189248f3;hp=f29a61d29e8a35b7731340fe9df87a0457a13343;hpb=0368a9c1972ff5fd6cd8984901ff0bdbd86afc9f;p=builder.git

diff --git a/source/internaltask.h b/source/internaltask.h
index f29a61d..608ea96 100644
--- a/source/internaltask.h
+++ b/source/internaltask.h
@@ -1,6 +1,7 @@
 #ifndef INTERNALTASK_H_
 #define INTERNALTASK_H_
 
+#include <functional>
 #include <msp/core/thread.h>
 #include "task.h"
 
@@ -11,30 +12,34 @@ or ERROR before terminating.
 */
 class InternalTask: public Task
 {
-public:
+private:
 	class Worker: public Msp::Thread
 	{
 		friend class InternalTask;
 
-	protected:
-		Status status;
-
-		Worker();
+	private:
+		std::function<bool()> func;
+		volatile Status status = Task::RUNNING;
 
 	public:
+		Worker(std::function<bool()> f): func(f) { }
+
 		Status get_status() const { return status; }
+
+	private:
+		void main() override;
 	};
 
-private:
-	Worker *worker;
+	Worker worker;
 
 public:
-	InternalTask(Worker *);
+	InternalTask(std::function<bool()> f): worker(f) { }
 	~InternalTask();
 
-	virtual std::string get_command() const { return "<internal>"; }
-	virtual void start();
-	virtual Status check();
+	std::string get_command() const override { return "<internal>"; }
+	void start() override;
+	Status check() override;
+	Status wait() override;
 };
 
 #endif