Add workqueue system

2020-07-10 03:16:51 -04:00
parent 02d361b541
commit 1f3fa53721
15 changed files with 529 additions and 15 deletions
@@ -0,0 +1,67 @@
+/*
+ * Copyright 2019-2020 PixlOne
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+#ifndef LOGID_TASK_H
+#define LOGID_TASK_H
+
+#include <functional>
+#include <memory>
+#include <future>
+#include "ExceptionHandler.h"
+
+namespace logid
+{
+    class task
+    {
+    public:
+        enum Status
+        {
+            Waiting,
+            Running,
+            Completed
+        };
+
+        explicit task(const std::function<void()>& function,
+                        const std::function<void(std::exception&)>&
+                        exception_handler={[](std::exception& e)
+                                           {ExceptionHandler::Default(e);}});
+
+        Status getStatus();
+
+        void run(); // Runs synchronously
+        void wait();
+
+        /* This function spawns a new task into the least used worker queue
+         * and forgets about it.
+         */
+        static void spawn(const std::function<void()>& function,
+                          const std::function<void(std::exception&)>&
+                          exception_handler={[](std::exception& e)
+                                             {ExceptionHandler::Default(e);}});
+
+        static void autoQueue(std::shared_ptr<task>);
+
+    private:
+        std::shared_ptr<std::function<void()>> _function;
+        std::shared_ptr<std::function<void(std::exception&)>>
+                _exception_handler;
+        std::atomic<Status> _status;
+        std::packaged_task<void()> _task_pkg;
+    };
+}
+
+#endif //LOGID_TASK_H