#include "BaseThread.h"


BaseThread::BaseThread() {
	m_thread = unique_ptr<thread>(new thread(&BaseThread::Run, this));
	execFunc = nullptr;
	m_thread->detach();
}

void BaseThread::ThreadStart(){
	unique_lock<mutex> lock(mtx);
	m_state = RunningState::START;
	cond.notify_one();
}

void BaseThread::ThreadPause() {
	unique_lock<mutex> lock(mtx);
	m_state = RunningState::PAUSE;
}

void BaseThread::ThreadStop() {
	unique_lock<mutex> lock(mtx);
	m_state = RunningState::STOP;
	cond.notify_one();
	cond_taskEmpty.notify_one();
}

BaseThread::RunningState BaseThread::GetRunningState() {
	return m_state;
}

void BaseThread::Run() {
	m_state = RunningState::START;
	cout << "thread ID:" << this_thread::get_id << " is start!\n";
	// 之所以这种方法终止线程安全，是因为该执行的模块都一起执行完了才退出循环。
	while (GetRunningState() != RunningState::STOP) {
		WaitIfPaused();
		if (GetRunningState() == RunningState::START) TaskAction();
	}
	cout << "Thread ID : " << this_thread::get_id << " is stop.\n";
}

void BaseThread::TaskAction() {
	unique_lock<mutex> lock(mtx);
	// 限制在此地WAIT而别的地方没有该置位，是因为希望在START过程中发出需要任务的请求而非在PAUSE过程中也可以添加任务
	m_state = RunningState::WAIT;
	cond_taskEmpty.wait(lock, [this] {return execFunc!=nullptr || m_state == RunningState::STOP; });
	if (m_state == RunningState::STOP) return;
	m_state = RunningState::START;
	if(execFunc()) cout << "thread: " << this_thread::get_id() << " task exec successful!\n";
	else cout << "thread: " << this_thread::get_id() << " task exec failed!\n";
	execFunc = nullptr;
	cond_taskFull.notify_one();
}

void BaseThread::WaitIfPaused() {
	unique_lock<mutex> lock(mtx);
	cond.wait(lock, [this] {return GetRunningState() != RunningState::PAUSE; });	//不为暂停
}



bool BaseThread::AddTask(function<bool()> func) {
	unique_lock<mutex> lock(mtx);
	cond_taskFull.wait(lock, [this] {return execFunc == nullptr; });
	execFunc = func;
	cond_taskEmpty.notify_one();
	return true;
}