ThreadWorkUnit.h
#pragma once
#include <queue>
#include <pthread.h>
#include "SafeQueue.h"
namespace Extralib
{
namespace Thread
{
//消息結構體
struct sWorkItem
{
sWorkItem()
{
type = 0;
nDataSize = 0;
pData = NULL;
param1 = 0;
}
void Set(char type,int param1,int size,char* data)
{
this->type = type;
this->param1 = param1;
nDataSize = size;
pData = data;
}
short type; //类型取决于不同系统的设定，并不固定
int param1; //消息id
unsigned int nDataSize; //消息包大小
char* pData; //消息包
};
class ThreadWorkUnit
{
public:
ThreadWorkUnit();
virtual ~ThreadWorkUnit();
//线程循环处理函数
virtual int svc();
void PushItem(const sWorkItem& data);
bool PopItem(sWorkItem& data);
bool GetItem(sWorkItem& data);
void PopAllItems(std::list<sWorkItem>& result_queue);
bool isStop(){ return m_bStop;}
/*
* 启动工作线程
*/
int Start();
/*
* 结束工作线程(可重复启动)
*/
void Stop();
private:
SafeQueue::safe_queue<sWorkItem> m_msg;
pthread_t tid;
bool m_bStop;
};
}
}ThreadWorkUnit.cpp
#include "ThreadWorkUnit.h"
#include "AutoLock.h"
#include "../OS/OS.h" //只是一个sleep头文件
namespace Extralib
{
namespace Thread
{
template <typename TYPE, int (TYPE::*svc)() >
void* _thread_t(void* param)
{
TYPE* This = (TYPE*)param;
This->svc();
return NULL;
}
ThreadWorkUnit::ThreadWorkUnit()
{
m_bStop = true;
}
ThreadWorkUnit::~ThreadWorkUnit()
{
}
int ThreadWorkUnit::svc()
{
return 0;
}
void ThreadWorkUnit::PushItem(const sWorkItem& data)
{
m_msg.push(data);
}
bool ThreadWorkUnit::PopItem(sWorkItem& data)
{
return m_msg.try_pop(data);
}
void ThreadWorkUnit::PopAllItems(std::list<sWorkItem>& result_queue)
{
m_msg.exchange(result_queue);
}
int ThreadWorkUnit::Start()
{
m_bStop = false;
pthread_create(&tid, NULL, _thread_t<ThreadWorkUnit, &ThreadWorkUnit::svc>, this);
return 1;
}
void ThreadWorkUnit::Stop()
{
m_bStop = true;
}
bool ThreadWorkUnit::GetItem( sWorkItem& data )
{
if(!m_msg.empty())
{
data = m_msg.front();
return true;
}
else
{
return false;
}
}
}
}
