[簡單多線程控制]_[模擬殺毒軟件線程控制]

場景:
1.在做業務庫時,目前有2種常見的狀態，開始，停止。假如需要增加一個暫停狀態的話。就需要線程控制。

要求:
1.參考殺毒的開始殺毒，繼續殺毒，暫停殺毒，停止殺毒4個狀態設計一個多線程程序。主線程控制工作線程的暫停，繼續和停止。注意，只能兩個線程,注意，工作線程只能是detach，不能是join,因爲主線程不能被掛起.
2.必須使用pthread線程來實現(網站研發的可以用.net自帶線程庫)，有時間自己可以實現win32版本。寫一個控制檯或界面程序，界面程序可以模仿例子的，控制檯程序可以輸入數字來代替按鈕按下。如:
1.開始掃描.
2.繼續掃描->暫停掃描切換.
3.停止掃描.

 

#include <stdio.h>
#include <iostream>

#include "pthread.h"

typedef enum ButtonType
{
	kStart = 0,  //開始
	kPause,  //暫停
	kStop,  //停止
	kContinue,  //繼續
}ButtonType;

typedef struct ThreadData
{
	ButtonType type_;
	pthread_mutex_t mutex_;
	pthread_cond_t mian_cv_;
	pthread_cond_t work_cv_;
}ThreadData;

void *WorkThread(void *data)
{
	ThreadData *user_data = (ThreadData *)data;
	while (1)
	{
		switch (user_data->type_)
		{
		case kPause:
		{
			Sleep(1000);
			pthread_mutex_lock(&user_data->mutex_);
			//告訴主線程可以繼續
			pthread_cond_signal(&user_data->mian_cv_);
			//線程阻塞

			pthread_cond_wait(&user_data->work_cv_, &user_data->mutex_);

			pthread_mutex_unlock(&user_data->mutex_);

			if(user_data->type_ == kContinue)
			{
				pthread_mutex_lock(&user_data->mutex_);
				pthread_cond_signal(&user_data->mian_cv_);
				pthread_mutex_unlock(&user_data->mutex_);
			}

			//pause
			if (user_data->type_ == kStop)
			{
				pthread_mutex_lock(&user_data->mutex_);
				pthread_cond_signal(&user_data->mian_cv_);
				pthread_mutex_unlock(&user_data->mutex_);
				return 0;
			}

			break;
		}
		case kStop:
		{
			//告訴主線程可以繼續
			pthread_mutex_lock(&user_data->mutex_);
			pthread_cond_signal(&user_data->mian_cv_);
			pthread_mutex_unlock(&user_data->mutex_);
			
			//結束進程
			return 0;
		}
		default:
		{
			break;
		}
		}

		//Sleep(1000);
	}
}

void PrintHelp()
{
	std::cout<<"0  start"<<std::endl;
	std::cout<<"1  pause<->continue"<<std::endl;
	std::cout<<"2  stop"<<std::endl;
	std::cout<<"q  quit"<<std::endl;
}


//開始狀態只能一次
//暫停<->繼續->停止
//當工作線程處於正確狀態時，主線程可以工作別的
//主線程控制工作線程狀態
int main(int argc, char *argv[])
{

	ThreadData data;
	data.type_ = kStop;
	pthread_cond_init(&data.mian_cv_, NULL);
	pthread_cond_init(&data.work_cv_, NULL);

	pthread_mutex_init(&data.mutex_, NULL);

	//打印幫助信息
	PrintHelp();

	int value;
	while (1)
	{
		value = getchar();
		switch (value)
		{
		case '0':  //0  start
		{
			if (data.type_ != kStop)
			{
				std::cout<<"already start !"<<std::endl;
				break;
			}

			std::cout<<"start"<<std::endl;
			data.type_ = kStart;
			pthread_t id;
			pthread_create(&id, NULL, WorkThread, (void *)&data);
			pthread_detach(id);
			break;
		}
		case '1':  //1  pause<->continue
		{
			if (data.type_ == kStop)
			{
				std::cout<<"not start !"<<std::endl;
				break;
			}

			pthread_mutex_lock(&data.mutex_);
			if (data.type_ == kPause)
			{
				std::cout<<"cotinue"<<std::endl;
				data.type_ = kContinue;
				pthread_cond_signal(&data.work_cv_);
			}
			else
			{
				std::cout<<"pause"<<std::endl;
				data.type_ = kPause;
			}

			pthread_cond_wait(&data.mian_cv_, &data.mutex_);
			pthread_mutex_unlock(&data.mutex_);
			break;
		}
		case '2':  //2  stop
		{
			if (data.type_ == kStop)
			{
				std::cout<<"already finish !"<<std::endl;
				break;
			}

			std::cout<<"stop"<<std::endl;
			pthread_mutex_lock(&data.mutex_);

			pthread_cond_signal(&data.work_cv_);
			data.type_ = kStop;
			pthread_cond_wait(&data.mian_cv_, &data.mutex_);
			pthread_mutex_unlock(&data.mutex_);
			break;
		}
		case 'q':  //quit
			break;
		default:
			break;
		}

		if (value == 'q')
		{
			break;
		}
	}
	
	pthread_cond_destroy(&data.mian_cv_);
	pthread_cond_destroy(&data.work_cv_);
	pthread_mutex_destroy(&data.mutex_);

	return 0;
}

 

 

參考文章：

http://blog.csdn.net/ffilman/article/details/4871920