1 files changed, 46 insertions, 0 deletions
diff --git a/lib/worker.cpp b/lib/worker.cpp
new file mode 100644
index 0000000..3c12caa
--- /dev/null
+++ b/lib/worker.cpp
@@ -0,0 +1,46 @@
+#include "worker.h"
+#include "work.h"
+#include <algorithm>
+#include <iterator>
+#include <mutex>
+
+Worker::Worker() : todoLen {0}
+{
+	std::generate_n(std::back_inserter(threads), std::thread::hardware_concurrency(), [this]() {
+		return std::thread {&Worker::worker, this};
+	});
+}
+
+Worker::~Worker()
+{
+	todoLen.release(std::thread::hardware_concurrency());
+	std::for_each(threads.begin(), threads.end(), [](auto & th) {
+		th.join();
+	});
+}
+
+void
+Worker::addWork(WorkPtr j)
+{
+	std::lock_guard<std::mutex> lck {todoMutex};
+	todoLen.release();
+	todo.emplace_back(std::move(j));
+}
+
+void
+Worker::worker()
+{
+	auto job = [this]() {
+		todoLen.acquire();
+		std::lock_guard<std::mutex> lck {todoMutex};
+		if (todo.size()) {
+			WorkPtr x = std::move(todo.front());
+			todo.pop_front();
+			return x;
+		}
+		return WorkPtr {};
+	};
+	while (auto j = job()) {
+		j->doWork();
+	}
+}