linux简单的任务服务模型

#include "type.h"
#include <pthread.h>
#include <list>
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <unistd.h>
#include "ThreadMutex.h"
using namespace std;

typedef struct tagWorkInfo
{
	tagWorkInfo()
	{
		m_nIndex = 0;
		m_strInfo.clear();
	}

	int		m_nIndex;
	string	m_strInfo;
}WorkInfo, *LPWorkInfo;

CCrtSession		g_crtPrint;
CCrtSession		g_crtWork;
list<WorkInfo>	g_lsWork;

void AddWork(const WorkInfo& info);
BOOL GetWork(WorkInfo& info);

void Print(const WorkInfo& info);
void Print(const char* lpszInfo);

void* AddTaskProc(void* pVoid);
void* DoTaskProc(void* pVoid);

int main()
{
	pthread_t th1;
	pthread_t th2;
	void* pRet = NULL;

	//创建生产者和消费者线程
	pthread_create(&th1, NULL, &AddTaskProc, NULL);
	pthread_create(&th2, NULL, &DoTaskProc, NULL);

	//等待两个线程结束
	pthread_join(th1, &pRet);
	pthread_join(th2, &pRet);

	return 0;
}

void Print(const WorkInfo& info)
{
	CThreadMutex mutex(g_crtPrint.GetMutex());
	string strPrint;

	char szIndex[20] = {0};
	//itoa(info.m_nIndex, szIndex, 10);
	sprintf(szIndex, "index:%d\t info:", info.m_nIndex);

	strPrint = szIndex;
	strPrint += info.m_strInfo;

    printf("%s\n", strPrint.c_str());
}

void Print(const char* lpszInfo)
{
	CThreadMutex mutex(g_crtPrint.GetMutex());

	if (NULL == lpszInfo)
	{
		return;
	}

	printf("%s\n", lpszInfo);
}

void AddWork(const WorkInfo& info)
{
	CThreadMutex mutex(g_crtWork.GetMutex());
	g_lsWork.push_back(info);
}

BOOL GetWork(WorkInfo& info)
{
	CThreadMutex mutex(g_crtWork.GetMutex());

	if (g_lsWork.size() <= 0)
	{
		return FALSE;
	}

	list<WorkInfo>::iterator it = g_lsWork.begin();
	info = *it;
	g_lsWork.erase(it);

	return TRUE;
}

void* AddTaskProc(void* pVoid)
{
	printf("start AddTaskProc work....\n");
	int i = 0;
	while (TRUE)
	{
		WorkInfo info;
		info.m_nIndex = i++;
		info.m_strInfo = "test info";

        AddWork(info);

		Print("push new work into list");
	}

	return NULL;
}

void* DoTaskProc(void* pVoid)
{
	printf("start DoTaskProc work....\n");

	while (TRUE)
	{
		WorkInfo info;
		if (GetWork(info))
		{
			Print(info);
			Print("get new work from list");
		}
		else
		{
			Print("list is empty.. sleep 1 second!");
			sleep(1);
		}
	}

	return NULL;
}