// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include <zencore/zencore.h>

#include <zencore/blockingqueue.h>
#include <zencore/refcount.h>

#include <exception>
#include <functional>
#include <system_error>
#include <thread>
#include <vector>

namespace zen {

struct IWork : public RefCounted
{
	virtual void Execute() = 0;

	inline std::exception_ptr GetException() { return m_Exception; }

private:
	std::exception_ptr m_Exception;

	friend class WorkerThreadPool;
};

class WorkerThreadPool
{
public:
	WorkerThreadPool(int InThreadCount);
	~WorkerThreadPool();

	void ScheduleWork(Ref<IWork> Work);
	void ScheduleWork(std::function<void()>&& Work);

	void WorkerThreadFunction();

private:
	std::vector<std::thread>  m_WorkerThreads;
	BlockingQueue<Ref<IWork>> m_WorkQueue;
};

}  // namespace zen