74 lines
1.9 KiB
C++
74 lines
1.9 KiB
C++
#pragma once
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#include <process.h>
|
|
#elif __linux__
|
|
#include <pthread.h>
|
|
#else
|
|
#error "이 플랫폼은 지원되지 않습니다."
|
|
#endif
|
|
#include <functional>
|
|
#include <future>
|
|
|
|
namespace Chattr {
|
|
|
|
template<typename F, typename... Args>
|
|
concept ReturnsVoidPtr = std::is_same_v<
|
|
std::invoke_result_t<F, Args...>,
|
|
void*
|
|
>;
|
|
|
|
class Thread {
|
|
public:
|
|
#ifdef _WIN32
|
|
static unsigned __stdcall thread_func(LPVOID param) {
|
|
// auto task = static_cast<std::function<void *()>*>(param);
|
|
std::unique_ptr<std::function<void*()>> task(static_cast<std::function<void* ()>*>(param));
|
|
(*task)();
|
|
return 0;
|
|
}
|
|
#elif __linux__
|
|
static void* thread_func(void *param) {
|
|
auto task(static_cast<std::packaged_task<void*()>*>(param));
|
|
(*task)();
|
|
delete task;
|
|
return 0;
|
|
}
|
|
#endif
|
|
Thread(Thread&&) noexcept;
|
|
Thread& operator=(Thread&&) noexcept;
|
|
|
|
Thread(const Thread&) = delete;
|
|
Thread& operator=(Thread&) = delete;
|
|
|
|
template<typename _Callable, typename... _Args>
|
|
requires ReturnsVoidPtr<_Callable, _Args...> && (!std::is_same_v<std::decay_t<_Callable>, Thread>)
|
|
Thread(_Callable&& __f, _Args&&... __args) {
|
|
auto boundFunc = [this, __f = std::move(__f), ... __args = std::move(__args)]() mutable -> void* {
|
|
void* ret = __f(std::move(__args)...);
|
|
returnValue = ret;
|
|
return ret;
|
|
};
|
|
std::function<void* ()> funcPtr = std::move(boundFunc);
|
|
#ifdef _WIN32
|
|
handle_ = (HANDLE)_beginthreadex(nullptr, 0, thread_func, new std::function<void *()>(std::move(funcPtr)), 0, nullptr);
|
|
#elif __linux__
|
|
pthread_create(&handle_, NULL, thread_func, funcPtr);
|
|
#endif
|
|
}
|
|
~Thread();
|
|
|
|
void* join();
|
|
void detach();
|
|
|
|
private:
|
|
#ifdef _WIN32
|
|
HANDLE handle_;
|
|
#elif __linux__
|
|
pthread_t handle_;
|
|
#endif
|
|
void* returnValue = nullptr;
|
|
bool detached = false;
|
|
};
|
|
|
|
} // namespace Chattr
|